summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/Rhymen/go-whatsapp/media.go
diff options
context:
space:
mode:
authorWim <wim@42.be>2022-01-31 00:27:37 +0100
committerWim <wim@42.be>2022-03-20 14:57:48 +0100
commite3cafeaf9292f67459ff1d186f68283bfaedf2ae (patch)
treeb69c39620aa91dba695b3b935c6651c0fb37ce75 /vendor/github.com/Rhymen/go-whatsapp/media.go
parente7b193788a56ee7cdb02a87a9db0ad6724ef66d5 (diff)
downloadmatterbridge-msglm-e3cafeaf9292f67459ff1d186f68283bfaedf2ae.tar.gz
matterbridge-msglm-e3cafeaf9292f67459ff1d186f68283bfaedf2ae.tar.bz2
matterbridge-msglm-e3cafeaf9292f67459ff1d186f68283bfaedf2ae.zip
Add dependencies/vendor (whatsapp)
Diffstat (limited to 'vendor/github.com/Rhymen/go-whatsapp/media.go')
-rw-r--r--vendor/github.com/Rhymen/go-whatsapp/media.go221
1 files changed, 0 insertions, 221 deletions
diff --git a/vendor/github.com/Rhymen/go-whatsapp/media.go b/vendor/github.com/Rhymen/go-whatsapp/media.go
deleted file mode 100644
index f72e7922..00000000
--- a/vendor/github.com/Rhymen/go-whatsapp/media.go
+++ /dev/null
@@ -1,221 +0,0 @@
-package whatsapp
-
-import (
- "bytes"
- "crypto/hmac"
- "crypto/rand"
- "crypto/sha256"
- "encoding/base64"
- "encoding/json"
- "fmt"
- "io"
- "io/ioutil"
- "net"
- "net/http"
- "net/url"
- "time"
-
- "github.com/Rhymen/go-whatsapp/crypto/cbc"
- "github.com/Rhymen/go-whatsapp/crypto/hkdf"
-)
-
-func Download(url string, mediaKey []byte, appInfo MediaType, fileLength int) ([]byte, error) {
- if url == "" {
- return nil, fmt.Errorf("no url present")
- }
- file, mac, err := downloadMedia(url)
- if err != nil {
- return nil, err
- }
- iv, cipherKey, macKey, _, err := getMediaKeys(mediaKey, appInfo)
- if err != nil {
- return nil, err
- }
- if err = validateMedia(iv, file, macKey, mac); err != nil {
- return nil, err
- }
- data, err := cbc.Decrypt(cipherKey, iv, file)
- if err != nil {
- return nil, err
- }
- if len(data) != fileLength {
- return nil, fmt.Errorf("file length does not match. Expected: %v, got: %v", fileLength, len(data))
- }
- return data, nil
-}
-
-func validateMedia(iv []byte, file []byte, macKey []byte, mac []byte) error {
- h := hmac.New(sha256.New, macKey)
- n, err := h.Write(append(iv, file...))
- if err != nil {
- return err
- }
- if n < 10 {
- return fmt.Errorf("hash to short")
- }
- if !hmac.Equal(h.Sum(nil)[:10], mac) {
- return fmt.Errorf("invalid media hmac")
- }
- return nil
-}
-
-func getMediaKeys(mediaKey []byte, appInfo MediaType) (iv, cipherKey, macKey, refKey []byte, err error) {
- mediaKeyExpanded, err := hkdf.Expand(mediaKey, 112, string(appInfo))
- if err != nil {
- return nil, nil, nil, nil, err
- }
- return mediaKeyExpanded[:16], mediaKeyExpanded[16:48], mediaKeyExpanded[48:80], mediaKeyExpanded[80:], nil
-}
-
-func downloadMedia(url string) (file []byte, mac []byte, err error) {
- resp, err := http.Get(url)
- if err != nil {
- return nil, nil, err
- }
- defer resp.Body.Close()
- if resp.StatusCode != http.StatusOK {
- return nil, nil, fmt.Errorf("download failed with status code %d", resp.StatusCode)
- }
- if resp.ContentLength <= 10 {
- return nil, nil, fmt.Errorf("file to short")
- }
- data, err := ioutil.ReadAll(resp.Body)
- if err != nil {
- return nil, nil, err
- }
-
- n := len(data)
- return data[:n-10], data[n-10 : n], nil
-}
-
-type MediaConn struct {
- Status int `json:"status"`
- MediaConn struct {
- Auth string `json:"auth"`
- TTL int `json:"ttl"`
- Hosts []struct {
- Hostname string `json:"hostname"`
- IPs []struct {
- IP4 net.IP `json:"ip4"`
- IP6 net.IP `json:"ip6"`
- } `json:"ips"`
- } `json:"hosts"`
- } `json:"media_conn"`
-}
-
-func (wac *Conn) queryMediaConn() (hostname, auth string, ttl int, err error) {
- queryReq := []interface{}{"query", "mediaConn"}
- ch, err := wac.writeJson(queryReq)
- if err != nil {
- return "", "", 0, err
- }
-
- var resp MediaConn
- select {
- case r := <-ch:
- if err = json.Unmarshal([]byte(r), &resp); err != nil {
- return "", "", 0, fmt.Errorf("error decoding query media conn response: %v", err)
- }
- case <-time.After(wac.msgTimeout):
- return "", "", 0, fmt.Errorf("query media conn timed out")
- }
-
- if resp.Status != http.StatusOK {
- return "", "", 0, fmt.Errorf("query media conn responded with %d", resp.Status)
- }
-
- for _, h := range resp.MediaConn.Hosts {
- if h.Hostname != "" {
- return h.Hostname, resp.MediaConn.Auth, resp.MediaConn.TTL, nil
- }
- }
-
- return "", "", 0, fmt.Errorf("query media conn responded with no host")
-}
-
-var mediaTypeMap = map[MediaType]string{
- MediaImage: "/mms/image",
- MediaVideo: "/mms/video",
- MediaDocument: "/mms/document",
- MediaAudio: "/mms/audio",
-}
-
-func (wac *Conn) Upload(reader io.Reader, appInfo MediaType) (downloadURL string, mediaKey []byte, fileEncSha256 []byte, fileSha256 []byte, fileLength uint64, err error) {
- data, err := ioutil.ReadAll(reader)
- if err != nil {
- return "", nil, nil, nil, 0, err
- }
-
- mediaKey = make([]byte, 32)
- rand.Read(mediaKey)
-
- iv, cipherKey, macKey, _, err := getMediaKeys(mediaKey, appInfo)
- if err != nil {
- return "", nil, nil, nil, 0, err
- }
-
- enc, err := cbc.Encrypt(cipherKey, iv, data)
- if err != nil {
- return "", nil, nil, nil, 0, err
- }
-
- fileLength = uint64(len(data))
-
- h := hmac.New(sha256.New, macKey)
- h.Write(append(iv, enc...))
- mac := h.Sum(nil)[:10]
-
- sha := sha256.New()
- sha.Write(data)
- fileSha256 = sha.Sum(nil)
-
- sha.Reset()
- sha.Write(append(enc, mac...))
- fileEncSha256 = sha.Sum(nil)
-
- hostname, auth, _, err := wac.queryMediaConn()
- if err != nil {
- return "", nil, nil, nil, 0, err
- }
-
- token := base64.URLEncoding.EncodeToString(fileEncSha256)
- q := url.Values{
- "auth": []string{auth},
- "token": []string{token},
- }
- path := mediaTypeMap[appInfo]
- uploadURL := url.URL{
- Scheme: "https",
- Host: hostname,
- Path: fmt.Sprintf("%s/%s", path, token),
- RawQuery: q.Encode(),
- }
-
- body := bytes.NewReader(append(enc, mac...))
-
- req, err := http.NewRequest(http.MethodPost, uploadURL.String(), body)
- if err != nil {
- return "", nil, nil, nil, 0, err
- }
-
- req.Header.Set("Origin", "https://web.whatsapp.com")
- req.Header.Set("Referer", "https://web.whatsapp.com/")
-
- client := &http.Client{}
- // Submit the request
- res, err := client.Do(req)
- if err != nil {
- return "", nil, nil, nil, 0, err
- }
-
- if res.StatusCode != http.StatusOK {
- return "", nil, nil, nil, 0, fmt.Errorf("upload failed with status code %d", res.StatusCode)
- }
-
- var jsonRes map[string]string
- if err := json.NewDecoder(res.Body).Decode(&jsonRes); err != nil {
- return "", nil, nil, nil, 0, err
- }
-
- return jsonRes["url"], mediaKey, fileEncSha256, fileSha256, fileLength, nil
-}