lunar-tear/server/internal/service/listbin.go

package service

import (
	"encoding/json"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"sync"
	"unicode/utf8"
)

// listBinEntry holds path (')' as segment separator) and size from list.bin; Size is 0 when not present.
type listBinEntry struct {
	Path string
	Size int64
	MD5  string
}

// listBinIndex caches object_id → entry for a revision.
type listBinIndex map[string]listBinEntry

type infoAlias struct {
	ToName     string
	ToRevision string
	MD5        string
}

type assetCandidate struct {
	Path        string
	Revision    string
	Source      string
	ExpectedMD5 string
}

type listBinLoad struct {
	done chan struct{}
	idx  listBinIndex
	ok   bool
}

type infoLoad struct {
	done chan struct{}
	m    map[string]infoAlias
}

var (
	listBinCache    = make(map[string]listBinIndex) // revision → index
	listBinInflight = make(map[string]*listBinLoad)
	listBinCacheMu  sync.RWMutex
	infoCache       = make(map[string]map[string]infoAlias) // revision → from-name → duplicate target
	infoInflight    = make(map[string]*infoLoad)
	infoCacheMu     sync.RWMutex
)

// infoJSONEntry is one entry from assets/revisions/{rev}/info.json (duplicate files: serve to-name when asked for from-name).
type infoJSONEntry struct {
	FromName   string `json:"from-name"`
	ToName     string `json:"to-name"`
	ToRevision *int   `json:"to-revision"`
	MD5        string `json:"md5"`
}

// readVarint reads a protobuf varint from b, returns value and number of bytes consumed.
func readVarint(b []byte) (value int, n int) {
	for i := 0; i < len(b) && i < 10; i++ {
		value |= int(b[i]&0x7f) << (7 * i)
		n = i + 1
		if b[i]&0x80 == 0 {
			return value, n
		}
	}
	return 0, 0
}

func skipProtoField(wireType int, data []byte, offset int) (int, bool) {
	switch wireType {
	case 0:
		_, n := readVarint(data[offset:])
		if n == 0 {
			return 0, false
		}
		return offset + n, true
	case 1:
		if offset+8 > len(data) {
			return 0, false
		}
		return offset + 8, true
	case 2:
		length, n := readVarint(data[offset:])
		if n == 0 || length < 0 || offset+n+length > len(data) {
			return 0, false
		}
		return offset + n + length, true
	case 5:
		if offset+4 > len(data) {
			return 0, false
		}
		return offset + 4, true
	default:
		return 0, false
	}
}

func parseListBinEntry(data []byte) (objectId string, entry listBinEntry, ok bool) {
	i := 0
	for i < len(data) {
		tag, n := readVarint(data[i:])
		if n == 0 {
			return "", listBinEntry{}, false
		}
		i += n
		fieldNum := tag >> 3
		wireType := tag & 0x7

		switch fieldNum {
		case 3: // path
			if wireType != 2 {
				return "", listBinEntry{}, false
			}
			length, vn := readVarint(data[i:])
			if vn == 0 || length < 0 || i+vn+length > len(data) {
				return "", listBinEntry{}, false
			}
			entry.Path = string(data[i+vn : i+vn+length])
			i += vn + length
		case 4: // size
			if wireType != 0 {
				return "", listBinEntry{}, false
			}
			size, vn := readVarint(data[i:])
			if vn == 0 {
				return "", listBinEntry{}, false
			}
			if size >= 256 {
				entry.Size = int64(size)
			}
			i += vn
		case 10: // md5
			if wireType != 2 {
				return "", listBinEntry{}, false
			}
			length, vn := readVarint(data[i:])
			if vn == 0 || length < 0 || i+vn+length > len(data) {
				return "", listBinEntry{}, false
			}
			entry.MD5 = string(data[i+vn : i+vn+length])
			i += vn + length
		case 11: // object_id
			if wireType != 2 {
				return "", listBinEntry{}, false
			}
			length, vn := readVarint(data[i:])
			if vn == 0 || length <= 0 || i+vn+length > len(data) {
				return "", listBinEntry{}, false
			}
			objectId = string(data[i+vn : i+vn+length])
			i += vn + length
		default:
			next, ok := skipProtoField(wireType, data, i)
			if !ok {
				return "", listBinEntry{}, false
			}
			i = next
		}
	}

	if objectId == "" || entry.Path == "" {
		return "", listBinEntry{}, false
	}
	return objectId, entry, true
}

// parseListBin reads list.bin and builds object_id (6-byte string) → entry (path, size, md5).
// The file is a protobuf message with repeated nested entry messages, so we parse each entry
// boundary first instead of doing a flat scan across the whole file.
func parseListBin(data []byte) listBinIndex {
	idx := make(listBinIndex)
	i := 0
	for i < len(data) {
		tag, n := readVarint(data[i:])
		if n == 0 {
			break
		}
		i += n
		wireType := tag & 0x7

		if wireType == 2 {
			length, vn := readVarint(data[i:])
			if vn == 0 || length < 0 || i+vn+length > len(data) {
				break
			}
			entryBytes := data[i+vn : i+vn+length]
			objectId, entry, ok := parseListBinEntry(entryBytes)
			if ok {
				idx[objectId] = entry
				i += vn + length
				continue
			}
		}

		next, ok := skipProtoField(wireType, data, i)
		if !ok {
			break
		}
		i = next
	}
	return idx
}

