slidge-whatsapp 0.2.2__cp313-cp313-manylinux_2_36_aarch64.whl → 0.2.5__cp313-cp313-manylinux_2_36_aarch64.whl

slidge_whatsapp/vendor/go.mau.fi/whatsmeow/socket/framesocket.go

@@ -0,0 +1,235 @@
+// Copyright (c) 2021 Tulir Asokan
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package socket
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"net/http"
+	"sync"
+	"time"
+
+	"github.com/gorilla/websocket"
+
+	waLog "go.mau.fi/whatsmeow/util/log"
+)
+
+type FrameSocket struct {
+	conn   *websocket.Conn
+	ctx    context.Context
+	cancel func()
+	log    waLog.Logger
+	lock   sync.Mutex
+
+	URL         string
+	HTTPHeaders http.Header
+
+	Frames       chan []byte
+	OnDisconnect func(remote bool)
+	WriteTimeout time.Duration
+
+	Header []byte
+	Dialer websocket.Dialer
+
+	incomingLength int
+	receivedLength int
+	incoming       []byte
+	partialHeader  []byte
+}
+
+func NewFrameSocket(log waLog.Logger, dialer websocket.Dialer) *FrameSocket {
+	return &FrameSocket{
+		conn:   nil,
+		log:    log,
+		Header: WAConnHeader,
+		Frames: make(chan []byte),
+
+		URL:         URL,
+		HTTPHeaders: http.Header{"Origin": {Origin}},
+
+		Dialer: dialer,
+	}
+}
+
+func (fs *FrameSocket) IsConnected() bool {
+	return fs.conn != nil
+}
+
+func (fs *FrameSocket) Context() context.Context {
+	return fs.ctx
+}
+
+func (fs *FrameSocket) Close(code int) {
+	fs.lock.Lock()
+	defer fs.lock.Unlock()
+
+	if fs.conn == nil {
+		return
+	}
+
+	if code > 0 {
+		message := websocket.FormatCloseMessage(code, "")
+		err := fs.conn.WriteControl(websocket.CloseMessage, message, time.Now().Add(time.Second))
+		if err != nil {
+			fs.log.Warnf("Error sending close message: %v", err)
+		}
+	}
+
+	fs.cancel()
+	err := fs.conn.Close()
+	if err != nil {
+		fs.log.Errorf("Error closing websocket: %v", err)
+	}
+	fs.conn = nil
+	fs.ctx = nil
+	fs.cancel = nil
+	if fs.OnDisconnect != nil {
+		go fs.OnDisconnect(code == 0)
+	}
+}
+
+func (fs *FrameSocket) Connect() error {
+	fs.lock.Lock()
+	defer fs.lock.Unlock()
+
+	if fs.conn != nil {
+		return ErrSocketAlreadyOpen
+	}
+	ctx, cancel := context.WithCancel(context.Background())
+
+	fs.log.Debugf("Dialing %s", fs.URL)
+	conn, _, err := fs.Dialer.Dial(fs.URL, fs.HTTPHeaders)
+	if err != nil {
+		cancel()
+		return fmt.Errorf("couldn't dial whatsapp web websocket: %w", err)
+	}
+
+	fs.ctx, fs.cancel = ctx, cancel
+	fs.conn = conn
+	conn.SetCloseHandler(func(code int, text string) error {
+		fs.log.Debugf("Server closed websocket with status %d/%s", code, text)
+		cancel()
+		// from default CloseHandler
+		message := websocket.FormatCloseMessage(code, "")
+		_ = conn.WriteControl(websocket.CloseMessage, message, time.Now().Add(time.Second))
+		return nil
+	})
+
+	go fs.readPump(conn, ctx)
+	return nil
+}
+
+func (fs *FrameSocket) SendFrame(data []byte) error {
+	conn := fs.conn
+	if conn == nil {
+		return ErrSocketClosed
+	}
+	dataLength := len(data)
+	if dataLength >= FrameMaxSize {
+		return fmt.Errorf("%w (got %d bytes, max %d bytes)", ErrFrameTooLarge, len(data), FrameMaxSize)
+	}
+
+	headerLength := len(fs.Header)
+	// Whole frame is header + 3 bytes for length + data
+	wholeFrame := make([]byte, headerLength+FrameLengthSize+dataLength)
+
+	// Copy the header if it's there
+	if fs.Header != nil {
+		copy(wholeFrame[:headerLength], fs.Header)
+		// We only want to send the header once
+		fs.Header = nil
+	}
+
+	// Encode length of frame
+	wholeFrame[headerLength] = byte(dataLength >> 16)
+	wholeFrame[headerLength+1] = byte(dataLength >> 8)
+	wholeFrame[headerLength+2] = byte(dataLength)
+
+	// Copy actual frame data
+	copy(wholeFrame[headerLength+FrameLengthSize:], data)
+
+	if fs.WriteTimeout > 0 {
+		err := conn.SetWriteDeadline(time.Now().Add(fs.WriteTimeout))
+		if err != nil {
+			fs.log.Warnf("Failed to set write deadline: %v", err)
+		}
+	}
+	return conn.WriteMessage(websocket.BinaryMessage, wholeFrame)
+}
+
+func (fs *FrameSocket) frameComplete() {
+	data := fs.incoming
+	fs.incoming = nil
+	fs.partialHeader = nil
+	fs.incomingLength = 0
+	fs.receivedLength = 0
+	fs.Frames <- data
+}
+
+func (fs *FrameSocket) processData(msg []byte) {
+	for len(msg) > 0 {
+		// This probably doesn't happen a lot (if at all), so the code is unoptimized
+		if fs.partialHeader != nil {
+			msg = append(fs.partialHeader, msg...)
+			fs.partialHeader = nil
+		}
+		if fs.incoming == nil {
+			if len(msg) >= FrameLengthSize {
+				length := (int(msg[0]) << 16) + (int(msg[1]) << 8) + int(msg[2])
+				fs.incomingLength = length
+				fs.receivedLength = len(msg)
+				msg = msg[FrameLengthSize:]
+				if len(msg) >= length {
+					fs.incoming = msg[:length]
+					msg = msg[length:]
+					fs.frameComplete()
+				} else {
+					fs.incoming = make([]byte, length)
+					copy(fs.incoming, msg)
+					msg = nil
+				}
+			} else {
+				fs.log.Warnf("Received partial header (report if this happens often)")
+				fs.partialHeader = msg
+				msg = nil
+			}
+		} else {
+			if fs.receivedLength+len(msg) >= fs.incomingLength {
+				copy(fs.incoming[fs.receivedLength:], msg[:fs.incomingLength-fs.receivedLength])
+				msg = msg[fs.incomingLength-fs.receivedLength:]
+				fs.frameComplete()
+			} else {
+				copy(fs.incoming[fs.receivedLength:], msg)
+				fs.receivedLength += len(msg)
+				msg = nil
+			}
+		}
+	}
+}
+
+func (fs *FrameSocket) readPump(conn *websocket.Conn, ctx context.Context) {
+	fs.log.Debugf("Frame websocket read pump starting %p", fs)
+	defer func() {
+		fs.log.Debugf("Frame websocket read pump exiting %p", fs)
+		go fs.Close(0)
+	}()
+	for {
+		msgType, data, err := conn.ReadMessage()
+		if err != nil {
+			// Ignore the error if the context has been closed
+			if !errors.Is(ctx.Err(), context.Canceled) {
+				fs.log.Errorf("Error reading from websocket: %v", err)
+			}
+			return
+		} else if msgType != websocket.BinaryMessage {
+			fs.log.Warnf("Got unexpected websocket message type %d", msgType)
+			continue
+		}
+		fs.processData(data)
+	}
+}

