Dump state to disk

Author: tulir

appstate.go

@@ -9,11 +9,14 @@ package whatsmeow
 import (
 	"context"
 	"encoding/hex"
+	"encoding/json"
 	"errors"
 	"fmt"
+	"os"
 	"time"
 
 	"github.com/rs/zerolog"
+	"go.mau.fi/util/exerrors"
 	"go.mau.fi/util/exslices"
 	"go.mau.fi/util/ptr"
 
@@ -22,6 +25,7 @@ import (
 	"go.mau.fi/whatsmeow/proto/waE2E"
 	"go.mau.fi/whatsmeow/proto/waServerSync"
 	"go.mau.fi/whatsmeow/store"
+	"go.mau.fi/whatsmeow/store/sqlstore"
 	"go.mau.fi/whatsmeow/types"
 	"go.mau.fi/whatsmeow/types/events"
 )
@@ -39,6 +43,8 @@ func (cli *Client) FetchAppState(ctx context.Context, name appstate.WAPatchName,
 	return nil
 }
 
+var DumpAppStateToDisk = false
+
 func (cli *Client) fetchAppState(ctx context.Context, name appstate.WAPatchName, fullSync, onlyIfNotSynced bool) ([]any, error) {
 	if cli == nil {
 		return nil, ErrClientIsNil
@@ -79,6 +85,17 @@ func (cli *Client) fetchAppState(ctx context.Context, name appstate.WAPatchName,
 		} else if patches.Snapshot != nil && state != (appstate.HashState{}) {
 			return nil, fmt.Errorf("unexpected non-empty input state (v%d) for %s when applying snapshot", state.Version, name)
 		}
+		if DumpAppStateToDisk {
+			f := exerrors.Must(os.OpenFile(fmt.Sprintf("/tmp/appstate-%s-v%d.json", name, state.Version), os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0600))
+			_ = json.NewEncoder(f).Encode(map[string]any{
+				"name":     name,
+				"patches":  patches.Patches,
+				"snapshot": patches.Snapshot,
+				"keys":     exerrors.Must(cli.Store.AppStateKeys.GetAllAppStateSyncKeys(ctx)),
+				"lids":     exerrors.Must(cli.Store.LIDs.(*sqlstore.CachedLIDMap).GetAll(ctx)),
+			})
+			_ = f.Close()
+		}
 		wantSnapshot = false
 		hasMore = patches.HasMorePatches
 		state, err = cli.applyAppStatePatches(ctx, name, state, patches, fullSync, eventsToDispatchPtr)

store/sqlstore/lidmap.go

@@ -12,6 +12,7 @@ import (
 	"database/sql"
 	"errors"
 	"fmt"
+	"maps"
 	"slices"
 	"strings"
 	"sync"
@@ -59,6 +60,9 @@ const (
 func (s *CachedLIDMap) FillCache(ctx context.Context) error {
 	s.lidCacheLock.Lock()
 	defer s.lidCacheLock.Unlock()
+	if s.cacheFilled {
+		return nil
+	}
 	rows, err := s.db.Query(ctx, getAllLIDMappingsQuery)
 	if err != nil {
 		return err
@@ -68,6 +72,11 @@ func (s *CachedLIDMap) FillCache(ctx context.Context) error {
 	return err
 }
 
+func (s *CachedLIDMap) GetAll(ctx context.Context) (map[string]string, error) {
+	err := s.FillCache(ctx)
+	return maps.Clone(s.lidToPNCache), err
+}
+
 func (s *CachedLIDMap) scanManyLids(rows dbutil.Rows, fn func(lid, pn string)) error {
 	if fn == nil {
 		fn = func(lid, pn string) {}

store/sqlstore/store.go

@@ -442,7 +442,7 @@ const (
 			SET key_data=excluded.key_data, timestamp=excluded.timestamp, fingerprint=excluded.fingerprint
 			WHERE excluded.timestamp > whatsmeow_app_state_sync_keys.timestamp
 	`
-	getAllAppStateSyncKeysQuery     = `SELECT key_data, timestamp, fingerprint FROM whatsmeow_app_state_sync_keys WHERE jid=$1 ORDER BY timestamp DESC`
+	getAllAppStateSyncKeysQuery     = `SELECT key_id, key_data, timestamp, fingerprint FROM whatsmeow_app_state_sync_keys WHERE jid=$1 ORDER BY timestamp DESC`
 	getAppStateSyncKeyQuery         = `SELECT key_data, timestamp, fingerprint FROM whatsmeow_app_state_sync_keys WHERE jid=$1 AND key_id=$2`
 	getLatestAppStateSyncKeyIDQuery = `SELECT key_id FROM whatsmeow_app_state_sync_keys WHERE jid=$1 ORDER BY timestamp DESC LIMIT 1`
 )
@@ -460,7 +460,7 @@ func (s *SQLStore) GetAllAppStateSyncKeys(ctx context.Context) ([]*store.AppStat
 	var out []*store.AppStateSyncKey
 	for rows.Next() {
 		var item store.AppStateSyncKey
-		err = rows.Scan(&item.Data, &item.Timestamp, &item.Fingerprint)
+		err = rows.Scan(&item.KeyID, &item.Data, &item.Timestamp, &item.Fingerprint)
 		if err != nil {
 			return nil, err
 		}

store/store.go

@@ -52,6 +52,7 @@ type SenderKeyStore interface {
 }
 
 type AppStateSyncKey struct {
+	KeyID       []byte
 	Data        []byte
 	Fingerprint []byte
 	Timestamp   int64