func loadListBinIndex(baseDir, revision string) (listBinIndex, bool) {
	listBinCacheMu.RLock()
	idx, ok := listBinCache[revision]
	listBinCacheMu.RUnlock()
	if ok {
		return idx, true
	}

	listBinCacheMu.Lock()
	if idx, ok := listBinCache[revision]; ok {
		listBinCacheMu.Unlock()
		return idx, true
	}
	if load := listBinInflight[revision]; load != nil {
		listBinCacheMu.Unlock()
		<-load.done
		return load.idx, load.ok
	}
	load := &listBinLoad{done: make(chan struct{})}
	listBinInflight[revision] = load
	listBinCacheMu.Unlock()

	filePath := filepath.Join(baseDir, "assets", "revisions", revision, "list.bin")
	data, err := os.ReadFile(filePath)
	if err != nil {
		listBinCacheMu.Lock()
		delete(listBinInflight, revision)
		close(load.done)
		listBinCacheMu.Unlock()
		return nil, false
	}
	idx = parseListBin(data)
	load.idx = idx
	load.ok = true
	listBinCacheMu.Lock()
	listBinCache[revision] = idx
	delete(listBinInflight, revision)
	close(load.done)
	listBinCacheMu.Unlock()
	return idx, true
}

func loadInfoIndex(baseDir, revision string) map[string]infoAlias {
	infoCacheMu.RLock()
	m, ok := infoCache[revision]
	infoCacheMu.RUnlock()
	if ok {
		return m
	}

	infoCacheMu.Lock()
	if m, ok := infoCache[revision]; ok {
		infoCacheMu.Unlock()
		return m
	}
	if load := infoInflight[revision]; load != nil {
		infoCacheMu.Unlock()
		<-load.done
		return load.m
	}
	load := &infoLoad{done: make(chan struct{})}
	infoInflight[revision] = load
	infoCacheMu.Unlock()

	filePath := filepath.Join(baseDir, "assets", "revisions", revision, "info.json")
	data, err := os.ReadFile(filePath)
	if err != nil {
		infoCacheMu.Lock()
		infoCache[revision] = nil
		delete(infoInflight, revision)
		close(load.done)
		infoCacheMu.Unlock()
		return nil
	}
	var entries []infoJSONEntry
	if err := json.Unmarshal(data, &entries); err != nil {
		infoCacheMu.Lock()
		infoCache[revision] = nil
		delete(infoInflight, revision)
		close(load.done)
		infoCacheMu.Unlock()
		return nil
	}
	m = make(map[string]infoAlias)
	for _, e := range entries {
		if e.FromName != "" && e.ToName != "" {
			aliasRevision := revision
			if e.ToRevision != nil {
				aliasRevision = strconv.Itoa(*e.ToRevision)
			}
			m[e.FromName] = infoAlias{
				ToName:     e.ToName,
				ToRevision: aliasRevision,
				MD5:        e.MD5,
			}
		}
	}
	load.m = m
	infoCacheMu.Lock()
	infoCache[revision] = m
	delete(infoInflight, revision)
	close(load.done)
	infoCacheMu.Unlock()
	return m
}

type pathCandidate struct {
	Path             string
	IsLocaleFallback bool
}

// utf8ToMojibake re-encodes non-ASCII runes as if each of their UTF-8 bytes
// were a Latin-1 codepoint. This matches filenames extracted by tools that
// misinterpret UTF-8 paths as Latin-1 (double-encoding). For example,
// U+FF12 (fullwidth ２, bytes EF BC 92) becomes U+00EF U+00BC U+0092
// (bytes C3 AF C2 BC C2 92).
func utf8ToMojibake(s string) string {
	var b strings.Builder
	changed := false
	for _, r := range s {
		if r >= 0x80 {
			var buf [4]byte
			n := utf8.EncodeRune(buf[:], r)
			for i := 0; i < n; i++ {
				b.WriteRune(rune(buf[i]))
			}
			changed = true
		} else {
			b.WriteRune(r)
		}
	}
	if !changed {
		return s
	}
	return b.String()
}

// normalizeFullwidth replaces fullwidth Unicode characters (U+FF01–U+FF5E)
// with their ASCII equivalents (U+0021–U+007E).
func normalizeFullwidth(s string) string {
	var b strings.Builder
	changed := false
	for _, r := range s {
		if r >= 0xFF01 && r <= 0xFF5E {
			b.WriteByte(byte(r - 0xFF01 + 0x21))
			changed = true
		} else {
			b.WriteRune(r)
		}
	}
	if !changed {
		return s
	}
	return b.String()
}

func hasNonASCII(s string) bool {
	for _, r := range s {
		if r >= 0x80 {
			return true
		}
	}
	return false
}

