mautrix-whatsapp/provisioning.go

406 lines
12 KiB
Go
Raw Normal View History

2020-02-09 19:32:14 +01:00
// mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
2021-10-22 19:14:34 +02:00
// Copyright (C) 2021 Tulir Asokan
2020-02-09 19:32:14 +01:00
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
2020-02-09 19:32:14 +01:00
package main
import (
2021-02-09 16:03:34 +01:00
"bufio"
2020-02-09 19:32:14 +01:00
"context"
"encoding/json"
"errors"
2021-02-09 16:03:34 +01:00
"net"
2020-02-09 19:32:14 +01:00
"net/http"
"strings"
2021-02-09 16:03:34 +01:00
"time"
2020-02-09 19:32:14 +01:00
"github.com/gorilla/websocket"
2021-02-17 00:21:30 +01:00
log "maunium.net/go/maulogger/v2"
2021-05-12 13:54:40 +02:00
2021-02-17 00:21:30 +01:00
"maunium.net/go/mautrix/id"
2020-02-09 19:32:14 +01:00
)
type ProvisioningAPI struct {
bridge *Bridge
log log.Logger
}
func (prov *ProvisioningAPI) Init() {
prov.log = prov.bridge.Log.Sub("Provisioning")
prov.log.Debugln("Enabling provisioning API at", prov.bridge.Config.AppService.Provisioning.Prefix)
r := prov.bridge.AS.Router.PathPrefix(prov.bridge.Config.AppService.Provisioning.Prefix).Subrouter()
r.Use(prov.AuthMiddleware)
r.HandleFunc("/ping", prov.Ping).Methods(http.MethodGet)
r.HandleFunc("/login", prov.Login).Methods(http.MethodGet)
2020-02-09 19:32:14 +01:00
r.HandleFunc("/logout", prov.Logout).Methods(http.MethodPost)
r.HandleFunc("/delete_session", prov.DeleteSession).Methods(http.MethodPost)
r.HandleFunc("/delete_connection", prov.DeleteConnection).Methods(http.MethodPost)
r.HandleFunc("/disconnect", prov.Disconnect).Methods(http.MethodPost)
r.HandleFunc("/reconnect", prov.Reconnect).Methods(http.MethodPost)
prov.bridge.AS.Router.HandleFunc("/_matrix/app/com.beeper.asmux/ping", prov.BridgeStatePing).Methods(http.MethodPost)
prov.bridge.AS.Router.HandleFunc("/_matrix/app/com.beeper.bridge_state", prov.BridgeStatePing).Methods(http.MethodPost)
2020-02-09 19:32:14 +01:00
}
2021-02-09 16:03:34 +01:00
type responseWrap struct {
http.ResponseWriter
statusCode int
}
var _ http.Hijacker = (*responseWrap)(nil)
func (rw *responseWrap) WriteHeader(statusCode int) {
rw.ResponseWriter.WriteHeader(statusCode)
rw.statusCode = statusCode
}
func (rw *responseWrap) Hijack() (net.Conn, *bufio.ReadWriter, error) {
hijacker, ok := rw.ResponseWriter.(http.Hijacker)
if !ok {
return nil, nil, errors.New("response does not implement http.Hijacker")
}
return hijacker.Hijack()
}
2020-02-09 19:32:14 +01:00
func (prov *ProvisioningAPI) AuthMiddleware(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
auth := r.Header.Get("Authorization")
if len(auth) == 0 && strings.HasSuffix(r.URL.Path, "/login") {
authParts := strings.Split(r.Header.Get("Sec-WebSocket-Protocol"), ",")
for _, part := range authParts {
part = strings.TrimSpace(part)
if strings.HasPrefix(part, "net.maunium.whatsapp.auth-") {
auth = part[len("net.maunium.whatsapp.auth-"):]
break
}
}
} else if strings.HasPrefix(auth, "Bearer ") {
auth = auth[len("Bearer "):]
}
2020-02-09 19:32:14 +01:00
if auth != prov.bridge.Config.AppService.Provisioning.SharedSecret {
jsonResponse(w, http.StatusForbidden, map[string]interface{}{
"error": "Invalid auth token",
"errcode": "M_FORBIDDEN",
})
return
}
userID := r.URL.Query().Get("user_id")
2020-05-08 21:32:22 +02:00
user := prov.bridge.GetUserByMXID(id.UserID(userID))
2021-02-09 16:03:34 +01:00
start := time.Now()
wWrap := &responseWrap{w, 200}
h.ServeHTTP(wWrap, r.WithContext(context.WithValue(r.Context(), "user", user)))
duration := time.Now().Sub(start).Seconds()
prov.log.Infofln("%s %s from %s took %.2f seconds and returned status %d", r.Method, r.URL.Path, user.MXID, duration, wWrap.statusCode)
2020-02-09 19:32:14 +01:00
})
}
type Error struct {
Success bool `json:"success"`
Error string `json:"error"`
ErrCode string `json:"errcode"`
}
type Response struct {
Success bool `json:"success"`
Status string `json:"status"`
}
func (prov *ProvisioningAPI) DeleteSession(w http.ResponseWriter, r *http.Request) {
user := r.Context().Value("user").(*User)
2021-10-22 19:14:34 +02:00
if user.Session == nil && user.Client == nil {
2020-02-09 19:32:14 +01:00
jsonResponse(w, http.StatusNotFound, Error{
Error: "Nothing to purge: no session information stored and no active connection.",
ErrCode: "no session",
})
return
}
user.DeleteConnection()
2021-10-22 19:14:34 +02:00
user.DeleteSession()
2020-02-09 19:32:14 +01:00
jsonResponse(w, http.StatusOK, Response{true, "Session information purged"})
}
func (prov *ProvisioningAPI) DeleteConnection(w http.ResponseWriter, r *http.Request) {
user := r.Context().Value("user").(*User)
2021-10-22 19:14:34 +02:00
if user.Client == nil {
2020-02-09 19:32:14 +01:00
jsonResponse(w, http.StatusNotFound, Error{
Error: "You don't have a WhatsApp connection.",
ErrCode: "not connected",
})
return
}
user.DeleteConnection()
2020-02-09 19:32:14 +01:00
jsonResponse(w, http.StatusOK, Response{true, "Disconnected from WhatsApp and connection deleted"})
}
func (prov *ProvisioningAPI) Disconnect(w http.ResponseWriter, r *http.Request) {
user := r.Context().Value("user").(*User)
2021-10-22 19:14:34 +02:00
if user.Client == nil {
2020-02-09 19:32:14 +01:00
jsonResponse(w, http.StatusNotFound, Error{
Error: "You don't have a WhatsApp connection.",
ErrCode: "no connection",
})
return
}
2021-10-22 19:14:34 +02:00
user.DeleteConnection()
2020-02-09 19:32:14 +01:00
jsonResponse(w, http.StatusOK, Response{true, "Disconnected from WhatsApp"})
}
func (prov *ProvisioningAPI) Reconnect(w http.ResponseWriter, r *http.Request) {
user := r.Context().Value("user").(*User)
2021-10-22 19:14:34 +02:00
if user.Client == nil {
2020-02-09 19:32:14 +01:00
if user.Session == nil {
jsonResponse(w, http.StatusForbidden, Error{
Error: "No existing connection and no session. Please log in first.",
ErrCode: "no session",
})
} else {
user.Connect(false)
jsonResponse(w, http.StatusOK, Response{true, "Created connection to WhatsApp."})
}
return
}
2020-03-11 17:21:51 +01:00
2021-10-22 19:14:34 +02:00
// TODO reimplement
//user.log.Debugln("Received /reconnect request, disconnecting")
//wasConnected := true
//err := user.Conn.Disconnect()
//if err == whatsapp.ErrNotConnected {
// wasConnected = false
//} else if err != nil {
// user.log.Warnln("Error while disconnecting:", err)
//}
//
//user.log.Debugln("Restoring session for /reconnect")
//err = user.Conn.Restore(true, r.Context())
//user.log.Debugfln("Restore session for /reconnect responded with %v", err)
//if err == whatsapp.ErrInvalidSession {
// if user.Session != nil {
// user.log.Debugln("Got invalid session error when reconnecting, but user has session. Retrying using RestoreWithSession()...")
// user.Conn.SetSession(*user.Session)
// err = user.Conn.Restore(true, r.Context())
// } else {
// jsonResponse(w, http.StatusForbidden, Error{
// Error: "You're not logged in",
// ErrCode: "not logged in",
// })
// return
// }
//}
//if err == whatsapp.ErrLoginInProgress {
// jsonResponse(w, http.StatusConflict, Error{
// Error: "A login or reconnection is already in progress.",
// ErrCode: "login in progress",
// })
// return
//} else if err == whatsapp.ErrAlreadyLoggedIn {
// jsonResponse(w, http.StatusConflict, Error{
// Error: "You were already connected.",
// ErrCode: err.Error(),
// })
// return
//}
//if err != nil {
// user.log.Warnln("Error while reconnecting:", err)
// jsonResponse(w, http.StatusInternalServerError, Error{
// Error: fmt.Sprintf("Unknown error while reconnecting: %v", err),
// ErrCode: err.Error(),
// })
// user.log.Debugln("Disconnecting due to failed session restore in reconnect command...")
// err = user.Conn.Disconnect()
// if err != nil {
// user.log.Errorln("Failed to disconnect after failed session restore in reconnect command:", err)
// }
// return
//}
//user.ConnectionErrors = 0
//user.PostLogin()
//
//var msg string
//if wasConnected {
// msg = "Reconnected successfully."
//} else {
// msg = "Connected successfully."
//}
//
//jsonResponse(w, http.StatusOK, Response{true, msg})
2020-02-09 19:32:14 +01:00
}
func (prov *ProvisioningAPI) Ping(w http.ResponseWriter, r *http.Request) {
user := r.Context().Value("user").(*User)
wa := map[string]interface{}{
"has_session": user.Session != nil,
"management_room": user.ManagementRoom,
"conn": nil,
}
2021-10-22 19:14:34 +02:00
if user.JID.IsEmpty() {
wa["jid"] = user.JID.String()
}
if user.Client != nil {
2020-02-09 19:32:14 +01:00
wa["conn"] = map[string]interface{}{
2021-10-22 19:14:34 +02:00
"is_connected": user.Client.IsConnected(),
"is_logged_in": user.Client.IsLoggedIn,
}
2020-02-09 19:32:14 +01:00
}
resp := map[string]interface{}{
"mxid": user.MXID,
"admin": user.Admin,
"whitelisted": user.Whitelisted,
"relaybot_whitelisted": user.RelaybotWhitelisted,
"whatsapp": wa,
}
jsonResponse(w, http.StatusOK, resp)
}
func jsonResponse(w http.ResponseWriter, status int, response interface{}) {
w.Header().Add("Content-Type", "application/json")
w.WriteHeader(status)
_ = json.NewEncoder(w).Encode(response)
}
func (prov *ProvisioningAPI) Logout(w http.ResponseWriter, r *http.Request) {
user := r.Context().Value("user").(*User)
if user.Session == nil {
jsonResponse(w, http.StatusNotFound, Error{
Error: "You're not logged in",
ErrCode: "not logged in",
})
return
}
2020-11-19 19:08:27 +01:00
force := strings.ToLower(r.URL.Query().Get("force")) != "false"
2021-10-22 19:14:34 +02:00
if user.Client == nil {
if !force {
jsonResponse(w, http.StatusNotFound, Error{
Error: "You're not connected",
ErrCode: "not connected",
})
}
} else {
2021-10-22 19:14:34 +02:00
// TODO reimplement
//err := user.Client.Logout()
//if err != nil {
// user.log.Warnln("Error while logging out:", err)
// if !force {
// jsonResponse(w, http.StatusInternalServerError, Error{
// Error: fmt.Sprintf("Unknown error while logging out: %v", err),
// ErrCode: err.Error(),
// })
// return
// }
//}
user.DeleteConnection()
2020-02-09 19:32:14 +01:00
}
user.bridge.Metrics.TrackConnectionState(user.JID, false)
user.removeFromJIDMap(StateLoggedOut)
2021-10-22 19:14:34 +02:00
user.DeleteSession()
2020-02-09 19:32:14 +01:00
jsonResponse(w, http.StatusOK, Response{true, "Logged out successfully."})
}
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
Subprotocols: []string{"net.maunium.whatsapp.login"},
}
2020-02-09 19:32:14 +01:00
func (prov *ProvisioningAPI) Login(w http.ResponseWriter, r *http.Request) {
2021-10-22 19:14:34 +02:00
// TODO reimplement
//userID := r.URL.Query().Get("user_id")
//user := prov.bridge.GetUserByMXID(id.UserID(userID))
//
//c, err := upgrader.Upgrade(w, r, nil)
//if err != nil {
// prov.log.Errorln("Failed to upgrade connection to websocket:", err)
// return
//}
//defer c.Close()
//if !user.Connect(true) {
// user.log.Debugln("Connect() returned false, assuming error was logged elsewhere and canceling login.")
// _ = c.WriteJSON(Error{
// Error: "Failed to connect to WhatsApp",
// ErrCode: "connection error",
// })
// return
//}
//
//qrChan := make(chan string, 3)
//go func() {
// for code := range qrChan {
// if code == "stop" {
// return
// }
// _ = c.WriteJSON(map[string]interface{}{
// "code": code,
// })
// }
//}()
//
//go func() {
// // Read everything so SetCloseHandler() works
// for {
// _, _, err = c.ReadMessage()
// if err != nil {
// break
// }
// }
//}()
//ctx, cancel := context.WithCancel(context.Background())
//c.SetCloseHandler(func(code int, text string) error {
// user.log.Debugfln("Login websocket closed (%d), cancelling login", code)
// cancel()
// return nil
//})
//
//user.log.Debugln("Starting login via provisioning API")
//session, jid, err := user.Conn.Login(qrChan, ctx)
//qrChan <- "stop"
//if err != nil {
// var msg string
// if errors.Is(err, whatsapp.ErrAlreadyLoggedIn) {
// msg = "You're already logged in"
// } else if errors.Is(err, whatsapp.ErrLoginInProgress) {
// msg = "You have a login in progress already."
// } else if errors.Is(err, whatsapp.ErrLoginTimedOut) {
// msg = "QR code scan timed out. Please try again."
// } else if errors.Is(err, whatsapp.ErrInvalidWebsocket) {
// msg = "WhatsApp connection error. Please try again."
// // TODO might need to make sure it reconnects?
// } else if errors.Is(err, whatsapp.ErrMultiDeviceNotSupported) {
// msg = "WhatsApp multi-device is not currently supported. Please disable it and try again."
// } else {
// msg = fmt.Sprintf("Unknown error while logging in: %v", err)
// }
// user.log.Warnln("Failed to log in:", err)
// _ = c.WriteJSON(Error{
// Error: msg,
// ErrCode: err.Error(),
// })
// return
//}
//user.log.Debugln("Successful login as", jid, "via provisioning API")
//user.ConnectionErrors = 0
//user.JID = strings.Replace(jid, whatsapp.OldUserSuffix, whatsapp.NewUserSuffix, 1)
//user.addToJIDMap()
//user.SetSession(&session)
//_ = c.WriteJSON(map[string]interface{}{
// "success": true,
// "jid": user.JID,
//})
//user.PostLogin()
2020-02-09 19:32:14 +01:00
}