slidge_whatsapp/vendor/go.mau.fi/whatsmeow/socket/noisehandshake.go

@@ -0,0 +1,124 @@
+// Copyright (c) 2021 Tulir Asokan
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package socket
+
+import (
+	"crypto/cipher"
+	"crypto/sha256"
+	"fmt"
+	"io"
+	"sync/atomic"
+
+	"golang.org/x/crypto/curve25519"
+	"golang.org/x/crypto/hkdf"
+
+	"go.mau.fi/whatsmeow/util/gcmutil"
+)
+
+type NoiseHandshake struct {
+	hash    []byte
+	salt    []byte
+	key     cipher.AEAD
+	counter uint32
+}
+
+func NewNoiseHandshake() *NoiseHandshake {
+	return &NoiseHandshake{}
+}
+
+func sha256Slice(data []byte) []byte {
+	hash := sha256.Sum256(data)
+	return hash[:]
+}
+
+func (nh *NoiseHandshake) Start(pattern string, header []byte) {
+	data := []byte(pattern)
+	if len(data) == 32 {
+		nh.hash = data
+	} else {
+		nh.hash = sha256Slice(data)
+	}
+	nh.salt = nh.hash
+	var err error
+	nh.key, err = gcmutil.Prepare(nh.hash)
+	if err != nil {
+		panic(err)
+	}
+	nh.Authenticate(header)
+}
+
+func (nh *NoiseHandshake) Authenticate(data []byte) {
+	nh.hash = sha256Slice(append(nh.hash, data...))
+}
+
+func (nh *NoiseHandshake) postIncrementCounter() uint32 {
+	count := atomic.AddUint32(&nh.counter, 1)
+	return count - 1
+}
+
+func (nh *NoiseHandshake) Encrypt(plaintext []byte) []byte {
+	ciphertext := nh.key.Seal(nil, generateIV(nh.postIncrementCounter()), plaintext, nh.hash)
+	nh.Authenticate(ciphertext)
+	return ciphertext
+}
+
+func (nh *NoiseHandshake) Decrypt(ciphertext []byte) (plaintext []byte, err error) {
+	plaintext, err = nh.key.Open(nil, generateIV(nh.postIncrementCounter()), ciphertext, nh.hash)
+	if err == nil {
+		nh.Authenticate(ciphertext)
+	}
+	return
+}
+
+func (nh *NoiseHandshake) Finish(fs *FrameSocket, frameHandler FrameHandler, disconnectHandler DisconnectHandler) (*NoiseSocket, error) {
+	if write, read, err := nh.extractAndExpand(nh.salt, nil); err != nil {
+		return nil, fmt.Errorf("failed to extract final keys: %w", err)
+	} else if writeKey, err := gcmutil.Prepare(write); err != nil {
+		return nil, fmt.Errorf("failed to create final write cipher: %w", err)
+	} else if readKey, err := gcmutil.Prepare(read); err != nil {
+		return nil, fmt.Errorf("failed to create final read cipher: %w", err)
+	} else if ns, err := newNoiseSocket(fs, writeKey, readKey, frameHandler, disconnectHandler); err != nil {
+		return nil, fmt.Errorf("failed to create noise socket: %w", err)
+	} else {
+		return ns, nil
+	}
+}
+
+func (nh *NoiseHandshake) MixSharedSecretIntoKey(priv, pub [32]byte) error {
+	secret, err := curve25519.X25519(priv[:], pub[:])
+	if err != nil {
+		return fmt.Errorf("failed to do x25519 scalar multiplication: %w", err)
+	}
+	return nh.MixIntoKey(secret)
+}
+
+func (nh *NoiseHandshake) MixIntoKey(data []byte) error {
+	nh.counter = 0
+	write, read, err := nh.extractAndExpand(nh.salt, data)
+	if err != nil {
+		return fmt.Errorf("failed to extract keys for mixing: %w", err)
+	}
+	nh.salt = write
+	nh.key, err = gcmutil.Prepare(read)
+	if err != nil {
+		return fmt.Errorf("failed to create new cipher while mixing keys: %w", err)
+	}
+	return nil
+}
+
+func (nh *NoiseHandshake) extractAndExpand(salt, data []byte) (write []byte, read []byte, err error) {
+	h := hkdf.New(sha256.New, data, salt, nil)
+	write = make([]byte, 32)
+	read = make([]byte, 32)
+
+	if _, err = io.ReadFull(h, write); err != nil {
+		err = fmt.Errorf("failed to read write key: %w", err)
+	} else if _, err = io.ReadFull(h, read); err != nil {
+		err = fmt.Errorf("failed to read read key: %w", err)
+	}
+	return
+}

