go/uidmap/hardcoded_uids.go - master - githubmirror/client

go/uidmap/hardcoded_uids.go
163 строки4 KB
chrisnojima-zoom
Version 670 - clean2 (#29122)
08 июн 2026, 19:31
08 июн 2026, 19:311943197
package uidmap

import (
	"bytes"
	"compress/zlib"
	"fmt"
	"io"
	"sort"
	"sync"

	"github.com/keybase/client/go/libkb"
	keybase1 "github.com/keybase/client/go/protocol/keybase1"
)

var (
	offsets   []uint32
	usernames []byte
	once      sync.Once
)

// Map of sorted location -> unsorted location, so we can binary search the usernames
var (
	usernameSortOrder []uint16
	usoOnce           sync.Once
)

func findInit() {
	once.Do(func() {
		initUsernameOffsets()
		initUsernames()
	})
}

func usernameFindInit() {
	usoOnce.Do(initUsernameSort)
}

func usernameBytesAtSortedIndex(i int) []byte {
	j := usernameSortOrder[i]
	o := offsets[j]
	return usernames[o : o+uint32(lengths[j])]
}

func usernameAtSortedIndex(i int) string {
	return string(usernameBytesAtSortedIndex(i))
}

func initUsernameSort() {
	findInit()
	usernameSortOrder = make([]uint16, len(lengths))
	for i := range len(lengths) {
		usernameSortOrder[i] = uint16(i) //nolint:gosec // G115: Hardcoded username list is small and bounded, safe to convert
	}
	sort.SliceStable(usernameSortOrder, func(i, j int) bool {
		return bytes.Compare(usernameBytesAtSortedIndex(i), usernameBytesAtSortedIndex(j)) < 0
	})
}

type bufWriter struct {
	b []byte
	n int
}

func (b *bufWriter) Write(p []byte) (int, error) {
	copy(b.b[b.n:], p)
	b.n += len(p)
	return len(p), nil
}

func initUsernameOffsets() {
	var offset uint32
	offsets = make([]uint32, len(lengths))
	for index, length := range lengths {
		offsets[index] = offset
		offset += uint32(length)
	}
}

func initUsernames() {
	//
	// The list of usernames is computed as such:
	//  1. Sort all exceptions by UID
	//  2. concatenate all username strings after sorting in 1
	//  3. Tolower the blob
	//  4. run through zlib
	//  5. Also, along with (2), record the length of all of the usernames, and insert into an array
	//  6. Compute running offsets by cumsum'ing the values in 5.
	//
	zip, err := zlib.NewReader(bytes.NewReader(usernamesCompressed[:]))
	if err != nil {
		panic(err)
	}
	usernames = make([]byte, usernamesLen)
	buf := bufWriter{usernames, 0}
	//nolint:gosec // G110: Decompressing hardcoded embedded data with known size; bomb protection via usernamesLen check
	n, err := io.Copy(&buf, zip)
	if err != nil {
		panic(err)
	}
	if n != usernamesLen {
		panic(fmt.Sprintf("bad expansion: wanted %d bytes but got %d", usernamesLen, n))
	}
	zip.Close()
}

func uidBytesAt(i int) []byte {
	start := i * keybase1.UID_LEN
	return uids[start : start+keybase1.UID_LEN]
}

func uidAt(i int) keybase1.UID {
	ret, _ := keybase1.UIDFromSlice(uidBytesAt(i))
	return ret
}

func findHardcodedUsername(u libkb.NormalizedUsername) keybase1.UID {
	usernameFindInit()
	l := len(lengths)
	searchFor := []byte(u.String())
	doCmp := func(i int) int {
		ret := bytes.Compare(searchFor, usernameBytesAtSortedIndex(i))
		return ret
	}
	n := sort.Search(l, func(i int) bool {
		return doCmp(i) <= 0
	})
	if n == l || doCmp(n) != 0 {
		return keybase1.UID("")
	}
	return uidAt(int(usernameSortOrder[n]))
}

// Returns empty name if not found
func findHardcoded(uid keybase1.UID) libkb.NormalizedUsername {
	findInit()
	searchFor := uid.ToBytes()
	if len(searchFor) != keybase1.UID_LEN {
		// wrong uid size
		return libkb.NormalizedUsername("")
	}
	l := len(uids) / keybase1.UID_LEN
	doCmp := func(i int) int {
		return bytes.Compare(searchFor, uidBytesAt(i))
	}
	n := sort.Search(l, func(i int) bool {
		return doCmp(i) <= 0
	})
	if n == l || doCmp(n) != 0 {
		return libkb.NormalizedUsername("")
	}
	offset := offsets[n]
	s := string(usernames[offset : offset+uint32(lengths[n])])
	return libkb.NewNormalizedUsername(s)
}

func checkUIDAgainstUsername(uid keybase1.UID, un libkb.NormalizedUsername) bool {
	found := findHardcoded(uid)
	if !found.IsNil() {
		return found.Eq(un)
	}
	computed := libkb.UsernameToUID(un.String())
	return computed.Equal(uid)
}