// pathStrToFullPaths converts a list.bin path string (using ')' separators) into filesystem
// candidates. The original locale path is returned first; if the path contains ja or ko,
// an en locale fallback is appended (marked IsLocaleFallback so callers can skip MD5 validation).
// For paths with non-ASCII characters, mojibake (double-encoded) and fullwidth-to-ASCII
// variants are also tried.
func pathStrToFullPaths(baseDir, revision, assetType, pathStr string) []pathCandidate {
	fsPath := strings.ReplaceAll(pathStr, ")", "/")
	if strings.Contains(fsPath, "..") || filepath.IsAbs(fsPath) || strings.HasPrefix(fsPath, "/") {
		return nil
	}
	fsPath = filepath.Clean(fsPath)
	if strings.Contains(fsPath, "..") {
		return nil
	}
	type tagged struct {
		pathStr  string
		fallback bool
	}
	entries := []tagged{{pathStr, false}}
	if hasNonASCII(pathStr) {
		entries = append(entries, tagged{utf8ToMojibake(pathStr), false})
		entries = append(entries, tagged{normalizeFullwidth(pathStr), false})
	}
	if strings.Contains(pathStr, ")ja)") {
		entries = append(entries, tagged{strings.ReplaceAll(pathStr, ")ja)", ")en)"), true})
	}
	if strings.Contains(pathStr, ")ko)") {
		entries = append(entries, tagged{strings.ReplaceAll(pathStr, ")ko)", ")en)"), true})
	}
	base := filepath.Join(baseDir, "assets", "revisions", revision)
	var out []pathCandidate
	seen := make(map[string]bool)
	for _, e := range entries {
		cleaned := filepath.Clean(strings.ReplaceAll(e.pathStr, ")", "/"))
		if seen[cleaned] {
			continue
		}
		seen[cleaned] = true
		var fullPath string
		switch assetType {
		case "assetbundle":
			fullPath = filepath.Join(base, "assetbundle", cleaned+".assetbundle")
		case "resources":
			fullPath = filepath.Join(base, "resources", cleaned)
		default:
			return nil
		}
		out = append(out, pathCandidate{Path: fullPath, IsLocaleFallback: e.fallback})
	}
	return out
}

func appendUniqueCandidate(candidates []assetCandidate, seen map[string]bool, candidate assetCandidate) []assetCandidate {
	key := candidate.Revision + ":" + candidate.Path
	if seen[key] {
		return candidates
	}
	seen[key] = true
	return append(candidates, candidate)
}

func duplicateCandidatePath(baseDir string, candidate assetCandidate, assetType, targetRevision, targetBaseName string) string {
	root := filepath.Join(baseDir, "assets", "revisions", candidate.Revision, assetType)
	rel, err := filepath.Rel(root, candidate.Path)
	if err != nil || strings.HasPrefix(rel, "..") || filepath.IsAbs(rel) {
		return ""
	}
	return filepath.Join(baseDir, "assets", "revisions", targetRevision, assetType, filepath.Dir(rel), targetBaseName)
}

// objectIdToFilePathCandidates returns candidate file paths for the object: list.bin path, locale fallbacks
// (ja/ko -> en), then info.json duplicate mappings (from-name -> to-name, possibly in a different revision).
// The original locale path is tried first (with MD5 validation); locale fallbacks are tried after
// (without MD5 validation, since the hash in list.bin refers to the original locale's content).
// Callers should try each path until one exists on disk.
func objectIdToFilePathCandidates(baseDir, revision, assetType, objectId string) (candidates []assetCandidate, size int64, ok bool) {
	idx, ok := loadListBinIndex(baseDir, revision)
	if !ok || idx == nil {
		return nil, 0, false
	}
	entry, ok := idx[objectId]
	if !ok || entry.Path == "" {
		return nil, 0, false
	}
	paths := pathStrToFullPaths(baseDir, revision, assetType, entry.Path)
	if len(paths) == 0 {
		return nil, 0, false
	}
	seen := make(map[string]bool)
	for _, pc := range paths {
		md5 := entry.MD5
		if pc.IsLocaleFallback {
			md5 = ""
		}
		candidates = appendUniqueCandidate(candidates, seen, assetCandidate{
			Path:        pc.Path,
			Revision:    revision,
			Source:      "list.bin",
			ExpectedMD5: md5,
		})
	}
	infoIndex := loadInfoIndex(baseDir, revision)
	if len(infoIndex) > 0 {
		for _, c := range candidates {
			alias, ok := infoIndex[filepath.Base(c.Path)]
			if !ok || alias.ToName == "" {
				continue
			}
			alt := duplicateCandidatePath(baseDir, c, assetType, alias.ToRevision, alias.ToName)
			if alt == "" {
				continue
			}
			candidates = appendUniqueCandidate(candidates, seen, assetCandidate{
				Path:        alt,
				Revision:    alias.ToRevision,
				Source:      "info.json redirect",
				ExpectedMD5: alias.MD5,
			})
		}
	}
	return candidates, entry.Size, true
}