slidge_whatsapp/vendor/go.mau.fi/whatsmeow/socket/noisesocket.go

@@ -0,0 +1,108 @@
+// Copyright (c) 2021 Tulir Asokan
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package socket
+
+import (
+	"context"
+	"crypto/cipher"
+	"encoding/binary"
+	"sync"
+	"sync/atomic"
+
+	"github.com/gorilla/websocket"
+)
+
+type NoiseSocket struct {
+	fs           *FrameSocket
+	onFrame      FrameHandler
+	writeKey     cipher.AEAD
+	readKey      cipher.AEAD
+	writeCounter uint32
+	readCounter  uint32
+	writeLock    sync.Mutex
+	destroyed    atomic.Bool
+	stopConsumer chan struct{}
+}
+
+type DisconnectHandler func(socket *NoiseSocket, remote bool)
+type FrameHandler func([]byte)
+
+func newNoiseSocket(fs *FrameSocket, writeKey, readKey cipher.AEAD, frameHandler FrameHandler, disconnectHandler DisconnectHandler) (*NoiseSocket, error) {
+	ns := &NoiseSocket{
+		fs:           fs,
+		writeKey:     writeKey,
+		readKey:      readKey,
+		onFrame:      frameHandler,
+		stopConsumer: make(chan struct{}),
+	}
+	fs.OnDisconnect = func(remote bool) {
+		disconnectHandler(ns, remote)
+	}
+	go ns.consumeFrames(fs.ctx, fs.Frames)
+	return ns, nil
+}
+
+func (ns *NoiseSocket) consumeFrames(ctx context.Context, frames <-chan []byte) {
+	if ctx == nil {
+		// ctx being nil implies the connection already closed somehow
+		return
+	}
+	ctxDone := ctx.Done()
+	for {
+		select {
+		case frame := <-frames:
+			ns.receiveEncryptedFrame(frame)
+		case <-ctxDone:
+			return
+		case <-ns.stopConsumer:
+			return
+		}
+	}
+}
+
+func generateIV(count uint32) []byte {
+	iv := make([]byte, 12)
+	binary.BigEndian.PutUint32(iv[8:], count)
+	return iv
+}
+
+func (ns *NoiseSocket) Context() context.Context {
+	return ns.fs.Context()
+}
+
+func (ns *NoiseSocket) Stop(disconnect bool) {
+	if ns.destroyed.CompareAndSwap(false, true) {
+		close(ns.stopConsumer)
+		ns.fs.OnDisconnect = nil
+		if disconnect {
+			ns.fs.Close(websocket.CloseNormalClosure)
+		}
+	}
+}
+
+func (ns *NoiseSocket) SendFrame(plaintext []byte) error {
+	ns.writeLock.Lock()
+	ciphertext := ns.writeKey.Seal(nil, generateIV(ns.writeCounter), plaintext, nil)
+	ns.writeCounter++
+	err := ns.fs.SendFrame(ciphertext)
+	ns.writeLock.Unlock()
+	return err
+}
+
+func (ns *NoiseSocket) receiveEncryptedFrame(ciphertext []byte) {
+	count := atomic.AddUint32(&ns.readCounter, 1) - 1
+	plaintext, err := ns.readKey.Open(nil, generateIV(count), ciphertext, nil)
+	if err != nil {
+		ns.fs.log.Warnf("Failed to decrypt frame: %v", err)
+		return
+	}
+	ns.onFrame(plaintext)
+}
+
+func (ns *NoiseSocket) IsConnected() bool {
+	return ns.fs.IsConnected()
+}

slidge_whatsapp/vendor/go.mau.fi/whatsmeow/store/clientpayload.go

@@ -0,0 +1,185 @@
+// Copyright (c) 2021 Tulir Asokan
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package store
+
+import (
+	"crypto/md5"
+	"encoding/binary"
+	"fmt"
+	"strconv"
+	"strings"
+
+	"google.golang.org/protobuf/proto"
+
+	"go.mau.fi/libsignal/ecc"
+
+	"go.mau.fi/whatsmeow/proto/waCompanionReg"
+	"go.mau.fi/whatsmeow/proto/waWa6"
+	"go.mau.fi/whatsmeow/types"
+)
+
+// WAVersionContainer is a container for a WhatsApp web version number.
+type WAVersionContainer [3]uint32
+
+// ParseVersion parses a version string (three dot-separated numbers) into a WAVersionContainer.
+func ParseVersion(version string) (parsed WAVersionContainer, err error) {
+	var part1, part2, part3 int
+	if parts := strings.Split(version, "."); len(parts) != 3 {
+		err = fmt.Errorf("'%s' doesn't contain three dot-separated parts", version)
+	} else if part1, err = strconv.Atoi(parts[0]); err != nil {
+		err = fmt.Errorf("first part of '%s' is not a number: %w", version, err)
+	} else if part2, err = strconv.Atoi(parts[1]); err != nil {
+		err = fmt.Errorf("second part of '%s' is not a number: %w", version, err)
+	} else if part3, err = strconv.Atoi(parts[2]); err != nil {
+		err = fmt.Errorf("third part of '%s' is not a number: %w", version, err)
+	} else {
+		parsed = WAVersionContainer{uint32(part1), uint32(part2), uint32(part3)}
+	}
+	return
+}
+
+func (vc WAVersionContainer) LessThan(other WAVersionContainer) bool {
+	return vc[0] < other[0] ||
+		(vc[0] == other[0] && vc[1] < other[1]) ||
+		(vc[0] == other[0] && vc[1] == other[1] && vc[2] < other[2])
+}
+
+// IsZero returns true if the version is zero.
+func (vc WAVersionContainer) IsZero() bool {
+	return vc == [3]uint32{0, 0, 0}
+}
+
+// String returns the version number as a dot-separated string.
+func (vc WAVersionContainer) String() string {
+	parts := make([]string, len(vc))
+	for i, part := range vc {
+		parts[i] = strconv.Itoa(int(part))
+	}
+	return strings.Join(parts, ".")
+}
+
+// Hash returns the md5 hash of the String representation of this version.
+func (vc WAVersionContainer) Hash() [16]byte {
+	return md5.Sum([]byte(vc.String()))
+}
+
+func (vc WAVersionContainer) ProtoAppVersion() *waWa6.ClientPayload_UserAgent_AppVersion {
+	return &waWa6.ClientPayload_UserAgent_AppVersion{
+		Primary:   &vc[0],
+		Secondary: &vc[1],
+		Tertiary:  &vc[2],
+	}
+}
+
+// waVersion is the WhatsApp web client version
+var waVersion = WAVersionContainer{2, 3000, 1020322842}
+
+// waVersionHash is the md5 hash of a dot-separated waVersion
+var waVersionHash [16]byte
+
+func init() {
+	waVersionHash = waVersion.Hash()
+}
+
+// GetWAVersion gets the current WhatsApp web client version.
+func GetWAVersion() WAVersionContainer {
+	return waVersion
+}
+
+// SetWAVersion sets the current WhatsApp web client version.
+//
+// In general, you should keep the library up-to-date instead of using this,
+// as there may be code changes that are necessary too (like protobuf schema changes).
+func SetWAVersion(version WAVersionContainer) {
+	if version.IsZero() {
+		return
+	}
+	waVersion = version
+	waVersionHash = version.Hash()
+}
+
+var BaseClientPayload = &waWa6.ClientPayload{
+	UserAgent: &waWa6.ClientPayload_UserAgent{
+		Platform:       waWa6.ClientPayload_UserAgent_WEB.Enum(),
+		ReleaseChannel: waWa6.ClientPayload_UserAgent_RELEASE.Enum(),
+		AppVersion:     waVersion.ProtoAppVersion(),
+		Mcc:            proto.String("000"),
+		Mnc:            proto.String("000"),
+		OsVersion:      proto.String("0.1.0"),
+		Manufacturer:   proto.String(""),
+		Device:         proto.String("Desktop"),
+		OsBuildNumber:  proto.String("0.1.0"),
+
+		LocaleLanguageIso6391:       proto.String("en"),
+		LocaleCountryIso31661Alpha2: proto.String("en"),
+	},
+	WebInfo: &waWa6.ClientPayload_WebInfo{
+		WebSubPlatform: waWa6.ClientPayload_WebInfo_WEB_BROWSER.Enum(),
+	},
+	ConnectType:   waWa6.ClientPayload_WIFI_UNKNOWN.Enum(),
+	ConnectReason: waWa6.ClientPayload_USER_ACTIVATED.Enum(),
+}
+
+var DeviceProps = &waCompanionReg.DeviceProps{
+	Os: proto.String("whatsmeow"),
+	Version: &waCompanionReg.DeviceProps_AppVersion{
+		Primary:   proto.Uint32(0),
+		Secondary: proto.Uint32(1),
+		Tertiary:  proto.Uint32(0),
+	},
+	PlatformType:    waCompanionReg.DeviceProps_UNKNOWN.Enum(),
+	RequireFullSync: proto.Bool(false),
+}
+
+func SetOSInfo(name string, version [3]uint32) {
+	DeviceProps.Os = &name
+	DeviceProps.Version.Primary = &version[0]
+	DeviceProps.Version.Secondary = &version[1]
+	DeviceProps.Version.Tertiary = &version[2]
+	BaseClientPayload.UserAgent.OsVersion = proto.String(fmt.Sprintf("%d.%d.%d", version[0], version[1], version[2]))
+	BaseClientPayload.UserAgent.OsBuildNumber = BaseClientPayload.UserAgent.OsVersion
+}
+
+func (device *Device) getRegistrationPayload() *waWa6.ClientPayload {
+	payload := proto.Clone(BaseClientPayload).(*waWa6.ClientPayload)
+	regID := make([]byte, 4)
+	binary.BigEndian.PutUint32(regID, device.RegistrationID)
+	preKeyID := make([]byte, 4)
+	binary.BigEndian.PutUint32(preKeyID, device.SignedPreKey.KeyID)
+	deviceProps, _ := proto.Marshal(DeviceProps)
+	payload.DevicePairingData = &waWa6.ClientPayload_DevicePairingRegistrationData{
+		ERegid:      regID,
+		EKeytype:    []byte{ecc.DjbType},
+		EIdent:      device.IdentityKey.Pub[:],
+		ESkeyID:     preKeyID[1:],
+		ESkeyVal:    device.SignedPreKey.Pub[:],
+		ESkeySig:    device.SignedPreKey.Signature[:],
+		BuildHash:   waVersionHash[:],
+		DeviceProps: deviceProps,
+	}
+	payload.Passive = proto.Bool(false)
+	return payload
+}
+
+func (device *Device) getLoginPayload() *waWa6.ClientPayload {
+	payload := proto.Clone(BaseClientPayload).(*waWa6.ClientPayload)
+	payload.Username = proto.Uint64(device.ID.UserInt())
+	payload.Device = proto.Uint32(uint32(device.ID.Device))
+	payload.Passive = proto.Bool(true)
+	return payload
+}
+
+func (device *Device) GetClientPayload() *waWa6.ClientPayload {
+	if device.ID != nil {
+		if *device.ID == types.EmptyJID {
+			panic(fmt.Errorf("GetClientPayload called with empty JID"))
+		}
+		return device.getLoginPayload()
+	} else {
+		return device.getRegistrationPayload()
+	}
+}