01 / GAME DIRECTORY
先从游戏目录
开始。
选择已安装的游戏文件夹。
我们会检查游戏程序和资源文件，再继续设置连接。
BrownDust II · 客户端配置
Login complete. You can return to the game.
`) -} - -type providerIdentity struct{ issuer, subject string } - -type providerFailure struct { - Provider string - Stage string - Reason string - HTTPStatus int - OAuthError string -} - -func (e *providerFailure) Error() string { - return fmt.Sprintf("provider=%s stage=%s reason=%s status=%d oauth_error=%s", e.Provider, e.Stage, e.Reason, e.HTTPStatus, e.OAuthError) -} - -func networkProviderFailure(ctx context.Context, provider, stage string, err error) error { - reason := "network_error" - if errors.Is(ctx.Err(), context.DeadlineExceeded) || errors.Is(err, context.DeadlineExceeded) { - reason = "timeout" - } else if errors.Is(ctx.Err(), context.Canceled) || errors.Is(err, context.Canceled) { - reason = "cancelled" - } else { - var networkError net.Error - if errors.As(err, &networkError) && networkError.Timeout() { - reason = "timeout" - } - } - return &providerFailure{Provider: provider, Stage: stage, Reason: reason} -} - -func rejectedProviderFailure(provider, stage string, response *http.Response) error { - failure := &providerFailure{Provider: provider, Stage: stage, Reason: "http_rejected", HTTPStatus: response.StatusCode} - var body struct { - Error string `json:"error"` - } - decoder := json.NewDecoder(io.LimitReader(response.Body, 8<<10)) - if decoder.Decode(&body) == nil { - failure.OAuthError = safeOAuthError(body.Error) - } - return failure -} - -func invalidProviderResponse(provider, stage string) error { - return &providerFailure{Provider: provider, Stage: stage, Reason: "invalid_response", HTTPStatus: http.StatusOK} -} - -func safeOAuthError(value string) string { - switch value { - case "invalid_request", "invalid_client", "invalid_grant", "unauthorized_client", - "unsupported_grant_type", "invalid_scope", "access_denied", "server_error", "temporarily_unavailable": - return value - default: - return "unknown" - } -} - -func (s *Service) exchangeIdentity(ctx context.Context, provider, code, verifier, nonce string) (providerIdentity, error) { - values := url.Values{"client_id": {s.config.Providers[provider].ClientID}, "client_secret": {s.config.ProviderSecrets[provider]}, "grant_type": {"authorization_code"}, "code": {code}, "redirect_uri": {s.redirectURL(provider)}, "code_verifier": {verifier}} - request, _ := http.NewRequestWithContext(ctx, http.MethodPost, providerTokenURL(provider), strings.NewReader(values.Encode())) - request.Header.Set("Content-Type", "application/x-www-form-urlencoded") - response, err := s.client.Do(request) - if err != nil { - return providerIdentity{}, networkProviderFailure(ctx, provider, "token_exchange", err) - } - defer func() { _ = response.Body.Close() }() - if response.StatusCode != http.StatusOK { - return providerIdentity{}, rejectedProviderFailure(provider, "token_exchange", response) - } - var token struct { - AccessToken string `json:"access_token"` - IDToken string `json:"id_token"` - } - if err := decodeProviderJSON(response.Body, &token); err != nil || token.AccessToken == "" { - return providerIdentity{}, invalidProviderResponse(provider, "token_exchange") - } - if provider == "google" { - if token.IDToken == "" { - return providerIdentity{}, invalidProviderResponse(provider, "token_exchange") - } - identity, err := s.verifyGoogleIDToken(ctx, token.IDToken, nonce) - token.AccessToken, token.IDToken = "", "" - return identity, err - } - userinfo, _ := http.NewRequestWithContext(ctx, http.MethodGet, providerUserURL(provider), nil) - userinfo.Header.Set("Authorization", "Bearer "+token.AccessToken) - response, err = s.client.Do(userinfo) - token.AccessToken = "" - if err != nil { - return providerIdentity{}, networkProviderFailure(ctx, provider, "userinfo", err) - } - defer func() { _ = response.Body.Close() }() - if response.StatusCode != http.StatusOK { - return providerIdentity{}, rejectedProviderFailure(provider, "userinfo", response) - } - var user struct { - ID string `json:"id"` - Sub string `json:"sub"` - } - if err := decodeProviderJSON(response.Body, &user); err != nil { - return providerIdentity{}, invalidProviderResponse(provider, "userinfo") - } - if provider == "discord" && user.ID != "" { - return providerIdentity{issuer: "https://discord.com", subject: user.ID}, nil - } - return providerIdentity{}, invalidProviderResponse(provider, "userinfo") -} - -// verifyGoogleIDToken delegates signature and standard-claim verification to -// Google's HTTPS tokeninfo endpoint, then independently verifies this server's -// audience, nonce and expiry. The raw ID token is never persisted or logged. -func (s *Service) verifyGoogleIDToken(ctx context.Context, idToken, nonce string) (providerIdentity, error) { - endpoint := "https://oauth2.googleapis.com/tokeninfo?id_token=" + url.QueryEscape(idToken) - request, _ := http.NewRequestWithContext(ctx, http.MethodGet, endpoint, nil) - response, err := s.client.Do(request) - if err != nil { - return providerIdentity{}, networkProviderFailure(ctx, "google", "id_token_verify", err) - } - defer func() { _ = response.Body.Close() }() - if response.StatusCode != http.StatusOK { - return providerIdentity{}, rejectedProviderFailure("google", "id_token_verify", response) - } - var claims struct { - Issuer string `json:"iss"` - Audience string `json:"aud"` - Subject string `json:"sub"` - Nonce string `json:"nonce"` - Expires string `json:"exp"` - } - if err := decodeProviderJSON(response.Body, &claims); err != nil { - return providerIdentity{}, invalidProviderResponse("google", "id_token_verify") - } - expires, err := strconv.ParseInt(claims.Expires, 10, 64) - validIssuer := claims.Issuer == "https://accounts.google.com" || claims.Issuer == "accounts.google.com" - if err != nil || !validIssuer || claims.Audience != s.config.Providers["google"].ClientID || claims.Subject == "" || claims.Nonce != nonce || s.store.now().Unix() >= expires { - return providerIdentity{}, &providerFailure{Provider: "google", Stage: "id_token_verify", Reason: "invalid_claims"} - } - return providerIdentity{issuer: "https://accounts.google.com", subject: claims.Subject}, nil -} - -func decodeProviderJSON(reader io.Reader, target any) error { - data, err := io.ReadAll(io.LimitReader(reader, 1<<20+1)) - if err != nil { - return err - } - if len(data) > 1<<20 { - return errors.New("auth: provider response is too large") - } - decoder := json.NewDecoder(bytes.NewReader(data)) - if err := decoder.Decode(target); err != nil { - return err - } - var trailing any - if err := decoder.Decode(&trailing); !errors.Is(err, io.EOF) { - return errors.New("auth: provider response has trailing JSON") - } - return nil -} - -func (s *Service) completeDevice(deviceID, provider string, identity providerIdentity) error { - if err := validateProviderIdentity(provider, identity); err != nil { - return err - } - now := s.store.now() - tx, err := s.store.db.Begin() - if err != nil { - return err - } - defer func() { _ = tx.Rollback() }() - var accountID, status string - subjectHash := s.store.identityDigest(identity.issuer, identity.subject) - err = tx.QueryRow(`SELECT i.account_id,a.status FROM identities i JOIN accounts a ON a.id=i.account_id WHERE i.issuer=? AND i.subject_hash=?`, identity.issuer, subjectHash).Scan(&accountID, &status) - if errors.Is(err, sql.ErrNoRows) { - var count int - if err := tx.QueryRow(`SELECT COUNT(*) FROM accounts`).Scan(&count); err != nil { - return err - } - if count != 0 { - result, err := tx.Exec(`UPDATE devices SET status='failed',error_code='not_allowed',state_hash=NULL,verifier_cipher=NULL,nonce_cipher=NULL WHERE id=? AND provider=? AND status='authorizing'`, deviceID, provider) - if err != nil { - return err - } - if count, err := rowsAffected(result); err != nil { - return err - } else if count != 1 { - return ErrConsumed - } - if err := tx.Commit(); err != nil { - return err - } - return ErrNotAllowed - } - accountID, err = randomToken(18) - if err != nil { - return err - } - if _, err = tx.Exec(`INSERT INTO accounts(id,status,created_at,last_login_at) VALUES(?,'active',?,?)`, accountID, now.Unix(), now.Unix()); err != nil { - return err - } - if _, err = tx.Exec(`INSERT INTO identities(provider,issuer,subject_hash,account_id,created_at,last_login_at) VALUES(?,?,?,?,?,?)`, provider, identity.issuer, subjectHash, accountID, now.Unix(), now.Unix()); err != nil { - return err - } - status = "active" - } else if err != nil { - return err - } else { - if _, err = tx.Exec(`UPDATE identities SET last_login_at=? WHERE issuer=? AND subject_hash=?`, now.Unix(), identity.issuer, subjectHash); err != nil { - return err - } - } - if status != "active" { - result, err := tx.Exec(`UPDATE devices SET status='failed',error_code='not_allowed',state_hash=NULL,verifier_cipher=NULL,nonce_cipher=NULL WHERE id=? AND provider=? AND status='authorizing'`, deviceID, provider) - if err != nil { - return err - } - if count, err := rowsAffected(result); err != nil { - return err - } else if count != 1 { - return ErrConsumed - } - if err := tx.Commit(); err != nil { - return err - } - return ErrNotAllowed - } - result, familyID, err := s.issueTokens(tx, accountID, provider, now) - if err != nil { - return err - } - _ = familyID - payload, err := json.Marshal(result) - if err != nil { - return err - } - sealed, err := s.store.seal(deviceID, "result", payload) - clear(payload) - if err != nil { - return err - } - update, err := tx.Exec(`UPDATE devices SET result_cipher=?,status='complete',state_hash=NULL,verifier_cipher=NULL,nonce_cipher=NULL WHERE id=? AND provider=? AND status='authorizing'`, sealed, deviceID, provider) - if err != nil { - return err - } - if count, err := rowsAffected(update); err != nil { - return err - } else if count != 1 { - return ErrConsumed - } - return tx.Commit() -} - -func validateProviderIdentity(provider string, identity providerIdentity) error { - switch provider { - case "discord": - if identity.issuer != "https://discord.com" || len(identity.subject) == 0 || len(identity.subject) > 32 { - return errors.New("auth: invalid Discord identity") - } - for _, digit := range identity.subject { - if digit < '0' || digit > '9' { - return errors.New("auth: invalid Discord identity") - } - } - case "google": - if identity.issuer != "https://accounts.google.com" || len(identity.subject) == 0 || len(identity.subject) > 255 { - return errors.New("auth: invalid Google identity") - } - default: - return errors.New("auth: unsupported identity provider") - } - return nil -} - -func (s *Service) issueTokens(tx *sql.Tx, accountID, provider string, now time.Time) (deviceResult, string, error) { - familyID, err := randomToken(18) - if err != nil { - return deviceResult{}, "", err - } - access, err := randomToken(32) - if err != nil { - return deviceResult{}, "", err - } - refresh, err := randomToken(32) - if err != nil { - return deviceResult{}, "", err - } - if _, err := tx.Exec(`INSERT INTO families(id,account_id,provider,created_at,expires_at) VALUES(?,?,?,?,?)`, familyID, accountID, provider, now.Unix(), now.Add(s.config.RefreshTTL).Unix()); err != nil { - return deviceResult{}, "", err - } - if _, err := tx.Exec(`INSERT INTO access_tokens(token_hash,family_id,account_id,created_at,expires_at) VALUES(?,?,?,?,?)`, s.store.digest("access-token", access), familyID, accountID, now.Unix(), now.Add(s.config.AccessTTL).Unix()); err != nil { - return deviceResult{}, "", err - } - if _, err := tx.Exec(`INSERT INTO refresh_tokens(token_hash,family_id,created_at,expires_at) VALUES(?,?,?,?)`, s.store.digest("refresh-token", refresh), familyID, now.Unix(), now.Add(s.config.RefreshTTL).Unix()); err != nil { - return deviceResult{}, "", err - } - return deviceResult{Provider: provider, AccessToken: access, AccessExpiresIn: int64(s.config.AccessTTL.Seconds()), RefreshToken: refresh, RefreshExpiresIn: int64(s.config.RefreshTTL.Seconds())}, familyID, nil -} - -func (s *Service) poll(w http.ResponseWriter, r *http.Request) { - id := r.PathValue("id") - if !s.limits.allow("poll:"+remoteIP(r.RemoteAddr)+":"+id, s.store.now(), time.Minute, 60) { - w.Header().Set("Retry-After", "2") - http.Error(w, "poll rate exceeded", http.StatusTooManyRequests) - return - } - authorization := r.Header.Get("Authorization") - if !strings.HasPrefix(authorization, "Device ") { - http.Error(w, "invalid device transaction", http.StatusForbidden) - return - } - secret := strings.TrimPrefix(authorization, "Device ") - tx, err := s.store.db.Begin() - if err != nil { - http.Error(w, "login result unavailable", http.StatusInternalServerError) - return - } - defer func() { _ = tx.Rollback() }() - var storedHash, sealed []byte - var status, errorCode string - var expires int64 - err = tx.QueryRow(`SELECT secret_hash,status,COALESCE(result_cipher,X''),COALESCE(error_code,''),expires_at FROM devices WHERE id=?`, id).Scan(&storedHash, &status, &sealed, &errorCode, &expires) - if err != nil || subtle.ConstantTimeCompare(storedHash, s.store.digest("device-secret", secret)) != 1 { - http.Error(w, "invalid device transaction", http.StatusForbidden) - return - } - if s.store.now().Unix() >= expires { - http.Error(w, "device transaction expired", http.StatusGone) - return - } - switch status { - case "created", "authorizing": - _ = tx.Rollback() - writeJSON(w, http.StatusAccepted, map[string]any{"status": "pending", "retry_after": 2}) - case "failed": - _ = tx.Rollback() - writeJSON(w, http.StatusForbidden, map[string]string{"status": "failed", "error": errorCode}) - case "complete": - plain, err := s.store.open(id, "result", sealed) - if err != nil { - http.Error(w, "login result unavailable", http.StatusInternalServerError) - return - } - result, err := tx.Exec(`UPDATE devices SET result_cipher=NULL,status='consumed' WHERE id=? AND status='complete'`, id) - count, affectedErr := rowsAffected(result) - if err != nil || affectedErr != nil || count != 1 { - _ = tx.Rollback() - clear(plain) - http.Error(w, "login result already consumed", http.StatusGone) - return - } - if err := tx.Commit(); err != nil { - clear(plain) - http.Error(w, "login result unavailable", http.StatusInternalServerError) - return - } - w.Header().Set("Content-Type", "application/json; charset=utf-8") - _, _ = w.Write(plain) - clear(plain) - default: - _ = tx.Rollback() - http.Error(w, "device transaction consumed", http.StatusGone) - } -} - -func (l *requestLimiter) allow(key string, now time.Time, duration time.Duration, maximum int) bool { - l.mu.Lock() - defer l.mu.Unlock() - if l.windows == nil { - l.windows = make(map[string]limitWindow) - } - if l.lastSweep.IsZero() || now.Sub(l.lastSweep) >= time.Minute { - for candidate, window := range l.windows { - if now.Sub(window.started) >= duration { - delete(l.windows, candidate) - } - } - l.lastSweep = now - } - window, exists := l.windows[key] - if !exists || now.Sub(window.started) >= duration { - if !exists && len(l.windows) >= 4096 { - return false - } - l.windows[key] = limitWindow{started: now, count: 1} - return true - } - if window.count >= maximum { - return false - } - window.count++ - l.windows[key] = window - return true -} - -func remoteIP(remoteAddr string) string { - host, _, err := net.SplitHostPort(remoteAddr) - if err == nil && host != "" { - return host - } - return remoteAddr -} - -func cleanupExpired(tx *sql.Tx, now int64) error { - statements := []struct { - query string - args []any - }{ - {`DELETE FROM devices WHERE expires_at<=?`, []any{now}}, - {`DELETE FROM access_tokens WHERE expires_at<=? OR family_id IN (SELECT id FROM families WHERE expires_at<=?)`, []any{now, now}}, - {`DELETE FROM refresh_attempts WHERE expires_at<=? OR family_id IN (SELECT id FROM families WHERE expires_at<=?)`, []any{now, now}}, - // Used refresh rows remain until their family expires so their reuse can - // still revoke every credential in that family. - {`DELETE FROM refresh_tokens WHERE family_id IN (SELECT id FROM families WHERE expires_at<=?)`, []any{now}}, - {`DELETE FROM families WHERE expires_at<=?`, []any{now}}, - } - for _, statement := range statements { - if _, err := tx.Exec(statement.query, statement.args...); err != nil { - return err - } - } - return nil -} - -func (s *Service) refresh(w http.ResponseWriter, r *http.Request) { - var request struct { - RefreshToken string `json:"refresh_token"` - AttemptID string `json:"attempt_id"` - } - if !decodeJSON(w, r, &request) { - return - } - if request.RefreshToken == "" || !validRefreshAttemptID(request.AttemptID) { - http.Error(w, "refresh_token and valid attempt_id required", http.StatusBadRequest) - return - } - now := s.store.now() - requestTokenHash := s.store.digest("refresh-token", request.RefreshToken) - attemptHash := s.store.digest("refresh-attempt", request.AttemptID) - tx, err := s.store.db.Begin() - if err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - defer func() { _ = tx.Rollback() }() - if err := cleanupExpired(tx, now.Unix()); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - var familyID, accountID, provider, accountStatus string - var tokenExpires, familyExpires int64 - var usedAt, revokedAt sql.NullInt64 - err = tx.QueryRow(`SELECT r.family_id,f.account_id,f.provider,a.status,r.expires_at,f.expires_at,r.used_at,COALESCE(r.revoked_at,f.revoked_at) FROM refresh_tokens r JOIN families f ON f.id=r.family_id JOIN accounts a ON a.id=f.account_id WHERE r.token_hash=?`, requestTokenHash).Scan(&familyID, &accountID, &provider, &accountStatus, &tokenExpires, &familyExpires, &usedAt, &revokedAt) - if err != nil || revokedAt.Valid || accountStatus != "active" || now.Unix() >= familyExpires { - w.Header().Set("X-BD2-Refresh-Invalid", "1") - http.Error(w, "refresh token invalid", http.StatusUnauthorized) - return - } - var savedRequestHash, resultCipher []byte - err = tx.QueryRow(`SELECT request_token_hash,result_cipher FROM refresh_attempts WHERE family_id=? AND attempt_hash=?`, familyID, attemptHash).Scan(&savedRequestHash, &resultCipher) - if err == nil { - if subtle.ConstantTimeCompare(savedRequestHash, requestTokenHash) != 1 { - w.Header().Set("X-BD2-Refresh-Invalid", "1") - http.Error(w, "refresh attempt_id already belongs to another request", http.StatusConflict) - return - } - plain, openErr := s.store.open(refreshAttemptSealID(familyID, attemptHash), "result", resultCipher) - if openErr != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - var saved refreshAttemptResult - decodeErr := json.Unmarshal(plain, &saved) - clear(plain) - if decodeErr != nil || saved.Provider == "" || saved.AccessToken == "" || saved.RefreshToken == "" { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - writeJSON(w, http.StatusOK, saved.deviceResult(now.Unix())) - return - } - if !errors.Is(err, sql.ErrNoRows) { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - if now.Unix() >= tokenExpires { - w.Header().Set("X-BD2-Refresh-Invalid", "1") - http.Error(w, "refresh token expired", http.StatusUnauthorized) - return - } - if usedAt.Valid { - if _, err := tx.Exec(`UPDATE families SET revoked_at=? WHERE id=? AND revoked_at IS NULL`, now.Unix(), familyID); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - if err := tx.Commit(); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - w.Header().Set("X-BD2-Refresh-Invalid", "1") - http.Error(w, "refresh token replayed", http.StatusUnauthorized) - return - } - newAccess, err := randomToken(32) - if err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - newRefresh, err := randomToken(32) - if err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - updated, err := tx.Exec(`UPDATE refresh_tokens SET used_at=? WHERE token_hash=? AND used_at IS NULL AND revoked_at IS NULL`, now.Unix(), requestTokenHash) - if err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - if count, err := rowsAffected(updated); err != nil || count != 1 { - w.Header().Set("X-BD2-Refresh-Invalid", "1") - http.Error(w, "refresh token invalid", http.StatusUnauthorized) - return - } - if _, err = tx.Exec(`DELETE FROM access_tokens WHERE family_id=?`, familyID); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - if _, err = tx.Exec(`INSERT INTO access_tokens(token_hash,family_id,account_id,created_at,expires_at) VALUES(?,?,?,?,?)`, s.store.digest("access-token", newAccess), familyID, accountID, now.Unix(), now.Add(s.config.AccessTTL).Unix()); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - refreshExpiry := min(familyExpires, now.Add(s.config.RefreshTTL).Unix()) - if _, err = tx.Exec(`INSERT INTO refresh_tokens(token_hash,family_id,created_at,expires_at) VALUES(?,?,?,?)`, s.store.digest("refresh-token", newRefresh), familyID, now.Unix(), refreshExpiry); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - result := refreshAttemptResult{ - Provider: provider, AccessToken: newAccess, AccessExpiresAt: now.Add(s.config.AccessTTL).Unix(), - RefreshToken: newRefresh, RefreshExpiresAt: refreshExpiry, - } - plain, err := json.Marshal(result) - if err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - resultCipher, err = s.store.seal(refreshAttemptSealID(familyID, attemptHash), "result", plain) - clear(plain) - if err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - if _, err = tx.Exec(`INSERT INTO refresh_attempts(family_id,attempt_hash,request_token_hash,result_cipher,created_at,expires_at) VALUES(?,?,?,?,?,?)`, familyID, attemptHash, requestTokenHash, resultCipher, now.Unix(), refreshExpiry); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - if err = tx.Commit(); err != nil { - http.Error(w, "refresh unavailable", http.StatusInternalServerError) - return - } - writeJSON(w, http.StatusOK, result.deviceResult(now.Unix())) -} - -func validRefreshAttemptID(value string) bool { - if len(value) < 16 || len(value) > 128 { - return false - } - for _, item := range value { - if item < 'a' || item > 'z' { - if item < 'A' || item > 'Z' { - if item < '0' || item > '9' { - if item != '-' && item != '_' { - return false - } - } - } - } - } - return true -} - -func refreshAttemptSealID(familyID string, attemptHash []byte) string { - return familyID + ":" + base64.RawURLEncoding.EncodeToString(attemptHash) -} - -func (r refreshAttemptResult) deviceResult(now int64) deviceResult { - accessTTL := max(r.AccessExpiresAt-now, 0) - refreshTTL := max(r.RefreshExpiresAt-now, 0) - return deviceResult{ - Provider: r.Provider, AccessToken: r.AccessToken, AccessExpiresIn: accessTTL, - RefreshToken: r.RefreshToken, RefreshExpiresIn: refreshTTL, - } -} - -func (s *Service) revoke(w http.ResponseWriter, r *http.Request) { - authorization := r.Header.Get("Authorization") - if !strings.HasPrefix(authorization, "Bearer ") { - http.Error(w, "access token required", http.StatusUnauthorized) - return - } - token := strings.TrimPrefix(authorization, "Bearer ") - if token == "" { - http.Error(w, "access token required", http.StatusUnauthorized) - return - } - now := s.store.now().Unix() - tx, err := s.store.db.Begin() - if err != nil { - http.Error(w, "revocation unavailable", http.StatusInternalServerError) - return - } - defer func() { _ = tx.Rollback() }() - var familyID, accountStatus string - var expires int64 - var revoked sql.NullInt64 - err = tx.QueryRow(`SELECT t.family_id,a.status,t.expires_at,COALESCE(t.revoked_at,f.revoked_at) - FROM access_tokens t JOIN families f ON f.id=t.family_id JOIN accounts a ON a.id=t.account_id - WHERE t.token_hash=?`, s.store.digest("access-token", token)).Scan(&familyID, &accountStatus, &expires, &revoked) - if err != nil || accountStatus != "active" || revoked.Valid || now >= expires { - http.Error(w, "access token invalid", http.StatusUnauthorized) - return - } - result, err := tx.Exec(`UPDATE families SET revoked_at=? WHERE id=? AND revoked_at IS NULL`, now, familyID) - if err != nil { - http.Error(w, "revocation unavailable", http.StatusInternalServerError) - return - } - if count, err := rowsAffected(result); err != nil || count != 1 { - http.Error(w, "access token invalid", http.StatusUnauthorized) - return - } - if err := tx.Commit(); err != nil { - http.Error(w, "revocation unavailable", http.StatusInternalServerError) - return - } - w.WriteHeader(http.StatusNoContent) -} - -func (s *Service) ValidateAccess(token string) (string, error) { - if token == "" { - return "", ErrUnauthorized - } - var accountID, status string - var expires int64 - var revoked sql.NullInt64 - err := s.store.db.QueryRow(`SELECT t.account_id,a.status,t.expires_at,COALESCE(t.revoked_at,f.revoked_at) FROM access_tokens t JOIN families f ON f.id=t.family_id JOIN accounts a ON a.id=t.account_id WHERE t.token_hash=?`, s.store.digest("access-token", token)).Scan(&accountID, &status, &expires, &revoked) - if err != nil || status != "active" || revoked.Valid || s.store.now().Unix() >= expires { - return "", ErrUnauthorized - } - return accountID, nil -} - -// AuthenticateLogin validates LoginUserRequest.access_token (field 2) before -// the game session is established. -func (s *Service) AuthenticateLogin(request []byte) (string, error) { - token, found, err := wire.Bytes(request, 2) - if err != nil || !found { - return "", ErrUnauthorized - } - return s.ValidateAccess(string(token)) -} - -func (s *Service) redirectURL(provider string) string { - return s.config.PublicURL + "/auth/" + provider + "/callback" -} -func providerScope(provider string) string { - if provider == "discord" { - return "identify" - } - return "openid" -} -func providerAuthorizeURL(provider string) string { - if provider == "discord" { - return "https://discord.com/oauth2/authorize" - } - return "https://accounts.google.com/o/oauth2/v2/auth" -} -func providerTokenURL(provider string) string { - if provider == "discord" { - return "https://discord.com/api/v10/oauth2/token" - } - return "https://oauth2.googleapis.com/token" -} -func providerUserURL(provider string) string { - return "https://discord.com/api/v10/users/@me" -} -func rowsAffected(result sql.Result) (int64, error) { - if result == nil { - return 0, errors.New("auth: missing SQL result") - } - value, err := result.RowsAffected() - if err != nil { - return 0, fmt.Errorf("auth: count affected rows: %w", err) - } - return value, nil -} diff --git a/go/internal/server/auth/service_test.go b/go/internal/server/auth/service_test.go deleted file mode 100644 index 91bea3a..0000000 --- a/go/internal/server/auth/service_test.go +++ /dev/null @@ -1,672 +0,0 @@ -package auth - -import ( - "bytes" - "context" - "encoding/json" - "errors" - "io" - "net/http" - "net/http/httptest" - "net/url" - "path/filepath" - "strconv" - "strings" - "testing" - "time" - - "bd2server/internal/server/authconfig" -) - -const testNowUnix = int64(1_800_000_000) - -func testService(t *testing.T) (*Service, *Store) { - t.Helper() - master := bytes.Repeat([]byte{0x42}, 32) - store, err := Open(filepath.Join(t.TempDir(), "auth.db"), master) - if err != nil { - t.Fatal(err) - } - t.Cleanup(func() { _ = store.Close() }) - store.now = func() time.Time { return time.Unix(testNowUnix, 0) } - public, _ := url.Parse("https://login.example.test") - runtime := authconfig.Runtime{ - Config: authconfig.Config{ - Mode: "oauth", - PublicURL: "https://login.example.test", - Providers: map[string]authconfig.ProviderConfig{ - "discord": {ClientID: "discord-client", ClientSecretEnv: "DISCORD_SECRET"}, - "google": {ClientID: "google-client", ClientSecretEnv: "GOOGLE_SECRET"}, - }, - }, - PublicURLParsed: public, - MasterKey: master, - ProviderSecrets: map[string]string{"discord": "discord-secret", "google": "google-secret"}, - AccessTTL: 15 * time.Minute, - RefreshTTL: 30 * 24 * time.Hour, - DeviceTTL: 10 * time.Minute, - } - service, err := New(runtime, store) - if err != nil { - t.Fatal(err) - } - if service.config.MasterKey != nil { - t.Fatal("service retained the authentication master key") - } - for i, value := range master { - if value != 0 { - t.Fatalf("master key byte %d was not cleared", i) - } - } - return service, store -} - -func insertAuthorizingDevice(t *testing.T, store *Store, id, provider string) { - t.Helper() - _, err := store.db.Exec(`INSERT INTO devices(id,client_hash,secret_hash,start_hash,provider,status,created_at,expires_at) - VALUES(?,?,?,?,?,'authorizing',?,?)`, id, store.digest("client-ip", "192.0.2.1"), store.digest("device-secret", "device-secret"), store.digest("start-ticket", "start-ticket"), provider, testNowUnix, testNowUnix+600) - if err != nil { - t.Fatal(err) - } -} - -func completeAndPoll(t *testing.T, service *Service, store *Store, deviceID, provider, issuer, subject string) deviceResult { - t.Helper() - insertAuthorizingDevice(t, store, deviceID, provider) - if err := service.completeDevice(deviceID, provider, providerIdentity{issuer: issuer, subject: subject}); err != nil { - t.Fatal(err) - } - request := httptest.NewRequest(http.MethodPost, "/auth/device/"+deviceID+"/poll", nil) - request.Header.Set("Authorization", "Device device-secret") - response := httptest.NewRecorder() - service.Handler().ServeHTTP(response, request) - if response.Code != http.StatusOK { - t.Fatalf("poll status=%d body=%q", response.Code, response.Body.String()) - } - var result deviceResult - if err := json.Unmarshal(response.Body.Bytes(), &result); err != nil { - t.Fatal(err) - } - return result -} - -func postJSON(handler http.Handler, path string, value any) *httptest.ResponseRecorder { - body, _ := json.Marshal(value) - request := httptest.NewRequest(http.MethodPost, path, bytes.NewReader(body)) - request.Header.Set("Content-Type", "application/json") - response := httptest.NewRecorder() - handler.ServeHTTP(response, request) - return response -} - -func TestCompleteDeviceRequiresAuthorizingTransition(t *testing.T) { - service, store := testService(t) - insertAuthorizingDevice(t, store, "already-consumed", "discord") - if _, err := store.db.Exec(`UPDATE devices SET status='consumed' WHERE id='already-consumed'`); err != nil { - t.Fatal(err) - } - err := service.completeDevice("already-consumed", "discord", providerIdentity{issuer: "https://discord.com", subject: "123456789"}) - if !errors.Is(err, ErrConsumed) { - t.Fatalf("completeDevice error=%v, want ErrConsumed", err) - } - for _, table := range []string{"accounts", "identities", "families", "access_tokens", "refresh_tokens"} { - var count int - if err := store.db.QueryRow(`SELECT COUNT(*) FROM ` + table).Scan(&count); err != nil { - t.Fatal(err) - } - if count != 0 { - t.Fatalf("%s has %d rows after rejected completion", table, count) - } - } -} - -func TestPollConsumesEncryptedResultExactlyOnce(t *testing.T) { - service, store := testService(t) - result := completeAndPoll(t, service, store, "poll-once", "discord", "https://discord.com", "123456789") - if result.AccessToken == "" || result.RefreshToken == "" { - t.Fatal("poll omitted issued tokens") - } - request := httptest.NewRequest(http.MethodPost, "/auth/device/poll-once/poll", nil) - request.Header.Set("Authorization", "Device device-secret") - response := httptest.NewRecorder() - service.Handler().ServeHTTP(response, request) - if response.Code != http.StatusGone { - t.Fatalf("second poll status=%d body=%q", response.Code, response.Body.String()) - } - var status string - var cipher []byte - if err := store.db.QueryRow(`SELECT status,COALESCE(result_cipher,X'') FROM devices WHERE id='poll-once'`).Scan(&status, &cipher); err != nil { - t.Fatal(err) - } - if status != "consumed" || len(cipher) != 0 { - t.Fatalf("device status=%q result bytes=%d", status, len(cipher)) - } -} - -func TestRefreshRotationReplayRevokesFamily(t *testing.T) { - service, store := testService(t) - first := completeAndPoll(t, service, store, "refresh-device", "discord", "https://discord.com", "123456789") - handler := service.Handler() - response := postJSON(handler, "/auth/session/refresh", map[string]string{"refresh_token": first.RefreshToken, "attempt_id": "rotation-attempt-0001"}) - if response.Code != http.StatusOK { - t.Fatalf("refresh status=%d body=%q", response.Code, response.Body.String()) - } - var rotated deviceResult - if err := json.Unmarshal(response.Body.Bytes(), &rotated); err != nil { - t.Fatal(err) - } - if rotated.RefreshToken == "" || rotated.RefreshToken == first.RefreshToken || rotated.AccessToken == first.AccessToken { - t.Fatal("refresh did not rotate both credentials") - } - if _, err := service.ValidateAccess(first.AccessToken); err == nil { - t.Fatal("old access token survived refresh rotation") - } - if _, err := service.ValidateAccess(rotated.AccessToken); err != nil { - t.Fatalf("new access token rejected: %v", err) - } - - replay := postJSON(handler, "/auth/session/refresh", map[string]string{"refresh_token": first.RefreshToken, "attempt_id": "replay-attempt-000002"}) - if replay.Code != http.StatusUnauthorized { - t.Fatalf("replay status=%d body=%q", replay.Code, replay.Body.String()) - } - if _, err := service.ValidateAccess(rotated.AccessToken); err == nil { - t.Fatal("refresh replay did not revoke the token family") - } - next := postJSON(handler, "/auth/session/refresh", map[string]string{"refresh_token": rotated.RefreshToken, "attempt_id": "after-replay-attempt-3"}) - if next.Code != http.StatusUnauthorized { - t.Fatalf("family refresh after replay status=%d", next.Code) - } -} - -func TestRevokeInvalidatesAccessAndRefreshFamily(t *testing.T) { - service, store := testService(t) - tokens := completeAndPoll(t, service, store, "revoke-device", "discord", "https://discord.com", "123456789") - request := httptest.NewRequest(http.MethodPost, "/auth/session/revoke", nil) - request.Header.Set("Authorization", "Bearer "+tokens.AccessToken) - response := httptest.NewRecorder() - service.Handler().ServeHTTP(response, request) - if response.Code != http.StatusNoContent { - t.Fatalf("revoke status=%d body=%q", response.Code, response.Body.String()) - } - if _, err := service.ValidateAccess(tokens.AccessToken); err == nil { - t.Fatal("revoked access token remained valid") - } - refresh := postJSON(service.Handler(), "/auth/session/refresh", map[string]string{"refresh_token": tokens.RefreshToken, "attempt_id": "revoked-attempt-0001"}) - if refresh.Code != http.StatusUnauthorized { - t.Fatalf("revoked refresh status=%d", refresh.Code) - } -} - -func TestRefreshRetryWithSameAttemptReturnsCommittedRotation(t *testing.T) { - service, store := testService(t) - first := completeAndPoll(t, service, store, "refresh-retry-device", "discord", "https://discord.com", "123456789") - handler := service.Handler() - request := map[string]string{"refresh_token": first.RefreshToken, "attempt_id": "stable-attempt-000001"} - response := postJSON(handler, "/auth/session/refresh", request) - if response.Code != http.StatusOK { - t.Fatalf("first refresh status=%d body=%q", response.Code, response.Body.String()) - } - var rotated deviceResult - if err := json.Unmarshal(response.Body.Bytes(), &rotated); err != nil { - t.Fatal(err) - } - retry := postJSON(handler, "/auth/session/refresh", request) - if retry.Code != http.StatusOK { - t.Fatalf("retry status=%d body=%q", retry.Code, retry.Body.String()) - } - var replayed deviceResult - if err := json.Unmarshal(retry.Body.Bytes(), &replayed); err != nil { - t.Fatal(err) - } - if replayed.Provider != rotated.Provider || replayed.AccessToken != rotated.AccessToken || replayed.RefreshToken != rotated.RefreshToken { - t.Fatalf("retry returned a different rotation: first=%+v retry=%+v", rotated, replayed) - } - if _, err := service.ValidateAccess(rotated.AccessToken); err != nil { - t.Fatalf("idempotent retry revoked the family: %v", err) - } - var sealed []byte - if err := store.db.QueryRow(`SELECT result_cipher FROM refresh_attempts`).Scan(&sealed); err != nil { - t.Fatal(err) - } - if bytes.Contains(sealed, []byte(rotated.AccessToken)) || bytes.Contains(sealed, []byte(rotated.RefreshToken)) { - t.Fatal("refresh attempt result was stored outside AES-GCM ciphertext") - } -} - -func TestRefreshRequiresStableAttemptID(t *testing.T) { - service, store := testService(t) - tokens := completeAndPoll(t, service, store, "refresh-attempt-required", "discord", "https://discord.com", "123456789") - response := postJSON(service.Handler(), "/auth/session/refresh", map[string]string{"refresh_token": tokens.RefreshToken}) - if response.Code != http.StatusBadRequest { - t.Fatalf("missing attempt status=%d body=%q", response.Code, response.Body.String()) - } - if _, err := service.ValidateAccess(tokens.AccessToken); err != nil { - t.Fatalf("malformed refresh request changed credential family: %v", err) - } -} - -func TestRefreshCommittedAttemptReplaysAfterRequestTokenExpiry(t *testing.T) { - service, store := testService(t) - first := completeAndPoll(t, service, store, "refresh-expiry-device", "discord", "https://discord.com", "123456789") - request := map[string]string{"refresh_token": first.RefreshToken, "attempt_id": "expiry-replay-attempt-01"} - response := postJSON(service.Handler(), "/auth/session/refresh", request) - if response.Code != http.StatusOK { - t.Fatalf("first refresh status=%d body=%q", response.Code, response.Body.String()) - } - var rotated deviceResult - if err := json.Unmarshal(response.Body.Bytes(), &rotated); err != nil { - t.Fatal(err) - } - if _, err := store.db.Exec(`UPDATE refresh_tokens SET expires_at=? WHERE token_hash=?`, testNowUnix-1, store.digest("refresh-token", first.RefreshToken)); err != nil { - t.Fatal(err) - } - store.now = func() time.Time { return time.Unix(testNowUnix+int64((16*time.Minute).Seconds()), 0) } - replay := postJSON(service.Handler(), "/auth/session/refresh", request) - if replay.Code != http.StatusOK { - t.Fatalf("expired request token replay status=%d body=%q", replay.Code, replay.Body.String()) - } - var replayed deviceResult - if err := json.Unmarshal(replay.Body.Bytes(), &replayed); err != nil { - t.Fatal(err) - } - if replayed.AccessToken != rotated.AccessToken || replayed.RefreshToken != rotated.RefreshToken { - t.Fatalf("replay changed committed rotation: first=%+v replay=%+v", rotated, replayed) - } - if replayed.AccessExpiresIn != 0 { - t.Fatalf("expired replay access TTL=%d, want 0", replayed.AccessExpiresIn) - } -} - -func TestSensitiveAuthenticationMaterialIsNotStoredInPlaintext(t *testing.T) { - service, store := testService(t) - handler := service.Handler() - created := postJSON(handler, "/auth/device", map[string]string{"provider": "discord"}) - if created.Code != http.StatusCreated { - t.Fatalf("create status=%d body=%q", created.Code, created.Body.String()) - } - var device struct { - ID string `json:"transaction_id"` - Secret string `json:"device_secret"` - StartURL string `json:"start_url"` - } - if err := json.Unmarshal(created.Body.Bytes(), &device); err != nil { - t.Fatal(err) - } - startURL, _ := url.Parse(device.StartURL) - ticket := startURL.Query().Get("ticket") - start := httptest.NewRequest(http.MethodGet, startURL.RequestURI(), nil) - started := httptest.NewRecorder() - handler.ServeHTTP(started, start) - if started.Code != http.StatusFound { - t.Fatalf("start status=%d body=%q", started.Code, started.Body.String()) - } - authorize, _ := url.Parse(started.Header().Get("Location")) - state := authorize.Query().Get("state") - var secretHash, startHash, stateHash, verifierCipher, nonceCipher []byte - if err := store.db.QueryRow(`SELECT secret_hash,start_hash,state_hash,verifier_cipher,nonce_cipher FROM devices WHERE id=?`, device.ID). - Scan(&secretHash, &startHash, &stateHash, &verifierCipher, &nonceCipher); err != nil { - t.Fatal(err) - } - verifier, err := store.open(device.ID, "pkce", verifierCipher) - if err != nil { - t.Fatal(err) - } - nonce, err := store.open(device.ID, "nonce", nonceCipher) - if err != nil { - t.Fatal(err) - } - for name, pair := range map[string]struct{ stored, raw []byte }{ - "device secret": {secretHash, []byte(device.Secret)}, - "start ticket": {startHash, []byte(ticket)}, - "oauth state": {stateHash, []byte(state)}, - "pkce verifier": {verifierCipher, verifier}, - "oidc nonce": {nonceCipher, nonce}, - } { - if bytes.Equal(pair.stored, pair.raw) || bytes.Contains(pair.stored, pair.raw) { - t.Fatalf("%s was stored in plaintext", name) - } - } - - const providerSubject = "987654321012345678" - if err := service.completeDevice(device.ID, "discord", providerIdentity{issuer: "https://discord.com", subject: providerSubject}); err != nil { - t.Fatal(err) - } - var subjectHash, sealedResult []byte - if err := store.db.QueryRow(`SELECT subject_hash FROM identities`).Scan(&subjectHash); err != nil { - t.Fatal(err) - } - if err := store.db.QueryRow(`SELECT result_cipher FROM devices WHERE id=?`, device.ID).Scan(&sealedResult); err != nil { - t.Fatal(err) - } - plainResult, err := store.open(device.ID, "result", sealedResult) - if err != nil { - t.Fatal(err) - } - var issued deviceResult - if err := json.Unmarshal(plainResult, &issued); err != nil { - t.Fatal(err) - } - if bytes.Contains(subjectHash, []byte(providerSubject)) { - t.Fatal("provider subject was stored in plaintext") - } - for name, raw := range map[string]string{"access token": issued.AccessToken, "refresh token": issued.RefreshToken} { - if bytes.Contains(sealedResult, []byte(raw)) { - t.Fatalf("pending %s was stored outside AES-GCM ciphertext", name) - } - var count int - table := "access_tokens" - if name == "refresh token" { - table = "refresh_tokens" - } - if err := store.db.QueryRow(`SELECT COUNT(*) FROM `+table+` WHERE token_hash=?`, []byte(raw)).Scan(&count); err != nil { - t.Fatal(err) - } - if count != 0 { - t.Fatalf("%s was stored in plaintext", name) - } - } -} - -func TestJSONLimitsAndSecurityHeaders(t *testing.T) { - service, _ := testService(t) - handler := service.Handler() - for name, body := range map[string]struct { - body string - want int - }{ - "trailing": {`{"provider":"discord"}{}`, http.StatusBadRequest}, - "oversize": {`{"provider":"discord","padding":"` + strings.Repeat("x", 17<<10) + `"}`, http.StatusRequestEntityTooLarge}, - } { - t.Run(name, func(t *testing.T) { - request := httptest.NewRequest(http.MethodPost, "/auth/device", strings.NewReader(body.body)) - response := httptest.NewRecorder() - handler.ServeHTTP(response, request) - if response.Code != body.want { - t.Fatalf("status=%d body=%q", response.Code, response.Body.String()) - } - for header, want := range map[string]string{ - "Cache-Control": "no-store", - "Referrer-Policy": "no-referrer", - "X-Content-Type-Options": "nosniff", - "X-Frame-Options": "DENY", - } { - if got := response.Header().Get(header); got != want { - t.Fatalf("%s=%q want %q", header, got, want) - } - } - }) - } -} - -func TestCreateDeviceLimitsPendingTransactionsPerClient(t *testing.T) { - service, store := testService(t) - handler := service.Handler() - for i := range 5 { - response := postJSON(handler, "/auth/device", map[string]string{"provider": "discord"}) - if response.Code != http.StatusCreated { - t.Fatalf("create %d status=%d body=%q", i, response.Code, response.Body.String()) - } - } - response := postJSON(handler, "/auth/device", map[string]string{"provider": "discord"}) - if response.Code != http.StatusTooManyRequests { - t.Fatalf("pending limit status=%d body=%q", response.Code, response.Body.String()) - } - var count int - if err := store.db.QueryRow(`SELECT COUNT(*) FROM devices`).Scan(&count); err != nil { - t.Fatal(err) - } - if count != 5 { - t.Fatalf("device count=%d want 5", count) - } -} - -func TestDecodeProviderJSONRejectsOversizeAndTrailingValues(t *testing.T) { - var target map[string]any - if err := decodeProviderJSON(strings.NewReader(`{"id":"1"}{}`), &target); err == nil { - t.Fatal("accepted provider response with trailing JSON") - } - oversize := `{"padding":"` + strings.Repeat("x", 1<<20) + `"}` - if err := decodeProviderJSON(strings.NewReader(oversize), &target); err == nil { - t.Fatal("accepted oversized provider response") - } -} - -func TestRequestLimiterIsBoundedAndExpiresWindows(t *testing.T) { - limiter := requestLimiter{windows: make(map[string]limitWindow)} - now := time.Unix(testNowUnix, 0) - for i := range 4096 { - if !limiter.allow(strconv.Itoa(i), now, time.Minute, 1) { - t.Fatalf("rejected window %d before capacity", i) - } - } - if limiter.allow("overflow", now, time.Minute, 1) { - t.Fatal("accepted a limiter key beyond its bounded capacity") - } - if !limiter.allow("after-expiry", now.Add(time.Minute), time.Minute, 1) { - t.Fatal("did not clean expired limiter windows") - } -} - -func TestCleanupRetainsUsedRefreshForReplayUntilFamilyExpiry(t *testing.T) { - service, store := testService(t) - tokens := completeAndPoll(t, service, store, "cleanup-device", "discord", "https://discord.com", "123456789") - if _, err := store.db.Exec(`UPDATE refresh_tokens SET used_at=? WHERE token_hash=?`, testNowUnix, store.digest("refresh-token", tokens.RefreshToken)); err != nil { - t.Fatal(err) - } - tx, err := store.db.Begin() - if err != nil { - t.Fatal(err) - } - if err := cleanupExpired(tx, testNowUnix+int64((29*24*time.Hour).Seconds())); err != nil { - _ = tx.Rollback() - t.Fatal(err) - } - if err := tx.Commit(); err != nil { - t.Fatal(err) - } - var count int - if err := store.db.QueryRow(`SELECT COUNT(*) FROM refresh_tokens WHERE used_at IS NOT NULL`).Scan(&count); err != nil { - t.Fatal(err) - } - if count != 1 { - t.Fatal("used refresh token was removed before family expiry") - } - tx, err = store.db.Begin() - if err != nil { - t.Fatal(err) - } - if err := cleanupExpired(tx, testNowUnix+int64((31*24*time.Hour).Seconds())); err != nil { - _ = tx.Rollback() - t.Fatal(err) - } - if err := tx.Commit(); err != nil { - t.Fatal(err) - } - if err := store.db.QueryRow(`SELECT COUNT(*) FROM refresh_tokens`).Scan(&count); err != nil { - t.Fatal(err) - } - if count != 0 { - t.Fatal("expired family refresh token was not cleaned") - } -} - -type roundTripFunc func(*http.Request) (*http.Response, error) - -func (f roundTripFunc) RoundTrip(request *http.Request) (*http.Response, error) { return f(request) } - -func jsonResponse(status int, body string) *http.Response { - return &http.Response{StatusCode: status, Body: io.NopCloser(strings.NewReader(body)), Header: make(http.Header)} -} - -func TestProviderIdentityVerification(t *testing.T) { - service, _ := testService(t) - service.client = &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { - switch request.URL.Host + request.URL.Path { - case "discord.com/api/v10/oauth2/token": - return jsonResponse(http.StatusOK, `{"access_token":"provider-access"}`), nil - case "discord.com/api/v10/users/@me": - if request.Header.Get("Authorization") != "Bearer provider-access" { - t.Fatal("Discord bearer token missing") - } - return jsonResponse(http.StatusOK, `{"id":"123456789"}`), nil - case "oauth2.googleapis.com/token": - return jsonResponse(http.StatusOK, `{"access_token":"provider-access","id_token":"signed-id-token"}`), nil - case "oauth2.googleapis.com/tokeninfo": - return jsonResponse(http.StatusOK, `{"iss":"https://accounts.google.com","aud":"google-client","sub":"google-subject","nonce":"expected-nonce","exp":"1900000000"}`), nil - default: - t.Fatalf("unexpected provider request %s", request.URL) - return nil, nil - } - })} - discord, err := service.exchangeIdentity(context.Background(), "discord", "code", "verifier", "nonce") - if err != nil || discord.issuer != "https://discord.com" || discord.subject != "123456789" { - t.Fatalf("Discord identity=%+v err=%v", discord, err) - } - google, err := service.exchangeIdentity(context.Background(), "google", "code", "verifier", "expected-nonce") - if err != nil || google.issuer != "https://accounts.google.com" || google.subject != "google-subject" { - t.Fatalf("Google identity=%+v err=%v", google, err) - } - if _, err := service.exchangeIdentity(context.Background(), "google", "code", "verifier", "wrong-nonce"); err == nil { - t.Fatal("Google identity accepted the wrong OIDC nonce") - } -} - -func TestProviderFailureIsStructuredAndSanitized(t *testing.T) { - service, _ := testService(t) - secretDescription := "provider leaked secret sentinel" - service.client = &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { - return jsonResponse(http.StatusUnauthorized, `{"error":"invalid_client","error_description":"`+secretDescription+`"}`), nil - })} - _, err := service.exchangeIdentity(context.Background(), "discord", "code", "verifier", "nonce") - var failure *providerFailure - if !errors.As(err, &failure) { - t.Fatalf("error %T does not expose provider failure", err) - } - if failure.Provider != "discord" || failure.Stage != "token_exchange" || failure.Reason != "http_rejected" || - failure.HTTPStatus != http.StatusUnauthorized || failure.OAuthError != "invalid_client" { - t.Fatalf("failure=%+v", failure) - } - if strings.Contains(err.Error(), secretDescription) { - t.Fatal("provider error description leaked through diagnostic error") - } -} - -func TestProviderFailureRejectsUntrustedOAuthError(t *testing.T) { - service, _ := testService(t) - service.client = &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { - return jsonResponse(http.StatusBadRequest, `{"error":"access-token-sentinel"}`), nil - })} - _, err := service.exchangeIdentity(context.Background(), "discord", "code", "verifier", "nonce") - var failure *providerFailure - if !errors.As(err, &failure) || failure.OAuthError != "unknown" { - t.Fatalf("failure=%+v err=%v", failure, err) - } - if strings.Contains(err.Error(), "access-token-sentinel") { - t.Fatal("untrusted provider error leaked through diagnostic error") - } -} - -func TestGoogleProviderNetworkFailureDoesNotLeakIDTokenURL(t *testing.T) { - service, _ := testService(t) - idToken := "signed-id-token-sentinel" - service.client = &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { - if request.URL.Path == "/token" { - return jsonResponse(http.StatusOK, `{"access_token":"provider-access","id_token":"`+idToken+`"}`), nil - } - return nil, &url.Error{Op: "Get", URL: "https://oauth2.googleapis.com/tokeninfo?id_token=" + idToken, Err: errors.New("transport sentinel")} - })} - _, err := service.exchangeIdentity(context.Background(), "google", "code", "verifier", "nonce") - var failure *providerFailure - if !errors.As(err, &failure) || failure.Provider != "google" || failure.Stage != "id_token_verify" || failure.Reason != "network_error" { - t.Fatalf("failure=%+v err=%v", failure, err) - } - if strings.Contains(err.Error(), idToken) || strings.Contains(err.Error(), "transport sentinel") { - t.Fatal("Google ID token URL or transport details leaked through diagnostic error") - } -} - -func TestCallbackFailureClearsShortLivedOAuthMaterial(t *testing.T) { - service, store := testService(t) - insertAuthorizingDevice(t, store, "failed-device", "discord") - state := "failed-state" - verifier, err := store.seal("failed-device", "pkce", []byte("verifier")) - if err != nil { - t.Fatal(err) - } - nonce, err := store.seal("failed-device", "nonce", []byte("nonce")) - if err != nil { - t.Fatal(err) - } - if _, err := store.db.Exec(`UPDATE devices SET state_hash=?,verifier_cipher=?,nonce_cipher=? WHERE id='failed-device'`, store.digest("oauth-state", state), verifier, nonce); err != nil { - t.Fatal(err) - } - service.client = &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { - return jsonResponse(http.StatusUnauthorized, `{"error":"invalid_client"}`), nil - })} - request := httptest.NewRequest(http.MethodGet, "/auth/discord/callback?code=failed-code&state="+url.QueryEscape(state), nil) - response := httptest.NewRecorder() - service.Handler().ServeHTTP(response, request) - if response.Code != http.StatusBadGateway || !strings.Contains(response.Body.String(), "server OAuth configuration is invalid") { - t.Fatalf("status=%d body=%q", response.Code, response.Body.String()) - } - var status string - var stateHash, verifierCipher, nonceCipher []byte - if err := store.db.QueryRow(`SELECT status,COALESCE(state_hash,X''),COALESCE(verifier_cipher,X''),COALESCE(nonce_cipher,X'') FROM devices WHERE id='failed-device'`).Scan(&status, &stateHash, &verifierCipher, &nonceCipher); err != nil { - t.Fatal(err) - } - if status != "failed" || len(stateHash) != 0 || len(verifierCipher) != 0 || len(nonceCipher) != 0 { - t.Fatalf("status=%q state=%d verifier=%d nonce=%d", status, len(stateHash), len(verifierCipher), len(nonceCipher)) - } -} - -func TestProviderScopesUseLeastPrivilege(t *testing.T) { - if got := providerScope("discord"); got != "identify" { - t.Fatalf("Discord scope=%q, want identify", got) - } - if got := providerScope("google"); got != "openid" { - t.Fatalf("Google scope=%q, want openid", got) - } -} - -func TestProviderErrorConsumesAuthorizationStateWithoutExchange(t *testing.T) { - service, store := testService(t) - insertAuthorizingDevice(t, store, "cancelled-device", "discord") - state := "cancelled-oauth-state" - if _, err := store.db.Exec(`UPDATE devices SET state_hash=? WHERE id='cancelled-device'`, store.digest("oauth-state", state)); err != nil { - t.Fatal(err) - } - service.client = &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { - t.Fatal("provider error callback attempted a token exchange") - return nil, nil - })} - request := httptest.NewRequest(http.MethodGet, "/auth/discord/callback?error=access_denied&state="+url.QueryEscape(state), nil) - response := httptest.NewRecorder() - service.Handler().ServeHTTP(response, request) - if response.Code != http.StatusBadRequest { - t.Fatalf("status=%d body=%q", response.Code, response.Body.String()) - } - var status, errorCode string - var stateHash, verifier, nonce []byte - if err := store.db.QueryRow(`SELECT status,error_code,COALESCE(state_hash,X''),COALESCE(verifier_cipher,X''),COALESCE(nonce_cipher,X'') FROM devices WHERE id='cancelled-device'`). - Scan(&status, &errorCode, &stateHash, &verifier, &nonce); err != nil { - t.Fatal(err) - } - if status != "failed" || errorCode != "provider_cancelled" || len(stateHash) != 0 || len(verifier) != 0 || len(nonce) != 0 { - t.Fatalf("cancelled device status=%q error=%q state=%d verifier=%d nonce=%d", status, errorCode, len(stateHash), len(verifier), len(nonce)) - } -} - -func TestCompleteDeviceRejectsMalformedProviderIdentity(t *testing.T) { - service, store := testService(t) - insertAuthorizingDevice(t, store, "bad-identity", "discord") - if err := service.completeDevice("bad-identity", "discord", providerIdentity{issuer: "https://discord.com", subject: "not-a-snowflake"}); err == nil { - t.Fatal("accepted malformed Discord identity") - } - var count int - if err := store.db.QueryRow(`SELECT COUNT(*) FROM identities`).Scan(&count); err != nil { - t.Fatal(err) - } - if count != 0 { - t.Fatal("malformed identity was persisted") - } -} diff --git a/go/internal/server/auth/store_test.go b/go/internal/server/auth/store_test.go deleted file mode 100644 index 5bb2410..0000000 --- a/go/internal/server/auth/store_test.go +++ /dev/null @@ -1,97 +0,0 @@ -package auth - -import ( - "bytes" - "database/sql" - "path/filepath" - "testing" -) - -func TestStoreRequiresAndClearsExactMasterKey(t *testing.T) { - if _, err := Open(filepath.Join(t.TempDir(), "short.db"), make([]byte, 31)); err == nil { - t.Fatal("store accepted a non-256-bit master key") - } - key := bytes.Repeat([]byte{0x7a}, 32) - store, err := Open(filepath.Join(t.TempDir(), "auth.db"), key) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := store.Close(); err != nil { - t.Error(err) - } - }() - for index, value := range key { - if value != 0 { - t.Fatalf("master key byte %d was retained by the caller buffer", index) - } - } -} - -func TestStoreMigratesSchemaV1ToV2(t *testing.T) { - path := filepath.Join(t.TempDir(), "auth.db") - store, err := Open(path, bytes.Repeat([]byte{0x61}, 32)) - if err != nil { - t.Fatal(err) - } - if err := store.Close(); err != nil { - t.Fatal(err) - } - db, err := sql.Open("sqlite", path) - if err != nil { - t.Fatal(err) - } - if _, err := db.Exec(`DROP TABLE refresh_attempts; UPDATE metadata SET value='1' WHERE key='schema_version'`); err != nil { - if err := db.Close(); err != nil { - t.Error(err) - } - t.Fatal(err) - } - if err := db.Close(); err != nil { - t.Fatal(err) - } - reopened, err := Open(path, bytes.Repeat([]byte{0x61}, 32)) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := reopened.Close(); err != nil { - t.Error(err) - } - }() - var version int - if err := reopened.db.QueryRow(`SELECT CAST(value AS INTEGER) FROM metadata WHERE key='schema_version'`).Scan(&version); err != nil { - t.Fatal(err) - } - if version != 2 { - t.Fatalf("schema_version=%d, want 2", version) - } - var table string - if err := reopened.db.QueryRow(`SELECT name FROM sqlite_master WHERE type='table' AND name='refresh_attempts'`).Scan(&table); err != nil { - t.Fatal(err) - } -} - -func TestStorePersistsExplicitSchemaVersionAndRejectsUnknownVersion(t *testing.T) { - path := filepath.Join(t.TempDir(), "auth.db") - store, err := Open(path, bytes.Repeat([]byte{0x35}, 32)) - if err != nil { - t.Fatal(err) - } - var version int - if err := store.db.QueryRow(`SELECT CAST(value AS INTEGER) FROM metadata WHERE key='schema_version'`).Scan(&version); err != nil { - t.Fatal(err) - } - if version != schemaVersion { - t.Fatalf("schema_version=%d, want %d", version, schemaVersion) - } - if _, err := store.db.Exec(`UPDATE metadata SET value='999' WHERE key='schema_version'`); err != nil { - t.Fatal(err) - } - if err := store.Close(); err != nil { - t.Fatal(err) - } - if _, err := Open(path, bytes.Repeat([]byte{0x35}, 32)); err == nil { - t.Fatal("store accepted an unknown schema version") - } -} diff --git a/go/internal/server/authconfig/config_test.go b/go/internal/server/authconfig/config_test.go deleted file mode 100644 index 3dc09fc..0000000 --- a/go/internal/server/authconfig/config_test.go +++ /dev/null @@ -1,99 +0,0 @@ -package authconfig - -import ( - "encoding/base64" - "encoding/json" - "os" - "path/filepath" - "strings" - "testing" -) - -func TestLoad(t *testing.T) { - path := filepath.Join(t.TempDir(), FileName) - if err := os.WriteFile(path, []byte(`{"mode":"oauth","public_url":"https://example.com","master_key_env":"MASTER","providers":{"discord":{"client_id":"d","client_secret_env":"DS"},"google":{"client_id":"g","client_secret_env":"GS"}}}`), 0o600); err != nil { - t.Fatal(err) - } - config, err := Load(path) - if err != nil { - t.Fatal(err) - } - if config.Mode != "oauth" || len(config.Providers) != 2 || config.Providers["discord"].ClientID != "d" || config.Providers["google"].ClientID != "g" { - t.Fatalf("unexpected config: %+v", config) - } -} - -func TestValidateRejectsUnsafePolicies(t *testing.T) { - for name, config := range map[string]Config{ - "unknown mode": {Mode: "disabled"}, - "local providers": {Mode: "local", Providers: map[string]ProviderConfig{"discord": {ClientID: "d", ClientSecretEnv: "DS"}}}, - "empty oauth": {Mode: "oauth"}, - "unknown provider": {Mode: "oauth", PublicURL: "https://example.com", MasterKeyEnv: "MASTER", Providers: map[string]ProviderConfig{"github": {ClientID: "g", ClientSecretEnv: "GS"}}}, - } { - t.Run(name, func(t *testing.T) { - if err := config.Validate(); err == nil { - t.Fatal("accepted invalid authentication policy") - } - }) - } -} - -func TestResolveEnvironmentKeepsSecretsOutOfPublicView(t *testing.T) { - master := make([]byte, 32) - for i := range master { - master[i] = byte(i + 1) - } - t.Setenv("AUTH_MASTER", base64.StdEncoding.EncodeToString(master)) - t.Setenv("DISCORD_SECRET", "private-discord-secret") - config := Config{ - Mode: "oauth", - PublicURL: "https://example.com", - MasterKeyEnv: "AUTH_MASTER", - Providers: map[string]ProviderConfig{ - "discord": {ClientID: "public-client-id", ClientSecretEnv: "DISCORD_SECRET"}, - }, - } - runtime, err := config.ResolveEnvironment() - if err != nil { - t.Fatal(err) - } - if len(runtime.MasterKey) != 32 || runtime.ProviderSecrets["discord"] != "private-discord-secret" { - t.Fatal("runtime did not resolve authentication secrets") - } - publicJSON, err := json.Marshal(config.Public()) - if err != nil { - t.Fatal(err) - } - for _, forbidden := range []string{"AUTH_MASTER", "DISCORD_SECRET", "private-discord-secret", "public-client-id"} { - if strings.Contains(string(publicJSON), forbidden) { - t.Fatalf("public authentication view leaked %q: %s", forbidden, publicJSON) - } - } -} - -func TestValidateRejectsUnsafePublicURLsAndTTLs(t *testing.T) { - base := Config{ - Mode: "oauth", - PublicURL: "https://example.com", - MasterKeyEnv: "MASTER", - Providers: map[string]ProviderConfig{ - "discord": {ClientID: "d", ClientSecretEnv: "DS"}, - }, - } - for name, mutate := range map[string]func(*Config){ - "http public": func(c *Config) { c.PublicURL = "http://example.com" }, - "path": func(c *Config) { c.PublicURL = "https://example.com/auth" }, - "query": func(c *Config) { c.PublicURL = "https://example.com?x=y" }, - "missing secret": func(c *Config) { c.Providers["discord"] = ProviderConfig{ClientID: "d"} }, - "invalid ttl": func(c *Config) { c.Session.AccessTTL = "0s" }, - } { - t.Run(name, func(t *testing.T) { - candidate := base - candidate.Providers = map[string]ProviderConfig{"discord": base.Providers["discord"]} - mutate(&candidate) - if err := candidate.Validate(); err == nil { - t.Fatal("accepted unsafe authentication configuration") - } - }) - } -} diff --git a/go/internal/server/battle/battle_test.go b/go/internal/server/battle/battle_test.go deleted file mode 100644 index 3d30ef0..0000000 --- a/go/internal/server/battle/battle_test.go +++ /dev/null @@ -1,258 +0,0 @@ -package battle - -import ( - "fmt" - "testing" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func request(seq uint64) []byte { return wire.AppendVarint(nil, 1, seq) } - -func TestQuestBattleSelectsAndRetainsDifficultyDeck(t *testing.T) { - s := NewService("design", "version", nil, func() (int, error) { return 1, nil }) - s.AttachCurrentDifficulty(func() (uint64, error) { return 2, nil }) - s.loadDifficultyDeck = func(_, _ string, pack int, deck, difficulty uint64) (uint64, error) { - if pack != 1 || deck != 1 || difficulty != 2 { - t.Fatalf("selection %d/%d/%d", pack, deck, difficulty) - } - return 200001, nil - } - s.loadPhases = func(_, _ string, pack int, monster, deck uint64) ([]gamedata.BattlePhase, error) { - if pack != 1 || deck != 200001 { - t.Fatalf("phase selection %d/%d", pack, deck) - } - return nil, nil - } - enter := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(request(1), 3, 1), 4, 1), 5, 1) - _, response, _, err := s.Handle("/BattleEnter", enter) - if err != nil { - t.Fatal(err) - } - deck, _, _ := wire.Varint(response, 2) - if deck != 200001 { - t.Fatalf("deck %d", deck) - } - start := wire.AppendBytes(wire.AppendVarint(request(2), 2, 200001), 5, wire.AppendVarint(nil, 2, 101)) - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - _, response, _, err = s.Handle("/BattleRetry", wire.AppendVarint(request(3), 2, 200001)) - if err != nil { - t.Fatal(err) - } - deck, _, _ = wire.Varint(response, 3) - if deck != 200001 { - t.Fatalf("retry deck %d", deck) - } -} - -func TestLocalBattleLifecycle(t *testing.T) { - s := &Service{} - enter := request(1) - enter = wire.AppendVarint(enter, 4, 1) - enter = wire.AppendVarint(enter, 5, 1) - code, response, ok, err := s.Handle("/BattleEnter", enter) - if err != nil || !ok || code != 52 { - t.Fatalf("enter: %d %v %v", code, ok, err) - } - engine, found, _ := wire.Varint(response, 6) - if !found || engine != 1 { - t.Fatalf("engine=%d/%v", engine, found) - } - red := wire.AppendVarint(nil, 1, 1) - blue := wire.AppendVarint(nil, 1, 6) - start := request(2) - start = wire.AppendVarint(start, 2, 1) - start = wire.AppendBytes(start, 4, red) - start = wire.AppendBytes(start, 5, blue) - code, response, _, err = s.Handle("/BattleStart", start) - if err != nil || code != 14 { - t.Fatalf("start: %d %v", code, err) - } - if _, found, _ := wire.Bytes(response, 1); !found { - t.Fatal("red state not echoed") - } - if _, found, _ := wire.Bytes(response, 2); !found { - t.Fatal("blue state not echoed") - } - code, response, _, err = s.Handle("/BattleVerifyState", request(3)) - state, found, _ := wire.Varint(response, 1) - if err != nil || code != 142 || !found || state != 3 { - t.Fatalf("verify: code=%d state=%d/%v err=%v", code, state, found, err) - } - end := request(4) - end = wire.AppendVarint(end, 2, 2) - end = wire.AppendBytes(end, 3, blue) - code, response, _, err = s.Handle("/BattleEnd", end) - result, found, _ := wire.Varint(response, 1) - if err != nil || code != 15 || !found || result != 2 { - t.Fatalf("end: code=%d result=%d/%v err=%v", code, result, found, err) - } -} - -func TestBattleOrdering(t *testing.T) { - s := &Service{} - start := request(1) - start = wire.AppendVarint(start, 2, 1) - if _, _, _, err := s.Handle("/BattleStart", start); err == nil { - t.Fatal("start without enter accepted") - } -} - -func TestBattleRetryRestoresFirstSubmittedBlueTeam(t *testing.T) { - s := &Service{} - enter := wire.AppendVarint(wire.AppendVarint(request(1), 4, 4), 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - first := wire.AppendVarint(wire.AppendVarint(nil, 2, 101), 4, 513) - second := wire.AppendVarint(wire.AppendVarint(nil, 2, 102), 4, 200) - start := wire.AppendVarint(request(2), 2, 77) - start = wire.AppendBytes(start, 5, first) - start = wire.AppendBytes(start, 5, second) - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - retry := wire.AppendVarint(request(3), 2, 77) - code, response, handled, err := s.Handle("/BattleRetry", retry) - if err != nil || !handled || code != 58 { - t.Fatalf("retry code=%d handled=%v err=%v", code, handled, err) - } - index, found, err := wire.Varint(response, 3) - if err != nil || !found || index != 77 { - t.Fatalf("retry battle index=%d found=%v err=%v", index, found, err) - } - var restored [][]byte - if err := wire.Walk(response, func(field wire.Field) error { - if field.Number == 2 { - restored = append(restored, field.Value) - } - return nil - }); err != nil || len(restored) != 2 || string(restored[0]) != string(first) || string(restored[1]) != string(second) { - t.Fatalf("retry team=%x err=%v", restored, err) - } -} - -func TestBattleEnterUsesSamePictorialSnapshotAsAllCharRefresh(t *testing.T) { - s := &Service{} - s.AttachPictorialBuffs(func() ([]gamedata.PictorialBuffStat, error) { - return []gamedata.PictorialBuffStat{{StatType: 2, Value: .0175}, {StatType: 4, Value: .01}}, nil - }) - enter := wire.AppendVarint(wire.AppendVarint(request(9), 4, 1), 5, 1) - code, response, handled, err := s.Handle("/BattleEnter", enter) - if err != nil || !handled || code != 52 { - t.Fatalf("battle entry code=%d handled=%v err=%v", code, handled, err) - } - var count int - if err := wire.Walk(response, func(field wire.Field) error { - if field.Number == 4 { - count++ - stat, found, err := wire.Varint(field.Value, 1) - if err != nil || !found || (stat != 2 && stat != 4) { - t.Fatalf("invalid battle buff stat=%d found=%v err=%v", stat, found, err) - } - } - return nil - }); err != nil || count != 2 { - t.Fatalf("battle buffs count=%d err=%v", count, err) - } -} - -func TestBattleVictoryLocksPackAtEnterForRewardsAndIdentity(t *testing.T) { - storage := stateio.NewMemory() - inventory, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - currentPack := 22 - s := NewService("test-root", "test-version", inventory, func() (int, error) { - return currentPack, nil - }) - var loadedPack int - var loadedDeck uint64 - s.loadPhases = func(string, string, int, uint64, uint64) ([]gamedata.BattlePhase, error) { return nil, nil } - s.loadRewards = func(_, _ string, packID int, deckID uint64) ([]gamedata.BattleReward, error) { - loadedPack, loadedDeck = packID, deckID - return []gamedata.BattleReward{{Type: 8, ID: 8, Count: 3}}, nil - } - - enter := wire.AppendVarint(request(1), 3, 7) - enter = wire.AppendVarint(enter, 4, 9) - enter = wire.AppendVarint(enter, 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - // A later world transition must not change the identity of an in-flight - // battle; the pack is captured at BattleEnter. - currentPack = 21 - end := wire.AppendVarint(request(2), 2, 1) - if _, _, _, err := s.Handle("/BattleEnd", end); err != nil { - t.Fatal(err) - } - if loadedPack != 22 || loadedDeck != 9 { - t.Fatalf("reward lookup pack/deck=%d/%d, want 22/9", loadedPack, loadedDeck) - } - if got := inventory.GrantedItems("pack22:monster7:deck9"); len(got) != 1 || got[0].ID != 8 || got[0].Count != 3 { - t.Fatalf("pack22 reward grant=%+v", got) - } - if got := inventory.GrantedItems("pack21:monster7:deck9"); len(got) != 0 { - t.Fatalf("reward leaked into pack21 identity: %+v", got) - } -} - -func TestNewGameSessionDiscardsUnfinishedBattleWithoutRewards(t *testing.T) { - s := NewService("", "", nil, nil) - s.BeginSession("session-a") - enter := wire.AppendVarint(request(1), 4, 10) - enter = wire.AppendVarint(enter, 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - start := wire.AppendVarint(request(2), 2, 99) - start = wire.AppendBytes(start, 5, wire.AppendVarint(nil, 1, 123)) - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - s.BeginSession("session-b") - end := wire.AppendVarint(request(3), 2, 1) - if _, _, _, err := s.Handle("/BattleEnd", end); err == nil { - t.Fatal("new session resumed unfinished battle") - } -} - -func TestBattleStateIsIsolatedPerGameSession(t *testing.T) { - s := NewService("", "", nil, nil) - enter := wire.AppendVarint(request(1), 4, 10) - enter = wire.AppendVarint(enter, 5, 1) - s.BeginSession("session-a") - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - s.BeginSession("session-b") - end := wire.AppendVarint(request(2), 2, 1) - if _, _, _, err := s.Handle("/BattleEnd", end); err == nil { - t.Fatal("session-b observed session-a battle") - } - s.BeginSession("session-a") - if _, _, _, err := s.Handle("/BattleEnd", end); err != nil { - t.Fatalf("session-a lost its own battle: %v", err) - } -} - -func TestRepeatedCurrentSessionSelectionDoesNotEvictOtherBattles(t *testing.T) { - service := NewService("", "", nil, nil) - for index := range 1024 { - service.BeginSession(fmt.Sprintf("session-%d", index)) - } - service.BeginSession("session-1023") - if len(service.states) != 1024 { - t.Fatalf("reselecting current session changed cache size: %d", len(service.states)) - } - if service.states["session-1023"] == nil { - t.Fatal("reselecting current session evicted its battle state") - } -} diff --git a/go/internal/server/battle/event_test.go b/go/internal/server/battle/event_test.go deleted file mode 100644 index c3f4c63..0000000 --- a/go/internal/server/battle/event_test.go +++ /dev/null @@ -1,72 +0,0 @@ -package battle - -import ( - "bytes" - "fmt" - "testing" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -type fakeEventBattle struct { - enter, end string - rejected bool -} - -func (*fakeEventBattle) HandlesBattle(mode uint64) bool { return mode == 17 } -func (e *fakeEventBattle) EnterBattle(_ []byte, receipt string) ([]byte, error) { - e.enter = receipt - if e.rejected { - return nil, fmt.Errorf("stage locked") - } - return wire.AppendVarint(nil, 1, 7), nil -} -func (e *fakeEventBattle) CompleteBattle(_ []byte, receipt string) ([]byte, error) { - e.end = receipt - return wire.AppendBytes(nil, 16, wire.AppendVarint(nil, 3, 77)), nil -} - -func TestEventBattleOwnsSettlementAndAvoidsWorldDeck(t *testing.T) { - s := NewService("unused", "unused", nil, func() (int, error) { return 21, nil }) - s.BeginSession("event-session") - e := &fakeEventBattle{} - s.AttachEventBattle(e) - s.loadPhases = func(string, string, int, uint64, uint64) ([]gamedata.BattlePhase, error) { - t.Fatal("event read ordinary world deck") - return nil, nil - } - s.AttachCommittedHealth(func(map[uint64]uint64) error { t.Fatal("event altered world health"); return nil }) - enter := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(request(1), 3, 1), 4, 73), 5, 17) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/BattleStart", wire.AppendVarint(request(2), 2, 73)); err != nil { - t.Fatal(err) - } - code, out, _, err := s.Handle("/BattleEnd", wire.AppendVarint(request(3), 2, 1)) - if err != nil || code != 15 { - t.Fatalf("settlement: %d %v", code, err) - } - if e.enter != "event-session:1" || e.end != e.enter { - t.Fatalf("receipt %q/%q", e.enter, e.end) - } - row, _, _ := wire.Bytes(out, 16) - if !bytes.Equal(row, wire.AppendVarint(nil, 3, 77)) { - t.Fatal("event progress lost") - } - if s.Active() { - t.Fatal("event battle remained active") - } - _, retry, _, err := s.Handle("/BattleEnd", wire.AppendVarint(request(3), 2, 1)) - if err != nil || !bytes.Equal(retry, out) { - t.Fatal("settlement retry lost reply") - } - if _, _, _, err = s.Handle("/BattleEnd", wire.AppendVarint(request(3), 2, 2)); err == nil { - t.Fatal("changed settlement retry accepted") - } - e.rejected = true - if _, _, _, err = s.Handle("/BattleEnter", enter); err == nil { - t.Fatal("locked stage entered") - } -} diff --git a/go/internal/server/battle/field_buff_test.go b/go/internal/server/battle/field_buff_test.go deleted file mode 100644 index 5699810..0000000 --- a/go/internal/server/battle/field_buff_test.go +++ /dev/null @@ -1,56 +0,0 @@ -package battle - -import ( - "bd2server/internal/server/wire" - "errors" - "testing" -) - -func TestFieldBuffConsumptionAfterValidationAndBeforeBattleActivation(t *testing.T) { - s := NewService("", "", nil, nil) - s.BeginSession("first-session") - remaining := 3 - used := map[string]bool{} - fail := false - s.AttachFieldBuffConsume(func(identity string) error { - if fail { - return errors.New("field buff persistence failed") - } - if !used[identity] { - remaining-- - used[identity] = true - } - return nil - }) - enter := wire.AppendVarint(wire.AppendVarint(request(1), 4, 1), 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", request(1)); err == nil || remaining != 3 { - t.Fatal("invalid battle consumed buff") - } - fail = true - if _, _, _, err := s.Handle("/BattleEnter", enter); err == nil || s.Active() { - t.Fatal("persistence failure activated battle") - } - fail = false - for range 2 { - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - } - if remaining != 2 { - t.Fatal("same enter receipt consumed twice") - } - start := wire.AppendBytes(wire.AppendVarint(request(2), 2, 1), 5, wire.AppendVarint(nil, 1, 6)) - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/BattleRetry", wire.AppendVarint(request(3), 2, 1)); err != nil { - t.Fatal(err) - } - if remaining != 2 { - t.Fatal("retry consumed extra field buff") - } - s.BeginSession("second-session") - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil || remaining != 1 { - t.Fatal("new battle did not consume once", err) - } -} diff --git a/go/internal/server/battle/health_test.go b/go/internal/server/battle/health_test.go deleted file mode 100644 index 382da98..0000000 --- a/go/internal/server/battle/health_test.go +++ /dev/null @@ -1,132 +0,0 @@ -package battle - -import ( - "strconv" - "testing" - - "bd2server/internal/server/wire" -) - -func TestBattleDefeatAndRunCommitZeroAndInjuredHealth(t *testing.T) { - // EBattleResultType: RED_WIN=2, CONDITION_FAIL=3, RUN=4. A dead - // CharDBInfo omits hp because proto3's default is zero; it must be committed - // rather than ignored and replaced with an earlier full-health value. - for _, result := range []uint64{2, 3, 4} { - t.Run(strconv.FormatUint(result, 10), func(t *testing.T) { - s := &Service{} - calls := 0 - var saved map[uint64]uint64 - s.AttachCommittedHealth(func(health map[uint64]uint64) error { calls++; saved = health; return nil }) - s.BeginSession("settlement") - enter := wire.AppendVarint(wire.AppendVarint(request(1), 4, 1), 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - start := wire.AppendVarint(request(2), 2, 11) - start = wire.AppendBytes(start, 5, wire.AppendVarint(wire.AppendVarint(nil, 2, 100), 4, 100)) - start = wire.AppendBytes(start, 5, wire.AppendVarint(wire.AppendVarint(nil, 2, 200), 4, 100)) - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - end := wire.AppendVarint(request(3), 2, result) - end = wire.AppendBytes(end, 3, wire.AppendVarint(nil, 1, 100)) - end = wire.AppendBytes(end, 3, wire.AppendVarint(wire.AppendVarint(nil, 1, 200), 3, 21)) - _, response, _, err := s.Handle("/BattleEnd", end) - if err != nil { - t.Fatal(err) - } - if calls != 1 || len(saved) != 2 || saved[100] != 0 || saved[200] != 21 { - t.Fatalf("result=%d saved health=%v calls=%d", result, saved, calls) - } - if err := wire.Walk(response, func(field wire.Field) error { - if field.Number == 3 { - index, _, _ := wire.Varint(field.Value, 1) - hp, _, _ := wire.Varint(field.Value, 3) - if hp != saved[index] { - t.Fatal("response differs from settled HP") - } - } - return nil - }); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/BattleExit", request(4)); err != nil { - t.Fatal(err) - } - if calls != 1 { - t.Fatal("exit wrote health after settlement") - } - }) - } -} - -func TestBattleHealthCommitsOnlyAtEndAndChecksParticipants(t *testing.T) { - s := &Service{} - calls := 0 - var health map[uint64]uint64 - s.AttachCommittedHealth(func(values map[uint64]uint64) error { - calls++ - for index, hp := range values { - values[index] = min(hp, 50) - } - health = values - return nil - }) - s.BeginSession("first") - enter := wire.AppendVarint(wire.AppendVarint(request(1), 4, 1), 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - start := wire.AppendVarint(request(2), 2, 11) - blue := wire.AppendVarint(wire.AppendVarint(nil, 2, 100), 4, 60) - start = wire.AppendBytes(start, 5, blue) - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - if calls != 0 || !s.Active() { - t.Fatal("round state was committed or battle was not active") - } - invalid := wire.AppendVarint(request(3), 2, 1) - invalid = wire.AppendBytes(invalid, 3, wire.AppendVarint(nil, 1, 999)) - if _, _, _, err := s.Handle("/BattleEnd", invalid); err == nil || calls != 0 { - t.Fatal("accepted health for a character outside the blue team") - } - result := wire.AppendVarint(request(4), 2, 1) - character := wire.AppendVarint(wire.AppendVarint(nil, 1, 100), 3, 17) - result = wire.AppendBytes(result, 3, character) - if _, _, _, err := s.Handle("/BattleEnd", result); err != nil { - t.Fatal(err) - } - if calls != 1 || health[100] != 17 || s.Active() { - t.Fatalf("health=%v calls=%d active=%t", health, calls, s.Active()) - } - // A second battle interrupted by a login never commits its round health. - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - s.BeginSession("reconnected") - if calls != 1 || s.Active() { - t.Fatal("reconnect persisted unfinished battle health") - } - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/BattleStart", start); err != nil { - t.Fatal(err) - } - result = wire.AppendVarint(request(5), 2, 1) - character = wire.AppendVarint(wire.AppendVarint(nil, 1, 100), 3, 999) - result = wire.AppendBytes(result, 3, character) - _, body, _, err := s.Handle("/BattleEnd", result) - if err != nil { - t.Fatal(err) - } - returned, _, _ := wire.Bytes(body, 3) - hp, _, _ := wire.Varint(returned, 3) - if health[100] != 50 || hp != 50 { - t.Fatalf("committed and returned HP differ: saved=%v response=%d", health, hp) - } -} diff --git a/go/internal/server/battle/hunting_test.go b/go/internal/server/battle/hunting_test.go deleted file mode 100644 index 453c2ef..0000000 --- a/go/internal/server/battle/hunting_test.go +++ /dev/null @@ -1,69 +0,0 @@ -package battle - -import ( - "bytes" - "errors" - "testing" - - "bd2server/internal/server/wire" -) - -type huntingFixture struct { - validationError error - settlementError error - pack int - mode, monster, deck uint64 - receipt string - settlements int -} - -func (h *huntingFixture) ValidateBattle(pack int, mode, monster, deck uint64) error { - h.pack, h.mode, h.monster, h.deck = pack, mode, monster, deck - return h.validationError -} -func (h *huntingFixture) CompleteBattle(pack int, mode, monster, deck uint64, receipt string) ([]byte, [][]byte, error) { - h.pack, h.mode, h.monster, h.deck, h.receipt = pack, mode, monster, deck, receipt - h.settlements++ - return wire.AppendVarint(nil, 2, 7), [][]byte{wire.AppendVarint(nil, 1, monster)}, h.settlementError -} - -func TestHuntingBattleUsesLockedEncounterAndSessionReceipt(t *testing.T) { - pack := 1 - s := NewService("", "", nil, func() (int, error) { return pack, nil }) - h := &huntingFixture{} - s.AttachHunting(h) - s.BeginSession("login-A") - enter := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(request(1), 3, 11), 4, 12), 5, huntingGroundMode) - h.validationError = errors.New("locked difficulty") - if _, _, _, err := s.Handle("/BattleEnter", enter); err == nil || s.Active() { - t.Fatal("invalid encounter accepted") - } - h.validationError = nil - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - pack = 2 - h.settlementError = errors.New("not enough AP") - end := wire.AppendVarint(request(2), 2, 1) - if _, _, _, err := s.Handle("/BattleEnd", end); err == nil || !s.Active() { - t.Fatal("failed settlement ended battle") - } - h.settlementError = nil - code, response, _, err := s.Handle("/BattleEnd", end) - if err != nil || code != 15 { - t.Fatalf("settle: %d %v", code, err) - } - if h.pack != 1 || h.mode != huntingGroundMode || h.monster != 11 || h.deck != 12 || h.receipt != "login-A:2" { - t.Fatalf("settled wrong encounter: %+v", h) - } - if _, found, _ := wire.Bytes(response, 4); !found { - t.Fatal("missing monster progress") - } - if bundle, found, _ := wire.Bytes(response, 5); !found || len(bundle) == 0 { - t.Fatal("missing hunting reward") - } - settlements := h.settlements - if _, retry, _, err := s.Handle("/BattleEnd", end); err != nil || !bytes.Equal(retry, response) || h.settlements != settlements { - t.Fatal("completed battle retry settled twice or lost response") - } -} diff --git a/go/internal/server/battle/monster_hunt_test.go b/go/internal/server/battle/monster_hunt_test.go deleted file mode 100644 index 7203a74..0000000 --- a/go/internal/server/battle/monster_hunt_test.go +++ /dev/null @@ -1,72 +0,0 @@ -package battle - -import ( - "errors" - "testing" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -type monsterHuntFixture struct { - enter, end []byte - enterReceipt, endReceipt string - err error -} - -func (h *monsterHuntFixture) EnterBattle(request []byte, receipt string) ([]byte, error) { - h.enter, h.enterReceipt = append([]byte(nil), request...), receipt - return wire.AppendBytes(nil, 5, wire.AppendVarint(nil, 1, 11)), h.err -} -func (h *monsterHuntFixture) CompleteBattle(request []byte, receipt string) ([]byte, error) { - h.end, h.endReceipt = append([]byte(nil), request...), receipt - return wire.AppendBytes(nil, 13, wire.AppendVarint(nil, 1, 11)), h.err -} - -func TestMonsterHuntBattleUsesSpecialRuntimeAndLeavesFieldHealthAlone(t *testing.T) { - for _, mode := range []uint64{8, 24} { - s := NewService("", "", nil, func() (int, error) { return 1, nil }) - h := &monsterHuntFixture{} - s.AttachMonsterHunt(h) - s.BeginSession("login-A") - s.loadPhases = func(string, string, int, uint64, uint64) ([]gamedata.BattlePhase, error) { - t.Fatal("monster hunt used pack phases") - return nil, nil - } - s.AttachCommittedHealth(func(map[uint64]uint64) error { t.Fatal("monster hunt changed field HP"); return nil }) - s.AttachMonsterWinMission(func() error { t.Fatal("monster hunt awarded field monster mission"); return nil }) - s.AttachTutorialWin(func() error { t.Fatal("monster hunt awarded tutorial progress"); return nil }) - enter := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(request(1), 4, 12), 5, mode), 6, 91) - h.err = errors.New("hunt unavailable") - if _, _, _, err := s.Handle("/BattleEnter", enter); err == nil || s.Active() { - t.Fatal("rejected hunt entered") - } - h.err = nil - code, response, _, err := s.Handle("/BattleEnter", enter) - if err != nil || code != 52 { - t.Fatalf("enter: %d %v", code, err) - } - if _, found, _ := wire.Bytes(response, 5); !found { - t.Fatal("missing hunt user info at enter") - } - end := wire.AppendVarint(wire.AppendVarint(request(2), 2, 1), 7, 1234) - h.err = errors.New("invalid hunt settlement") - if _, _, _, err := s.Handle("/BattleEnd", end); err == nil || !s.Active() { - t.Fatal("failed hunt settlement finished battle") - } - h.err = nil - code, response, _, err = s.Handle("/BattleEnd", end) - if err != nil || code != 15 || h.enterReceipt != "login-A:1" || h.endReceipt != h.enterReceipt { - t.Fatalf("end: %d %v %+v", code, err, h) - } - if _, found, _ := wire.Bytes(response, 13); !found { - t.Fatal("missing hunt settlement progress") - } - if _, found, _ := wire.Bytes(response, 5); found { - t.Fatal("hunt used ordinary pack reward bundle") - } - if s.Active() { - t.Fatal("hunt battle still active") - } - } -} diff --git a/go/internal/server/battle/phase_test.go b/go/internal/server/battle/phase_test.go deleted file mode 100644 index 912253e..0000000 --- a/go/internal/server/battle/phase_test.go +++ /dev/null @@ -1,208 +0,0 @@ -package battle - -import ( - "bytes" - "testing" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func phaseService(t *testing.T) *Service { - t.Helper() - s := &Service{loadPhases: func(string, string, int, uint64, uint64) ([]gamedata.BattlePhase, error) { - return []gamedata.BattlePhase{{GroupID: 70, ID: 101, DeckID: 11}, {GroupID: 70, ID: 205, DeckID: 12}, {GroupID: 70, ID: 309, DeckID: 13}}, nil - }} - enter := wire.AppendVarint(wire.AppendVarint(request(1), 4, 11), 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - return s -} - -func phaseStart(t *testing.T, s *Service, seq, deck uint64, blue []byte) []byte { - t.Helper() - start := wire.AppendVarint(request(seq), 2, deck) - start = wire.AppendBytes(start, 5, blue) - code, reply, _, err := s.Handle("/BattleStart", start) - if err != nil || code != 14 { - t.Fatalf("start %d: code=%d err=%v", deck, code, err) - } - return reply -} - -func TestPhaseLifecyclePreservesClientStateAndRejectsSkipping(t *testing.T) { - s := phaseService(t) - blue := wire.AppendVarint(wire.AppendVarint(nil, 2, 400), 4, 1234) - if _, _, _, err := s.Handle("/BattlePhaseChange", request(2)); err == nil { - t.Fatal("phase advanced before start") - } - phaseStart(t, s, 3, 11, blue) - commits := 0 - s.AttachCommittedHealth(func(map[uint64]uint64) error { commits++; return nil }) - win := wire.AppendVarint(request(4), 2, 1) - if _, _, _, err := s.Handle("/BattleEnd", win); err == nil { - t.Fatal("early victory accepted") - } - if commits != 0 { - t.Fatal("early victory committed health") - } - code, reply, _, err := s.Handle("/BattlePhaseChange", request(5)) - if err != nil || code != 632 { - t.Fatalf("phase response=%d %v", code, err) - } - for field, want := range map[int]uint64{1: 70, 2: 205, 8: 12} { - got, found, err := wire.Varint(reply, field) - if err != nil || !found || got != want { - t.Fatalf("field %d = %d/%v: %v", field, got, found, err) - } - } - if result, found, _ := wire.Bytes(reply, 3); found && len(result) != 0 { - t.Fatal("fabricated verified battle result") - } - verify, _, _ := wire.Varint(reply, 7) - if verify != 0 { - t.Fatal("verification enabled without authoritative combat state") - } - _, replay, _, err := s.Handle("/BattlePhaseChange", request(5)) - if err != nil || !bytes.Equal(reply, replay) || s.stateLocked().phase != 1 { - t.Fatal("replay advanced phase") - } - if _, _, _, err := s.Handle("/BattlePhaseChange", request(6)); err == nil { - t.Fatal("skipped unstarted phase") - } - if _, _, _, err := s.Handle("/BattleStart", wire.AppendVarint(request(7), 2, 13)); err == nil { - t.Fatal("skipped deck accepted") - } - changedBlue := wire.AppendVarint(wire.AppendVarint(nil, 2, 400), 4, 600) - response := phaseStart(t, s, 8, 12, changedBlue) - got, _, _ := wire.Bytes(response, 2) - if !bytes.Equal(got, changedBlue) || !bytes.Equal(s.stateLocked().initialBlue[0], blue) { - t.Fatal("phase start reset player state or retry baseline") - } - if _, _, _, err := s.Handle("/BattlePhaseChange", request(9)); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/BattleEnd", win); err == nil { - t.Fatal("victory before final start") - } - phaseStart(t, s, 10, 13, changedBlue) - if _, _, _, err := s.Handle("/BattlePhaseChange", request(11)); err == nil { - t.Fatal("advanced past final phase") - } - if _, _, _, err := s.Handle("/BattleEnd", win); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/BattlePhaseChange", request(12)); err == nil { - t.Fatal("inactive phase request accepted") - } -} - -func TestPhaseRetryRestoresFirstDeckAndInitialBlue(t *testing.T) { - s := phaseService(t) - blue := wire.AppendVarint(wire.AppendVarint(nil, 2, 401), 4, 200) - phaseStart(t, s, 2, 11, blue) - if _, _, _, err := s.Handle("/BattlePhaseChange", request(3)); err != nil { - t.Fatal(err) - } - phaseStart(t, s, 4, 12, wire.AppendVarint(nil, 4, 1)) - if _, _, _, err := s.Handle("/BattleRetry", wire.AppendVarint(request(5), 2, 11)); err == nil { - t.Fatal("retry accepted wrong current deck") - } - code, response, _, err := s.Handle("/BattleRetry", wire.AppendVarint(request(6), 2, 12)) - index, _, _ := wire.Varint(response, 3) - restored, _, _ := wire.Bytes(response, 2) - if err != nil || code != 58 || index != 11 || !bytes.Equal(restored, blue) { - t.Fatalf("retry: code=%d index=%d blue=%x err=%v", code, index, restored, err) - } - state := s.stateLocked() - if state.phase != 0 || state.deck != 11 || state.phaseStarted || state.phaseReply != nil { - t.Fatal("retry did not reset phase state") - } - if _, _, _, err := s.Handle("/BattlePhaseChange", request(7)); err == nil { - t.Fatal("retry advanced before start") - } -} - -func TestMalformedPhaseStartDoesNotConsumeRound(t *testing.T) { - s := phaseService(t) - start := append(wire.AppendVarint(request(2), 2, 11), 0x2a, 0x80) - if _, _, _, err := s.Handle("/BattleStart", start); err == nil { - t.Fatal("malformed start accepted") - } - if state := s.stateLocked(); state.round != 0 || state.index != 0 || state.phaseStarted { - t.Fatal("malformed request mutated battle") - } -} - -func TestPhaseEntryRejectsLaterDeckWithoutReplacingActiveBattle(t *testing.T) { - s := phaseService(t) - phaseStart(t, s, 2, 11, wire.AppendVarint(nil, 2, 400)) - enter := wire.AppendVarint(wire.AppendVarint(request(3), 4, 12), 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err == nil { - t.Fatal("entered a later phase directly") - } - state := s.stateLocked() - if state.deck != 11 || state.index != 11 || !state.phaseStarted { - t.Fatal("rejected enter replaced active battle") - } -} - -func TestOrdinaryBattleRejectsPhaseChange(t *testing.T) { - s := &Service{} - enter := wire.AppendVarint(wire.AppendVarint(request(1), 4, 11), 5, 1) - if _, _, _, err := s.Handle("/BattleEnter", enter); err != nil { - t.Fatal(err) - } - phaseStart(t, s, 2, 11, wire.AppendVarint(nil, 2, 400)) - if _, _, _, err := s.Handle("/BattlePhaseChange", request(3)); err == nil { - t.Fatal("ordinary battle accepted phase change") - } -} - -func TestPhaseVictoryRewardsOnlyFinalDeck(t *testing.T) { - s := phaseService(t) - var err error - s.inventory, err = player.OpenInventory(stateio.NewMemory(), &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - s.gameDataRoot = "test-root" - s.stateLocked().pack, s.stateLocked().monster = 77, 42 - loaded := uint64(0) - s.loadRewards = func(_, _ string, pack int, deck uint64) ([]gamedata.BattleReward, error) { - if pack != 77 { - t.Fatalf("reward pack = %d", pack) - } - loaded = deck - return []gamedata.BattleReward{{Type: 8, ID: 8, Count: 1}}, nil - } - blue := wire.AppendVarint(nil, 2, 400) - phaseStart(t, s, 2, 11, blue) - win := wire.AppendVarint(request(3), 2, 1) - if _, _, _, err := s.Handle("/BattleEnd", win); err == nil { - t.Fatal("early win accepted") - } - if loaded != 0 { - t.Fatal("early win loaded rewards") - } - if _, _, _, err := s.Handle("/BattlePhaseChange", request(4)); err != nil { - t.Fatal(err) - } - phaseStart(t, s, 5, 12, blue) - if _, _, _, err := s.Handle("/BattlePhaseChange", request(6)); err != nil { - t.Fatal(err) - } - phaseStart(t, s, 7, 13, blue) - if _, _, _, err := s.Handle("/BattleEnd", win); err != nil { - t.Fatal(err) - } - if loaded != 13 || len(s.inventory.GrantedItems("pack77:monster42:deck13")) != 1 { - t.Fatalf("final reward deck=%d", loaded) - } - if len(s.inventory.GrantedItems("pack77:monster42:deck11")) != 0 { - t.Fatal("granted first phase reward") - } -} diff --git a/go/internal/server/bootstrap/bootstrap_test.go b/go/internal/server/bootstrap/bootstrap_test.go deleted file mode 100644 index 3e07e1a..0000000 --- a/go/internal/server/bootstrap/bootstrap_test.go +++ /dev/null @@ -1,82 +0,0 @@ -package bootstrap - -import ( - "testing" - "time" - - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" -) - -func TestMaintenance(t *testing.T) { - req := wire.AppendVarint(nil, 1, 2) - req = wire.AppendVarint(req, 2, 8) - response, err := Maintenance(versionconfig.Game(), versionconfig.Bundle(), req) - if err != nil { - t.Fatal(err) - } - market, found, err := wire.Bytes(response, 1) - if err != nil || !found { - t.Fatalf("market info missing: %v", err) - } - typ, _, _ := wire.Varint(market, 1) - if typ != 4 { - t.Fatalf("market type does not match response contract: %d", typ) - } - connect, _, _ := wire.Varint(response, 3) - user, _, _ := wire.Varint(response, 4) - if connect != 1 || user == 0 { - t.Fatalf("client would enter update branch: connect=%d user=%d", connect, user) - } -} - -func TestServerInfoNoOfficialEndpoints(t *testing.T) { - c := Config{BaseURL: "http://127.0.0.1:8080/game/", CDNURL: "http://127.0.0.1:8080/assets/ServerData", Version: versionconfig.Game(), BundleVer: versionconfig.Bundle()} - if err := c.Validate(); err != nil { - t.Fatal(err) - } - response := ServerInfo(c) - info, _, _ := wire.Bytes(response, 1) - address, _, _ := wire.Bytes(info, 2) - if string(address) != c.BaseURL { - t.Fatalf("game URL: %q", address) - } - if _, exists, _ := wire.Bytes(info, 8); exists { - t.Fatal("unexpected GameData URL") - } - if v, _, _ := wire.Varint(ServerNowTime(time.UnixMilli(1234567)), 1); v != 1234567 { - t.Fatalf("time: %d", v) - } -} - -func TestConfigAcceptsHTTPSPublicOrigin(t *testing.T) { - cfg := Config{ - BaseURL: "https://bd2.example.com/game/", CDNURL: "https://bd2.example.com/assets/ServerData", - Version: "client", BundleVer: "bundle", - } - if err := cfg.Validate(); err != nil { - t.Fatalf("HTTPS self-hosted server config rejected: %v", err) - } -} - -func TestServerInfoIncludesLocalGameData(t *testing.T) { - c := Config{ - BaseURL: "http://127.0.0.1:8080/game/", - CDNURL: "http://127.0.0.1:8080/assets/ServerData", - Version: versionconfig.Game(), BundleVer: versionconfig.Bundle(), - GameDataURL: "http://127.0.0.1:8080/assets/GameData", - GameDataVer: "20260921140855", - } - if err := c.Validate(); err != nil { - t.Fatal(err) - } - info, found, err := wire.Bytes(ServerInfo(c), 1) - if err != nil || !found { - t.Fatalf("server info missing: %v", err) - } - url, _, _ := wire.Bytes(info, 8) - version, _, _ := wire.Bytes(info, 9) - if string(url) != c.GameDataURL || string(version) != c.GameDataVer { - t.Fatalf("GameData mismatch: url=%q version=%q", url, version) - } -} diff --git a/go/internal/server/calendar/binary.go b/go/internal/server/calendar/binary.go deleted file mode 100644 index dd0e2a5..0000000 --- a/go/internal/server/calendar/binary.go +++ /dev/null @@ -1,194 +0,0 @@ -package calendar - -import ( - "bytes" - "crypto/sha256" - "encoding/binary" - "fmt" - "unicode/utf8" -) - -const MaxFileSize = 8 * 1024 * 1024 -const maxRows = 100000 -const maxString = 64 * 1024 -const headerSize = 46 - -var magic = []byte{'B', 'D', '2', 'S', 'C', 'H', 0, 0} - -type encoder struct { - data []byte - err error - rows uint64 -} - -func (e *encoder) u64(v uint64) { - if e.room(8) { - e.data = binary.LittleEndian.AppendUint64(e.data, v) - } -} -func (e *encoder) u32(v uint32) { - if e.room(4) { - e.data = binary.LittleEndian.AppendUint32(e.data, v) - } -} -func (e *encoder) b(v bool) { - if !e.room(1) { - return - } - if v { - e.data = append(e.data, 1) - } else { - e.data = append(e.data, 0) - } -} -func (e *encoder) str(v string) { - if e.err != nil { - return - } - if len(v) > maxString || !utf8.ValidString(v) { - e.err = fmt.Errorf("calendar: invalid/oversized UTF8 string") - return - } - if !e.room(4 + len(v)) { - return - } - e.u32(uint32(len(v))) - e.data = append(e.data, v...) -} -func (e *encoder) count(n int) { - if e.err != nil { - return - } - e.rows += uint64(n) - if n > maxRows || e.rows > maxRows { - e.err = fmt.Errorf("calendar: row limit exceeded") - return - } - e.u32(uint32(n)) -} -func (e *encoder) room(n int) bool { - if e.err != nil { - return false - } - if n > MaxFileSize-headerSize-len(e.data) { - e.err = fmt.Errorf("calendar: file size limit exceeded") - return false - } - return true -} - -type decoder struct { - data []byte - pos int - err error - rows uint64 -} - -func (d *decoder) take(n int) []byte { - if d.err != nil { - return nil - } - if n < 0 || n > len(d.data)-d.pos { - d.err = fmt.Errorf("calendar: truncated payload") - return nil - } - v := d.data[d.pos : d.pos+n] - d.pos += n - return v -} -func (d *decoder) u64() uint64 { - v := d.take(8) - if len(v) != 8 { - return 0 - } - return binary.LittleEndian.Uint64(v) -} -func (d *decoder) u32() uint32 { - v := d.take(4) - if len(v) != 4 { - return 0 - } - return binary.LittleEndian.Uint32(v) -} -func (d *decoder) b() bool { - v := d.take(1) - if len(v) != 1 { - return false - } - if v[0] > 1 { - d.err = fmt.Errorf("calendar: invalid boolean/presence") - } - return v[0] == 1 -} -func (d *decoder) str() string { - n := d.u32() - if n > maxString { - d.err = fmt.Errorf("calendar: string limit exceeded") - return "" - } - v := d.take(int(n)) - if !utf8.Valid(v) { - d.err = fmt.Errorf("calendar: invalid UTF8") - } - return string(v) -} -func (d *decoder) count() int { - n := d.u32() - d.rows += uint64(n) - if n > maxRows || d.rows > maxRows || uint64(n) > uint64(len(d.data)-d.pos) { - d.err = fmt.Errorf("calendar: row count limit/truncation") - return 0 - } - return int(n) -} - -// MarshalBinary emits format 1's fixed-order record payload and checksum. -func MarshalBinary(m Manifest) ([]byte, error) { - if m.SchemaVersion != 1 { - return nil, fmt.Errorf("calendar: unsupported schema") - } - e := &encoder{} - e.manifest(m) - if e.err != nil { - return nil, e.err - } - if len(e.data) > MaxFileSize-headerSize { - return nil, fmt.Errorf("calendar: file size limit exceeded") - } - out := append([]byte(nil), magic...) - out = binary.LittleEndian.AppendUint16(out, 1) - out = binary.LittleEndian.AppendUint32(out, uint32(len(e.data))) - sum := sha256.Sum256(e.data) - out = append(out, sum[:]...) - return append(out, e.data...), nil -} - -// UnmarshalBinary rejects damaged, unsupported, oversized and trailing data. -func UnmarshalBinary(raw []byte) (Manifest, error) { - var m Manifest - if len(raw) < headerSize || len(raw) > MaxFileSize { - return m, fmt.Errorf("calendar: invalid file size") - } - if !bytes.Equal(raw[:8], magic) || binary.LittleEndian.Uint16(raw[8:10]) != 1 { - return m, fmt.Errorf("calendar: unsupported magic/format") - } - n := binary.LittleEndian.Uint32(raw[10:14]) - if uint64(n) != uint64(len(raw)-headerSize) { - return m, fmt.Errorf("calendar: payload length mismatch") - } - payload := raw[headerSize:] - sum := sha256.Sum256(payload) - if !bytes.Equal(raw[14:46], sum[:]) { - return m, fmt.Errorf("calendar: checksum mismatch") - } - d := &decoder{data: payload} - m = d.manifest() - if d.err != nil { - return Manifest{}, d.err - } - if d.pos != len(payload) { - return Manifest{}, fmt.Errorf("calendar: trailing payload") - } - m.SchemaVersion = 1 - return m, nil -} diff --git a/go/internal/server/calendar/binary_test.go b/go/internal/server/calendar/binary_test.go deleted file mode 100644 index 4c6c80b..0000000 --- a/go/internal/server/calendar/binary_test.go +++ /dev/null @@ -1,115 +0,0 @@ -package calendar - -import ( - "bytes" - "crypto/sha256" - "encoding/binary" - "os" - "path/filepath" - "reflect" - "testing" -) - -func TestInstalledBinaryFilesReencodeExactly(t *testing.T) { - paths, e := filepath.Glob("../../../../schedules/*.bd2schedule") - if e != nil || len(paths) != 6 { - t.Fatalf("installed files=%d err=%v", len(paths), e) - } - for _, path := range paths { - raw, e := os.ReadFile(path) - if e != nil { - t.Fatal(e) - } - m, e := UnmarshalBinary(raw) - if e != nil { - t.Fatal(e) - } - actual, e := MarshalBinary(m) - if e != nil { - t.Fatal(e) - } - if !bytes.Equal(raw, actual) { - t.Fatalf("canonical bytes changed %s", path) - } - } -} -func TestBinaryRejectsInvalidPresence(t *testing.T) { - raw, e := MarshalBinary(baseManifest()) - if e != nil { - t.Fatal(e) - } - d := &decoder{data: raw[headerSize:]} - d.str() - d.str() - d.str() - d.count() - d.count() - d.count() - raw[headerSize+d.pos] = 2 - sum := sha256.Sum256(raw[headerSize:]) - copy(raw[14:46], sum[:]) - if _, e = UnmarshalBinary(raw); e == nil { - t.Fatal("invalid optional presence accepted despite valid checksum") - } -} - -func TestBinaryRoundTripAllRecords(t *testing.T) { - m := baseManifest() - m.Events = []Event{{UID: 5, Type: 11, ID: 7, Start: "2026-01-01T00:00:00Z", End: "2026-02-01T00:00:00Z"}} - m.Gacha = []Gacha{{GroupID: 2, FreeCountBonus: true}} - m.StepUps = []Gacha{{GroupID: 3}} - m.Regular = &Regular{Contents: []Content{{ID: 4, Current: Season{ID: 8, Error: true}, Next: Season{ID: 9, Return: true}}}} - m.MonsterHunt = &MonsterHunt{Seasons: []Hunt{{HuntID: 6, CostumeBanIDs: []uint64{5}, BurstBanIDs: []uint64{7}, IndependentFlag: true}}, History: []HuntHistory{{Season: 5, HuntID: 6, Hidden: true}}} - m.CashProducts = []CashProduct{{GroupID: 1, ProductID: 2, EventIndex: 3}} - m.EventHubs = []EventHub{{HubID: 4, Settings: []HubSetting{{Slot: 5, EventUIDs: []uint64{6, 7}}}}} - m.MiniGameHubs = []MiniGameHub{{Slot: 8, EventUID: 5}} - raw, e := MarshalBinary(m) - if e != nil { - t.Fatal(e) - } - actual, e := UnmarshalBinary(raw) - if e != nil { - t.Fatal(e) - } - if !reflect.DeepEqual(actual, m) { - t.Fatalf("roundtrip changed records: %+v", actual) - } -} -func TestBinaryRejectsCorruptionLimitsAndTrailing(t *testing.T) { - raw, e := MarshalBinary(baseManifest()) - if e != nil { - t.Fatal(e) - } - for n := range len(raw) { - if _, e = UnmarshalBinary(raw[:n]); e == nil { - t.Fatalf("accepted truncation %d", n) - } - } - for _, offset := range []int{0, 8, 10, 14, 46} { - bad := append([]byte(nil), raw...) - bad[offset] ^= 0xff - if _, e = UnmarshalBinary(bad); e == nil { - t.Fatalf("accepted corruption %d", offset) - } - } - if _, e = UnmarshalBinary(append(raw, 0)); e == nil { - t.Fatal("accepted trailing data") - } - m := baseManifest() - m.Revision = string(make([]byte, maxString+1)) - if _, e = MarshalBinary(m); e == nil { - t.Fatal("oversized string accepted") - } - m = baseManifest() - m.Events = make([]Event, maxRows+1) - if _, e = MarshalBinary(m); e == nil { - t.Fatal("oversized count accepted") - } - bad := append([]byte(nil), raw...) - binary.LittleEndian.PutUint32(bad[46:50], ^uint32(0)) - sum := sha256.Sum256(bad[46:]) - copy(bad[14:46], sum[:]) - if _, e = UnmarshalBinary(bad); e == nil { - t.Fatal("overflow string length accepted") - } -} diff --git a/go/internal/server/calendar/design.go b/go/internal/server/calendar/design.go deleted file mode 100644 index f5769aa..0000000 --- a/go/internal/server/calendar/design.go +++ /dev/null @@ -1,133 +0,0 @@ -package calendar - -import ( - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "fmt" -) - -// ValidateDesign checks playable identities against their domain's installed -// GameData. Announcement-only types 3/6/14/15/16/18 have no local gameplay -// design resolver and receive only the manifest's protocol/format validation. -func (s *Set) ValidateDesign(root, version string) error { - cash, e := gamedata.LoadCashCatalog(root, version) - if e != nil { - return e - } - keys := map[gamedata.CashProductKey]bool{} - for _, v := range cash.Products { - keys[v.Key] = true - } - for _, v := range s.CashProducts { - if !keys[gamedata.CashProductKey{GroupID: v.GroupID, ProductID: v.ProductID, SaleGroup: v.SaleGroup}] { - return fmt.Errorf("calendar: cash product %d/%d/%d missing GameData", v.GroupID, v.ProductID, v.SaleGroup) - } - } - if s.MonsterHunt != nil { - ids, e := gamedata.LoadMonsterHuntIDs(root, version) - if e != nil { - return e - } - for _, v := range s.MonsterHunt.Seasons { - if !ids[v.HuntID] { - return fmt.Errorf("calendar: active/playable hunt %d missing GameData", v.HuntID) - } - } - } - play, e := gamedata.LoadEventPlayCatalog(root, version) - if e != nil { - return e - } - tasks, e := gamedata.LoadEventTasksDesign(root, version) - if e != nil { - return e - } - exchange, e := gamedata.LoadEventExchangeCatalog(root, version) - if e != nil { - return e - } - actions, e := gamedata.LoadEventActionsDesign(root, version) - if e != nil { - return e - } - packs, e := gamedata.LoadCalendarPackIDs(root, version) - if e != nil { - return e - } - for _, h := range s.EventHubs { - if _, e = play.Row("PackEventHubTable", 14, h.HubID); e != nil { - return fmt.Errorf("calendar: hub %d: %w", h.HubID, e) - } - } - if err := s.validateMiniHubBindings(play); err != nil { - return err - } - uid := map[uint64]events.Schedule{} - for _, v := range s.Events { - if v.UID != 0 { - uid[v.UID] = v - } - } - for _, h := range s.MiniGameHubs { - v, ok := uid[h.EventUID] - if !ok || v.Type != 11 { - return fmt.Errorf("calendar: mini game slot %d references non-minigame event %d", h.Slot, h.EventUID) - } - if _, e := play.Row("PackEventMiniGameTable", 8, v.ID); e != nil { - return e - } - - } - for _, v := range s.Events { - valid := true - var err error - switch v.Type { - case 0: - _, valid = tasks.Attendance[v.ID] - case 1: - valid = false - for k := range tasks.LimitRewards { - if k[0] == v.ID { - valid = true - break - } - } - case 4: - _, valid = tasks.MissionGroups[v.ID] - case 5: - _, valid = tasks.Passes[v.ID] - case 7: - _, valid = exchange.Groups[v.ID] - case 8: - valid = packs[v.ID] - case 9: - _, err = play.Row("PackEventBattleGroupTable", 3, v.ID) - case 10: - _, err = play.Row("PackEventStoryGroupTable", 1, v.ID) - case 11: - _, err = play.Row("PackEventMiniGameTable", 8, v.ID) - case 12, 13, 17, 19: - _, err = gamedata.LoadEventGame(root, version, v.Type, v.ID) - case 20: - _, valid = actions.Row("TacticsBingoGroupTable", 3, v.ID) - case 21: - _, valid = actions.Row("FieldSpawnEventTable", 5, v.ID) - case 22: - _, valid = actions.Row("FireworksTable", 4, v.ID) - case 23: - _, valid = actions.Row("VotingEventTable", 6, v.ID) - case 24: - _, valid = actions.Row("FriendshipSpecialEpisodeTable", 5, v.ID) - case 25: - id := v.SubID - if id == 0 { - id = v.ID - } - _, valid = actions.Row("CafeteriaEventTable", 5, id) - } - if !valid || err != nil { - return fmt.Errorf("calendar: event uid %d type %d id %d missing GameData: %v", v.UID, v.Type, v.ID, err) - } - } - return nil -} diff --git a/go/internal/server/calendar/load_test.go b/go/internal/server/calendar/load_test.go deleted file mode 100644 index 008e393..0000000 --- a/go/internal/server/calendar/load_test.go +++ /dev/null @@ -1,138 +0,0 @@ -package calendar - -import ( - "bd2server/internal/server/readonly" - "bd2server/internal/server/wire" - "os" - "path/filepath" - "testing" -) - -func writeManifest(t *testing.T, dir, name string, m Manifest) { - t.Helper() - raw, e := MarshalBinary(m) - if e != nil { - t.Fatal(e) - } - if e = os.WriteFile(filepath.Join(dir, name), raw, 0600); e != nil { - t.Fatal(e) - } -} -func baseManifest() Manifest { - return Manifest{SchemaVersion: 1, Revision: "2026-10-05", GameVersion: "2.35.10", GameDataVersion: "2.35.10"} -} -func TestInstalledCalendarsContainAllRuntimeDomains(t *testing.T) { - set, e := LoadDirectory("../../../../schedules", "2.35.10", "20260923193640") - if e != nil { - t.Fatal(e) - } - if len(set.Events) != 54 || len(set.GachaSeed.Schedules) != 11 || len(set.GachaSeed.StepUps) != 2 || set.RegularService == nil || len(set.RegularService.Contents) != 9 || set.MonsterHunt == nil || len(set.MonsterHunt.History) != 80 || len(set.CashProducts) != 61 || len(set.EventHubs) != 4 || len(set.MiniGameHubs) != 6 { - t.Fatalf("installed calendar omitted a domain: events=%d gacha=%d steps=%d cash=%d hubs=%d mini=%d", len(set.Events), len(set.GachaSeed.Schedules), len(set.GachaSeed.StepUps), len(set.CashProducts), len(set.EventHubs), len(set.MiniGameHubs)) - } - var mooncake, chained *EventHub - for i := range set.EventHubs { - h := &set.EventHubs[i] - if h.UID == 3 && h.HubID == 1003 { - mooncake = h - } - if h.UID == 76 && h.HubID == 59 { - chained = h - } - } - if mooncake == nil || mooncake.PlayEnd != "2026-09-30T23:59:59.000Z" || len(mooncake.Settings) != 3 || mooncake.Settings[2].EventUIDs[0] != 10000034 { - t.Fatal("published Mooncake Wishes calendar omitted its deferred board slot") - } - if chained == nil || len(chained.Settings) != 6 || chained.Settings[2].EventUIDs[0] != 80 { - t.Fatal("published Chained Soldier hub omitted its event shop") - } -} -func TestMultiFileCalendarAtomicDeterministicAndIndependentRevision(t *testing.T) { - dir := t.TempDir() - a := baseManifest() - a.Revision = "a" - a.Gacha = []Gacha{{GroupID: 9, Start: "2026-10-01T00:00:00Z", End: "2026-10-02T00:00:00Z"}} - writeManifest(t, dir, "z.bd2schedule", a) - b := baseManifest() - b.Revision = "b" - b.Gacha = []Gacha{{GroupID: 1, Start: "2027-01-01T09:00:00+09:00", End: "2027-01-02T09:00:00+09:00"}, {GroupID: 9, Start: "2026-10-02T00:00:00Z", End: "2026-10-03T00:00:00Z"}} - writeManifest(t, dir, "a.bd2schedule", b) - set, e := LoadDirectory(dir, "2.35.10", "2.35.10") - if e != nil { - t.Fatal(e) - } - if len(set.GachaSeed.Schedules) != 3 || set.GachaSeed.Schedules[0].GroupID != 1 || set.Revisions[0] != "b" { - t.Fatalf("unexpected merged set %+v", set) - } - b.Revision = "changed without game update" - b.Gacha[0].End = "2027-01-03T09:00:00+09:00" - writeManifest(t, dir, "a.bd2schedule", b) - changed, e := LoadDirectory(dir, "2.35.10", "2.35.10") - if e != nil || changed.GachaSeed.Schedules[0].EndTime == set.GachaSeed.Schedules[0].EndTime { - t.Fatalf("calendar update ignored: %v", e) - } - if e = os.WriteFile(filepath.Join(dir, "broken.bd2schedule"), []byte("broken binary"), 0600); e != nil { - t.Fatal(e) - } - if partial, e := LoadDirectory(dir, "2.35.10", "2.35.10"); e == nil || partial != nil { - t.Fatal("bad file returned partial calendar") - } -} -func TestStrictCalendarRejectsConflictsAndInvalidData(t *testing.T) { - for _, mutate := range []func(*Manifest){func(m *Manifest) { m.GameDataVersion = "wrong" }, func(m *Manifest) { m.Gacha[0].Start = "2026-10-01" }, func(m *Manifest) { m.Gacha[0].End = m.Gacha[0].Start }, func(m *Manifest) { m.Gacha = append(m.Gacha, m.Gacha[0]) }, func(m *Manifest) { m.Gacha[0].GroupID = 1 << 32 }} { - dir := t.TempDir() - m := baseManifest() - m.Gacha = []Gacha{{GroupID: 1, Start: "2026-10-01T00:00:00Z", End: "2026-10-02T00:00:00Z"}} - mutate(&m) - writeManifest(t, dir, "one.bd2schedule", m) - if _, e := LoadDirectory(dir, "2.35.10", "2.35.10"); e == nil { - t.Fatal("invalid calendar accepted") - } - } - dir := t.TempDir() - m := baseManifest() - m.Events = []Event{{UID: 77, Type: 12, ID: 3, Start: "2026-10-01T00:00:00Z", End: "2026-10-02T00:00:00Z"}} - writeManifest(t, dir, "a.bd2schedule", m) - writeManifest(t, dir, "b.bd2schedule", m) - if _, e := LoadDirectory(dir, "2.35.10", "2.35.10"); e == nil { - t.Fatal("duplicate UID across files accepted") - } -} -func TestReadonlyCalendarReplacementPreservesStaticResponses(t *testing.T) { - set := &Set{CashProducts: []CashProduct{{GroupID: 11, ProductID: 22, EventIndex: 99}}, MonsterHunt: &MonsterHunt{StartRegularSeason: 5, History: []HuntHistory{{Season: 78, HuntID: 73, Hidden: true}}}} - old := &readonly.Seed{Version: "2.35.10", Responses: map[string]readonly.Response{"/Static": {PacketCode: 17, Fields: []readonly.Field{scalar(1, 10)}}, "/CashShopInfo": {PacketCode: 60, Fields: []readonly.Field{scalar(2, 1)}}}} - result, e := set.ApplyReadonly(old) - if e != nil { - t.Fatal(e) - } - if len(old.Responses["/CashShopInfo"].Fields) != 1 || result.Responses["/Static"].PacketCode != 17 { - t.Fatal("source seed changed/static endpoint lost") - } - _, raw, handled, e := result.Handle("/CashShopInfo", wire.AppendVarint(nil, 1, 1)) - if e != nil || !handled { - t.Fatal(e) - } - p, found, e := wire.Bytes(raw, 1) - if e != nil || !found { - t.Fatal("missing product") - } - id, _, _ := wire.Varint(p, 8) - if id != 99 { - t.Fatalf("event index=%d", id) - } - if _, found, _ = wire.Varint(raw, 2); found { - t.Fatal("stale reset timestamp retained") - } -} -func TestInstalledCalendarRealGameData(t *testing.T) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("BD2_REAL_GAMEDATA not configured") - } - set, e := LoadDirectory("../../../../schedules", "2.35.10", "20260923193640") - if e != nil { - t.Fatal(e) - } - if e = set.ValidateDesign(root, "20260923193640"); e != nil { - t.Fatal(e) - } -} diff --git a/go/internal/server/calendar/mini_hubs.go b/go/internal/server/calendar/mini_hubs.go deleted file mode 100644 index ba49c4e..0000000 --- a/go/internal/server/calendar/mini_hubs.go +++ /dev/null @@ -1,113 +0,0 @@ -package calendar - -import ( - "fmt" - - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -// validateMiniHubBindings rejects invalid published routes at startup instead -// of letting a client open an incompatible prefab or use another activity UID. -func (s *Set) validateMiniHubBindings(design *gamedata.EventPlayCatalog) error { - uid := make(map[uint64]events.Schedule) - for _, v := range s.Events { - if v.UID != 0 { - uid[v.UID] = v - } - } - value := func(row []byte, field int) uint64 { v, _, _ := wire.Varint(row, field); return v } - for _, hub := range s.EventHubs { - table, err := design.Row("PackEventHubTable", 14, hub.HubID) - if err != nil { - return err - } - if value(table, 13) != 1 { - continue - } - if hub.UID == 0 { - return fmt.Errorf("calendar: mini hub %d requires a nonzero UID", hub.HubID) - } - start, err := timestamp(hub.Start) - if err != nil { - return err - } - playEnd, err := timestamp(hub.PlayEnd) - if err != nil { - return err - } - end, err := timestamp(hub.End) - if err != nil { - return err - } - for _, binding := range hub.Settings { - var slot []byte - for _, row := range design.Rows("PackEventListTable", 6, hub.HubID) { - if value(row, 11) != binding.Slot { - continue - } - if slot != nil { - return fmt.Errorf("calendar: mini hub %d slot %d is ambiguous in GameData", hub.HubID, binding.Slot) - } - slot = row - } - if slot == nil { - return fmt.Errorf("calendar: mini hub %d slot %d missing GameData", hub.HubID, binding.Slot) - } - contentType, contentID := value(slot, 9), value(slot, 7) - if binding.ProgressType != contentType { - return fmt.Errorf("calendar: mini hub %d slot %d content type %d, want %d", hub.HubID, binding.Slot, binding.ProgressType, contentType) - } - eventType, supported := gamedata.MiniHubEventType(contentType) - // Mini stories and NPC quizzes have their own slot UID namespace, - // distinct from Define_EventType (13 there means bingo). Their group - // and availability derive from this static slot and the hub window. - if contentType == 13 || contentType == 14 { - if len(binding.EventUIDs) != 1 || binding.EventUIDs[0] == 0 { - return fmt.Errorf("calendar: mini hub %d slot %d requires one local content UID", hub.HubID, binding.Slot) - } - if _, collision := uid[binding.EventUIDs[0]]; collision { - return fmt.Errorf("calendar: mini hub %d slot %d content UID collides with global event", hub.HubID, binding.Slot) - } - if value(slot, 4) != 0 { - return fmt.Errorf("calendar: mini hub %d story/quiz slot has invalid end type", hub.HubID) - } - continue - } - if !supported { - return fmt.Errorf("calendar: mini hub %d slot %d content type %d has no supported scheduled route", hub.HubID, binding.Slot, contentType) - } - endType := value(slot, 4) - if endType > 1 { - return fmt.Errorf("calendar: mini hub %d slot %d unsupported end type %d", hub.HubID, binding.Slot, endType) - } - matches := 0 - for _, id := range binding.EventUIDs { - child, ok := uid[id] - if !ok || child.Type != eventType || child.ID != contentID || child.SubID != 0 { - return fmt.Errorf("calendar: mini hub %d slot %d references incompatible event UID %d", hub.HubID, binding.Slot, id) - } - childStart, childEnd := child.Start, child.End - if childStart < int64(start) { - childStart = int64(start) - } - if childEnd > int64(end) { - childEnd = int64(end) - } - // Project policy: EndDateType=0 closes the slot at PlayEnd; - // EndDateType=1 allows it through the final hub End window. - if endType == 0 && childEnd > int64(playEnd) { - childEnd = int64(playEnd) - } - if childStart < childEnd { - matches++ - } - } - if matches != 1 { - return fmt.Errorf("calendar: mini hub %d slot %d has %d schedules in its window, want one", hub.HubID, binding.Slot, matches) - } - } - } - return nil -} diff --git a/go/internal/server/calendar/mini_hubs_test.go b/go/internal/server/calendar/mini_hubs_test.go deleted file mode 100644 index 4902f87..0000000 --- a/go/internal/server/calendar/mini_hubs_test.go +++ /dev/null @@ -1,81 +0,0 @@ -package calendar - -import ( - "testing" - "time" - - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -func TestMiniHubPublishedRoutesValidateIndependentEnumsAndWindows(t *testing.T) { - row := func(fields ...uint64) []byte { - var b []byte - for i := 0; i < len(fields); i += 2 { - b = wire.AppendVarint(b, int(fields[i]), fields[i+1]) - } - return b - } - start := time.Date(2030, 1, 1, 0, 0, 0, 0, time.UTC) - format := func(d time.Duration) string { return start.Add(d).Format(time.RFC3339) } - design := &gamedata.EventPlayCatalog{Tables: map[string][][]byte{ - "PackEventHubTable": {row(14, 7, 13, 1)}, - "PackEventListTable": {row(6, 7, 10, 42, 11, 5, 9, 12, 7, 99, 4, 1)}, - }} - fresh := func() *Set { - return &Set{ - Events: []events.Schedule{{UID: 21, Type: 19, ID: 99, Start: start.UnixMilli(), End: start.Add(48 * time.Hour).UnixMilli()}}, - EventHubs: []EventHub{{UID: 1, HubID: 7, Start: format(0), PlayEnd: format(24 * time.Hour), End: format(48 * time.Hour), Settings: []HubSetting{{Slot: 5, ProgressType: 12, EventUIDs: []uint64{21}}}}}, - } - } - if err := fresh().validateMiniHubBindings(design); err != nil { - t.Fatal(err) - } - for _, test := range []struct { - name string - change func(*Set) - }{ - {"wrong content enum", func(s *Set) { s.EventHubs[0].Settings[0].ProgressType = 19 }}, - {"table id is not slot index", func(s *Set) { s.EventHubs[0].Settings[0].Slot = 42 }}, - {"missing UID", func(s *Set) { s.EventHubs[0].Settings[0].EventUIDs = []uint64{22} }}, - {"wrong event enum", func(s *Set) { s.Events[0].Type = 12 }}, - {"wrong design id", func(s *Set) { s.Events[0].ID = 7 }}, - {"wrong sub identity", func(s *Set) { s.Events[0].SubID = 7 }}, - {"no window overlap", func(s *Set) { - s.Events[0].Start = start.Add(72 * time.Hour).UnixMilli() - s.Events[0].End = start.Add(96 * time.Hour).UnixMilli() - }}, - {"ambiguous schedule", func(s *Set) { - other := s.Events[0] - other.UID = 22 - s.Events = append(s.Events, other) - s.EventHubs[0].Settings[0].EventUIDs = append(s.EventHubs[0].Settings[0].EventUIDs, 22) - }}, - } { - t.Run(test.name, func(t *testing.T) { - s := fresh() - test.change(s) - if err := s.validateMiniHubBindings(design); err == nil { - t.Fatal("invalid published route accepted") - } - }) - } - // A future hub must validate without comparing its dates to the current clock. - design.Tables["PackEventListTable"][0] = row(6, 7, 10, 42, 11, 5, 9, 13, 7, 99, 4, 1) - s := fresh() - s.EventHubs[0].Settings[0].ProgressType = 13 - if err := s.validateMiniHubBindings(design); err == nil { - t.Fatal("quiz borrowed global bingo event type") - } - design.Tables["PackEventListTable"][0] = row(6, 7, 10, 42, 11, 11, 9, 13, 7, 99) - s = fresh() - s.EventHubs[0].Settings = []HubSetting{{Slot: 11, ProgressType: 13, EventUIDs: []uint64{10000032}}} - if err := s.validateMiniHubBindings(design); err != nil { - t.Fatalf("independent quiz content UID rejected: %v", err) - } - s.EventHubs[0].Settings[0].EventUIDs = []uint64{21} - if err := s.validateMiniHubBindings(design); err == nil { - t.Fatal("quiz content UID collided with global event") - } -} diff --git a/go/internal/server/calendar/published_complete_test.go b/go/internal/server/calendar/published_complete_test.go deleted file mode 100644 index 6dc915c..0000000 --- a/go/internal/server/calendar/published_complete_test.go +++ /dev/null @@ -1,41 +0,0 @@ -package calendar - -import "testing" - -// Current-version publication facts were verified against the client schemas, -// GameData identities and 2026-10-05 schedule comparison. No capture is read. -func TestPublishedRegularAndGachaCalendarCompleteness(t *testing.T) { - set, err := LoadDirectory("../../../../schedules", "2.35.10", "20260923193640") - if err != nil { - t.Fatal(err) - } - want := map[uint64][2]uint64{1: {163, 164}, 2: {58, 58}, 3: {131, 999999}, 4: {877, 999999}, 5: {30, 31}, 6: {27, 28}, 7: {15, 999999}, 8: {44, 45}, 9: {2, 999999}} - if len(set.RegularService.Contents) != len(want) || set.RegularService.CalculateMilliseconds != 32400000 { - t.Fatal("regular content calendar incomplete") - } - for _, c := range set.RegularService.Contents { - if want[c.ID] != [2]uint64{c.Current.ID, c.Next.ID} { - t.Fatalf("content %d stale seasons %d/%d", c.ID, c.Current.ID, c.Next.ID) - } - } - current := map[uint64]bool{166: true, 30010: true, 30011: true, 206: true, 205: true, 208: true, 153: true, 72: true, 71: true, 207: true} - for _, g := range set.GachaSeed.Schedules { - if current[g.GroupID] { - delete(current, g.GroupID) - if g.EndTime != 1791417599000 { - t.Fatalf("gacha %d truncated window", g.GroupID) - } - } - } - if len(current) != 0 { - t.Fatalf("missing published gacha groups %v", current) - } - if len(set.GachaSeed.StepUps) != 2 { - t.Fatal("step-up schedule omitted") - } - for _, g := range set.GachaSeed.StepUps { - if g.GroupID != 29 && g.GroupID != 30 || g.EndTime != 1791417599000 { - t.Fatalf("step-up schedule incorrect %+v", g) - } - } -} diff --git a/go/internal/server/calendar/records.go b/go/internal/server/calendar/records.go deleted file mode 100644 index 02f1c03..0000000 --- a/go/internal/server/calendar/records.go +++ /dev/null @@ -1,312 +0,0 @@ -package calendar - -import "bd2server/internal/server/schedule" - -func (e *encoder) manifest(v Manifest) { - e.str(v.Revision) - e.str(v.GameVersion) - e.str(v.GameDataVersion) - e.count(len(v.Events)) - for _, item := range v.Events { - e.event(item) - } - e.count(len(v.Gacha)) - for _, item := range v.Gacha { - e.gacha(item) - } - e.count(len(v.StepUps)) - for _, item := range v.StepUps { - e.gacha(item) - } - e.b(v.Regular != nil) - if v.Regular != nil { - e.regular(*v.Regular) - } - e.b(v.MonsterHunt != nil) - if v.MonsterHunt != nil { - e.monsterhunt(*v.MonsterHunt) - } - e.count(len(v.CashProducts)) - for _, item := range v.CashProducts { - e.cashproduct(item) - } - e.count(len(v.EventHubs)) - for _, item := range v.EventHubs { - e.eventhub(item) - } - e.count(len(v.MiniGameHubs)) - for _, item := range v.MiniGameHubs { - e.minigamehub(item) - } -} -func (d *decoder) manifest() Manifest { - var v Manifest - v.Revision = d.str() - v.GameVersion = d.str() - v.GameDataVersion = d.str() - for n := d.count(); n > 0 && d.err == nil; n-- { - v.Events = append(v.Events, d.event()) - } - for n := d.count(); n > 0 && d.err == nil; n-- { - v.Gacha = append(v.Gacha, d.gacha()) - } - for n := d.count(); n > 0 && d.err == nil; n-- { - v.StepUps = append(v.StepUps, d.gacha()) - } - if d.b() { - item := d.regular() - v.Regular = &item - } - if d.b() { - item := d.monsterhunt() - v.MonsterHunt = &item - } - for n := d.count(); n > 0 && d.err == nil; n-- { - v.CashProducts = append(v.CashProducts, d.cashproduct()) - } - for n := d.count(); n > 0 && d.err == nil; n-- { - v.EventHubs = append(v.EventHubs, d.eventhub()) - } - for n := d.count(); n > 0 && d.err == nil; n-- { - v.MiniGameHubs = append(v.MiniGameHubs, d.minigamehub()) - } - return v -} -func (e *encoder) event(v Event) { - e.u64(v.UID) - e.u64(v.Type) - e.u64(v.ID) - e.u64(v.SubID) - e.str(v.Start) - e.str(v.End) -} -func (d *decoder) event() Event { - var v Event - v.UID = d.u64() - v.Type = d.u64() - v.ID = d.u64() - v.SubID = d.u64() - v.Start = d.str() - v.End = d.str() - return v -} -func (e *encoder) gacha(v Gacha) { - e.u64(v.GroupID) - e.str(v.Start) - e.str(v.End) - e.b(v.FreeCountBonus) - e.b(v.CashCountBonus) -} -func (d *decoder) gacha() Gacha { - var v Gacha - v.GroupID = d.u64() - v.Start = d.str() - v.End = d.str() - v.FreeCountBonus = d.b() - v.CashCountBonus = d.b() - return v -} -func (e *encoder) season(v Season) { - e.u64(v.ID) - e.str(v.Start) - e.str(v.End) - e.u64(v.RankRewardGroupID) - e.b(v.Error) - e.b(v.Return) -} -func (d *decoder) season() Season { - var v Season - v.ID = d.u64() - v.Start = d.str() - v.End = d.str() - v.RankRewardGroupID = d.u64() - v.Error = d.b() - v.Return = d.b() - return v -} -func (e *encoder) content(v Content) { - e.u64(v.ID) - e.season(v.Current) - e.season(v.Next) -} -func (d *decoder) content() Content { - var v Content - v.ID = d.u64() - v.Current = d.season() - v.Next = d.season() - return v -} -func (e *encoder) regular(v Regular) { - e.u64(v.CalculateMilliseconds) - e.count(len(v.Contents)) - for _, item := range v.Contents { - e.content(item) - } - e.count(len(v.Regular)) - for _, item := range v.Regular { - e.regularseason(item) - } -} -func (d *decoder) regular() Regular { - var v Regular - v.CalculateMilliseconds = d.u64() - for n := d.count(); n > 0 && d.err == nil; n-- { - v.Contents = append(v.Contents, d.content()) - } - for n := d.count(); n > 0 && d.err == nil; n-- { - v.Regular = append(v.Regular, d.regularseason()) - } - return v -} -func (e *encoder) hunt(v Hunt) { - e.season(v.Season) - e.u64(v.HuntID) - e.u64(v.InfoOpenDay) - e.str(v.CalculateEndAt) - e.b(v.ErrorFlag) - e.b(v.IndependentFlag) - e.u64(v.RankRewardGroupID) - e.count(len(v.CostumeBanIDs)) - for _, item := range v.CostumeBanIDs { - e.u64(item) - } - e.count(len(v.BurstBanIDs)) - for _, item := range v.BurstBanIDs { - e.u64(item) - } -} -func (d *decoder) hunt() Hunt { - var v Hunt - v.Season = d.season() - v.HuntID = d.u64() - v.InfoOpenDay = d.u64() - v.CalculateEndAt = d.str() - v.ErrorFlag = d.b() - v.IndependentFlag = d.b() - v.RankRewardGroupID = d.u64() - for n := d.count(); n > 0 && d.err == nil; n-- { - v.CostumeBanIDs = append(v.CostumeBanIDs, d.u64()) - } - for n := d.count(); n > 0 && d.err == nil; n-- { - v.BurstBanIDs = append(v.BurstBanIDs, d.u64()) - } - return v -} -func (e *encoder) hunthistory(v HuntHistory) { - e.u64(v.Season) - e.u64(v.HuntID) - e.b(v.ErrorFlag) - e.b(v.Hidden) -} -func (d *decoder) hunthistory() HuntHistory { - var v HuntHistory - v.Season = d.u64() - v.HuntID = d.u64() - v.ErrorFlag = d.b() - v.Hidden = d.b() - return v -} -func (e *encoder) monsterhunt(v MonsterHunt) { - e.count(len(v.Seasons)) - for _, item := range v.Seasons { - e.hunt(item) - } - e.u64(v.StartRegularSeason) - e.count(len(v.History)) - for _, item := range v.History { - e.hunthistory(item) - } -} -func (d *decoder) monsterhunt() MonsterHunt { - var v MonsterHunt - for n := d.count(); n > 0 && d.err == nil; n-- { - v.Seasons = append(v.Seasons, d.hunt()) - } - v.StartRegularSeason = d.u64() - for n := d.count(); n > 0 && d.err == nil; n-- { - v.History = append(v.History, d.hunthistory()) - } - return v -} -func (e *encoder) cashproduct(v CashProduct) { - e.u64(v.GroupID) - e.u64(v.ProductID) - e.u64(v.SaleGroup) - e.str(v.Start) - e.str(v.End) - e.u64(v.EndDelayMinutes) - e.u64(v.EventIndex) -} -func (d *decoder) cashproduct() CashProduct { - var v CashProduct - v.GroupID = d.u64() - v.ProductID = d.u64() - v.SaleGroup = d.u64() - v.Start = d.str() - v.End = d.str() - v.EndDelayMinutes = d.u64() - v.EventIndex = d.u64() - return v -} -func (e *encoder) hubsetting(v HubSetting) { - e.u64(v.Slot) - e.u64(v.ProgressType) - e.count(len(v.EventUIDs)) - for _, item := range v.EventUIDs { - e.u64(item) - } -} -func (d *decoder) hubsetting() HubSetting { - var v HubSetting - v.Slot = d.u64() - v.ProgressType = d.u64() - for n := d.count(); n > 0 && d.err == nil; n-- { - v.EventUIDs = append(v.EventUIDs, d.u64()) - } - return v -} -func (e *encoder) eventhub(v EventHub) { - e.u64(v.UID) - e.u64(v.HubID) - e.str(v.Start) - e.str(v.PlayEnd) - e.str(v.End) - e.count(len(v.Settings)) - for _, item := range v.Settings { - e.hubsetting(item) - } -} -func (d *decoder) eventhub() EventHub { - var v EventHub - v.UID = d.u64() - v.HubID = d.u64() - v.Start = d.str() - v.PlayEnd = d.str() - v.End = d.str() - for n := d.count(); n > 0 && d.err == nil; n-- { - v.Settings = append(v.Settings, d.hubsetting()) - } - return v -} -func (e *encoder) minigamehub(v MiniGameHub) { - e.u64(v.Slot) - e.u64(v.EventUID) - e.u64(v.ProgressType) -} -func (d *decoder) minigamehub() MiniGameHub { - var v MiniGameHub - v.Slot = d.u64() - v.EventUID = d.u64() - v.ProgressType = d.u64() - return v -} -func (e *encoder) regularseason(v schedule.RegularSeason) { - e.u64(v.ContentID) - e.u64(v.Season) -} -func (d *decoder) regularseason() schedule.RegularSeason { - var v schedule.RegularSeason - v.ContentID = d.u64() - v.Season = d.u64() - return v -} diff --git a/go/internal/server/commerce/attendance.go b/go/internal/server/commerce/attendance.go deleted file mode 100644 index 33dc84b..0000000 --- a/go/internal/server/commerce/attendance.go +++ /dev/null @@ -1,142 +0,0 @@ -package commerce - -import ( - "crypto/sha256" - "encoding/json" - "fmt" - - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -type attendanceHandler interface { - Handle(string, []byte) (int, []byte, bool, error) -} - -// AttendanceHandler preserves the original event progress response and adds -// subscription grants in extension fields understood by the commerce plugin. -// All operations execute inside the transport's account transaction. -type AttendanceHandler struct { - Events attendanceHandler - Economy *EntitlementEconomy - LoginPasses *LoginPasses - Store stateio.Store -} - -type attendanceReceipt struct { - Digest string `json:"digest"` - Bundle []byte `json:"bundle"` -} - -func (h AttendanceHandler) Handle(path string, request []byte) (int, []byte, bool, error) { - return h.HandleSession(path, request, "") -} - -func (h AttendanceHandler) HandleSession(path string, request []byte, session string) (int, []byte, bool, error) { - if path != "/Attendance" { - return 0, nil, false, nil - } - if session == "" || h.Events == nil || h.Economy == nil || h.Store == nil { - return 0, nil, true, fmt.Errorf("commerce: attendance dependencies/session unavailable") - } - seq, found, err := wire.Varint(request, 1) - if err != nil || !found || seq == 0 { - return 0, nil, true, fmt.Errorf("commerce: missing attendance sequence") - } - code, response, handled, err := h.Events.Handle(path, request) - if err != nil || !handled { - return code, response, handled, err - } - // The event handler may already have granted ordinary attendance rewards. - // The client accepts exactly one reward envelope, so combine every grant in - // execution order under this wrapper's replay identity. - response, eventBundle, err := takeAttendanceRewardEnvelope(response) - if err != nil { - return code, nil, true, err - } - key := fmt.Sprintf("commerce_attendance:%x:%d", sha256.Sum256([]byte(session)), seq) - digest := fmt.Sprintf("%x", sha256.Sum256(request)) - var receipt attendanceReceipt - previous, err := h.Store.Load(key) - if err != nil { - return code, nil, true, err - } - var loginBundle []byte - if h.LoginPasses != nil { - var infos [][]byte - loginBundle, infos, err = h.LoginPasses.ClaimAndInfo(key) - if err != nil { - return code, nil, true, err - } - for _, info := range infos { - response = wire.AppendBytes(response, 6, info) - } - } - if previous != nil { - if err = json.Unmarshal(previous, &receipt); err != nil || receipt.Digest != digest { - return code, nil, true, fmt.Errorf("commerce: conflicting attendance replay") - } - } else { - receipt.Digest = digest - subscriptionBundle, err := h.Economy.ClaimSubscriptions(key) - if err != nil { - return code, nil, true, err - } - // Preserve events -> login pass -> subscription execution order and all - // repeated reward entries in the single client envelope. - receipt.Bundle = append(receipt.Bundle, eventBundle...) - receipt.Bundle = append(receipt.Bundle, loginBundle...) - receipt.Bundle = append(receipt.Bundle, subscriptionBundle...) - raw, err := json.Marshal(receipt) - if err != nil { - return code, nil, true, err - } - if err = h.Store.Save(key, raw); err != nil { - return code, nil, true, err - } - } - response, err = h.Economy.MergeAttendance(response) - if err != nil { - return code, nil, true, err - } - if len(receipt.Bundle) != 0 { - response = wire.AppendBytes(response, 1001, receipt.Bundle) - response = wire.AppendString(response, 1002, key) - } - return code, response, true, nil -} - -// Strip the child envelope before adding the combined one. Keep native and -// unrelated unknown fields byte-for-byte; reject ambiguous child envelopes -// rather than returning a response the client would silently ignore. -func takeAttendanceRewardEnvelope(response []byte) ([]byte, []byte, error) { - var native, bundle []byte - var receipt string - var hasBundle, hasReceipt bool - err := wire.Walk(response, func(f wire.Field) error { - switch f.Number { - case 1001: - if f.Type != 2 || hasBundle { - return fmt.Errorf("commerce: ambiguous attendance reward bundle") - } - hasBundle = true - bundle = append([]byte(nil), f.Value...) - case 1002: - if f.Type != 2 || hasReceipt { - return fmt.Errorf("commerce: ambiguous attendance reward receipt") - } - hasReceipt = true - receipt = string(f.Value) - default: - native = append(native, response[f.Start:f.End]...) - } - return nil - }) - if err != nil { - return nil, nil, err - } - if hasBundle != hasReceipt || hasReceipt && (receipt == "" || len(receipt) > 1024) { - return nil, nil, fmt.Errorf("commerce: incomplete attendance reward envelope") - } - return native, bundle, nil -} diff --git a/go/internal/server/commerce/attendance_test.go b/go/internal/server/commerce/attendance_test.go deleted file mode 100644 index 07669e6..0000000 --- a/go/internal/server/commerce/attendance_test.go +++ /dev/null @@ -1,256 +0,0 @@ -package commerce - -import ( - "bytes" - "testing" - "time" - - "bd2server/internal/server/events" - "bd2server/internal/server/eventtasks" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" -) - -type originalAttendance struct{} - -func (originalAttendance) Handle(path string, _ []byte) (int, []byte, bool, error) { - return 0, wire.AppendVarint(nil, 55, 123), path == "/Attendance", nil -} - -type attendanceBalances struct { - balance uint64 - calls int -} - -func (e *attendanceBalances) Apply(_ string, _ []gamedata.Reward, rewards []gamedata.Reward) ([]byte, error) { - e.calls++ - var bundle []byte - for _, r := range rewards { - e.balance += r.Count - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(player.Item{Type: r.Type, ID: r.ID, Count: r.Count})) - } - return bundle, nil -} - -func TestAttendanceMailsOrdinaryRewardsAndCombinesLoginPassAndSubscription(t *testing.T) { - e, items, _, _, now := entitlementFixture(t) - // eventtasks uses the production clock; align the commerce fixture with it - // while buying the subscription on the previous reset day. - *now = time.Now().UTC().Add(-24 * time.Hour) - balances := &attendanceBalances{} - e.base = balances - if _, err := e.Apply("subscription", nil, []gamedata.Reward{{Type: 19, ID: 38, Count: 1}}); err != nil { - t.Fatal(err) - } - *now = now.Add(24 * time.Hour) - registry := events.NewRegistry() - if err := registry.Replace([]events.Schedule{{UID: 17, Type: 0, ID: 1, Start: 1, End: now.Add(time.Hour).UnixMilli()}}); err != nil { - t.Fatal(err) - } - design := &gamedata.EventTasksDesign{ - Attendance: map[uint64]gamedata.EventAttendance{1: {ID: 1, Group: 10}}, - AttendanceRewards: map[uint64][]gamedata.EventAttendanceReward{10: {{Group: 10, ID: 1, Day: 1, Basic: gamedata.Reward{Type: 3, Count: 2}}}}, - } - tasks, err := eventtasks.Open(e.store, design, registry, e) - if err != nil { - t.Fatal(err) - } - tasks.SetSession("combined-session") - mailWallet, err := player.OpenWallet(e.store, player.Currency{}) - if err != nil { - t.Fatal(err) - } - mailbox, err := mail.OpenService(e.store, &mail.Starter{Version: "2.35.10", MailCount: 1}, items, mailWallet) - if err != nil { - t.Fatal(err) - } - if err := mailbox.AttachAttendanceRewardEconomy(e); err != nil { - t.Fatal(err) - } - tasks.AttachAttendanceMail(mailbox) - passes, err := NewLoginPasses(e.store, &gamedata.LoginPassCatalog{Groups: map[uint64][]gamedata.LoginPassReward{ - 20: {{ID: 1, TicketID: 77, Free: gamedata.Reward{Type: 3, Count: 11}}}, - }}, e, items, func(uint64) bool { return true }) - if err != nil { - t.Fatal(err) - } - passes.SetClock(func() time.Time { return *now }, 0) - h := AttendanceHandler{Events: tasks, Economy: e, LoginPasses: passes, Store: e.store} - request := wire.AppendVarint(nil, 1, 1) - _, response, _, err := h.HandleSession("/Attendance", request, "combined-session") - if err != nil { - t.Fatal(err) - } - if balances.calls != 3 || balances.balance != 17 { - t.Fatalf("wrong combined grant: calls=%d balance=%d", balances.calls, balances.balance) - } - counts := map[int]int{} - if err := wire.Walk(response, func(f wire.Field) error { counts[f.Number]++; return nil }); err != nil { - t.Fatal(err) - } - if counts[1001] != 1 || counts[1002] != 1 || counts[5] != 1 || counts[6] != 1 { - t.Fatalf("client requires a single combined envelope and native notices: %v", counts) - } - bundle, _, _ := wire.Bytes(response, 1001) - var snapshots []uint64 - if err := wire.Walk(bundle, func(f wire.Field) error { - if f.Number == 1 { - count, _, err := wire.Varint(f.Value, 4) - if err != nil { - return err - } - snapshots = append(snapshots, count) - } - return nil - }); err != nil { - t.Fatal(err) - } - if len(snapshots) != 2 || snapshots[0] != 11 || snapshots[1] != 5 { - t.Fatalf("combined reward entries lost or reordered: %v", snapshots) - } - _, replay, _, err := h.HandleSession("/Attendance", request, "combined-session") - if err != nil || !bytes.Equal(response, replay) || balances.calls != 3 { - t.Fatal("combined retry changed rewards", err) - } - _, next, _, err := h.HandleSession("/Attendance", wire.AppendVarint(nil, 1, 2), "combined-session") - if err != nil || balances.calls != 3 { - t.Fatal("fresh request granted again", err) - } - if _, ok, _ := wire.Bytes(next, 1001); ok { - t.Fatal("fresh request replays old reward envelope") - } - if _, ok, _ := wire.Bytes(next, 5); ok { - t.Fatal("fresh request replays old attendance stamps") - } - _, mailInfo, _, err := mailbox.Handle("/MailInfo", wire.AppendVarint(nil, 1, 3)) - if err != nil { - t.Fatal(err) - } - mailCount := 0 - var mailID uint64 - if err := wire.Walk(mailInfo, func(f wire.Field) error { - if f.Number == 1 { - mailCount++ - mailID, _, _ = wire.Varint(f.Value, 1) - } - return nil - }); err != nil || mailCount != 1 { - t.Fatalf("ordinary attendance did not issue exactly one mail: count=%d err=%v", mailCount, err) - } - open := wire.AppendVarint(wire.AppendVarint(nil, 1, 4), 2, mailID) - if _, _, _, err := mailbox.Handle("/MailOpen", open); err != nil { - t.Fatal(err) - } - if balances.calls != 4 || balances.balance != 19 { - t.Fatalf("ordinary reward must arrive only after opening mail: calls=%d balance=%d", balances.calls, balances.balance) - } -} - -func TestAttendanceRejectsAmbiguousChildRewardEnvelope(t *testing.T) { - native := wire.AppendVarint(nil, 55, 123) - bundle := wire.AppendBytes(native, 1001, wire.AppendVarint(nil, 7, 1)) - valid := wire.AppendString(bundle, 1002, "event-receipt") - stripped, rewards, err := takeAttendanceRewardEnvelope(valid) - if err != nil || !bytes.Equal(stripped, native) || len(rewards) == 0 { - t.Fatal("valid envelope was not preserved", err) - } - for _, malformed := range [][]byte{ - bundle, - wire.AppendString(native, 1002, "event-receipt"), - wire.AppendBytes(valid, 1001, nil), - wire.AppendString(valid, 1002, "second"), - wire.AppendString(bundle, 1002, ""), - wire.AppendVarint(native, 1001, 1), - } { - if _, _, err := takeAttendanceRewardEnvelope(malformed); err == nil { - t.Fatalf("ambiguous envelope accepted: %x", malformed) - } - } -} - -func TestAttendanceExtensionReplaysRewardAndPreservesNativeResponse(t *testing.T) { - e, _, base, _, now := entitlementFixture(t) - if _, err := e.Apply("subscription", nil, []gamedata.Reward{{Type: 19, ID: 38, Count: 1}}); err != nil { - t.Fatal(err) - } - *now = now.Add(24 * time.Hour) - h := AttendanceHandler{Events: originalAttendance{}, Economy: e, Store: e.store} - request := wire.AppendVarint(nil, 1, 1) - _, response, handled, err := h.HandleSession("/Attendance", request, "session") - if err != nil || !handled { - t.Fatal(err) - } - if v, ok, _ := wire.Varint(response, 55); !ok || v != 123 { - t.Fatal("native response lost") - } - bundle, ok, err := wire.Bytes(response, 1001) - if err != nil || !ok || len(bundle) == 0 { - t.Fatal("reward extension missing") - } - if receipt, ok, _ := wire.Bytes(response, 1002); !ok || len(receipt) == 0 { - t.Fatal("receipt extension missing") - } - calls := base.Calls - _, replay, _, err := h.HandleSession("/Attendance", request, "session") - if err != nil || !bytes.Equal(response, replay) || base.Calls != calls { - t.Fatal("retry lost reward or granted twice", err) - } - _, next, _, err := h.HandleSession("/Attendance", wire.AppendVarint(nil, 1, 2), "session") - if err != nil { - t.Fatal(err) - } - if _, ok, _ := wire.Bytes(next, 1001); ok || base.Calls != calls { - t.Fatal("same day duplicate grant") - } - _, _, _, err = h.HandleSession("/Attendance", append(request, wire.AppendVarint(nil, 2, 1)...), "session") - if err == nil { - t.Fatal("conflicting sequence accepted") - } -} - -func TestAttendanceCombinesLoginPassAndSubscriptionWithoutDuplicateClaims(t *testing.T) { - e, items, base, _, now := entitlementFixture(t) - if _, err := e.Apply("subscription", nil, []gamedata.Reward{{Type: 19, ID: 38, Count: 1}}); err != nil { - t.Fatal(err) - } - *now = now.Add(24 * time.Hour) - design := &gamedata.LoginPassCatalog{Groups: map[uint64][]gamedata.LoginPassReward{ - 10: {{ID: 1, TicketID: 77, Free: gamedata.Reward{Type: 3, Count: 11}, Premium: gamedata.Reward{Type: 3, Count: 22}}}, - }} - passes, err := NewLoginPasses(e.store, design, e, items, func(group uint64) bool { return group == 10 }) - if err != nil { - t.Fatal(err) - } - passes.SetClock(func() time.Time { return *now }, 0) - h := AttendanceHandler{Events: originalAttendance{}, Economy: e, LoginPasses: passes, Store: e.store} - calls := base.Calls - request := wire.AppendVarint(nil, 1, 1) - _, response, _, err := h.HandleSession("/Attendance", request, "combined-session") - if err != nil || base.Calls != calls+2 { - t.Fatalf("daily claims not combined: calls=%d err=%v", base.Calls-calls, err) - } - info, ok, err := wire.Bytes(response, 6) - if err != nil || !ok { - t.Fatal("login-pass metadata lost", err) - } - if group, _, _ := wire.Varint(info, 1); group != 10 { - t.Fatal("wrong login-pass group") - } - bundle, ok, err := wire.Bytes(response, 1001) - if err != nil || !ok || len(bundle) == 0 { - t.Fatal("combined reward envelope missing", err) - } - _, replay, _, err := h.HandleSession("/Attendance", request, "combined-session") - if err != nil || !bytes.Equal(replay, response) || base.Calls != calls+2 { - t.Fatal("retry changed the envelope or repeated claims", err) - } - _, next, _, err := h.HandleSession("/Attendance", wire.AppendVarint(nil, 1, 2), "combined-session") - if err != nil || base.Calls != calls+2 { - t.Fatal("fresh request claimed the same day again", err) - } - if _, ok, _ := wire.Bytes(next, 1001); ok { - t.Fatal("fresh request replayed old rewards") - } -} diff --git a/go/internal/server/commerce/bonuses.go b/go/internal/server/commerce/bonuses.go deleted file mode 100644 index 1e6b64c..0000000 --- a/go/internal/server/commerce/bonuses.go +++ /dev/null @@ -1,216 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "crypto/sha256" - "encoding/hex" - "encoding/json" - "fmt" - "math" - "sort" - "sync" -) - -type bonusGroup struct{ ProductGroup, ContentsGroup uint64 } -type bonusPurchases interface { - LifetimePurchaseTotal([]gamedata.CashProductKey) (uint64, error) -} -type bonusState struct { - Claims map[string]bool `json:"claims"` - Receipts map[string]purchaseReceipt `json:"receipts"` -} -type CashBonuses struct { - mu sync.Mutex - store stateio.Store - economy Economy - counts bonusPurchases - groups map[bonusGroup][]gamedata.CashProductKey - rewards map[bonusGroup][]gamedata.CashBonusReward - ordered []bonusGroup - claimIDs map[string]bool -} - -func bonusClaimID(group bonusGroup, id uint64) string { - return fmt.Sprintf("%d:%d:%d", group.ProductGroup, group.ContentsGroup, id) -} - -func NewCashBonuses(store stateio.Store, economy Economy, counts bonusPurchases, design *gamedata.CashBonusCatalog, packages []gamedata.CashPackageDesign) (*CashBonuses, error) { - if store == nil || economy == nil || counts == nil || design == nil { - return nil, fmt.Errorf("commerce: missing cash bonus dependency") - } - s := &CashBonuses{store: store, economy: economy, counts: counts, groups: map[bonusGroup][]gamedata.CashProductKey{}, rewards: map[bonusGroup][]gamedata.CashBonusReward{}, claimIDs: map[string]bool{}} - seen := map[gamedata.CashProductKey]bool{} - for _, p := range packages { - // EPackages.BonusBundleGroup=8; unrelated packages can reuse a contents ID. - if p.PackageType != 8 || len(design.Groups[p.ContentsGroupID]) == 0 { - continue - } - k := gamedata.CashProductKey{GroupID: p.GroupID, ProductID: p.ID, SaleGroup: p.SaleGroup} - if seen[k] { - return nil, fmt.Errorf("commerce: duplicate cash bonus product %+v", k) - } - seen[k] = true - group := bonusGroup{p.GroupID, p.ContentsGroupID} - s.groups[group] = append(s.groups[group], k) - s.rewards[group] = design.Groups[p.ContentsGroupID] - } - for group, rows := range s.rewards { - s.ordered = append(s.ordered, group) - for _, row := range rows { - s.claimIDs[bonusClaimID(group, row.ID)] = true - } - } - sort.Slice(s.ordered, func(i, j int) bool { - if s.ordered[i].ProductGroup != s.ordered[j].ProductGroup { - return s.ordered[i].ProductGroup < s.ordered[j].ProductGroup - } - return s.ordered[i].ContentsGroup < s.ordered[j].ContentsGroup - }) - _, err := s.load() - return s, err -} - -func (s *CashBonuses) load() (bonusState, error) { - v := bonusState{Claims: map[string]bool{}, Receipts: map[string]purchaseReceipt{}} - raw, err := s.store.Load("commerce_cash_bonuses") - if err != nil || raw == nil { - return v, err - } - if err = stateio.RequireExactJSONObject(raw, "claims", "receipts"); err != nil { - return v, err - } - if err = json.Unmarshal(raw, &v); err != nil { - return v, err - } - if v.Claims == nil || v.Receipts == nil { - return v, fmt.Errorf("commerce: malformed cash bonus state") - } - for id, claimed := range v.Claims { - if !claimed || !s.claimIDs[id] { - return v, fmt.Errorf("commerce: invalid cash bonus claim %s", id) - } - } - for id, r := range v.Receipts { - digest, err := hex.DecodeString(r.Digest) - if id == "" || err != nil || len(digest) != sha256.Size || len(r.Response) == 0 { - return v, fmt.Errorf("commerce: invalid cash bonus receipt") - } - } - return v, nil -} - -func (s *CashBonuses) rewarded(v bonusState, group bonusGroup, field int) []byte { - var out []byte - for _, row := range s.rewards[group] { - if v.Claims[bonusClaimID(group, row.ID)] { - out = wire.AppendVarint(out, field, row.ID) - } - } - return out -} - -func (s *CashBonuses) Handle(path string, request []byte) (int, []byte, bool, error) { - return s.HandleSession(path, request, "") -} - -func (s *CashBonuses) HandleSession(path string, request []byte, session string) (int, []byte, bool, error) { - code := 588 - if path == "/CashBonusReward" { - code = 589 - } else if path != "/CashBonusInfo" { - return 0, nil, false, nil - } - seq, present, err := wire.Varint(request, 1) - if err != nil || !present || seq == 0 || seq > math.MaxInt32 { - return code, nil, true, fmt.Errorf("commerce: invalid cash bonus sequence") - } - if err = wire.Walk(request, func(wire.Field) error { return nil }); err != nil { - return code, nil, true, err - } - s.mu.Lock() - defer s.mu.Unlock() - v, err := s.load() - if err != nil { - return code, nil, true, err - } - if path == "/CashBonusInfo" { - var out []byte - for _, group := range s.ordered { - n, err := s.counts.LifetimePurchaseTotal(s.groups[group]) - if err != nil { - return code, nil, true, err - } - // Preserve the native empty response for groups with no purchases. - if n == 0 { - continue - } - b := wire.AppendVarint(nil, 1, group.ProductGroup) - b = wire.AppendVarint(b, 2, group.ContentsGroup) - b = wire.AppendVarint(b, 3, n) - b = append(b, s.rewarded(v, group, 4)...) - out = wire.AppendBytes(out, 1, b) - } - return code, out, true, nil - } - if session == "" { - return code, nil, true, fmt.Errorf("commerce: cash bonus session unavailable") - } - identity := fmt.Sprintf("cash-bonus:%x:%d", sha256.Sum256([]byte(session)), seq) - digest := fmt.Sprintf("%x", sha256.Sum256(request)) - if receipt, ok := v.Receipts[identity]; ok { - if receipt.Digest != digest { - return code, nil, true, fmt.Errorf("commerce: conflicting cash bonus replay") - } - return code, append([]byte(nil), receipt.Response...), true, nil - } - groupID, _, err := wire.Varint(request, 2) - if err != nil { - return code, nil, true, err - } - contents, _, err := wire.Varint(request, 3) - if err != nil { - return code, nil, true, err - } - id, _, err := wire.Varint(request, 4) - if err != nil { - return code, nil, true, err - } - group := bonusGroup{groupID, contents} - var chosen *gamedata.CashBonusReward - for _, row := range s.rewards[group] { - if row.ID == id { - copy := row - chosen = © - break - } - } - if chosen == nil { - return code, nil, true, fmt.Errorf("commerce: unknown cash bonus group=%d contents=%d bonus=%d", groupID, contents, id) - } - n, err := s.counts.LifetimePurchaseTotal(s.groups[group]) - if err != nil { - return code, nil, true, err - } - if n < chosen.RequireCount { - return code, nil, true, fmt.Errorf("commerce: cash bonus purchase threshold not reached group=%d contents=%d bonus=%d count=%d require=%d", groupID, contents, id, n, chosen.RequireCount) - } - claimID := bonusClaimID(group, id) - var bundle []byte - if !v.Claims[claimID] { - bundle, err = s.economy.Apply("cash-bonus-grant:"+claimID, nil, []gamedata.Reward{chosen.Reward}) - if err != nil { - return code, nil, true, err - } - v.Claims[claimID] = true - } - response := wire.AppendBytes(nil, 1, bundle) - response = append(response, s.rewarded(v, group, 2)...) - v.Receipts[identity] = purchaseReceipt{Digest: digest, Response: response} - raw, err := json.Marshal(v) - if err == nil { - err = s.store.Save("commerce_cash_bonuses", raw) - } - return code, response, true, err -} diff --git a/go/internal/server/commerce/bonuses_test.go b/go/internal/server/commerce/bonuses_test.go deleted file mode 100644 index 13638b6..0000000 --- a/go/internal/server/commerce/bonuses_test.go +++ /dev/null @@ -1,114 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" - "bytes" - "testing" -) - -func bonusRequest(seq, group, contents, bonus uint64) []byte { - b := wire.AppendVarint(nil, 1, seq) - b = wire.AppendVarint(b, 2, group) - b = wire.AppendVarint(b, 3, contents) - return wire.AppendVarint(b, 4, bonus) -} - -func TestCashBonusUsesCommittedSeriesPurchasesAndNativeClaimProtocol(t *testing.T) { - shop, eco, store := serviceFixture(t, 1) - packages := []gamedata.CashPackageDesign{ - {GroupID: 2, ID: 1, PackageType: 8, ContentsGroupID: 7}, - {GroupID: 2, ID: 1, SaleGroup: 1, PackageType: 8, ContentsGroupID: 7}, - // Another package kind can reuse contents=7 but must not count. - {GroupID: 1, ID: 1, PackageType: 3, ContentsGroupID: 7}, - } - design := &gamedata.CashBonusCatalog{Groups: map[uint64][]gamedata.CashBonusReward{ - 7: {{ID: 1, RequireCount: 2, Reward: gamedata.Reward{Type: 8, ID: 1000, Count: 2}}, {ID: 2, RequireCount: 3, Reward: gamedata.Reward{Type: 8, ID: 1000, Count: 15}}}, - }} - s, err := NewCashBonuses(store, eco, shop, design, packages) - if err != nil { - t.Fatal(err) - } - infoRequest := wire.AppendVarint(nil, 1, 1) - if code, info, handled, err := s.Handle("/CashBonusInfo", infoRequest); err != nil || code != 588 || !handled || len(info) != 0 { - t.Fatal("new account bonus info must be empty", code, info, err) - } - claim := bonusRequest(1, 2, 7, 1) - if _, _, _, err := s.HandleSession("/CashBonusReward", claim, "s"); err == nil || eco.calls != 0 { - t.Fatal("unearned bonus granted") - } - for seq, sale := range []uint64{0, 1} { - if _, _, _, err := shop.HandleSession("/CashShopBuy", buyRequest(uint64(seq+1), 2, 1, sale, ""), "s"); err != nil { - t.Fatal(err) - } - } - // Normal recharge is not a member of the bonus series. - if _, _, _, err := shop.HandleSession("/CashShopBuy", buyRequest(3, 1, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - _, info, _, err := s.Handle("/CashBonusInfo", infoRequest) - if err != nil { - t.Fatal(err) - } - row, present, _ := wire.Bytes(info, 1) - count, _, _ := wire.Varint(row, 3) - if !present || count != 2 { - t.Fatal("incorrect committed series purchase count", count) - } - calls := eco.calls - code, reply, handled, err := s.HandleSession("/CashBonusReward", claim, "s") - if err != nil || code != 589 || !handled || eco.calls != calls+1 || len(eco.rewards) != 1 || eco.rewards[0].Count != 2 || len(eco.costs) != 0 { - t.Fatal("earned bonus not granted through native protocol", code, err, eco.rewards) - } - if id, _, _ := wire.Varint(reply, 2); id != 1 { - t.Fatal("native rewarded ID list missing") - } - reopened, err := NewCashBonuses(store, eco, shop, design, packages) - if err != nil { - t.Fatal(err) - } - _, replay, _, err := reopened.HandleSession("/CashBonusReward", claim, "s") - if err != nil || !bytes.Equal(reply, replay) || eco.calls != calls+1 { - t.Fatal("restart replay duplicated bonus", err) - } - _, noOp, _, err := reopened.HandleSession("/CashBonusReward", bonusRequest(2, 2, 7, 1), "s") - bundle, present, _ := wire.Bytes(noOp, 1) - if err != nil || !present || len(bundle) != 0 || eco.calls != calls+1 { - t.Fatal("fresh duplicate must return empty grant with claimed IDs", err) - } - for _, request := range [][]byte{bonusRequest(1, 2, 7, 2), bonusRequest(4, 1, 7, 1), bonusRequest(5, 2, 7, 99), bonusRequest(6, 2, 7, 2)} { - if _, _, _, err := reopened.HandleSession("/CashBonusReward", request, "s"); err == nil || eco.calls != calls+1 { - t.Fatal("conflicting replay, wrong series or unmet bonus accepted") - } - } - _, info, _, err = reopened.Handle("/CashBonusInfo", infoRequest) - row, _, _ = wire.Bytes(info, 1) - if id, _, _ := wire.Varint(row, 4); err != nil || id != 1 { - t.Fatal("claimed IDs missing after restart", err) - } -} - -func TestCashBonusFailedGrantDoesNotMarkClaimed(t *testing.T) { - shop, eco, store := serviceFixture(t, 0) - if _, _, _, err := shop.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - design := &gamedata.CashBonusCatalog{Groups: map[uint64][]gamedata.CashBonusReward{7: {{ID: 1, RequireCount: 1, Reward: gamedata.Reward{Type: 8, ID: 1000, Count: 2}}}}} - s, err := NewCashBonuses(store, eco, shop, design, []gamedata.CashPackageDesign{{GroupID: 2, ID: 1, PackageType: 8, ContentsGroupID: 7}}) - if err != nil { - t.Fatal(err) - } - eco.fail = true - request := bonusRequest(1, 2, 7, 1) - if _, _, _, err := s.HandleSession("/CashBonusReward", request, "s"); err == nil { - t.Fatal("failed grant accepted") - } - v, err := s.load() - if err != nil || len(v.Claims) != 0 || len(v.Receipts) != 0 { - t.Fatal("failed grant marked bonus claimed", err) - } - eco.fail = false - if _, _, _, err := s.HandleSession("/CashBonusReward", request, "s"); err != nil { - t.Fatal("failed grant could not be retried", err) - } -} diff --git a/go/internal/server/commerce/catalog_test.go b/go/internal/server/commerce/catalog_test.go deleted file mode 100644 index e47d5e8..0000000 --- a/go/internal/server/commerce/catalog_test.go +++ /dev/null @@ -1,95 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gameconfig" - "bd2server/internal/server/gamedata" - "math" - "testing" -) - -func testDesign() *gamedata.CashCatalog { - return &gamedata.CashCatalog{Products: []gamedata.CashProductDesign{ - {Key: gamedata.CashProductKey{GroupID: 1, ProductID: 1, SaleGroup: 1}, GoogleSKU: "recharge", AppleSKU: "recharge-ios", PriceType: 1, PriceCount: 2200, Recharge: true, NominalPaidDiamonds: 100}, - {Key: gamedata.CashProductKey{GroupID: 1, ProductID: 2, SaleGroup: 1}, GoogleSKU: "recharge", AppleSKU: "recharge-ios", PriceType: 1, PriceCount: 2200, Recharge: true, NominalPaidDiamonds: 100}, - {Key: gamedata.CashProductKey{GroupID: 2, ProductID: 1, SaleGroup: 1}, GoogleSKU: "skin", AppleSKU: "skin-ios", PriceType: 1, PriceCount: 5501}, - {Key: gamedata.CashProductKey{GroupID: 3, ProductID: 1}, GoogleSKU: "native", PriceType: 2, PriceCount: 100}, - }} -} -func TestCashPricesAliasesAndIsolation(t *testing.T) { - for _, currency := range []string{"free", "gold", "diamonds", "ban", ""} { - cfg := gameconfig.Default().Purchases - cfg.DiamondRecharge.Currency = currency - c, err := NewCatalog("test", testDesign(), cfg) - if err != nil { - t.Fatal(err) - } - m := c.Manifest() - if m.SchemaVersion != 1 || len(m.Products) != 6 { - t.Fatalf("manifest %+v", m) - } - skin, err := c.Quote(gamedata.CashProductKey{GroupID: 2, ProductID: 1, SaleGroup: 1}, 2) - if err != nil || skin.Cost != 502 || skin.ItemType != 2 || skin.Currency != "paid_diamonds" { - t.Fatalf("ordinary goods affected by %q: %+v %v", currency, skin, err) - } - recharge, err := c.Quote(gamedata.CashProductKey{GroupID: 1, ProductID: 1, SaleGroup: 1}, 1) - if currency == "ban" || currency == "" { - if err == nil { - t.Fatal("disabled recharge accepted") - } - continue - } - if err != nil { - t.Fatal(err) - } - want := uint64(0) - if currency == "gold" { - want = 100000 - } - if currency == "diamonds" { - want = 100 - } - if recharge.Cost != want { - t.Fatalf("currency %s cost=%d", currency, recharge.Cost) - } - second, _ := c.Quote(gamedata.CashProductKey{GroupID: 1, ProductID: 2, SaleGroup: 1}, 1) - if second.Cost != recharge.Cost { - t.Fatal("first purchase incorrectly doubled cost") - } - m.Products[0].SKU = "changed" - if c.Manifest().Products[0].SKU == "changed" { - t.Fatal("manifest mutable through caller") - } - if _, err = c.Quote(gamedata.CashProductKey{GroupID: 3, ProductID: 1}, 1); err == nil { - t.Fatal("native wallet item intercepted") - } - } -} -func TestMonetaryBoundsAndBaseline(t *testing.T) { - for _, v := range [][3]uint64{{math.MaxUint64, 2, 1}, {1, 0, 1}, {1, 1, 0}, {math.MaxInt32 + 1, 1, 1}} { - if _, err := ceilRatio(v[0], v[1], v[2]); err == nil { - t.Fatalf("accepted %v", v) - } - } - d := testDesign() - d.Products[0].PriceCount = 2000 - d.Products[1].PriceCount = 2000 - c, err := NewCatalog("test", d, gameconfig.Default().Purchases) - if err != nil { - t.Fatal(err) - } - p, _ := c.Quote(d.Products[2].Key, 1) - if p.Cost != 276 { - t.Fatalf("baseline not derived dynamically: %+v", p) - } - if _, err = c.Quote(d.Products[2].Key, math.MaxInt32); err == nil { - t.Fatal("overflow accepted") - } - if _, err = c.Quote(d.Products[2].Key, 0); err == nil { - t.Fatal("zero purchase accepted") - } - d.Products[0].Recharge = false - d.Products[1].Recharge = false - if _, err = NewCatalog("test", d, gameconfig.Default().Purchases); err == nil { - t.Fatal("missing baseline accepted") - } -} diff --git a/go/internal/server/commerce/clear_packages.go b/go/internal/server/commerce/clear_packages.go deleted file mode 100644 index 0f17aa4..0000000 --- a/go/internal/server/commerce/clear_packages.go +++ /dev/null @@ -1,258 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "encoding/binary" - "encoding/json" - "fmt" - "math" - "sort" - "sync" - "time" -) - -type ClearPackageInventory interface{ All() []player.Item } -type clearClaim struct{ Kind, GroupID, TicketID, TargetID, Level uint64 } -type clearClaimReceipt struct { - Claim clearClaim `json:"claim"` - Response []byte `json:"response"` -} -type ClearPackages struct { - mu sync.Mutex - store stateio.Store - economy Economy - items ClearPackageInventory - design map[clearClaim]gamedata.ClearPackageRewardDesign - packCleared func(uint64, uint64) bool - towerCleared func(uint64, uint64) bool - now func() time.Time -} - -func NewClearPackages(store stateio.Store, design *gamedata.ClearPackageCatalog, economy Economy, items ClearPackageInventory) (*ClearPackages, error) { - if store == nil || design == nil || economy == nil || items == nil { - return nil, fmt.Errorf("commerce: invalid clear package dependencies") - } - s := &ClearPackages{store: store, economy: economy, items: items, design: map[clearClaim]gamedata.ClearPackageRewardDesign{}, now: time.Now} - for _, r := range design.Rewards { - k := clearClaim{r.Kind, r.GroupID, r.TicketID, r.TargetID, r.Level} - if _, ok := s.design[k]; ok { - return nil, fmt.Errorf("commerce: duplicate clear reward") - } - s.design[k] = r - } - _, err := s.load() - return s, err -} -func (s *ClearPackages) AttachProgress(pack, tower func(uint64, uint64) bool) { - s.packCleared = pack - s.towerCleared = tower -} -func (s *ClearPackages) load() (map[string]clearClaimReceipt, error) { - v := map[string]clearClaimReceipt{} - raw, err := s.store.Load("commerce_clear_claims") - if err != nil || raw == nil { - return v, err - } - err = json.Unmarshal(raw, &v) - if err == nil && v == nil { - err = fmt.Errorf("commerce: invalid clear claim state") - } - if err == nil { - for identity, receipt := range v { - if identity != clearClaimID(receipt.Claim) || len(receipt.Response) == 0 { - err = fmt.Errorf("commerce: invalid saved clear claim") - break - } - if _, ok := s.design[receipt.Claim]; !ok { - err = fmt.Errorf("commerce: unknown saved clear claim") - break - } - } - } - return v, err -} -func clearClaimID(c clearClaim) string { - return fmt.Sprintf("clear-package:%d:%d:%d:%d:%d", c.Kind, c.GroupID, c.TicketID, c.TargetID, c.Level) -} -func (s *ClearPackages) entitled(ticket uint64) bool { - for _, item := range s.items.All() { - if item.Type == 19 && item.ID == ticket && item.Count > 0 && (item.ExpiryTime == 0 || item.ExpiryTime > uint64(s.now().UnixMilli())) { - return true - } - } - return false -} -func (s *ClearPackages) Handle(path string, request []byte) (int, []byte, bool, error) { - return s.HandleSession(path, request, "") -} - -// Both clear-info fields are ordinary proto3 messages, not a oneof. The native -// client creates an empty message for the inactive kind (CommonPacket), and -// omits scalar zero values such as ClearPackagePack and the normal pack level. -func parseClearClaim(request []byte) (clearClaim, error) { - var scalar [2]uint64 - var rows [2][]byte - var seen [4]bool - err := wire.Walk(request, func(f wire.Field) error { - if f.Number < 1 || f.Number > 4 { - return nil - } - if seen[f.Number-1] { - return fmt.Errorf("commerce: duplicate clear claim field %d", f.Number) - } - seen[f.Number-1] = true - if f.Number <= 2 { - if f.Type != 0 { - return fmt.Errorf("commerce: invalid clear claim scalar %d", f.Number) - } - scalar[f.Number-1], _ = binary.Uvarint(f.Value) - if scalar[f.Number-1] > math.MaxInt32 { - return fmt.Errorf("commerce: clear claim scalar %d exceeds int32", f.Number) - } - } else { - if f.Type != 2 { - return fmt.Errorf("commerce: invalid clear claim row %d", f.Number) - } - rows[f.Number-3] = f.Value - } - return nil - }) - if err != nil { - return clearClaim{}, err - } - if scalar[0] == 0 { - return clearClaim{}, fmt.Errorf("commerce: invalid clear claim sequence") - } - kind := scalar[1] - if kind > 1 { - return clearClaim{}, fmt.Errorf("commerce: invalid clear claim type") - } - if !seen[kind+2] { - return clearClaim{}, fmt.Errorf("commerce: missing clear claim row") - } - var active [4]uint64 - for i, row := range rows { - var values [4]uint64 - var fields [4]bool - if err := wire.Walk(row, func(f wire.Field) error { - if f.Number < 1 || f.Number > 4 { - // Unknown active fields retain normal protobuf compatibility. An - // inactive row must contain only the schema's default scalars. - if uint64(i) != kind { - return fmt.Errorf("commerce: nonempty inactive clear claim row") - } - return nil - } - if fields[f.Number-1] || f.Type != 0 { - return fmt.Errorf("commerce: invalid clear claim row scalar %d", f.Number) - } - fields[f.Number-1] = true - v, _ := binary.Uvarint(f.Value) - if v > math.MaxInt32 { - return fmt.Errorf("commerce: clear claim row scalar %d exceeds int32", f.Number) - } - if uint64(i) != kind && v != 0 { - return fmt.Errorf("commerce: conflicting inactive clear claim row") - } - values[f.Number-1] = v - return nil - }); err != nil { - return clearClaim{}, err - } - if uint64(i) == kind { - active = values - } - } - if active[0] == 0 || active[1] == 0 { - return clearClaim{}, fmt.Errorf("commerce: invalid clear claim identity") - } - return clearClaim{kind, active[0], active[1], active[2], active[3]}, nil -} - -func (s *ClearPackages) HandleSession(path string, request []byte, session string) (int, []byte, bool, error) { - if path != "/ClearPackageReward" { - return 0, nil, false, nil - } - claim, err := parseClearClaim(request) - if err != nil { - return 286, nil, true, err - } - d, ok := s.design[claim] - if !ok { - return 286, nil, true, fmt.Errorf("commerce: unknown clear reward") - } - s.mu.Lock() - defer s.mu.Unlock() - v, err := s.load() - if err != nil { - return 286, nil, true, err - } - identity := clearClaimID(claim) - if receipt, ok := v[identity]; ok { - return 286, append([]byte(nil), receipt.Response...), true, nil - } - if d.Type == 1 && !s.entitled(d.TicketID) { - return 286, nil, true, fmt.Errorf("commerce: clear reward premium ticket required") - } - proof := s.packCleared - if claim.Kind == 1 { - proof = s.towerCleared - } - if proof == nil || !proof(d.TargetID, d.Level) { - return 286, nil, true, fmt.Errorf("commerce: clear reward progression incomplete") - } - rewards := []gamedata.Reward{{Type: 9, ID: d.RandomBoxID, Count: 1}} - var bundle []byte - if delivery, ok := s.economy.(interface { - ApplyPurchase(string, []gamedata.Reward, []gamedata.Reward) ([]byte, error) - }); ok { - // Clear reward groups carry MailId just like cash products. The - // delivery resolver selects the versioned mail template and keeps - // mailed contents out of the direct inventory response. - bundle, err = delivery.ApplyPurchase(identity, nil, rewards) - } else { - bundle, err = s.economy.Apply(identity, nil, rewards) - } - if err != nil { - return 286, nil, true, err - } - response := wire.AppendBytes(nil, 1, bundle) - v[identity] = clearClaimReceipt{Claim: claim, Response: response} - raw, err := json.Marshal(v) - if err == nil { - err = s.store.Save("commerce_clear_claims", raw) - } - return 286, response, true, err -} - -// RewardDBInfos is attached to PackInfoResponse fields 3 and 4 so reconnects -// restore claimed reward buttons from server state. -func (s *ClearPackages) RewardDBInfos() (pack, evil [][]byte, err error) { - s.mu.Lock() - defer s.mu.Unlock() - v, err := s.load() - if err != nil { - return nil, nil, err - } - keys := make([]string, 0, len(v)) - for k := range v { - keys = append(keys, k) - } - sort.Strings(keys) - for _, k := range keys { - c := v[k].Claim - b := wire.AppendVarint(nil, 1, c.GroupID) - b = wire.AppendVarint(b, 2, c.TicketID) - b = wire.AppendVarint(b, 3, c.TargetID) - b = wire.AppendVarint(b, 4, c.Level) - if c.Kind == 0 { - pack = append(pack, b) - } else { - evil = append(evil, b) - } - } - return pack, evil, nil -} diff --git a/go/internal/server/commerce/clear_packages_test.go b/go/internal/server/commerce/clear_packages_test.go deleted file mode 100644 index a79eb7c..0000000 --- a/go/internal/server/commerce/clear_packages_test.go +++ /dev/null @@ -1,478 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "encoding/binary" - "errors" - "fmt" - "math" - "os" - "path/filepath" - "reflect" - "testing" - "time" -) - -type clearInventory struct{ items []player.Item } - -func (i *clearInventory) All() []player.Item { return i.items } -func clearRequest(kind, ticket uint64) []byte { - b := wire.AppendVarint(nil, 1, 1) - if kind != 0 { - b = wire.AppendVarint(b, 2, kind) - } - row := wire.AppendVarint(nil, 1, 10) - row = wire.AppendVarint(row, 2, ticket) - row = wire.AppendVarint(row, 3, 2) - b = wire.AppendBytes(b, 3+int(kind), row) - return wire.AppendBytes(b, 4-int(kind), nil) -} - -func TestClearPackageNativeProtoDefaultsAndInactiveRows(t *testing.T) { - for _, kind := range []uint64{0, 1} { - for _, placeholder := range []bool{false, true} { - b := wire.AppendVarint(nil, 1, 1) - if kind != 0 { - b = wire.AppendVarint(b, 2, kind) - } - row := wire.AppendVarint(nil, 1, 10) - row = wire.AppendVarint(row, 2, 12) - row = wire.AppendVarint(row, 3, 2) - if placeholder { - // The native tower request serializes its empty pack placeholder - // before the active tower row; pack serializes it afterwards. - if kind == 1 { - b = wire.AppendBytes(b, 3, nil) - } - } - b = wire.AppendBytes(b, int(kind)+3, row) - if placeholder && kind == 0 { - b = wire.AppendBytes(b, 4, nil) - } - got, err := parseClearClaim(b) - want := clearClaim{kind, 10, 12, 2, 0} - if err != nil || got != want { - t.Fatalf("kind %d placeholder %t: %+v %v", kind, placeholder, got, err) - } - } - } - // Explicit defaults are also valid protobuf; they must not look like a - // second active claim, nor be confused with absent active message presence. - row := wire.AppendVarint(nil, 1, 10) - row = wire.AppendVarint(row, 2, 12) - row = wire.AppendVarint(row, 3, 2) - b := wire.AppendVarint(nil, 1, 1) - b = wire.AppendVarint(b, 2, 0) - b = wire.AppendBytes(b, 3, wire.AppendVarint(row, 4, 0)) - var defaults []byte - for f := 1; f <= 4; f++ { - defaults = wire.AppendVarint(defaults, f, 0) - } - if _, err := parseClearClaim(wire.AppendBytes(b, 4, defaults)); err != nil { - t.Fatal(err) - } -} - -func TestClearPackageRejectsConflictingMalformedAndSpoofedRows(t *testing.T) { - row := wire.AppendVarint(nil, 1, 10) - row = wire.AppendVarint(row, 2, 12) - row = wire.AppendVarint(row, 3, 2) - prefix := wire.AppendVarint(nil, 1, 1) - row = row[:len(row):len(row)] //nolint:modernize // slicesclip - prefix = prefix[:len(prefix):len(prefix)] //nolint:modernize // slicesclip - active := wire.AppendBytes(append([]byte(nil), prefix...), 3, row) - active = active[:len(active):len(active)] //nolint:modernize // slicesclip - requests := map[string][]byte{ - "missing row": prefix, - "empty active": wire.AppendBytes(prefix, 3, nil), - "only inactive": wire.AppendBytes(prefix, 4, nil), - "conflicting inactive": wire.AppendBytes(active, 4, row), - "duplicate active": wire.AppendBytes(active, 3, row), - "duplicate inactive": wire.AppendBytes(wire.AppendBytes(active, 4, nil), 4, nil), - "duplicate kind": wire.AppendVarint(wire.AppendVarint(active, 2, 0), 2, 1), - "duplicate seq": wire.AppendVarint(active, 1, 1), - "kind wire type": wire.AppendBytes(active, 2, nil), - "active wire type": wire.AppendVarint(prefix, 3, 1), - "inactive wire type": wire.AppendVarint(active, 4, 0), - "row scalar wire type": wire.AppendBytes(prefix, 3, wire.AppendBytes(row, 4, nil)), - "duplicate row scalar": wire.AppendBytes(prefix, 3, wire.AppendVarint(row, 3, 2)), - "overflow seq": wire.AppendBytes(wire.AppendVarint(nil, 1, uint64(math.MaxInt32)+1), 3, row), - "negative kind": wire.AppendVarint(active, 2, math.MaxUint64), - "negative level": wire.AppendBytes(prefix, 3, wire.AppendVarint(row, 4, math.MaxUint64)), - "truncated inactive": append(append([]byte(nil), active...), 34, 2, 8), - "unknown inactive data": wire.AppendBytes(active, 4, wire.AppendVarint(nil, 5, 1)), - "unknown static row": wire.AppendBytes(prefix, 3, wire.AppendVarint(row, 4, 1)), - } - for name, request := range requests { - t.Run(name, func(t *testing.T) { - eco := &purchaseEconomy{} - design := &gamedata.ClearPackageCatalog{Rewards: []gamedata.ClearPackageRewardDesign{{GroupID: 10, TicketID: 12, TargetID: 2, RandomBoxID: 100}}} - s, err := NewClearPackages(stateio.NewMemory(), design, eco, &clearInventory{}) - if err != nil { - t.Fatal(err) - } - s.AttachProgress(func(uint64, uint64) bool { return true }, nil) - if _, _, _, err := s.Handle("/ClearPackageReward", request); err == nil || eco.calls != 0 { - t.Fatal("invalid request granted a reward", err, eco.calls) - } - }) - } -} - -func TestClearPackagePremiumTicketExpiryAndTowerProof(t *testing.T) { - items := &clearInventory{items: []player.Item{{Type: 19, ID: 77, Count: 1, ExpiryTime: 1000}}} - eco := &purchaseEconomy{} - design := &gamedata.ClearPackageCatalog{Rewards: []gamedata.ClearPackageRewardDesign{{Kind: 0, GroupID: 10, TicketID: 77, TargetID: 2, RandomBoxID: 100, Type: 1}, {Kind: 1, GroupID: 10, TicketID: 12, TargetID: 2, RandomBoxID: 101}}} - s, err := NewClearPackages(stateio.NewMemory(), design, eco, items) - if err != nil { - t.Fatal(err) - } - s.now = func() time.Time { return time.UnixMilli(1000) } - s.AttachProgress(func(uint64, uint64) bool { return true }, nil) - if _, _, _, err := s.Handle("/ClearPackageReward", clearRequest(0, 77)); err == nil || eco.calls != 0 { - t.Fatal("expired premium accepted", err) - } - if _, _, _, err := s.Handle("/ClearPackageReward", clearRequest(1, 12)); err == nil || eco.calls != 0 { - t.Fatal("pack proof authorized tower", err) - } - s.AttachProgress(nil, func(tower, floor uint64) bool { return tower == 2 && floor == 0 }) - if _, _, _, err := s.Handle("/ClearPackageReward", clearRequest(1, 12)); err != nil { - t.Fatal(err) - } - p, e, err := s.RewardDBInfos() - if err != nil || len(p) != 0 || len(e) != 1 { - t.Fatal("tower receipt not separated", p, e, err) - } - if got, _, _ := wire.Varint(e[0], 3); got != 2 { - t.Fatal("tower_type field wrong", got) - } -} - -type clearReceiptFailure struct { - stateio.Store - fail bool -} - -func (s *clearReceiptFailure) Save(name string, payload []byte) error { - if s.fail && name == "commerce_clear_claims" { - return fmt.Errorf("injected clear receipt failure") - } - return s.Store.Save(name, payload) -} - -func TestClearPackageMailSQLiteAtomicRetryAndReconnect(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - design := &gamedata.ClearPackageCatalog{Rewards: []gamedata.ClearPackageRewardDesign{{GroupID: 10, TicketID: 12, TargetID: 2, RandomBoxID: 100}}} - graph := &deliveryGraph{} - open := func(fail bool) (*accountstate.Repository, *ClearPackages, *mail.Service, *player.Wallet, *player.Inventory) { - t.Helper() - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - t.Cleanup(func() { _ = repo.Close() }) - items, err := player.OpenInventory(repo, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(repo, player.Currency{}) - if err != nil { - t.Fatal(err) - } - for _, persist := range []func() error{items.EnsurePersisted, wallet.EnsurePersisted} { - if err := persist(); err != nil { - t.Fatal(err) - } - } - economy, err := NewEntitlementEconomy(repo, deliveryBase{wallet, items}, graph, items, &gamedata.CashEntitlementDesign{}) - if err != nil { - t.Fatal(err) - } - mailbox, err := mail.OpenService(repo, &mail.Starter{Version: "2.35.10", MailCount: 1}, items, wallet) - if err != nil { - t.Fatal(err) - } - if err := mailbox.AttachCashRewards(economy, map[uint64]bool{40: true}); err != nil { - t.Fatal(err) - } - if err := economy.AttachCashMail(mailbox); err != nil { - t.Fatal(err) - } - s, err := NewClearPackages(&clearReceiptFailure{Store: repo, fail: fail}, design, economy, items) - if err != nil { - t.Fatal(err) - } - s.AttachProgress(func(pack, level uint64) bool { return pack == 2 && level == 0 }, nil) - return repo, s, mailbox, wallet, items - } - repo, s, _, _, _ := open(true) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/ClearPackageReward", clearRequest(0, 12)); err == nil { - t.Fatal("receipt failure lost") - } - if err := op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, s, mailbox, wallet, items := open(false) - if ids, _ := cashList(t, mailbox, 0, 100); len(ids) != 0 { - t.Fatal("mail survived failed transaction", ids) - } - for _, name := range []string{"commerce_clear_claims", "commerce_entitlements"} { - if raw, err := repo.Load(name); err != nil || raw != nil { - t.Fatal("receipt survived rollback", name, err) - } - } - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - _, response, _, err := s.Handle("/ClearPackageReward", clearRequest(0, 12)) - if err != nil { - _ = op.Rollback() - t.Fatal(err) - } - if err := op.Commit(); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Gold != 0 || len(items.All()) != 0 { - t.Fatal("mail contents directly granted") - } - bundle, ok, err := wire.Bytes(response, 1) - if err != nil || !ok { - t.Fatal("response bundle missing", err) - } - if _, direct, _ := wire.Bytes(bundle, 1); direct { - t.Fatal("mailed items returned as inventory") - } - // CommonPacket updates the claimed row from the request and shows the - // localized "sent to mail" notice. An empty bundle is valid for mail-only - // delivery; it must not claim that those attachments entered inventory. - if len(bundle) != 0 { - t.Fatal("mail-only clear returned direct rewards", bundle) - } - ids, _ := cashList(t, mailbox, 0, 100) - if len(ids) != 1 { - t.Fatal("one clear reward should issue one mail", ids) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - _, s, mailbox, _, _ = open(false) - // A replay must restore the exact response without requiring progress again - // or issuing a second mail, including after sequence/session changes. - s.AttachProgress(nil, nil) - request, _, err := wire.ReplaceVarint(clearRequest(0, 12), 1, 9) - if err != nil { - t.Fatal(err) - } - _, replay, _, err := s.HandleSession("/ClearPackageReward", request, "reconnected") - if err != nil || !bytes.Equal(replay, response) { - t.Fatal("reconnect response differs", err) - } - if got, _ := cashList(t, mailbox, 0, 100); !reflect.DeepEqual(got, ids) { - t.Fatal("retry duplicated mail", got, ids) - } - p, e, err := s.RewardDBInfos() - if err != nil || len(p) != 1 || len(e) != 0 { - t.Fatal("claimed row not restored", p, e, err) - } - for f, want := range []uint64{10, 12, 2, 0} { - if got, _, err := wire.Varint(p[0], f+1); err != nil || got != want { - t.Fatal("claim protocol field", f+1, got, want, err) - } - } -} - -func TestInstalledClearPackageDeliveryAndNativeClaims23510(t *testing.T) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("BD2_REAL_GAMEDATA not configured") - } - const version = "20260923193640" - design, err := gamedata.LoadClearPackageCatalog(root, version) - if err != nil { - t.Fatal(err) - } - resolver, err := gamedata.LoadCashRewardResolver(root, version) - if err != nil { - t.Fatal(err) - } - templates, err := gamedata.LoadCashMailTemplates(root, version) - if err != nil { - t.Fatal(err) - } - store := stateio.NewMemory() - items, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(store, player.Currency{}) - if err != nil { - t.Fatal(err) - } - eco, err := NewEntitlementEconomy(store, deliveryBase{wallet, items}, resolver, items, &gamedata.CashEntitlementDesign{}) - if err != nil { - t.Fatal(err) - } - mailbox, err := mail.OpenService(store, &mail.Starter{Version: "2.35.10", MailCount: 1}, items, wallet) - if err != nil { - t.Fatal(err) - } - if err := mailbox.AttachCashRewards(eco, templates); err != nil { - t.Fatal(err) - } - if err := eco.AttachCashMail(mailbox); err != nil { - t.Fatal(err) - } - s, err := NewClearPackages(store, design, eco, items) - if err != nil { - t.Fatal(err) - } - // Current design has normal and premium rows for both pack and tower. - // Every row must derive its mail from GameData and retain its claim identity. - counts := [2][2]int{} - for index, row := range design.Rewards { - counts[row.Kind][row.Type]++ - plan, err := resolver.ResolveDelivery([]gamedata.BattleReward{{Type: 9, ID: row.RandomBoxID, Count: 1}}) - if err != nil || len(plan.Direct) != 0 || len(plan.Mail) != 1 || plan.Mail[0].TemplateID == 0 || len(plan.Mail[0].Rewards) == 0 { - t.Fatalf("clear row %+v delivery %+v: %v", row, plan, err) - } - if row.Type == 1 { - if _, err := items.GrantOnce(fmt.Sprintf("installed-clear-ticket:%d", index), []gamedata.BattleReward{{Type: 19, ID: row.TicketID, Count: 1}}); err != nil { - t.Fatal(err) - } - } - proof := func(target, level uint64) bool { return target == row.TargetID && level == row.Level } - if row.Kind == 0 { - s.AttachProgress(proof, nil) - } else { - s.AttachProgress(nil, proof) - } - request := wire.AppendVarint(nil, 1, 1) - if row.Kind != 0 { - request = wire.AppendVarint(request, 2, row.Kind) - } - var nested []byte - for f, value := range []uint64{row.GroupID, row.TicketID, row.TargetID, row.Level} { - if value != 0 { - nested = wire.AppendVarint(nested, f+1, value) - } - } - request = wire.AppendBytes(request, 3+int(row.Kind), nested) - request = wire.AppendBytes(request, 4-int(row.Kind), nil) - _, response, _, err := s.Handle("/ClearPackageReward", request) - if err != nil { - t.Fatalf("native clear row %+v: %v", row, err) - } - bundle, present, err := wire.Bytes(response, 1) - if err != nil || !present || len(bundle) != 0 { - t.Fatal("mail-only native response", row, response, err) - } - _, replay, _, err := s.Handle("/ClearPackageReward", request) - if err != nil || !bytes.Equal(response, replay) { - t.Fatal("native clear retry", row, err) - } - listRequest := wire.AppendVarint(nil, 1, 1) - listRequest = wire.AppendVarint(listRequest, 3, 1) - _, list, _, err := mailbox.Handle("/CashMailInfo", listRequest) - if err != nil { - t.Fatal(err) - } - if total, _, err := wire.Varint(list, 2); err != nil || total != uint64(index+1) { - t.Fatal("mail duplicated or missing", row, total, err) - } - entry, present, err := wire.Bytes(list, 1) - if err != nil || !present { - t.Fatal("newest clear mail missing", row, err) - } - if template, _, err := wire.Varint(entry, 3); err != nil || template != plan.Mail[0].TemplateID { - t.Fatal("GameData clear mail template mismatch", row, template, plan.Mail[0].TemplateID, err) - } - // MailDBInfo represents attachment vectors as packed repeated int32. - for field, expected := range map[int]func(gamedata.BattleReward) uint64{ - 8: func(r gamedata.BattleReward) uint64 { return r.Type }, - 9: func(r gamedata.BattleReward) uint64 { return r.ID }, - 10: func(r gamedata.BattleReward) uint64 { return r.Count }, - } { - packed, present, err := wire.Bytes(entry, field) - if err != nil || !present { - t.Fatal("clear mail attachments missing", row, field, err) - } - for _, reward := range plan.Mail[0].Rewards { - value, n := binary.Uvarint(packed) - if n <= 0 || value != expected(reward) { - t.Fatal("clear mail attachment mismatch", row, field, value, reward) - } - packed = packed[n:] - } - if len(packed) != 0 { - t.Fatal("extra clear mail attachment", row, field) - } - } - } - if wallet.Snapshot() != (player.Currency{}) { - t.Fatal("mail contents directly credited", wallet.Snapshot()) - } - p, e, err := s.RewardDBInfos() - if err != nil || len(p) != counts[0][0]+counts[0][1] || len(e) != counts[1][0]+counts[1][1] { - t.Fatal("native claims not restored by kind", len(p), len(e), counts, err) - } - for kind, byType := range counts { - for typ, n := range byType { - if n == 0 { - t.Fatalf("missing kind %d type %d design rows", kind, typ) - } - } - } -} -func TestClearPackageRequiresServerProgressAndPremiumEntitlement(t *testing.T) { - store := stateio.NewMemory() - eco := &purchaseEconomy{} - items := &clearInventory{} - design := &gamedata.ClearPackageCatalog{Rewards: []gamedata.ClearPackageRewardDesign{{Kind: 0, GroupID: 10, TicketID: 12, TargetID: 2, RandomBoxID: 100, Type: 0}, {Kind: 0, GroupID: 10, TicketID: 77, TargetID: 2, RandomBoxID: 101, Type: 1}, {Kind: 1, GroupID: 10, TicketID: 12, TargetID: 2, RandomBoxID: 102, Type: 0}}} - s, err := NewClearPackages(store, design, eco, items) - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/ClearPackageReward", clearRequest(0, 12)); err == nil || eco.calls != 0 { - t.Fatal("client spoofed clear progress") - } - s.AttachProgress(func(pack, level uint64) bool { return pack == 2 && level == 0 }, nil) - if _, _, _, err = s.Handle("/ClearPackageReward", clearRequest(0, 77)); err == nil { - t.Fatal("unpaid premium claim accepted") - } - code, _, ok, err := s.Handle("/ClearPackageReward", clearRequest(0, 12)) - if err != nil || !ok || code != 286 || eco.rewards[0].ID != 100 { - t.Fatal(code, err) - } - items.items = []player.Item{{Type: 19, ID: 77, Count: 1}} - if _, _, _, err = s.Handle("/ClearPackageReward", clearRequest(0, 77)); err != nil { - t.Fatal(err) - } - reloaded, err := NewClearPackages(store, design, eco, items) - if err != nil { - t.Fatal(err) - } - if _, _, _, err = reloaded.Handle("/ClearPackageReward", clearRequest(0, 77)); err != nil || eco.calls != 2 { - t.Fatal("restart duplicated grant", err, eco.calls) - } - p, e, err := reloaded.RewardDBInfos() - if err != nil || len(p) != 2 || len(e) != 0 { - t.Fatal(p, e, err) - } - if _, _, _, err = reloaded.Handle("/ClearPackageReward", clearRequest(1, 12)); err == nil { - t.Fatal("unimplemented evil progress granted reward") - } -} diff --git a/go/internal/server/commerce/delivery.go b/go/internal/server/commerce/delivery.go deleted file mode 100644 index 4168194..0000000 --- a/go/internal/server/commerce/delivery.go +++ /dev/null @@ -1,95 +0,0 @@ -package commerce - -import ( - "bytes" - "crypto/sha256" - "encoding/json" - "fmt" - "time" - - "bd2server/internal/server/gamedata" -) - -type CashMailIssuer interface { - IssueCashOnce(string, uint64, []gamedata.Reward, time.Time) error -} - -func (e *EntitlementEconomy) AttachCashMail(issuer CashMailIssuer) error { - if issuer == nil { - return fmt.Errorf("commerce: cash mail issuer unavailable") - } - if _, ok := e.graph.(interface { - ResolveDelivery([]gamedata.BattleReward) (gamedata.CashDelivery, error) - }); !ok { - return fmt.Errorf("commerce: cash delivery resolver unavailable") - } - e.mail = issuer - return nil -} - -// ApplyPurchase selects rewards once, activates direct entitlements, and issues -// durable cash mail. All writes run in the enclosing account transaction. -func (e *EntitlementEconomy) ApplyPurchase(identity string, costs, rewards []gamedata.Reward) ([]byte, error) { - e.mu.Lock() - defer e.mu.Unlock() - if e.mail == nil { - return nil, fmt.Errorf("commerce: cash mail issuer unavailable") - } - definition, _ := json.Marshal(struct{ Costs, Rewards []gamedata.Reward }{costs, rewards}) - digest := sha256.Sum256(definition) - key := identity + ":delivery" - state, err := e.load() - if err != nil { - return nil, err - } - if receipt, ok := state.Receipts[key]; ok { - if !bytes.Equal(receipt.Definition, digest[:]) { - return nil, fmt.Errorf("commerce: delivery identity reused") - } - return append([]byte(nil), receipt.Bundle...), nil - } - resolver, ok := e.graph.(interface { - ResolveDelivery([]gamedata.BattleReward) (gamedata.CashDelivery, error) - }) - if !ok { - return nil, fmt.Errorf("commerce: missing cash delivery resolver") - } - input := make([]gamedata.BattleReward, len(rewards)) - for i, r := range rewards { - input[i] = gamedata.BattleReward(r) - } - plan, err := resolver.ResolveDelivery(input) - if err != nil { - return nil, err - } - direct := make([]gamedata.Reward, len(plan.Direct)) - for i, r := range plan.Direct { - direct[i] = gamedata.Reward(r) - } - var mailed []gamedata.BattleReward - for _, mail := range plan.Mail { - mailed = append(mailed, mail.Rewards...) - } - bundle, err := e.applyPrepared(identity+":direct", costs, direct, true, mailed) - if err != nil { - return nil, err - } - for i, mail := range plan.Mail { - attachments := make([]gamedata.Reward, len(mail.Rewards)) - for j, r := range mail.Rewards { - attachments[j] = gamedata.Reward(r) - } - if err := e.mail.IssueCashOnce(fmt.Sprintf("%s:%d", identity, i), mail.TemplateID, attachments, e.now()); err != nil { - return nil, err - } - } - state, err = e.load() - if err != nil { - return nil, err - } - state.Receipts[key] = entitlementReceipt{Definition: digest[:], Bundle: append([]byte(nil), bundle...)} - if err = e.save(state); err != nil { - return nil, err - } - return bundle, nil -} diff --git a/go/internal/server/commerce/delivery_test.go b/go/internal/server/commerce/delivery_test.go deleted file mode 100644 index cef016d..0000000 --- a/go/internal/server/commerce/delivery_test.go +++ /dev/null @@ -1,320 +0,0 @@ -package commerce - -import ( - "bytes" - "errors" - "fmt" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -type deliveryGraph struct{ selections int } - -func (g *deliveryGraph) ResolveGranted(r []gamedata.BattleReward) ([]gamedata.BattleReward, error) { - return r, nil -} -func (g *deliveryGraph) ResolveDelivery(r []gamedata.BattleReward) (gamedata.CashDelivery, error) { - g.selections++ - return gamedata.CashDelivery{Mail: []gamedata.CashMailReward{{TemplateID: 40, Rewards: []gamedata.BattleReward{{Type: 4, Count: 40}, {Type: 9, ID: 200, Count: 1}}}}}, nil -} - -type deliveryBase struct { - wallet *player.Wallet - items *player.Inventory -} - -func (e deliveryBase) Apply(identity string, costs, rewards []gamedata.Reward) ([]byte, error) { - var currencies []gamedata.Reward - for _, r := range rewards { - if r.Type != 9 { - currencies = append(currencies, r) - } - } - if err := e.wallet.ExchangeOnce(identity, costs, currencies); err != nil { - return nil, err - } - var items []gamedata.BattleReward - var bundle []byte - for _, r := range rewards { - if r.Type == 9 { - items = append(items, gamedata.BattleReward(r)) - } else { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(player.Item{Type: r.Type, ID: r.ID, Count: r.Count})) - } - } - granted, err := e.items.GrantOnce(identity, items) - if err != nil { - return nil, err - } - if len(granted) == 0 { - granted = e.items.GrantedItems(identity) - } - for _, item := range granted { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(item)) - } - return bundle, nil -} -func cashList(t *testing.T, m *mail.Service, start, count uint64) ([]uint64, []byte) { - t.Helper() - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendVarint(request, 2, start) - request = wire.AppendVarint(request, 3, count) - code, response, handled, err := m.Handle("/CashMailInfo", request) - if err != nil || !handled || code != 140 { - t.Fatalf("cash list: %d %t %v", code, handled, err) - } - var ids []uint64 - _ = wire.Walk(response, func(f wire.Field) error { - if f.Number == 1 && f.Type == 2 { - id, _, _ := wire.Varint(f.Value, 1) - template, _, _ := wire.Varint(f.Value, 3) - typ, _, _ := wire.Varint(f.Value, 2) - cash, _, _ := wire.Varint(f.Value, 15) - if id == 0 || template != 40 || typ != 0 || cash != 1 { - t.Fatalf("invalid cash mail %x", f.Value) - } - ids = append(ids, id) - } - return nil - }) - return ids, response -} -func TestCashPurchaseMailSQLiteRollbackRestartBatchAndReplay(t *testing.T) { - fixture, _, _ := serviceFixture(t, 0) - nativeKey := gamedata.CashProductKey{GroupID: 44, ProductID: 1} - fixture.catalog.designs[nativeKey] = gamedata.CashProductDesign{Key: nativeKey, PriceType: 2, PriceCount: 1000, RandomBoxID: 20} - path := filepath.Join(t.TempDir(), "state.db") - graph := &deliveryGraph{} - open := func() (*accountstate.Repository, *Service, *mail.Service, *player.Wallet, *player.Inventory) { - t.Helper() - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - t.Cleanup(func() { _ = repo.Close() }) - wallet, err := player.OpenWallet(repo, player.Currency{Jewelry: 4000}) - if err != nil { - t.Fatal(err) - } - items, err := player.OpenInventory(repo, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - if err = wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - if err = items.EnsurePersisted(); err != nil { - t.Fatal(err) - } - economy, err := NewEntitlementEconomy(repo, deliveryBase{wallet, items}, graph, items, &gamedata.CashEntitlementDesign{}) - if err != nil { - t.Fatal(err) - } - mailbox, err := mail.OpenService(repo, &mail.Starter{Version: "2.35.10", MailCount: 1}, items, wallet) - if err != nil { - t.Fatal(err) - } - if err = mailbox.AttachCashRewards(economy, map[uint64]bool{40: true}); err != nil { - t.Fatal(err) - } - if err = economy.AttachCashMail(mailbox); err != nil { - t.Fatal(err) - } - shop, err := NewService(fixture.catalog, repo, economy) - if err != nil { - t.Fatal(err) - } - return repo, shop, mailbox, wallet, items - } - rollback := func(op stateio.RequestOperation) { - t.Helper() - if err := op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - } - var mailbox *mail.Service - var wallet *player.Wallet - var items *player.Inventory - repo, shop, _, _, _ := open() - shop.AttachPurchaseHook(func(string, gamedata.CashProductDesign, uint64) error { - return fmt.Errorf("injected failure after issuing mail") - }) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = shop.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 0, ""), "s"); err == nil { - t.Fatal("failure ignored") - } - rollback(op) - _ = repo.Close() - repo, shop, mailbox, wallet, items = open() - if wallet.Snapshot().Jewelry != 4000 || wallet.Snapshot().Gold != 0 || len(items.All()) != 0 { - t.Fatal("rollback retained debit/grant") - } - if ids, _ := cashList(t, mailbox, 0, 20); len(ids) != 0 { - t.Fatal("rollback retained cash mail") - } - for seq := uint64(1); seq <= 2; seq++ { - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if err = mailbox.BeforeDispatch("/CashShopBuy", nil); err != nil { - t.Fatal(err) - } - request := buyRequest(seq, 2, 1, 0, "") - if seq == 2 { - request = nativeBuyRequest(seq, 44, [3]uint64{1, 0, 1}) - } - _, response, _, err := shop.HandleSession("/CashShopBuy", request, "s") - if err != nil { - t.Fatal(err) - } - bundle, present, _ := wire.Bytes(response, 1) - if !present || len(bundle) != 0 { - t.Fatalf("mail-only purchase must not return direct rewards: %x", response) - } - notify, err := mailbox.AfterDispatch("/CashShopBuy", nil, nil) - flag, _, _ := wire.Varint(notify, 1) - if err != nil || flag != 1 { - t.Fatal("new mail notification missing") - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - } - if wallet.Snapshot().Jewelry != 2000 || wallet.Snapshot().Gold != 0 || len(items.All()) != 0 { - t.Fatal("unclaimed attachments granted early") - } - ids, list := cashList(t, mailbox, 0, 1) - total, _, _ := wire.Varint(list, 2) - if len(ids) != 1 || ids[0] != 2 || total != 2 { - t.Fatal("cash first page", ids, total) - } - next, _ := cashList(t, mailbox, ids[0], 1) - if len(next) != 1 || next[0] != 1 { - t.Fatal("cash cursor skipped/duplicated", next) - } - _, ordinary, _, err := mailbox.Handle("/MailInfo", wire.AppendVarint(nil, 1, 1)) - if err != nil { - t.Fatal(err) - } - if _, present, _ := wire.Bytes(ordinary, 1); present { - t.Fatal("cash mail leaked into ordinary list") - } - beforeSelections := graph.selections - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = shop.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if graph.selections != beforeSelections || wallet.Snapshot().Jewelry != 2000 { - t.Fatal("purchase replay selected or charged again") - } - _ = repo.Close() - repo, _, mailbox, _, _ = open() - claim := wire.AppendVarint(nil, 1, 3) - claim = wire.AppendVarint(claim, 2, 1) - claim = wire.AppendVarint(claim, 2, 2) - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = mailbox.Handle("/MailOpen", claim); err != nil { - t.Fatal(err) - } - rollback(op) - _ = repo.Close() - repo, _, mailbox, wallet, items = open() - if wallet.Snapshot().Gold != 0 || len(items.All()) != 0 { - t.Fatal("claim rollback retained rewards") - } - if ids, _ := cashList(t, mailbox, 0, 20); len(ids) != 2 { - t.Fatal("claim rollback hid mails") - } - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - _, claimed, _, err := mailbox.Handle("/MailOpen", claim) - if err != nil { - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - countBoxes := func(items *player.Inventory) uint64 { - var n uint64 - for _, r := range items.All() { - if r.Type == 9 && r.ID == 200 { - n += r.Count - } - } - return n - } - if wallet.Snapshot().Gold != 80 || countBoxes(items) != 2 { - t.Fatal("batch claim failed", wallet.Snapshot(), items.All()) - } - _ = repo.Close() - repo, _, mailbox, wallet, items = open() - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - _, replayed, _, err := mailbox.Handle("/MailOpen", claim) - if err != nil { - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if !bytes.Equal(claimed, replayed) || wallet.Snapshot().Gold != 80 || countBoxes(items) != 2 { - t.Fatal("restart replay duplicated claim") - } - if ids, _ := cashList(t, mailbox, 0, 20); len(ids) != 0 { - t.Fatal("claimed mails still visible") - } - historyReq := wire.AppendVarint(nil, 1, 4) - historyReq = wire.AppendVarint(historyReq, 3, 20) - _, history, _, err := mailbox.Handle("/MailHistoryInfo", historyReq) - n, _, _ := wire.Varint(history, 2) - if err != nil || n != 2 { - t.Fatal("claim history missing", n, err) - } -} -func TestDirectSubscriptionDoesNotDuplicateFirstDayAlreadyInMail(t *testing.T) { - e, _, base, _, _ := entitlementFixture(t) - e.mu.Lock() - _, err := e.applyPrepared("direct", nil, []gamedata.Reward{{Type: 19, ID: 38, Count: 1}}, true, []gamedata.BattleReward{{Type: 3, Count: 1}}) - e.mu.Unlock() - if err != nil || len(base.Rewards) != 0 { - t.Fatal("mailed first day was granted directly", base.Rewards, err) - } - if _, err = e.ApplyResolved("mail", nil, []gamedata.Reward{{Type: 3, Count: 1}}); err != nil { - t.Fatal(err) - } - if len(base.Rewards) != 1 || base.Rewards[0].Count != 1 { - t.Fatal("claim did not grant exactly one first day", base.Rewards) - } - if _, err = e.ClaimSubscriptions("same day"); err != nil || len(base.Rewards) != 1 { - t.Fatal("same-day login duplicated first day", err) - } -} diff --git a/go/internal/server/commerce/entitlements.go b/go/internal/server/commerce/entitlements.go deleted file mode 100644 index ed2074c..0000000 --- a/go/internal/server/commerce/entitlements.go +++ /dev/null @@ -1,331 +0,0 @@ -package commerce - -import ( - "bytes" - "crypto/sha256" - "encoding/json" - "fmt" - "math" - "sort" - "strconv" - "sync" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -type grantedResolver interface { - ResolveGranted([]gamedata.BattleReward) ([]gamedata.BattleReward, error) -} -type entitlementReceipt struct { - Definition []byte `json:"definition"` - Bundle []byte `json:"bundle"` -} -type cashSubscription struct { - Start, Expiry int64 - Claimed uint64 - LastDay string -} -type entitlementState struct { - Receipts map[string]entitlementReceipt `json:"receipts"` - Subscriptions map[string]cashSubscription `json:"subscriptions"` -} -type EntitlementEconomy struct { - mu sync.Mutex - store stateio.Store - base Economy - graph grantedResolver - items *player.Inventory - design *gamedata.CashEntitlementDesign - now func() time.Time - resetSeconds int64 - mail CashMailIssuer -} - -func NewEntitlementEconomy(store stateio.Store, base Economy, graph grantedResolver, items *player.Inventory, design *gamedata.CashEntitlementDesign) (*EntitlementEconomy, error) { - if store == nil || base == nil || graph == nil || items == nil || design == nil { - return nil, fmt.Errorf("commerce: missing entitlement dependency") - } - e := &EntitlementEconomy{store: store, base: base, graph: graph, items: items, design: design, now: time.Now} - _, err := e.load() - return e, err -} -func (e *EntitlementEconomy) SetClock(now func() time.Time, resetSeconds int64) { - e.now = now - e.resetSeconds = resetSeconds -} -func (e *EntitlementEconomy) day() string { - return e.now().UTC().Add(-time.Duration(e.resetSeconds) * time.Second).Format("2006-01-02") -} -func (e *EntitlementEconomy) load() (entitlementState, error) { - s := entitlementState{Receipts: map[string]entitlementReceipt{}, Subscriptions: map[string]cashSubscription{}} - raw, err := e.store.Load("commerce_entitlements") - if err != nil || raw == nil { - return s, err - } - if err = stateio.RequireExactJSONObject(raw, "receipts", "subscriptions"); err != nil { - return s, err - } - if err = json.Unmarshal(raw, &s); err != nil { - return s, err - } - if s.Receipts == nil || s.Subscriptions == nil { - return s, fmt.Errorf("commerce: malformed entitlement state") - } - for id, r := range s.Receipts { - if id == "" || len(r.Definition) != sha256.Size { - return s, fmt.Errorf("commerce: malformed entitlement receipt") - } - } - for id, sub := range s.Subscriptions { - ticket, err := strconv.ParseUint(id, 10, 64) - if err != nil || len(e.design.Attendance[ticket]) == 0 || sub.Start <= 0 || sub.Expiry != 0 && sub.Expiry < sub.Start || sub.Claimed == 0 || sub.LastDay == "" { - return s, fmt.Errorf("commerce: malformed subscription") - } - } - return s, nil -} -func (e *EntitlementEconomy) save(s entitlementState) error { - b, err := json.Marshal(s) - if err != nil { - return err - } - return e.store.Save("commerce_entitlements", b) -} -func (e *EntitlementEconomy) Apply(identity string, costs, rewards []gamedata.Reward) ([]byte, error) { - e.mu.Lock() - defer e.mu.Unlock() - return e.apply(identity, costs, rewards) -} -func (e *EntitlementEconomy) ApplyResolved(identity string, costs, rewards []gamedata.Reward) ([]byte, error) { - e.mu.Lock() - defer e.mu.Unlock() - return e.applyPrepared(identity, costs, rewards, true, nil) -} -func (e *EntitlementEconomy) apply(identity string, costs, rewards []gamedata.Reward) ([]byte, error) { - return e.applyPrepared(identity, costs, rewards, false, nil) -} -func (e *EntitlementEconomy) applyPrepared(identity string, costs, rewards []gamedata.Reward, resolved bool, mailed []gamedata.BattleReward) ([]byte, error) { - if identity == "" { - return nil, fmt.Errorf("commerce: missing entitlement identity") - } - definition, _ := json.Marshal(struct{ Costs, Rewards []gamedata.Reward }{costs, rewards}) - digest := sha256.Sum256(definition) - s, err := e.load() - if err != nil { - return nil, err - } - if r, ok := s.Receipts[identity]; ok { - if !bytes.Equal(r.Definition, digest[:]) { - return nil, fmt.Errorf("commerce: entitlement identity reused") - } - return append([]byte(nil), r.Bundle...), nil - } - input := make([]gamedata.BattleReward, len(rewards)) - for i, r := range rewards { - input[i] = gamedata.BattleReward{Type: r.Type, ID: r.ID, Count: r.Count} //nolint:staticcheck // S1016 - } - leaves := input - if !resolved { - leaves, err = e.graph.ResolveGranted(input) - } - if err != nil { - return nil, err - } - // Some product boxes include the first attendance reward (draw-ticket - // subscriptions); others include only the upfront paid currency. Add only - // the missing first-day components on activation, then mark row one claimed. - initial := append([]gamedata.BattleReward(nil), leaves...) - for _, r := range initial { - rows := e.design.Attendance[r.ID] - if r.Type != 19 || len(rows) == 0 { - continue - } - sub := s.Subscriptions[strconv.FormatUint(r.ID, 10)] - if sub.Start != 0 && (sub.Expiry == 0 || sub.Expiry > e.now().UnixMilli()) { - continue - } - first, resolveErr := e.graph.ResolveGranted([]gamedata.BattleReward{rows[0].Reward}) - if resolveErr != nil { - return nil, resolveErr - } - available := map[[2]uint64]uint64{} - for _, leaf := range append(append([]gamedata.BattleReward(nil), leaves...), mailed...) { - k := [2]uint64{leaf.Type, leaf.ID} - available[k] += leaf.Count - } - for _, leaf := range first { - k := [2]uint64{leaf.Type, leaf.ID} - if available[k] < leaf.Count { - leaf.Count -= available[k] - leaves = append(leaves, leaf) - } - } - } - var regular []gamedata.Reward - var special []player.Item - for _, r := range leaves { - if r.Count == 0 || r.Count > math.MaxInt32 { - return nil, fmt.Errorf("commerce: invalid entitlement quantity") - } - switch { - case r.Type == 62: - if !e.design.AvatarSets[r.ID] { - return nil, fmt.Errorf("commerce: unknown avatar set %d", r.ID) - } - // The shared gameplay economy expands AvatarSetTable members and - // emits real AvatarItem/AvatarMotion/AvatarChar ownership. - regular = append(regular, gamedata.Reward(r)) - case r.Type == 19 && e.design.TicketTypes[r.ID] == 2: - if len(e.design.Attendance[r.ID]) == 0 { - return nil, fmt.Errorf("commerce: subscription reward schedule missing") - } - now := e.now().UnixMilli() - expiry := max(int64(e.items.ContentTicketExpiry(r.ID)), now) - if r.Count > uint64((math.MaxInt64-expiry)/(30*86400000)) { - return nil, fmt.Errorf("commerce: subscription expiry overflow") - } - expiry += int64(r.Count) * 30 * 86400000 - special = append(special, player.Item{Type: 19, ID: r.ID, Count: r.Count, ExpiryTime: uint64(expiry), TimeValue: uint64(e.now().UnixMilli())}) - key := strconv.FormatUint(r.ID, 10) - sub := s.Subscriptions[key] - if sub.Start == 0 || sub.Expiry <= now { - sub = cashSubscription{Start: now, Claimed: 1, LastDay: e.day()} - } - sub.Expiry = expiry - s.Subscriptions[key] = sub - default: - regular = append(regular, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 - if r.Type == 19 && e.design.TicketTypes[r.ID] == 3 && len(e.design.Attendance[r.ID]) > 0 { - key := strconv.FormatUint(r.ID, 10) - if _, exists := s.Subscriptions[key]; !exists { - s.Subscriptions[key] = cashSubscription{Start: e.now().UnixMilli(), Claimed: 1, LastDay: e.day()} - } - } - } - } - bundle, err := e.base.Apply(identity+":base", costs, regular) - if err != nil { - return nil, err - } - items, err := e.items.GrantCommerceOnce(identity+":special", special) - if err != nil { - return nil, err - } - for _, i := range items { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(i)) - bundle = wire.AppendBytes(bundle, 6, player.ItemWire(player.Item{ID: i.ID, Type: i.Type, Count: i.Count})) - } - s.Receipts[identity] = entitlementReceipt{Definition: append([]byte(nil), digest[:]...), Bundle: append([]byte(nil), bundle...)} - if err = e.save(s); err != nil { - return nil, err - } - return bundle, nil -} - -// ClaimSubscriptions grants the next row once per server reset day. Missing -// login days are not retroactively claimed. First-row purchase rewards are -// already present in the product box, matching the client's first-row marker. -func (e *EntitlementEconomy) ClaimSubscriptions(identity string) ([]byte, error) { - e.mu.Lock() - defer e.mu.Unlock() - if identity == "" { - return nil, fmt.Errorf("commerce: missing attendance receipt identity") - } - s, err := e.load() - if err != nil { - return nil, err - } - receiptKey := "attendance-reply:" + identity - if receipt, ok := s.Receipts[receiptKey]; ok { - return append([]byte(nil), receipt.Bundle...), nil - } - keys := make([]string, 0, len(s.Subscriptions)) - for k := range s.Subscriptions { - keys = append(keys, k) - } - sort.Strings(keys) - var bundle []byte - for _, key := range keys { - sub := s.Subscriptions[key] - ticket, _ := strconv.ParseUint(key, 10, 64) - rows := e.design.Attendance[ticket] - if sub.LastDay == e.day() || sub.Expiry != 0 && sub.Expiry <= e.now().UnixMilli() || sub.Expiry == 0 && sub.Claimed >= uint64(len(rows)) { - continue - } - index := sub.Claimed % uint64(len(rows)) - r := rows[index] - grant, err := e.apply(fmt.Sprintf("commerce:attendance:%s:%s", key, e.day()), nil, []gamedata.Reward{{Type: r.Reward.Type, ID: r.Reward.ID, Count: r.Reward.Count}}) - if err != nil { - return nil, err - } - bundle = append(bundle, grant...) - // apply persists its receipt; reload before saving progress so it survives. - latest, err := e.load() - if err != nil { - return nil, err - } - sub.Claimed++ - sub.LastDay = e.day() - latest.Subscriptions[key] = sub - if err = e.save(latest); err != nil { - return nil, err - } - s = latest - } - latest, err := e.load() - if err != nil { - return nil, err - } - digest := sha256.Sum256([]byte(receiptKey)) - latest.Receipts[receiptKey] = entitlementReceipt{Definition: append([]byte(nil), digest[:]...), Bundle: append([]byte(nil), bundle...)} - if err = e.save(latest); err != nil { - return nil, err - } - return bundle, nil -} -func (e *EntitlementEconomy) MergeAttendance(response []byte) ([]byte, error) { - e.mu.Lock() - defer e.mu.Unlock() - s, err := e.load() - if err != nil { - return nil, err - } - out := append([]byte(nil), response...) - keys := make([]string, 0, len(s.Subscriptions)) - for k := range s.Subscriptions { - keys = append(keys, k) - } - sort.Strings(keys) - for _, key := range keys { - sub := s.Subscriptions[key] - ticket, _ := strconv.ParseUint(key, 10, 64) - if sub.Expiry != 0 { - b := wire.AppendVarint(nil, 1, ticket) - b = wire.AppendVarint(b, 2, uint64(sub.Start)) - b = wire.AppendVarint(b, 3, uint64(sub.Expiry)) - out = wire.AppendBytes(out, 3, b) - } - if typ := e.design.AttendanceTypes[ticket]; typ != 0 { - rewards := []byte{} - n := min(sub.Claimed, uint64(len(e.design.Attendance[ticket]))) - for i := uint64(1); i <= n; i++ { - rewards = wire.AppendVarint(rewards, 1, i) - } - entry := wire.AppendVarint(nil, 1, typ) - entry = wire.AppendBytes(entry, 2, rewards) - out = wire.AppendBytes(out, 7, entry) - } else { - for i := uint64(1); i <= sub.Claimed; i++ { - b := wire.AppendVarint(nil, 1, ticket) - b = wire.AppendVarint(b, 2, i) - b = wire.AppendVarint(b, 3, 1) - out = wire.AppendBytes(out, 4, b) - } - } - } - return out, nil -} diff --git a/go/internal/server/commerce/entitlements_test.go b/go/internal/server/commerce/entitlements_test.go deleted file mode 100644 index 7b5a23e..0000000 --- a/go/internal/server/commerce/entitlements_test.go +++ /dev/null @@ -1,143 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "testing" - "time" -) - -type entitlementGraph struct{ Calls int } - -func (g *entitlementGraph) ResolveGranted(r []gamedata.BattleReward) ([]gamedata.BattleReward, error) { - g.Calls++ - return r, nil -} - -type entitlementBase struct { - Calls int - Rewards []gamedata.Reward -} - -func (b *entitlementBase) Apply(_ string, _ []gamedata.Reward, r []gamedata.Reward) ([]byte, error) { - b.Calls++ - b.Rewards = append(b.Rewards, r...) - return wire.AppendVarint(nil, 7, 1), nil -} -func entitlementFixture(t *testing.T) (*EntitlementEconomy, *player.Inventory, *entitlementBase, *entitlementGraph, *time.Time) { - t.Helper() - store := stateio.NewMemory() - items, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - d := &gamedata.CashEntitlementDesign{AvatarSets: map[uint64]bool{10: true}, TicketTypes: map[uint64]uint64{38: 2, 99: 3}, Attendance: map[uint64][]gamedata.CashAttendanceReward{38: {{ID: 1, Reward: gamedata.BattleReward{Type: 3, Count: 1}}, {ID: 2, Reward: gamedata.BattleReward{Type: 3, Count: 5}}}, 99: {{ID: 1, Reward: gamedata.BattleReward{Type: 3, Count: 1}}, {ID: 2, Reward: gamedata.BattleReward{Type: 3, Count: 7}}}}, AttendanceTypes: map[uint64]uint64{38: 1}} - base := &entitlementBase{} - graph := &entitlementGraph{} - e, err := NewEntitlementEconomy(store, base, graph, items, d) - if err != nil { - t.Fatal(err) - } - now := time.Date(2026, 10, 5, 12, 0, 0, 0, time.UTC) - e.SetClock(func() time.Time { return now }, 0) - return e, items, base, graph, &now -} -func TestEntitlementTypedSkinExpiryRetryAndRenewal(t *testing.T) { - e, items, base, graph, now := entitlementFixture(t) - rewards := []gamedata.Reward{{Type: 62, ID: 10, Count: 1}, {Type: 19, ID: 38, Count: 1}, {Type: 45, ID: 999, Count: 1}} - first, err := e.Apply("buy1", nil, rewards) - if err != nil { - t.Fatal(err) - } - if items.ContentTicketExpiry(38) != uint64(now.UnixMilli()+30*86400000) || len(base.Rewards) != 3 || base.Rewards[0].Type != 62 || base.Rewards[1].Type != 45 { - t.Fatal("expiry or prestige dispatch invalid") - } - for _, item := range items.All() { - if item.Type == 62 { - t.Fatal("commerce stored synthetic avatar set") - } - } - again, err := e.Apply("buy1", nil, rewards) - if err != nil || !bytes.Equal(first, again) || graph.Calls != 2 || base.Calls != 1 { - t.Fatal("retry rerolled or double granted") - } - if _, err = e.Apply("buy1", nil, nil); err == nil { - t.Fatal("identity reuse accepted") - } - if _, err = e.Apply("invalid", nil, []gamedata.Reward{{Type: 62, ID: 888, Count: 1}}); err == nil || base.Calls != 1 { - t.Fatal("unknown skin partially applied") - } - if _, err = e.Apply("buy2", nil, []gamedata.Reward{{Type: 19, ID: 38, Count: 1}}); err != nil { - t.Fatal(err) - } - if items.ContentTicketExpiry(38) != uint64(now.UnixMilli()+60*86400000) { - t.Fatal("renewal lost prior period") - } - merged, err := e.MergeAttendance(nil) - if err != nil { - t.Fatal(err) - } - if _, ok, _ := wire.Bytes(merged, 3); !ok { - t.Fatal("subscription info missing") - } - if _, ok, _ := wire.Bytes(merged, 7); !ok { - t.Fatal("monthly reward map missing") - } -} -func TestSubscriptionDailyClaimResetExpiryAndPermanentCompletion(t *testing.T) { - e, _, base, _, now := entitlementFixture(t) - _, err := e.Apply("buy", nil, []gamedata.Reward{{Type: 19, ID: 38, Count: 1}, {Type: 19, ID: 99, Count: 1}}) - if err != nil { - t.Fatal(err) - } - calls := base.Calls - if _, err = e.ClaimSubscriptions("login"); err != nil || base.Calls != calls { - t.Fatal("purchase day double claim") - } - *now = now.Add(24 * time.Hour) - if _, err = e.ClaimSubscriptions("login2"); err != nil || base.Calls != calls+2 { - t.Fatalf("daily claims %v calls%d", err, base.Calls) - } - calls = base.Calls - if _, err = e.ClaimSubscriptions("retry"); err != nil || base.Calls != calls { - t.Fatal("daily retry grants again") - } - *now = now.Add(31 * 24 * time.Hour) - if _, err = e.ClaimSubscriptions("expired"); err != nil || base.Calls != calls { - t.Fatal("expired or completed claim") - } - merged, err := e.MergeAttendance(nil) - if err != nil { - t.Fatal(err) - } - if _, ok, _ := wire.Bytes(merged, 4); !ok { - t.Fatal("permanent attendance markers missing") - } -} -func TestSubscriptionFirstRewardAndLostReply(t *testing.T) { - e, _, base, _, now := entitlementFixture(t) - _, err := e.Apply("covered", nil, []gamedata.Reward{{Type: 19, ID: 38, Count: 1}, {Type: 3, Count: 1}}) - if err != nil { - t.Fatal(err) - } - if len(base.Rewards) != 1 || base.Rewards[0].Count != 1 { - t.Fatalf("first-day reward duplicated: %+v", base.Rewards) - } - *now = now.Add(24 * time.Hour) - first, err := e.ClaimSubscriptions("same-session-request") - if err != nil || len(first) == 0 { - t.Fatalf("claim %x %v", first, err) - } - calls := base.Calls - replay, err := e.ClaimSubscriptions("same-session-request") - if err != nil || !bytes.Equal(first, replay) || base.Calls != calls { - t.Fatal("lost reply not replayed exactly") - } - fresh, err := e.ClaimSubscriptions("new-request-same-day") - if err != nil || len(fresh) != 0 || base.Calls != calls { - t.Fatal("new same-day request repeated award") - } -} diff --git a/go/internal/server/commerce/event_shops.go b/go/internal/server/commerce/event_shops.go deleted file mode 100644 index 64c7105..0000000 --- a/go/internal/server/commerce/event_shops.go +++ /dev/null @@ -1,74 +0,0 @@ -package commerce - -import ( - "fmt" - "sort" - - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -// AttachEventShopSchedules joins event type 15 to EventShopTable and then to -// CashProductTable. Hub IDs are presentation identities, never product groups. -// All scheduled goods are published; purchase handlers enforce their windows. -// Call once before serving sessions. -func (s *Service) AttachEventShopSchedules(design *gamedata.CashCatalog, schedules []events.Schedule) error { - if design == nil || s.shopWindows == nil { - return fmt.Errorf("commerce: missing event shop catalog or shop seed") - } - shops := map[uint64]uint64{} - groups := map[uint64]bool{} - for _, shop := range design.EventShops { - if shop.ID == 0 || shop.ProductGroupID == 0 || shops[shop.ID] != 0 || groups[shop.ProductGroupID] { - return fmt.Errorf("commerce: invalid or duplicate event shop identity") - } - shops[shop.ID], groups[shop.ProductGroupID] = shop.ProductGroupID, true - } - windows := map[gamedata.CashProductKey][][2]uint64{} - var products [][]byte - for _, raw := range s.shopProducts { - group, _, _ := wire.Varint(raw, 1) - if !groups[group] { - products = append(products, raw) - } - } - rows := append([]events.Schedule(nil), schedules...) - sort.Slice(rows, func(i, j int) bool { return rows[i].UID < rows[j].UID }) - for _, row := range rows { - if row.Type != 15 { - continue - } - group := shops[row.ID] - if group == 0 || row.UID == 0 || row.Start <= 0 || row.Start >= row.End { - return fmt.Errorf("commerce: invalid event shop schedule %d", row.UID) - } - found := false - start, end := uint64(row.Start), uint64(row.End) - for _, product := range s.catalog.Designs() { - key := product.Key - if key.GroupID != group { - continue - } - found = true - for _, prior := range windows[key] { - if start < prior[1] && prior[0] < end { - return fmt.Errorf("commerce: overlapping event shop product windows") - } - } - windows[key] = append(windows[key], [2]uint64{start, end}) - raw := wire.AppendVarint(nil, 1, key.GroupID) - raw = wire.AppendVarint(raw, 2, key.ProductID) - raw = wire.AppendVarint(raw, 3, key.SaleGroup) - raw = wire.AppendVarint(raw, 4, start) - raw = wire.AppendVarint(raw, 5, end) - raw = wire.AppendVarint(raw, 8, row.UID) - products = append(products, raw) - } - if !found { - return fmt.Errorf("commerce: event shop %d has no products", row.ID) - } - } - s.shopProducts, s.eventShopWindows, s.eventShopGroups = products, windows, groups - return nil -} diff --git a/go/internal/server/commerce/event_shops_test.go b/go/internal/server/commerce/event_shops_test.go deleted file mode 100644 index 65d47be..0000000 --- a/go/internal/server/commerce/event_shops_test.go +++ /dev/null @@ -1,166 +0,0 @@ -package commerce - -import ( - "bytes" - "errors" - "fmt" - "path/filepath" - "testing" - "time" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func TestEventShopPublishesAllSchedulesAndNativePurchaseReplay(t *testing.T) { - s, eco, store := serviceFixture(t, 0) - key := gamedata.CashProductKey{GroupID: 700, ProductID: 3} - d := gamedata.CashProductDesign{Key: key, PriceType: 8, PriceID: 99, PriceCount: 5, RandomBoxID: 20, PurchaseLimitType: 4, PurchaseLimitCount: 2, BulkOrderAvailability: 1} - s.catalog.designs[key] = d - s.shopWindows = map[gamedata.CashProductKey][2]uint64{} - design := &gamedata.CashCatalog{EventShops: []gamedata.EventShopDesign{{ID: 7, ProductGroupID: 700}}} - rows := []events.Schedule{{UID: 9, Type: 15, ID: 7, Start: 100, End: 200}} - if err := s.AttachEventShopSchedules(design, rows); err != nil { - t.Fatal(err) - } - for _, now := range []int64{50, 150, 250} { - s.SetClock(func() time.Time { return time.UnixMilli(now) }, 0) - _, out, _, err := s.HandleSession("/CashShopInfo", wire.AppendVarint(nil, 1, 1), "s") - raw, found, _ := wire.Bytes(out, 1) - group, _, _ := wire.Varint(raw, 1) - uid, _, _ := wire.Varint(raw, 8) - if err != nil || !found || group != 700 || uid != 9 { - t.Fatalf("calendar filtered at %d: %v", now, err) - } - if s.IsAvailable(key) != (now == 150) { - t.Fatal("window authorization differs") - } - } - req := wire.AppendVarint(nil, 1, 1) - req = wire.AppendVarint(req, 3, 700) - line := wire.AppendVarint(nil, 1, 3) - line = wire.AppendVarint(line, 3, 2) - req = wire.AppendBytes(req, 4, line) - if _, _, handled, err := s.HandleSession("/CashShopBuy", req, "s"); !handled || err == nil || eco.calls != 0 { - t.Fatal("expired purchase authorized") - } - s.SetClock(func() time.Time { return time.UnixMilli(150) }, 0) - _, out, handled, err := s.HandleSession("/CashShopBuy", req, "s") - if err != nil || !handled || eco.calls != 1 { - t.Fatal(err, handled, eco.calls) - } - if len(eco.costs) != 1 || eco.costs[0] != (gamedata.Reward{Type: 8, ID: 99, Count: 10}) || eco.rewards[0].Count != 2 { - t.Fatal(eco) - } - // Reload service state, then replay the same request without another grant. - s2, err := NewService(s.catalog, store, eco) - if err != nil { - t.Fatal(err) - } - s2.shopWindows = map[gamedata.CashProductKey][2]uint64{} - if err = s2.AttachEventShopSchedules(design, rows); err != nil { - t.Fatal(err) - } - s2.SetClock(func() time.Time { return time.UnixMilli(250) }, 0) - _, replay, _, err := s2.HandleSession("/CashShopBuy", req, "s") - if err != nil || !bytes.Equal(out, replay) || eco.calls != 1 { - t.Fatal("replay failed", err) - } - req[1] = 2 - if _, _, _, err = s.HandleSession("/CashShopBuy", req, "s"); err == nil || eco.calls != 1 { - t.Fatal("purchase limit ignored") - } -} - -func TestNativeEventShopSQLiteRollsBackDebitGrantAndPurchaseCount(t *testing.T) { - fixture, _, _ := serviceFixture(t, 0) - key := gamedata.CashProductKey{GroupID: 700, ProductID: 3} - fixture.catalog.designs[key] = gamedata.CashProductDesign{Key: key, PriceType: 3, PriceCount: 5, RandomBoxID: 20, PurchaseLimitType: 4, PurchaseLimitCount: 1} - design := &gamedata.CashCatalog{EventShops: []gamedata.EventShopDesign{{ID: 7, ProductGroupID: 700}}} - rows := []events.Schedule{{UID: 9, Type: 15, ID: 7, Start: 100, End: 200}} - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(repo, player.Currency{FreeJewelry: 10}) - if err != nil { - t.Fatal(err) - } - if err = wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - open := func() *Service { - t.Helper() - s, err := NewService(fixture.catalog, repo, walletPurchaseEconomy{wallet, repo}) - if err != nil { - t.Fatal(err) - } - s.shopWindows = map[gamedata.CashProductKey][2]uint64{} - if err = s.AttachEventShopSchedules(design, rows); err != nil { - t.Fatal(err) - } - s.SetClock(func() time.Time { return time.UnixMilli(150) }, 0) - return s - } - s := open() - s.AttachPurchaseHook(func(string, gamedata.CashProductDesign, uint64) error { return fmt.Errorf("grant hook failed") }) - req := wire.AppendVarint(nil, 1, 1) - req = wire.AppendVarint(req, 3, 700) - line := wire.AppendVarint(nil, 1, 3) - line = wire.AppendVarint(line, 3, 1) - req = wire.AppendBytes(req, 4, line) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, handled, err := s.HandleSession("/CashShopBuy", req, "s"); !handled || err == nil { - t.Fatal("hook failure swallowed") - } - if err = op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - wallet, err = player.OpenWallet(repo, player.Currency{}) - if err != nil { - t.Fatal(err) - } - if wallet.Snapshot().FreeJewelry != 10 { - t.Fatal("debit survived rollback") - } - for _, name := range []string{"commerce", "commerce_test_rewards"} { - raw, err := repo.Load(name) - if err != nil || raw != nil { - t.Fatal("grant or receipt survived rollback", name, err) - } - } - s = open() - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.HandleSession("/CashShopBuy", req, "s"); err != nil { - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().FreeJewelry != 5 || !s.HasPurchased(key) { - t.Fatal("retry did not persist") - } -} diff --git a/go/internal/server/commerce/installed_audit_test.go b/go/internal/server/commerce/installed_audit_test.go deleted file mode 100644 index 6cb2b45..0000000 --- a/go/internal/server/commerce/installed_audit_test.go +++ /dev/null @@ -1,484 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/calendar" - "bd2server/internal/server/events" - "bd2server/internal/server/gameconfig" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/hunting" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/readonly" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "os" - "path/filepath" - "reflect" - "strings" - "testing" - "time" -) - -func TestInstalledCashProductRewardCoverage23510(t *testing.T) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("BD2_REAL_GAMEDATA not configured") - } - catalog, err := gamedata.LoadCashCatalog(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - resolver, err := gamedata.LoadCashRewardResolver(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - entitlements, err := gamedata.LoadCashEntitlementDesign(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - // Infinite draw products grant the user's confirmed preview, rather than - // the placeholder cash box. Validate their dedicated design and route. - db, release, err := gamedata.OpenDatabase(root, "20260923193640", "common") - if err != nil { - t.Fatal(err) - } - rowsDB, err := db.Query("SELECT id,ProtoBuf FROM GachaGroupTable") - if err != nil { - release() - t.Fatal(err) - } - type drawGroup struct{ id, gacha uint64 } - var groups []drawGroup - for rowsDB.Next() { - var id uint64 - var raw []byte - if err := rowsDB.Scan(&id, &raw); err != nil { - t.Fatal(err) - } - typ, _, _ := wire.Varint(raw, 17) - sub, _, _ := wire.Varint(raw, 16) - if typ == 1 && sub == 5 { - gacha, _, _ := wire.Varint(raw, 33) - groups = append(groups, drawGroup{id, gacha}) - } - } - if err := rowsDB.Err(); err != nil { - t.Fatal(err) - } - if err := rowsDB.Close(); err != nil { - t.Fatal(err) - } - release() - policy, err := NewCatalog("2.35.10", catalog, gameconfig.Default().Purchases) - if err != nil { - t.Fatal(err) - } - shop, err := NewService(policy, stateio.NewMemory(), &purchaseEconomy{}) - if err != nil { - t.Fatal(err) - } - seed, err := readonly.Load(filepath.Join("..", "..", "..", "seed", "v2_35_10", "readonly.json")) - if err != nil { - t.Fatal(err) - } - calendars, err := calendar.LoadDirectory(filepath.Join("..", "..", "..", "..", "schedules"), "2.35.10", "20260923193640") - if err != nil { - t.Fatal(err) - } - seed, err = calendars.ApplyReadonly(seed) - if err != nil { - t.Fatal(err) - } - shop.SetClock(func() time.Time { return time.Date(2026, 10, 5, 5, 0, 0, 0, time.UTC) }, 0) - if err := shop.AttachShopSeed(seed); err != nil { - t.Fatal(err) - } - if err := shop.AttachEventShopSchedules(catalog, calendars.Events); err != nil { - t.Fatal(err) - } - graph, err := gamedata.LoadRewardGraph(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - costumes, err := gamedata.LoadRewardCostumeCatalog(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - equipment, err := gamedata.LoadRewardEquipmentCatalog(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - skins, err := gamedata.LoadPrestigeSkins(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - ownedDesign, err := gamedata.LoadOwnedEventItemDesign(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - avatarRewards, err := gamedata.LoadAvatarRewardDesign(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - templates, err := gamedata.LoadCashMailTemplates(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - newEconomy := func() (*EntitlementEconomy, *player.Wallet) { - store := stateio.NewMemory() - items, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(store, player.Currency{Jewelry: 100000000, FreeJewelry: 100000000, Gold: 100000000}) - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(store, nil) - if err != nil { - t.Fatal(err) - } - ownedEquipment, err := player.OpenEquipmentInventory(store) - if err != nil { - t.Fatal(err) - } - for _, persist := range []func() error{items.EnsurePersisted, wallet.EnsurePersisted, collection.EnsurePersisted, ownedEquipment.EnsurePersisted} { - if err := persist(); err != nil { - t.Fatal(err) - } - } - base, err := events.NewEconomy(store, items, wallet, collection, ownedEquipment, costumes, equipment, graph, nil) - if err != nil { - t.Fatal(err) - } - base.AttachPrestigeSkins(skins) - base.AttachOwnedItemDesign(ownedDesign) - base.AttachAvatarRewards(avatarRewards) - ap, err := hunting.Open(store, root, "20260923193640", items, wallet, func() (int, error) { return 21, nil }, 0, 0) - if err != nil { - t.Fatal(err) - } - base.AttachHuntingAP(ap) - e, err := NewEntitlementEconomy(store, base, resolver, items, entitlements) - if err != nil { - t.Fatal(err) - } - e.SetClock(func() time.Time { return time.Date(2026, 10, 5, 5, 0, 0, 0, time.UTC) }, 0) - mailbox, err := mail.OpenService(store, &mail.Starter{Version: "2.35.10", MailCount: 1}, items, wallet) - if err != nil { - t.Fatal(err) - } - if err = mailbox.AttachCashRewards(e, templates); err != nil { - t.Fatal(err) - } - if err = e.AttachCashMail(mailbox); err != nil { - t.Fatal(err) - } - return e, wallet - } - t.Run("event shop complete calendar and native purchase", func(t *testing.T) { - e, wallet := newEconomy() - shop, err := NewService(policy, e.store, e) - if err != nil { - t.Fatal(err) - } - if err = shop.AttachShopSeed(seed); err != nil { - t.Fatal(err) - } - rows := append([]events.Schedule(nil), calendars.Events...) - var current events.Schedule - var lastEnd int64 - for _, row := range rows { - if row.Type == 15 && row.ID == 71 { - current = row - if row.End > lastEnd { - lastEnd = row.End - } - } - } - if current.UID == 0 { - t.Fatal("real event shop 71 calendar missing") - } - future := events.Schedule{UID: 99999999, Type: 15, ID: 71, Start: lastEnd + 86400000, End: lastEnd + 2*86400000} - rows = append(rows, future) - if err = shop.AttachEventShopSchedules(catalog, rows); err != nil { - t.Fatal(err) - } - shop.SetClock(func() time.Time { return time.UnixMilli(current.Start + 1) }, 0) - _, info, handled, err := shop.HandleSession("/CashShopInfo", wire.AppendVarint(nil, 1, 1), "audit") - if err != nil || !handled { - t.Fatal(err) - } - actual := map[uint64]map[gamedata.CashProductKey]bool{} - _ = wire.Walk(info, func(f wire.Field) error { - if f.Number != 1 || f.Type != 2 { - return nil - } - uid, _, _ := wire.Varint(f.Value, 8) - if uid == 0 { - return nil - } - group, _, _ := wire.Varint(f.Value, 1) - id, _, _ := wire.Varint(f.Value, 2) - sale, _, _ := wire.Varint(f.Value, 3) - if actual[uid] == nil { - actual[uid] = map[gamedata.CashProductKey]bool{} - } - actual[uid][gamedata.CashProductKey{GroupID: group, ProductID: id, SaleGroup: sale}] = true - return nil - }) - seenOld := false - for _, row := range rows { - if row.Type != 15 { - continue - } - group := uint64(0) - for _, design := range catalog.EventShops { - if design.ID == row.ID { - group = design.ProductGroupID - } - } - if row.ID == 70 { - seenOld = true - } - if row.ID == 71 && group != 900071 { - t.Fatal("event shop 71 product group mismatch") - } - expected := 0 - for _, product := range catalog.Products { - if product.Key.GroupID == group { - expected++ - if !actual[row.UID][product.Key] { - t.Fatalf("shop %d UID %d missing product %+v", row.ID, row.UID, product.Key) - } - } - } - if len(actual[row.UID]) != expected { - t.Fatalf("shop %d UID %d count=%d expected=%d", row.ID, row.UID, len(actual[row.UID]), expected) - } - } - if !seenOld || len(actual[future.UID]) == 0 { - t.Fatal("old or future event shop filtered") - } - fund := []gamedata.Reward{{Type: 8, ID: 2074, Count: 100}} - if _, err = e.Apply("audit funds", nil, fund); err != nil { - t.Fatal(err) - } - beforeWallet := wallet.Snapshot() - countItem := func(id uint64) uint64 { - var n uint64 - for _, item := range e.items.All() { - if item.Type == 8 && item.ID == id { - n += item.Count - } - } - return n - } - beforeCoin, beforeReward := countItem(2074), countItem(710) - request := wire.AppendVarint(nil, 1, 2) - request = wire.AppendVarint(request, 3, 900071) - for _, id := range []uint64{1, 5} { - line := wire.AppendVarint(nil, 1, id) - line = wire.AppendVarint(line, 3, 1) - request = wire.AppendBytes(request, 4, line) - } - code, response, handled, err := shop.HandleSession("/CashShopBuy", request, "audit") - if err != nil || !handled || code != 61 { - t.Fatalf("native batch code=%d handled=%t err=%v", code, handled, err) - } - if wallet.Snapshot().FreeJewelry != beforeWallet.FreeJewelry-100 || countItem(2074) != beforeCoin-12 || countItem(710) != beforeReward+1 { - t.Fatal("native batch debit or real reward missing", wallet.Snapshot(), e.items.All()) - } - for _, id := range []uint64{1, 5} { - found := false - for _, raw := range shop.PurchaseCountDBInfos() { - g, _, _ := wire.Varint(raw, 1) - p, _, _ := wire.Varint(raw, 2) - n, _, _ := wire.Varint(raw, 4) - if g == 900071 && p == id && n == 1 { - found = true - } - } - if !found { - t.Fatal("native purchase count missing", id) - } - } - balance := wallet.Snapshot() - coins := countItem(2074) - rewardCount := countItem(710) - _, replay, _, err := shop.HandleSession("/CashShopBuy", request, "audit") - if err != nil || !bytes.Equal(response, replay) || wallet.Snapshot() != balance || countItem(2074) != coins || countItem(710) != rewardCount { - t.Fatal("native purchase replay duplicated", err) - } - }) - delegated := map[gamedata.CashProductKey]bool{} - retired := map[gamedata.CashProductKey]bool{} - for _, group := range groups { - d, err := gamedata.LoadInfiniteGachaForSchedules(root, "20260923193640", []uint64{group.id}) - if err != nil && strings.Contains(err.Error(), "infinite cash product missing") { - // Early infinite draws retain placeholder gold/count=0 boxes in this - // version. They are historical timed products, not grantable purchases. - found := false - for _, p := range catalog.Products { - if p.Key.ProductID != group.gacha { - continue - } - found = true - if p.TimeLimitType == 0 || shop.IsAvailable(p.Key) { - t.Fatalf("retired infinite product %+v is purchasable without a valid preview design", p.Key) - } - retired[p.Key] = true - } - if !found { - t.Fatalf("infinite group=%d has no matching retired product", group.id) - } - continue - } - if err != nil || d == nil || d.Count == 0 { - t.Fatalf("infinite draw group=%d cannot grant confirmed preview: %v", group.id, err) - } - delegated[gamedata.CashProductKey{GroupID: d.ProductGroupID, ProductID: d.ProductID, SaleGroup: d.SaleGroup}] = true - } - cash, recharge, delegateCount, retiredCount, mailedProducts := 0, 0, 0, 0, 0 - for _, p := range catalog.Products { - if p.PriceType != 1 { - continue - } - cash++ - if p.Recharge { - recharge++ - } - if delegated[p.Key] { - delegateCount++ - continue - } - if retired[p.Key] { - retiredCount++ - continue - } - if p.RandomBoxID == 0 { - t.Errorf("cash product %+v has no reward box", p.Key) - continue - } - for _, box := range []uint64{p.RandomBoxID, p.BonusRandomBoxID} { - if box == 0 { - continue - } - leaves, err := resolver.ResolveGranted([]gamedata.BattleReward{{Type: 9, ID: box, Count: 1}}) - if err != nil || len(leaves) == 0 { - t.Errorf("cash product %+v box=%d cannot grant rewards: %v", p.Key, box, err) - continue - } - for _, r := range leaves { - if r.Type == 19 && entitlements.TicketTypes[r.ID] == 0 || r.Type == 62 && !entitlements.AvatarSets[r.ID] { - t.Errorf("cash product %+v has unknown entitlement %+v", p.Key, r) - } - } - } - // Exercise the same grant and debit modules used by CashShopBuy against - // isolated in-memory accounts; no real save or authenticated API is touched. - q, err := policy.Quote(p.Key, 1) - if err != nil { - t.Fatal(err) - } - var costs []gamedata.Reward - if q.Cost > 0 { - costs = []gamedata.Reward{{Type: q.ItemType, Count: q.Cost}} - } - rewards := []gamedata.Reward{{Type: 9, ID: p.RandomBoxID, Count: 1}} - if p.BonusRandomBoxID != 0 { - rewards = append(rewards, gamedata.Reward{Type: 9, ID: p.BonusRandomBoxID, Count: 1}) - } - e, wallet := newEconomy() - bundle, err := e.ApplyPurchase("installed-audit", costs, rewards) - if err != nil { - t.Errorf("cash product %+v failed real debit/grant: %v", p.Key, err) - continue - } - _ = wire.Walk(bundle, func(f wire.Field) error { - if f.Number == 1 && f.Type == 2 { - typ, _, _ := wire.Varint(f.Value, 3) - if typ == 62 { - t.Errorf("cash product %+v returned synthetic avatar set", p.Key) - } - } - return nil - }) - balance := wallet.Snapshot() - replay, err := e.ApplyPurchase("installed-audit", costs, rewards) - if err != nil || !bytes.Equal(replay, bundle) || wallet.Snapshot() != balance { - t.Errorf("cash product %+v replay repeated debit/grant: %v", p.Key, err) - } - - mailbox := e.mail.(*mail.Service) - listReq := wire.AppendVarint(nil, 1, 1) - listReq = wire.AppendVarint(listReq, 3, 100) - code, list, ok, err := mailbox.Handle("/CashMailInfo", listReq) - if err != nil || !ok || code != 140 { - t.Fatalf("cash list %+v: %v", p.Key, err) - } - claim := wire.AppendVarint(nil, 1, 2) - mailCount := 0 - _ = wire.Walk(list, func(f wire.Field) error { - if f.Number == 1 && f.Type == 2 { - id, _, _ := wire.Varint(f.Value, 1) - template, _, _ := wire.Varint(f.Value, 3) - cash, _, _ := wire.Varint(f.Value, 15) - if !templates[template] || cash != 1 { - t.Errorf("invalid cash mail template/flag %+v", p.Key) - } - claim = wire.AppendVarint(claim, 2, id) - mailCount++ - } - return nil - }) - if mailCount > 0 { - mailedProducts++ - _, claimed, _, err := mailbox.Handle("/MailOpen", claim) - if err != nil { - t.Errorf("cash claim %+v: %v", p.Key, err) - continue - } - claimedBalance := wallet.Snapshot() - claimedItems := e.items.All() - _, replayed, _, err := mailbox.Handle("/MailOpen", claim) - if err != nil || !bytes.Equal(claimed, replayed) || wallet.Snapshot() != claimedBalance || !reflect.DeepEqual(e.items.All(), claimedItems) { - t.Errorf("cash claim replay %+v duplicated: %v", p.Key, err) - } - _, remaining, _, err := mailbox.Handle("/CashMailInfo", listReq) - n, _, _ := wire.Varint(remaining, 2) - if err != nil || n != 0 { - t.Errorf("claimed cash mail remains %+v", p.Key) - } - } - // Cash mailbox delivery must grant exactly what the previous immediate - // reward resolver granted, including subscriptions' first-day deduplication. - reference, referenceWallet := newEconomy() - if _, err := reference.Apply("reference", costs, rewards); err != nil { - t.Fatal(err) - } - itemTotals := func(items []player.Item) map[[3]uint64]uint64 { - totals := map[[3]uint64]uint64{} - for _, item := range items { - totals[[3]uint64{item.Type, item.ID, item.ExpiryTime}] += item.Count - } - return totals - } - if wallet.Snapshot() != referenceWallet.Snapshot() || !reflect.DeepEqual(itemTotals(e.items.All()), itemTotals(reference.items.All())) { - t.Errorf("cash product %+v mailed delivery differs from direct grant: wallet=%+v expected=%+v items=%+v expected=%+v", p.Key, wallet.Snapshot(), referenceWallet.Snapshot(), e.items.All(), reference.items.All()) - } - } - if cash != 614 || recharge != 14 || delegateCount != 3 || retiredCount != 5 { - t.Fatalf("cash coverage changed: cash=%d recharge=%d delegated=%d retired=%d", cash, recharge, delegateCount, retiredCount) - } - bonus, err := gamedata.LoadCashBonusCatalog(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - rows := bonus.Groups[1] - if len(rows) != 3 || rows[0].RequireCount != 2 || rows[0].Reward.Count != 2 || rows[1].RequireCount != 6 || rows[1].Reward.Count != 15 || rows[2].RequireCount != 8 || rows[2].Reward.Count != 30 { - t.Fatalf("cash bonus thresholds or rewards changed: %+v", rows) - } - t.Logf("validated %d cash rows: %d recharge variants, %d preview designs, %d unavailable historical draws; checked %d products through cash mail creation, batch claim, and replay with typed ticket/skin entitlements", cash, recharge, delegateCount, retiredCount, mailedProducts) -} diff --git a/go/internal/server/commerce/installed_native_test.go b/go/internal/server/commerce/installed_native_test.go deleted file mode 100644 index a894ae8..0000000 --- a/go/internal/server/commerce/installed_native_test.go +++ /dev/null @@ -1,235 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gameconfig" - "bd2server/internal/server/readonly" - "bd2server/internal/server/wire" - "bytes" - "fmt" - "os" - "testing" - "time" - - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/hunting" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" -) - -// Covers static native goods independently of their current publication -// windows. The account is an isolated memory store; no player saves are read. -func TestInstalledAllNativeProductRewardAndPriceDomains(t *testing.T) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("BD2_REAL_GAMEDATA not configured") - } - const version = "20260923193640" - catalog, err := gamedata.LoadCashCatalog(root, version) - if err != nil { - t.Fatal(err) - } - graph, err := gamedata.LoadRewardGraph(root, version) - if err != nil { - t.Fatal(err) - } - costumes, err := gamedata.LoadRewardCostumeCatalog(root, version) - if err != nil { - t.Fatal(err) - } - options, err := gamedata.LoadRewardEquipmentCatalog(root, version) - if err != nil { - t.Fatal(err) - } - owned, err := gamedata.LoadOwnedEventItemDesign(root, version) - if err != nil { - t.Fatal(err) - } - avatars, err := gamedata.LoadAvatarRewardDesign(root, version) - if err != nil { - t.Fatal(err) - } - skins, err := gamedata.LoadPrestigeSkins(root, version) - if err != nil { - t.Fatal(err) - } - newAccount := func() (*events.Economy, *player.Inventory, *player.Wallet, stateio.Store, *hunting.Service) { - store := stateio.NewMemory() - items, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(store, player.Currency{}) - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(store, nil) - if err != nil { - t.Fatal(err) - } - equipment, err := player.OpenEquipmentInventory(store) - if err != nil { - t.Fatal(err) - } - economy, err := events.NewEconomy(store, items, wallet, collection, equipment, costumes, options, graph, nil) - if err != nil { - t.Fatal(err) - } - economy.AttachOwnedItemDesign(owned) - economy.AttachAvatarRewards(avatars) - economy.AttachPrestigeSkins(skins) - ap, err := hunting.Open(store, root, version, items, wallet, func() (int, error) { return 21, nil }, 0, 0) - if err != nil { - t.Fatal(err) - } - economy.AttachHuntingAP(ap) - return economy, items, wallet, store, ap - } - economy, _, _, _, _ := newAccount() - families := map[uint64]int{} - free := 0 - native := 0 - // Each product reward is settled once through the actual shared economy. - // Prices are funded in that same typed domain and spent once separately, - // avoiding historical publication or predecessor constraints in this audit. - for _, product := range catalog.Products { - if product.PriceType == 1 { - continue - } - native++ - if native%50 == 0 { - economy, _, _, _, _ = newAccount() - } - families[product.PriceType]++ - identity := fmt.Sprintf("native-audit:%d:%d:%d", product.Key.GroupID, product.Key.ProductID, product.Key.SaleGroup) - var cost []gamedata.Reward - if product.PriceType == 0 { - free++ - if product.PriceID != 0 || product.PriceCount != 0 { - t.Errorf("free product has nonzero price %+v", product.Key) - } - } else { - cost = []gamedata.Reward{{Type: product.PriceType, ID: product.PriceID, Count: product.PriceCount}} - if _, err = economy.Apply(identity+":fund", nil, cost); err != nil { - t.Errorf("native price domain %+v failed: %v", product.Key, err) - continue - } - } - rewards := []gamedata.Reward{{Type: 9, ID: product.RandomBoxID, Count: 1}} - if product.BonusRandomBoxID != 0 { - rewards = append(rewards, gamedata.Reward{Type: 9, ID: product.BonusRandomBoxID, Count: 1}) - } - if _, err = economy.Apply(identity, cost, rewards); err != nil { - t.Errorf("native product %+v real reward/debit failed: %v", product.Key, err) - } - } - if native != 2274 || free != 100 || len(families) != 14 { - t.Fatalf("native design coverage changed: goods=%d free=%d families=%v", native, free, families) - } - t.Run("native dispatcher quantity free and total diamonds", func(t *testing.T) { - base, items, wallet, store, ap := newAccount() - entitlementDesign, err := gamedata.LoadCashEntitlementDesign(root, version) - if err != nil { - t.Fatal(err) - } - resolver, err := gamedata.LoadCashRewardResolver(root, version, graph) - if err != nil { - t.Fatal(err) - } - entitlement, err := NewEntitlementEconomy(store, base, resolver, items, entitlementDesign) - if err != nil { - t.Fatal(err) - } - mailbox, err := mail.OpenService(store, &mail.Starter{Version: "2.35.10", MailCount: 1}, items, wallet) - if err != nil { - t.Fatal(err) - } - templates, err := gamedata.LoadCashMailTemplates(root, version) - if err != nil { - t.Fatal(err) - } - if err = mailbox.AttachCashRewards(entitlement, templates); err != nil { - t.Fatal(err) - } - if err = entitlement.AttachCashMail(mailbox); err != nil { - t.Fatal(err) - } - policy, err := NewCatalog("2.35.10", catalog, gameconfig.Default().Purchases) - if err != nil { - t.Fatal(err) - } - service, err := NewService(policy, store, entitlement) - if err != nil { - t.Fatal(err) - } - field := func(n int, v uint64) readonly.Field { return readonly.Field{Number: n, Type: 0, Varint: v} } - seed := &readonly.Seed{Responses: map[string]readonly.Response{"/CashShopInfo": {Fields: []readonly.Field{{Number: 1, Type: 2, Fields: []readonly.Field{field(1, 1700001), field(2, 1), field(3, 0), field(4, 1), field(5, uint64(time.Now().Add(time.Hour).UnixMilli()))}}}}}} - if err = service.AttachShopSeed(seed); err != nil { - t.Fatal(err) - } - if _, err = base.Apply("fund diamonds", nil, []gamedata.Reward{{Type: 3, Count: 1000}, {Type: 2, Count: 50000}}); err != nil { - t.Fatal(err) - } - makeBuy := func(seq, group, id, count uint64) []byte { - r := wire.AppendVarint(nil, 1, seq) - r = wire.AppendVarint(r, 3, group) - line := wire.AppendVarint(nil, 1, id) - line = wire.AppendVarint(line, 3, count) - return wire.AppendBytes(r, 4, line) - } - request := makeBuy(1, 400001, 2, 300) - _, bonusBefore, err := ap.HuntingAP() - if err != nil { - t.Fatal(err) - } - code, response, handled, err := service.HandleSession("/CashShopBuy", request, "native-audit") - if err != nil || !handled || code != 61 { - t.Fatalf("native quantity300 code%d handled%t err%v", code, handled, err) - } - balance := wallet.Snapshot() - _, bonus, err := ap.HuntingAP() - if err != nil || balance.FreeJewelry != 0 || balance.Jewelry != 0 || bonus != bonusBefore+18000 { - t.Fatal("total diamond split or AP grant wrong", balance, bonus, err) - } - _, replay, _, err := service.HandleSession("/CashShopBuy", request, "native-audit") - _, againBonus, _ := ap.HuntingAP() - if err != nil || !bytes.Equal(response, replay) || againBonus != bonus || wallet.Snapshot() != balance { - t.Fatal("bulk native replay duplicated", err) - } - found := false - for _, raw := range service.PurchaseCountDBInfos() { - g, _, _ := wire.Varint(raw, 1) - id, _, _ := wire.Varint(raw, 2) - count, _, _ := wire.Varint(raw, 4) - if g == 400001 && id == 2 && count == 300 { - found = true - } - } - if !found { - t.Fatal("quantity300 count missing") - } - free := makeBuy(2, 1700001, 1, 1) - if code, _, handled, err = service.HandleSession("/CashShopBuy", free, "native-audit"); err != nil || !handled || code != 61 { - t.Fatal("free native goods failed", err) - } - if wallet.Snapshot().Jewelry != balance.Jewelry || wallet.Snapshot().FreeJewelry != balance.FreeJewelry { - t.Fatal("free goods charged currency") - } - if _, _, _, err = service.HandleSession("/CashShopBuy", makeBuy(3, 1700001, 1, 1), "native-audit"); err == nil { - t.Fatal("free limit bypassed") - } - if _, err = base.Apply("fund resource", nil, []gamedata.Reward{{Type: 8, ID: 1002, Count: 3000}}); err != nil { - t.Fatal(err) - } - if code, _, handled, err = service.HandleSession("/CashShopBuy", makeBuy(4, 700001, 2, 300), "native-audit"); err != nil || !handled || code != 61 { - t.Fatal("native resource quantity300 failed", err) - } - for _, item := range items.All() { - if item.Type == 8 && item.ID == 1002 && item.Count != 0 { - t.Fatal("resource price not consumed", item) - } - } - }) - t.Logf("validated %d native goods across %d typed price domains, including %d free goods", native, len(families), free) -} diff --git a/go/internal/server/commerce/installed_test.go b/go/internal/server/commerce/installed_test.go deleted file mode 100644 index de2a6ac..0000000 --- a/go/internal/server/commerce/installed_test.go +++ /dev/null @@ -1,80 +0,0 @@ -package commerce - -import ( - "os" - "testing" - - "bd2server/internal/server/gameconfig" - "bd2server/internal/server/gamedata" -) - -// Opt in against the installed release; synthetic tests also run without it. -func TestInstalledCashCatalogAndRewardPrograms(t *testing.T) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("BD2_REAL_GAMEDATA not set") - } - design, err := gamedata.LoadCashCatalog(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - catalog, err := NewCatalog("2.35.10", design, gameconfig.Default().Purchases) - if err != nil { - t.Fatal(err) - } - rewards, err := gamedata.LoadCashRewardResolver(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - entitlements, err := gamedata.LoadCashEntitlementDesign(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - counts, recharge := 0, 0 - for _, product := range design.Products { - if product.PriceType != 1 { - continue - } - counts++ - if product.Recharge { - recharge++ - } - quote, err := catalog.Quote(product.Key, 1) - if err != nil { - t.Fatalf("quote %+v: %v", product.Key, err) - } - if product.Recharge && (quote.ItemType != 0 || quote.Amount != 0) { - t.Fatalf("default recharge isn't free: %+v", quote) - } - if !product.Recharge && (quote.ItemType != 2 || quote.Amount == 0) { - t.Fatalf("cash goods aren't paid diamond: %+v", quote) - } - // Infinite preview products use the gacha delegate, whose selected - // costumes replace weighted/empty placeholder reward programs. - if product.Key.GroupID == 1100001 { - continue - } - leaves, err := rewards.ResolveGranted([]gamedata.BattleReward{{Type: 9, ID: product.RandomBoxID, Count: 1}}) - if err != nil || len(leaves) == 0 { - t.Fatalf("product %+v: no reward program: %v", product.Key, err) - } - for _, leaf := range leaves { - if leaf.Type == 19 && entitlements.TicketTypes[leaf.ID] == 2 && len(entitlements.Attendance[leaf.ID]) == 0 { - t.Fatalf("monthly reward schedule missing: %d", leaf.ID) - } - if leaf.Type == 62 && !entitlements.AvatarSets[leaf.ID] { - t.Fatalf("avatar set missing: %d", leaf.ID) - } - } - } - if counts != 614 || recharge != 14 { - t.Fatalf("unexpected cash coverage: goods=%d recharge=%d", counts, recharge) - } - for _, key := range []gamedata.CashProductKey{{GroupID: 1100001, ProductID: 9100033}, {GroupID: 310001, ProductID: 507}, {GroupID: 310001, ProductID: 518}} { - quote, err := catalog.Quote(key, 1) - if err != nil || quote.Amount != 1000 || quote.ItemType != 2 { - t.Fatalf("sample %+v: %+v %v", key, quote, err) - } - } - t.Logf("Verified all %d cash products, %d recharge variants and %d platform aliases", counts, recharge, len(catalog.Manifest().Products)) -} diff --git a/go/internal/server/commerce/login_pass.go b/go/internal/server/commerce/login_pass.go deleted file mode 100644 index 68b73f3..0000000 --- a/go/internal/server/commerce/login_pass.go +++ /dev/null @@ -1,157 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "encoding/json" - "fmt" - "slices" - "sync" - "time" -) - -type loginPassProgress struct { - Free, Premium uint64 - LastDay string -} -type loginPassReceipt struct { - Bundle []byte - Rewarded map[uint64]bool -} -type loginPassState struct { - Progress map[uint64]loginPassProgress - Receipts map[string]loginPassReceipt -} -type LoginPasses struct { - mu sync.Mutex - store stateio.Store - design *gamedata.LoginPassCatalog - economy Economy - items ClearPackageInventory - available func(uint64) bool - now func() time.Time - resetSeconds int64 -} - -func NewLoginPasses(store stateio.Store, design *gamedata.LoginPassCatalog, economy Economy, items ClearPackageInventory, available func(uint64) bool) (*LoginPasses, error) { - if store == nil || design == nil || economy == nil || items == nil || available == nil { - return nil, fmt.Errorf("commerce: invalid login-pass dependencies") - } - s := &LoginPasses{store: store, design: design, economy: economy, items: items, available: available, now: time.Now} - _, err := s.load() - return s, err -} -func (s *LoginPasses) SetClock(now func() time.Time, resetSeconds int64) { - s.now = now - s.resetSeconds = resetSeconds -} -func (s *LoginPasses) load() (loginPassState, error) { - v := loginPassState{Progress: map[uint64]loginPassProgress{}, Receipts: map[string]loginPassReceipt{}} - raw, err := s.store.Load("commerce_login_passes") - if err != nil || raw == nil { - return v, err - } - if err = json.Unmarshal(raw, &v); err != nil { - return v, err - } - if v.Progress == nil || v.Receipts == nil { - return v, fmt.Errorf("commerce: invalid login-pass state") - } - for group, p := range v.Progress { - rows, ok := s.design.Groups[group] - if !ok || p.Free > uint64(len(rows)) || p.Premium > p.Free || p.Free > 0 && p.LastDay == "" { - return v, fmt.Errorf("commerce: invalid login-pass progress") - } - } - return v, nil -} -func (s *LoginPasses) paid(ticket uint64) bool { - for _, i := range s.items.All() { - if i.Type == 19 && i.ID == ticket && i.Count > 0 && (i.ExpiryTime == 0 || i.ExpiryTime > uint64(s.now().UnixMilli())) { - return true - } - } - return false -} - -// ClaimAndInfo is called by AttendanceHandler inside the account transaction. -// Free progression advances once per observed reset day, and buying premium -// catches up the already earned rows without advancing the login day count. -func (s *LoginPasses) ClaimAndInfo(identity string) ([]byte, [][]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() - if identity == "" { - return nil, nil, fmt.Errorf("commerce: missing login-pass identity") - } - v, err := s.load() - if err != nil { - return nil, nil, err - } - keys := make([]uint64, 0, len(s.design.Groups)) - for group := range s.design.Groups { - keys = append(keys, group) - } - slices.Sort(keys) - r, seen := v.Receipts[identity] - if !seen { - r = loginPassReceipt{Rewarded: map[uint64]bool{}} - var rewards []gamedata.Reward - day := s.now().UTC().Add(-time.Duration(s.resetSeconds) * time.Second).Format("2006-01-02") - for _, group := range keys { - if !s.available(group) { - continue - } - rows := s.design.Groups[group] - if len(rows) == 0 { - continue - } - p := v.Progress[group] - if p.LastDay != day && p.Free < uint64(len(rows)) { - rewards = append(rewards, rows[p.Free].Free) - p.Free++ - p.LastDay = day - r.Rewarded[group] = true - } - if s.paid(rows[0].TicketID) { - for p.Premium < p.Free { - rewards = append(rewards, rows[p.Premium].Premium) - p.Premium++ - r.Rewarded[group] = true - } - } - v.Progress[group] = p - } - if len(rewards) > 0 { - r.Bundle, err = s.economy.Apply("login-pass:"+identity, nil, rewards) - if err != nil { - return nil, nil, err - } - } - v.Receipts[identity] = r - raw, err := json.Marshal(v) - if err != nil { - return nil, nil, err - } - if err = s.store.Save("commerce_login_passes", raw); err != nil { - return nil, nil, err - } - } - var infos [][]byte - for _, group := range keys { - if !s.available(group) { - continue - } - p := v.Progress[group] - if p.Free == 0 { - continue - } - b := wire.AppendVarint(nil, 1, group) - b = wire.AppendVarint(b, 2, p.Free) - if r.Rewarded[group] { - b = wire.AppendVarint(b, 3, 1) - } - infos = append(infos, b) - } - return append([]byte(nil), r.Bundle...), infos, nil -} diff --git a/go/internal/server/commerce/login_pass_test.go b/go/internal/server/commerce/login_pass_test.go deleted file mode 100644 index 1785c5f..0000000 --- a/go/internal/server/commerce/login_pass_test.go +++ /dev/null @@ -1,68 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "testing" - "time" -) - -type loginPassEconomy struct { - calls int - rewards [][]gamedata.Reward -} - -func (e *loginPassEconomy) Apply(_ string, _ []gamedata.Reward, r []gamedata.Reward) ([]byte, error) { - e.calls++ - e.rewards = append(e.rewards, append([]gamedata.Reward(nil), r...)) - return []byte{8, 1}, nil -} -func TestLoginPassFreeDaysPaidCatchupAndRestart(t *testing.T) { - d := &gamedata.LoginPassCatalog{Groups: map[uint64][]gamedata.LoginPassReward{10: {{ID: 1, TicketID: 77, Free: gamedata.Reward{Type: 9, ID: 100, Count: 1}, Premium: gamedata.Reward{Type: 9, ID: 200, Count: 1}}, {ID: 2, TicketID: 77, Free: gamedata.Reward{Type: 9, ID: 101, Count: 1}, Premium: gamedata.Reward{Type: 9, ID: 201, Count: 1}}}, 20: {{ID: 1, TicketID: 88, Free: gamedata.Reward{Type: 9, ID: 300, Count: 1}, Premium: gamedata.Reward{Type: 9, ID: 400, Count: 1}}}}} - store := stateio.NewMemory() - eco := &loginPassEconomy{} - items := &clearInventory{} - available := func(group uint64) bool { return group == 10 } - s, err := NewLoginPasses(store, d, eco, items, available) - if err != nil { - t.Fatal(err) - } - now := time.Date(2026, 10, 5, 0, 0, 0, 0, time.UTC) - s.SetClock(func() time.Time { return now }, 0) - bundle, infos, err := s.ClaimAndInfo("attendance1") - if err != nil || eco.calls != 1 || len(eco.rewards[0]) != 1 || eco.rewards[0][0].ID != 100 || len(infos) != 1 { - t.Fatal(err, eco.rewards, infos) - } - replay, _, err := s.ClaimAndInfo("attendance1") - if err != nil || !bytes.Equal(bundle, replay) || eco.calls != 1 { - t.Fatal("duplicate login grant", err) - } - _, infos, err = s.ClaimAndInfo("attendance2") - rewarded, _, _ := wire.Varint(infos[0], 3) - if err != nil || eco.calls != 1 || rewarded != 0 { - t.Fatal("same-day advanced") - } - items.items = []player.Item{{Type: 19, ID: 77, Count: 1}} - _, _, err = s.ClaimAndInfo("after-buy") - if err != nil || eco.calls != 2 || len(eco.rewards[1]) != 1 || eco.rewards[1][0].ID != 200 { - t.Fatal("premium catchup missing", err, eco.rewards) - } - now = now.Add(48 * time.Hour) - _, infos, err = s.ClaimAndInfo("next-login") - day, _, _ := wire.Varint(infos[0], 2) - if err != nil || day != 2 || eco.calls != 3 || len(eco.rewards[2]) != 2 { - t.Fatal("missed day incorrectly advanced", err, day) - } - s, err = NewLoginPasses(store, d, eco, items, available) - if err != nil { - t.Fatal(err) - } - s.SetClock(func() time.Time { return now }, 0) - _, _, err = s.ClaimAndInfo("restart") - if err != nil || eco.calls != 3 { - t.Fatal("restart advanced complete pass", err) - } -} diff --git a/go/internal/server/commerce/native_products_test.go b/go/internal/server/commerce/native_products_test.go deleted file mode 100644 index 00d4e9e..0000000 --- a/go/internal/server/commerce/native_products_test.go +++ /dev/null @@ -1,197 +0,0 @@ -package commerce - -import ( - "bytes" - "errors" - "fmt" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func nativeBuyRequest(seq, group uint64, lines ...[3]uint64) []byte { - b := wire.AppendVarint(nil, 1, seq) - b = wire.AppendVarint(b, 3, group) - for _, line := range lines { - row := wire.AppendVarint(nil, 1, line[0]) - row = wire.AppendVarint(row, 2, line[1]) - row = wire.AppendVarint(row, 3, line[2]) - b = wire.AppendBytes(b, 4, row) - } - // Native clients repeat these display fields on subsequent purchases. - return wire.AppendBytes(b, 7, wire.AppendString(nil, 1, "2")) -} - -func TestNativeGoodsSingleQuantityFreeClaimsAndDisplayBilling(t *testing.T) { - s, eco, _ := serviceFixture(t, 0) - key := gamedata.CashProductKey{GroupID: 44, ProductID: 2} - d := gamedata.CashProductDesign{Key: key, PriceType: 22, PriceCount: 10, RandomBoxID: 500, PurchaseLimitType: 3, PurchaseLimitCount: 301} - s.catalog.designs[key] = d - req := nativeBuyRequest(1, 44, [3]uint64{2, 0, 300}) - code, out, handled, err := s.HandleSession("/CashShopBuy", req, "s") - if err != nil || !handled || code != 61 || eco.calls != 1 { - t.Fatal(code, handled, err) - } - if eco.costs[0] != (gamedata.Reward{Type: 22, Count: 3000}) || eco.rewards[0].Count != 300 { - t.Fatal(eco) - } - _, replay, _, err := s.HandleSession("/CashShopBuy", req, "s") - if err != nil || !bytes.Equal(out, replay) || eco.calls != 1 { - t.Fatal("duplicate native request", err) - } - if _, _, _, err = s.HandleSession("/CashShopBuy", nativeBuyRequest(2, 44, [3]uint64{2, 0, 1}), "s"); err != nil { - t.Fatal("display billing treated as receipt", err) - } - if _, _, _, err = s.HandleSession("/CashShopBuy", nativeBuyRequest(3, 44, [3]uint64{2, 0, 1}), "s"); err == nil || eco.calls != 2 { - t.Fatal("monthly limit ignored") - } - freeKey := gamedata.CashProductKey{GroupID: 45, ProductID: 1} - s.catalog.designs[freeKey] = gamedata.CashProductDesign{Key: freeKey, RandomBoxID: 501, PurchaseLimitType: 4, PurchaseLimitCount: 1} - if _, _, _, err = s.HandleSession("/CashShopBuy", nativeBuyRequest(4, 45, [3]uint64{1, 0, 1}), "s"); err != nil || len(eco.costs) != 0 { - t.Fatal("free claim failed", err) - } - if _, _, _, err = s.HandleSession("/CashShopBuy", nativeBuyRequest(5, 45, [3]uint64{1, 0, 1}), "s"); err == nil { - t.Fatal("free claimed twice") - } -} - -func TestNativeMultipleProductsAndSpecialPreviewsAreValidatedBeforeGrant(t *testing.T) { - s, eco, _ := serviceFixture(t, 0) - key := gamedata.CashProductKey{GroupID: 44, ProductID: 1} - other := gamedata.CashProductKey{GroupID: 44, ProductID: 2} - s.catalog.designs[key] = gamedata.CashProductDesign{Key: key, PriceType: 4, PriceCount: 1, RandomBoxID: 1} - s.catalog.designs[other] = gamedata.CashProductDesign{Key: other, PriceType: 4, PriceCount: 1, RandomBoxID: 2, BulkOrderAvailability: 1} - if _, _, _, err := s.HandleSession("/CashShopBuy", nativeBuyRequest(1, 44, [3]uint64{1, 0, 1}, [3]uint64{2, 0, 1}), "s"); err == nil || eco.calls != 0 { - t.Fatal("ineligible batch executed") - } - if err := s.AttachSpecialProducts([]gamedata.CashProductKey{key}); err != nil { - t.Fatal(err) - } - delegates := 0 - s.AttachDelegate(func(k gamedata.CashProductKey, _ []byte) ([]byte, bool, error) { - delegates++ - if k != key { - return nil, false, fmt.Errorf("unexpected delegate key") - } - return []byte{8, 1}, true, nil - }) - if _, _, _, err := s.HandleSession("/CashShopBuy", nativeBuyRequest(2, 44, [3]uint64{1, 0, 2}), "s"); err == nil || delegates != 0 || eco.calls != 0 { - t.Fatal("invalid preview mutated state") - } - if _, _, _, err := s.HandleSession("/CashShopBuy", nativeBuyRequest(3, 44, [3]uint64{1, 0, 1}), "s"); err != nil || delegates != 1 || len(eco.rewards) != 0 { - t.Fatal("preview did not use delegate", err) - } - if _, _, _, err := s.HandleSession("/CashShopBuy", nativeBuyRequest(4, 44, [3]uint64{2, 0, 2}), "s"); err != nil || delegates != 1 { - t.Fatal("ordinary native delegated", err) - } -} - -type nativeTestGraph struct{} - -func (nativeTestGraph) Resolve(rewards []gamedata.BattleReward) ([]gamedata.BattleReward, error) { - var out []gamedata.BattleReward - for _, r := range rewards { - out = append(out, gamedata.BattleReward{Type: 8, ID: 710, Count: r.Count}) - } - return out, nil -} - -func TestNativeTotalDiamondPurchaseSQLiteRollbackAndRestartReplay(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(repo, player.Currency{FreeJewelry: 5, Jewelry: 10}) - if err != nil { - t.Fatal(err) - } - if err = wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - fixture, _, _ := serviceFixture(t, 0) - key := gamedata.CashProductKey{GroupID: 44, ProductID: 2} - fixture.catalog.designs[key] = gamedata.CashProductDesign{Key: key, PriceType: 3, PriceCount: 7, RandomBoxID: 1} - open := func() (*Service, *player.Inventory) { - t.Helper() - items, err := player.OpenInventory(repo, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(repo, nil) - if err != nil { - t.Fatal(err) - } - equipment, err := player.OpenEquipmentInventory(repo) - if err != nil { - t.Fatal(err) - } - economy, err := events.NewEconomy(repo, items, wallet, collection, equipment, &gamedata.RegularGachaCatalog{}, &gamedata.EquipmentGachaCatalog{}, nativeTestGraph{}, nil) - if err != nil { - t.Fatal(err) - } - s, err := NewService(fixture.catalog, repo, economy) - if err != nil { - t.Fatal(err) - } - return s, items - } - s, _ := open() - s.AttachPurchaseHook(func(string, gamedata.CashProductDesign, uint64) error { return fmt.Errorf("grant failed") }) - req := nativeBuyRequest(1, 44, [3]uint64{2, 0, 1}) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err := s.HandleSession("/CashShopBuy", req, "s"); err == nil { - t.Fatal("hook failure swallowed") - } - if err = op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - wallet, err = player.OpenWallet(repo, player.Currency{}) - if err != nil { - t.Fatal(err) - } - s, items := open() - if wallet.Snapshot().FreeJewelry != 5 || wallet.Snapshot().Jewelry != 10 || len(items.All()) != 0 || s.HasPurchased(key) { - t.Fatal("rollback retained partial purchase") - } - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - _, out, _, err := s.HandleSession("/CashShopBuy", req, "s") - if err != nil { - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().FreeJewelry != 0 || wallet.Snapshot().Jewelry != 8 || len(items.All()) != 1 { - t.Fatal("total diamonds not spent free-first") - } - s, items = open() - _, replay, _, err := s.HandleSession("/CashShopBuy", req, "s") - if err != nil || !bytes.Equal(out, replay) || wallet.Snapshot().Jewelry != 8 || len(items.All()) != 1 { - t.Fatal("restart replay repeated debit/grant", err) - } -} diff --git a/go/internal/server/commerce/pack_info.go b/go/internal/server/commerce/pack_info.go deleted file mode 100644 index c4130ba..0000000 --- a/go/internal/server/commerce/pack_info.go +++ /dev/null @@ -1,39 +0,0 @@ -package commerce - -import "bd2server/internal/server/wire" - -type PackInfoHandler struct { - World attendanceHandler - Claims *ClearPackages -} - -func (h PackInfoHandler) Handle(path string, request []byte) (int, []byte, bool, error) { - if path != "/PackInfo" { - return 0, nil, false, nil - } - code, response, handled, err := h.World.Handle(path, request) - if err != nil || !handled { - return code, response, handled, err - } - pack, evil, err := h.Claims.RewardDBInfos() - if err != nil { - return code, nil, true, err - } - var result []byte - err = wire.Walk(response, func(field wire.Field) error { - if field.Number != 3 && field.Number != 4 { - result = append(result, response[field.Start:field.End]...) - } - return nil - }) - if err != nil { - return code, nil, true, err - } - for _, info := range pack { - result = wire.AppendBytes(result, 3, info) - } - for _, info := range evil { - result = wire.AppendBytes(result, 4, info) - } - return code, result, true, nil -} diff --git a/go/internal/server/commerce/service.go b/go/internal/server/commerce/service.go deleted file mode 100644 index e719c44..0000000 --- a/go/internal/server/commerce/service.go +++ /dev/null @@ -1,764 +0,0 @@ -package commerce - -import ( - "crypto/sha256" - "encoding/hex" - "encoding/json" - "fmt" - "math" - "sort" - "strconv" - "strings" - "sync" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/readonly" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -// Economy and the receipt store must participate in the caller's account -// transaction. A failed hook, delegate or save then rolls back all purchases. -type Economy interface { - Apply(string, []gamedata.Reward, []gamedata.Reward) ([]byte, error) -} -type CountProvider interface{ PurchaseCountDBInfos() [][]byte } -type PurchaseDelegate func(gamedata.CashProductKey, []byte) ([]byte, bool, error) -type PurchaseHook func(string, gamedata.CashProductDesign, uint64) error -type purchaseReceipt struct { - Digest string `json:"digest"` - Response []byte `json:"response"` -} -type purchaseCount struct { - Count uint64 `json:"count"` - Period string `json:"period"` - Lifetime uint64 `json:"lifetime"` -} -type purchaseState struct { - Receipts map[string]purchaseReceipt `json:"receipts"` - Billing map[string]string `json:"billing"` - Counts map[string]purchaseCount `json:"counts"` -} -type Service struct { - mu sync.Mutex - catalog *Catalog - store stateio.Store - economy Economy - now func() time.Time - resetSeconds int64 - delegate PurchaseDelegate - specialProducts map[gamedata.CashProductKey]bool - hook PurchaseHook - legacy CountProvider - shopProducts [][]byte - shopWindows map[gamedata.CashProductKey][2]uint64 - eventShopWindows map[gamedata.CashProductKey][][2]uint64 - eventShopGroups map[uint64]bool - predecessors map[gamedata.CashProductKey][]gamedata.CashProductKey -} - -func NewService(catalog *Catalog, store stateio.Store, economy Economy) (*Service, error) { - if catalog == nil || store == nil || economy == nil { - return nil, fmt.Errorf("commerce: invalid service dependencies") - } - s := &Service{catalog: catalog, store: store, economy: economy, now: time.Now} - _, err := s.load() - return s, err -} -func (s *Service) AttachDelegate(d PurchaseDelegate) { s.delegate = d } -func (s *Service) AttachPurchaseHook(h PurchaseHook) { s.hook = h } -func (s *Service) AttachLegacyCounts(p CountProvider) { s.legacy = p } - -// AttachPackageRules enforces the client's type-2/type-8 step and relay -// ordering using versioned contentsGroupId/contentsSortId, never SKU numbers. -func (s *Service) AttachPackageRules(packages []gamedata.CashPackageDesign) error { - rules := map[gamedata.CashProductKey][]gamedata.CashProductKey{} - for _, p := range packages { - if (p.PackageType != 2 && p.PackageType != 8) || p.ContentsGroupID == 0 || p.ContentsSortID <= 1 { - continue - } - key := gamedata.CashProductKey{GroupID: p.GroupID, ProductID: p.ID, SaleGroup: p.SaleGroup} - var prior []gamedata.CashProductKey - for _, q := range packages { - if q.PackageType == p.PackageType && q.ContentsGroupID == p.ContentsGroupID && q.ContentsSortID == p.ContentsSortID-1 { - prior = append(prior, gamedata.CashProductKey{GroupID: q.GroupID, ProductID: q.ID, SaleGroup: q.SaleGroup}) - } - } - // Some current-version packages have a sort label above one but an - // independent contents group. Only a real same-group predecessor is a - // progression rule; do not invent links between adjacent product IDs. - if len(prior) == 0 { - continue - } - rules[key] = prior - } - s.predecessors = rules - return nil -} -func (s *Service) predecessorBought(v purchaseState, key gamedata.CashProductKey) bool { - prior := s.predecessors[key] - if len(prior) == 0 { - return true - } - for _, p := range prior { - if v.Counts[skuKey(p)].Lifetime > 0 { - return true - } - } - if s.legacy != nil { - for _, raw := range s.legacy.PurchaseCountDBInfos() { - g, _, _ := wire.Varint(raw, 1) - id, _, _ := wire.Varint(raw, 2) - sale, _, _ := wire.Varint(raw, 3) - n, _, _ := wire.Varint(raw, 4) - if n > 0 { - for _, p := range prior { - if p.GroupID == g && p.ProductID == id && p.SaleGroup == sale { - return true - } - } - } - } - } - return false -} - -// AttachShopSeed retains versioned dynamic windows and event identities. It -// must run before serving sessions; absent windows never authorize timed goods. -func (s *Service) AttachShopSeed(seed *readonly.Seed) error { - if seed == nil { - return fmt.Errorf("commerce: missing shop seed") - } - _, raw, handled, err := seed.Handle("/CashShopInfo", wire.AppendVarint(nil, 1, 1)) - if err != nil || !handled { - return fmt.Errorf("commerce: invalid shop seed: %w", err) - } - windows := map[gamedata.CashProductKey][2]uint64{} - var products [][]byte - err = wire.Walk(raw, func(f wire.Field) error { - if f.Number != 1 { - return nil - } - if f.Type != 2 { - return fmt.Errorf("commerce: invalid shop product") - } - g, _, e := wire.Varint(f.Value, 1) - if e != nil { - return e - } - id, _, e := wire.Varint(f.Value, 2) - if e != nil { - return e - } - sale, _, e := wire.Varint(f.Value, 3) - if e != nil { - return e - } - start, _, e := wire.Varint(f.Value, 4) - if e != nil { - return e - } - end, _, e := wire.Varint(f.Value, 5) - if e != nil { - return e - } - k := gamedata.CashProductKey{GroupID: g, ProductID: id, SaleGroup: sale} - if _, ok := windows[k]; ok { - return fmt.Errorf("commerce: duplicate shop schedule") - } - if end != 0 && start >= end { - return fmt.Errorf("commerce: invalid shop window") - } - windows[k] = [2]uint64{start, end} - products = append(products, append([]byte(nil), f.Value...)) - return nil - }) - if err != nil { - return err - } - s.shopProducts = products - s.shopWindows = windows - return nil -} -func (s *Service) available(d gamedata.CashProductDesign) bool { - if s.eventShopGroups[d.Key.GroupID] { - now := uint64(s.now().UnixMilli()) - for _, w := range s.eventShopWindows[d.Key] { - if w[0] <= now && now < w[1] { - return true - } - } - return false - } - w, ok := s.shopWindows[d.Key] - if !ok { - return d.TimeLimitType == 0 - } - now := uint64(s.now().UnixMilli()) - return w[0] <= now && (w[1] == 0 || now < w[1]) -} -func (s *Service) IsAvailable(key gamedata.CashProductKey) bool { - d, ok := s.catalog.Design(key) - return ok && s.available(d) -} -func (s *Service) shopInfo() []byte { - var response []byte - for _, raw := range s.shopProducts { - g, _, _ := wire.Varint(raw, 1) - id, _, _ := wire.Varint(raw, 2) - sale, _, _ := wire.Varint(raw, 3) - _, ok := s.catalog.Design(gamedata.CashProductKey{GroupID: g, ProductID: id, SaleGroup: sale}) - if ok { - response = wire.AppendBytes(response, 1, raw) - } - } - t := s.now().UTC().Add(-time.Duration(s.resetSeconds) * time.Second) - day := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, time.UTC) - daily := day.AddDate(0, 0, 1) - days := (8 - int(day.Weekday())) % 7 - if days == 0 { - days = 7 - } - weekly := day.AddDate(0, 0, days) - monthly := time.Date(t.Year(), t.Month()+1, 1, 0, 0, 0, 0, time.UTC) - for i, reset := range []time.Time{daily, weekly, monthly} { - response = wire.AppendVarint(response, i+2, uint64(reset.Add(time.Duration(s.resetSeconds)*time.Second).UnixMilli())) - } - return response -} -func (s *Service) SetClock(now func() time.Time, resetSeconds int64) { - s.now = now - s.resetSeconds = resetSeconds -} -func skuKey(k gamedata.CashProductKey) string { - return fmt.Sprintf("%d:%d:%d", k.GroupID, k.ProductID, k.SaleGroup) -} -func (s *Service) load() (purchaseState, error) { - v := purchaseState{Receipts: map[string]purchaseReceipt{}, Billing: map[string]string{}, Counts: map[string]purchaseCount{}} - b, e := s.store.Load("commerce") - if e != nil || b == nil { - return v, e - } - if e = stateio.RequireExactJSONObject(b, "receipts", "billing", "counts"); e != nil { - return v, e - } - if e = json.Unmarshal(b, &v); e != nil { - return v, e - } - if v.Receipts == nil || v.Billing == nil || v.Counts == nil { - return v, fmt.Errorf("commerce: invalid state") - } - for identity, receipt := range v.Receipts { - digest, err := hex.DecodeString(receipt.Digest) - if identity == "" || err != nil || len(digest) != sha256.Size || len(receipt.Response) == 0 { - return v, fmt.Errorf("commerce: invalid saved receipt") - } - } - for _, identity := range v.Billing { - if _, ok := v.Receipts[identity]; !ok { - return v, fmt.Errorf("commerce: billing identity has no purchase receipt") - } - } - for key, count := range v.Counts { - if key == "" || count.Period == "" || count.Count == 0 || count.Count > math.MaxInt32 || count.Lifetime < count.Count { - return v, fmt.Errorf("commerce: invalid saved purchase count") - } - } - return v, nil -} -func (s *Service) period(typ uint64) string { - t := s.now().UTC().Add(-time.Duration(s.resetSeconds) * time.Second) - switch typ { - case 1: - return t.Format("2006-01-02") - case 2: - y, w := t.ISOWeek() - return fmt.Sprintf("%d-W%02d", y, w) - case 3: - return t.Format("2006-01") - default: - return "account" - } -} -func (s *Service) count(v purchaseState, d gamedata.CashProductDesign) uint64 { - c := v.Counts[skuKey(d.Key)] - if c.Period != s.period(d.PurchaseLimitType) { - return 0 - } - return c.Count -} -func (s *Service) HandleSession(path string, request []byte, session string) (int, []byte, bool, error) { - switch path { - case "/CashShopInfo": - if s.shopWindows == nil { - return 0, nil, false, nil - } - if seq, ok, err := wire.Varint(request, 1); err != nil || !ok || seq == 0 || seq > math.MaxInt32 { - return 60, nil, true, fmt.Errorf("commerce: invalid shop sequence") - } - return 60, s.shopInfo(), true, nil - case "/CashShopPurchaseCountInfo": - if seq, ok, err := wire.Varint(request, 1); err != nil || !ok || seq == 0 || seq > math.MaxInt32 { - return 432, nil, true, fmt.Errorf("commerce: invalid count sequence") - } - infos, err := s.purchaseCountDBInfos() - if err != nil { - return 0, nil, true, err - } - var b []byte - for _, i := range infos { - b = wire.AppendBytes(b, 1, i) - } - return 432, b, true, nil - case "/CashShopBuy": - return s.buy(session, request) - default: - return 0, nil, false, nil - } -} -func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) { - return s.HandleSession(path, request, "") -} - -type buyLine struct { - Key gamedata.CashProductKey - Count uint64 -} - -func parseBuy(request []byte) (uint64, []buyLine, []string, error) { - seq, found, e := wire.Varint(request, 1) - if e != nil || !found || seq == 0 || seq > math.MaxInt32 { - return 0, nil, nil, fmt.Errorf("commerce: invalid sequence") - } - group, found, e := wire.Varint(request, 3) - if e != nil || !found || group == 0 { - return 0, nil, nil, fmt.Errorf("commerce: invalid product group") - } - var lines []buyLine - var billing []string - seen := map[gamedata.CashProductKey]bool{} - e = wire.Walk(request, func(f wire.Field) error { - switch f.Number { - case 4: - if f.Type != 2 { - return fmt.Errorf("commerce: invalid buy info") - } - id, ok, err := wire.Varint(f.Value, 1) - if err != nil || !ok || id == 0 { - return fmt.Errorf("commerce: invalid product id") - } - sale, _, err := wire.Varint(f.Value, 2) - if err != nil { - return err - } - n, ok, err := wire.Varint(f.Value, 3) - if err != nil || !ok || n == 0 || n > math.MaxInt32 { - return fmt.Errorf("commerce: invalid buy count") - } - k := gamedata.CashProductKey{GroupID: group, ProductID: id, SaleGroup: sale} - if seen[k] { - return fmt.Errorf("commerce: duplicate buy info") - } - seen[k] = true - lines = append(lines, buyLine{k, n}) - case 7: - if f.Type != 2 { - return fmt.Errorf("commerce: invalid billing info") - } - return wire.Walk(f.Value, func(b wire.Field) error { - if b.Number == 1 || b.Number == 2 { - if b.Type != 2 { - return fmt.Errorf("commerce: invalid billing identity") - } - if len(b.Value) > 16384 { - return fmt.Errorf("commerce: excessive billing identity") - } - if len(b.Value) > 0 { - h := sha256.Sum256(b.Value) - billing = append(billing, fmt.Sprintf("%d:%x", b.Number, h)) - } - } - return nil - }) - } - return nil - }) - if e == nil && len(lines) == 0 { - e = fmt.Errorf("commerce: empty purchase") - } - return seq, lines, billing, e -} -func (s *Service) buy(session string, request []byte) (int, []byte, bool, error) { - if session == "" { - return 61, nil, true, fmt.Errorf("commerce: authenticated session required") - } - seq, lines, billing, e := parseBuy(request) - if e != nil { - return 61, nil, true, e - } - var native, cash bool - for _, line := range lines { - if design, ok := s.catalog.Design(line.Key); ok { - if design.PriceType == 1 { - cash = true - } else { - native = true - } - } - } - if native && cash { - return 61, nil, true, fmt.Errorf("commerce: mixed cash and native purchase") - } - if !cash { - // Native BillingInfo contains display identifiers (often a product ID), - // not a unique payment receipt. Request identity provides replay safety. - billing = nil - } - h := sha256.Sum256(request) - digest := hex.EncodeToString(h[:]) - sessionHash := sha256.Sum256([]byte(session)) - identity := fmt.Sprintf("commerce:%x:%d", sessionHash, seq) - s.mu.Lock() - defer s.mu.Unlock() - v, e := s.load() - if e != nil { - return 61, nil, true, e - } - if r, ok := v.Receipts[identity]; ok { - if r.Digest != digest { - return 61, nil, true, fmt.Errorf("commerce: sequence reused with different purchase") - } - return 61, append([]byte(nil), r.Response...), true, nil - } - for _, b := range billing { - if _, ok := v.Billing[b]; ok { - return 61, nil, true, fmt.Errorf("commerce: billing receipt already used") - } - } - // Validate every line before charging the first one. - designs := make([]gamedata.CashProductDesign, len(lines)) - quotes := make([]Product, len(lines)) - for i, l := range lines { - if !s.predecessorBought(v, l.Key) { - return 61, nil, true, fmt.Errorf("commerce: preceding package purchase required") - } - d, ok := s.catalog.Design(l.Key) - if !ok { - return 61, nil, true, fmt.Errorf("commerce: unknown product") - } - if !s.available(d) { - return 61, nil, true, fmt.Errorf("commerce: product is not currently available") - } - if s.specialProducts[l.Key] && (len(lines) != 1 || l.Count != 1) { - return 61, nil, true, fmt.Errorf("commerce: special purchase requires one product") - } - var q Product - if d.PriceType == 1 { - var err error - q, err = s.catalog.Quote(l.Key, l.Count) - if err != nil { - return 61, nil, true, err - } - } else { - cost, err := nativePrice(d, l.Count) - if err != nil { - return 61, nil, true, err - } - q = Product{Enabled: true, ItemType: d.PriceType, Cost: cost} - } - if !q.Enabled { - return 61, nil, true, fmt.Errorf("commerce: product disabled") - } - if len(lines) > 1 && d.BulkOrderAvailability != 1 { - return 61, nil, true, fmt.Errorf("commerce: bulk purchase disabled") - } - n := s.count(v, d) - if s.legacy != nil { - for _, raw := range s.legacy.PurchaseCountDBInfos() { - g, _, _ := wire.Varint(raw, 1) - id, _, _ := wire.Varint(raw, 2) - sale, _, _ := wire.Varint(raw, 3) - legacyCount, _, _ := wire.Varint(raw, 4) - if g == l.Key.GroupID && id == l.Key.ProductID && sale == l.Key.SaleGroup && legacyCount > n { - n = legacyCount - } - } - } - if d.PurchaseLimitType > 4 { - return 61, nil, true, fmt.Errorf("commerce: unknown purchase limit") - } - if n > math.MaxInt32-l.Count { - return 61, nil, true, fmt.Errorf("commerce: purchase count overflow") - } - if d.PurchaseLimitType != 0 && (d.PurchaseLimitCount == 0 || n+l.Count > d.PurchaseLimitCount) { - return 61, nil, true, fmt.Errorf("commerce: purchase limit exceeded") - } - designs[i] = d - quotes[i] = q - } - if cash { - e = validateAcceptedQuote(request, quotes) - } - if e != nil { - return 61, nil, true, e - } - var bundle []byte - for i, l := range lines { - d, q := designs[i], quotes[i] - operation := fmt.Sprintf("%s:%d", identity, i) - var costs []gamedata.Reward - if q.Cost > 0 { - if d.PriceType != 1 { - costs = []gamedata.Reward{{Type: d.PriceType, ID: d.PriceID, Count: q.Cost}} - if resolver, ok := s.economy.(nativeCostResolver); ok { - var err error - costs, err = resolver.NativePurchaseCosts(costs[0]) - if err != nil { - return 61, nil, true, err - } - } - } else { - var typ uint64 - switch q.Currency { - case "paid_diamonds": - typ = 2 - case "diamonds": - typ = 3 - case "gold": - typ = 4 - default: - return 61, nil, true, fmt.Errorf("commerce: unknown currency %q", q.Currency) - } - costs = []gamedata.Reward{{Type: typ, Count: q.Cost}} - } - } - var special []byte - handled := false - // Delegate must only select known special products. Parent account transaction - // guarantees its grant and the subsequent debit commit together. - if s.delegate != nil && (d.PriceType == 1 || s.specialProducts[l.Key]) { - special, handled, e = s.delegate(l.Key, request) - if e != nil { - return 61, nil, true, e - } - if handled && (len(lines) != 1 || l.Count != 1) { - return 61, nil, true, fmt.Errorf("commerce: special purchase requires one product") - } - } - var rewards []gamedata.Reward - if !handled { - if d.RandomBoxID == 0 { - return 61, nil, true, fmt.Errorf("commerce: product reward missing") - } - rewards = append(rewards, gamedata.Reward{Type: 9, ID: d.RandomBoxID, Count: l.Count}) - if d.BonusRandomBoxID != 0 { - rewards = append(rewards, gamedata.Reward{Type: 9, ID: d.BonusRandomBoxID, Count: l.Count}) - } - } - var b []byte - var err error - if delivery, ok := s.economy.(interface { - ApplyPurchase(string, []gamedata.Reward, []gamedata.Reward) ([]byte, error) - }); ok { - b, err = delivery.ApplyPurchase(operation, costs, rewards) - } else { - b, err = s.economy.Apply(operation, costs, rewards) - } - if err != nil { - return 61, nil, true, err - } - bundle = append(bundle, b...) - bundle = append(bundle, special...) - if s.hook != nil { - if err = s.hook(operation, d, l.Count); err != nil { - return 61, nil, true, err - } - } - old := v.Counts[skuKey(l.Key)] - if old.Lifetime > math.MaxUint64-l.Count { - return 61, nil, true, fmt.Errorf("commerce: lifetime purchase count overflow") - } - v.Counts[skuKey(l.Key)] = purchaseCount{Count: s.count(v, d) + l.Count, Period: s.period(d.PurchaseLimitType), Lifetime: old.Lifetime + l.Count} - } - response := wire.AppendBytes(nil, 1, bundle) - v.Receipts[identity] = purchaseReceipt{digest, response} - for _, b := range billing { - v.Billing[b] = identity - } - encoded, e := json.Marshal(v) - if e == nil { - e = s.store.Save("commerce", encoded) - } - return 61, response, true, e -} - -// The receipt repeats the client-visible quote for stale-catalog detection. -// Prices always come from Catalog, and a mismatch never changes the debit. -func validateAcceptedQuote(request []byte, quotes []Product) error { - var pay, receipt string - seen := false - err := wire.Walk(request, func(f wire.Field) error { - if f.Number != 7 { - return nil - } - if seen || f.Type != 2 { - return fmt.Errorf("commerce: invalid billing info") - } - seen = true - return wire.Walk(f.Value, func(b wire.Field) error { - if b.Number == 1 { - if b.Type != 2 || pay != "" { - return fmt.Errorf("commerce: invalid pay id") - } - pay = string(b.Value) - } - if b.Number == 2 { - if b.Type != 2 || receipt != "" { - return fmt.Errorf("commerce: invalid receipt") - } - receipt = string(b.Value) - } - return nil - }) - }) - if err != nil { - return err - } - parts := strings.Split(receipt, ":") - if len(parts) != 4 || parts[0] != "bd2-local-commerce-v1" || parts[1] != pay { - return fmt.Errorf("commerce: local billing quote required") - } - payID, err := strconv.ParseUint(pay, 10, 64) - if err != nil || payID == 0 || strconv.FormatUint(payID, 10) != pay { - return fmt.Errorf("commerce: invalid local pay id") - } - typ, err := strconv.ParseUint(parts[2], 10, 64) - if err != nil || strconv.FormatUint(typ, 10) != parts[2] { - return fmt.Errorf("commerce: invalid accepted quote currency") - } - amount, err := strconv.ParseUint(parts[3], 10, 64) - if err != nil || strconv.FormatUint(amount, 10) != parts[3] { - return fmt.Errorf("commerce: invalid accepted quote amount") - } - var expected uint64 - for _, q := range quotes { - if q.ItemType != typ || q.Cost > math.MaxInt32-expected { - return fmt.Errorf("commerce: accepted quote currency mismatch") - } - expected += q.Cost - } - if expected != amount { - return fmt.Errorf("commerce: purchase quote changed; refresh the shop") - } - return nil -} -func (s *Service) purchaseCountDBInfos() ([][]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() - v, e := s.load() - if e != nil { - return nil, e - } - counts := map[gamedata.CashProductKey]uint64{} - if s.legacy != nil { - for _, raw := range s.legacy.PurchaseCountDBInfos() { - g, _, _ := wire.Varint(raw, 1) - id, _, _ := wire.Varint(raw, 2) - sale, _, _ := wire.Varint(raw, 3) - n, _, _ := wire.Varint(raw, 4) - counts[gamedata.CashProductKey{GroupID: g, ProductID: id, SaleGroup: sale}] = n - } - } - for _, d := range s.catalog.Designs() { - if n := s.count(v, d); n > counts[d.Key] { - counts[d.Key] = n - } - } - keys := make([]gamedata.CashProductKey, 0, len(counts)) - for k := range counts { - keys = append(keys, k) - } - sort.Slice(keys, func(i, j int) bool { - a, b := keys[i], keys[j] - if a.GroupID != b.GroupID { - return a.GroupID < b.GroupID - } - if a.ProductID != b.ProductID { - return a.ProductID < b.ProductID - } - return a.SaleGroup < b.SaleGroup - }) - var out [][]byte - for _, k := range keys { - b := wire.AppendVarint(nil, 1, k.GroupID) - b = wire.AppendVarint(b, 2, k.ProductID) - b = wire.AppendVarint(b, 3, k.SaleGroup) - b = wire.AppendVarint(b, 4, counts[k]) - out = append(out, b) - } - return out, nil -} -func (s *Service) PurchaseCountDBInfos() [][]byte { v, _ := s.purchaseCountDBInfos(); return v } - -// LifetimePurchaseTotal derives cumulative bonuses from committed purchases, -// including previous reset periods, without keeping a second purchase counter. -func (s *Service) LifetimePurchaseTotal(keys []gamedata.CashProductKey) (uint64, error) { - s.mu.Lock() - defer s.mu.Unlock() - v, err := s.load() - if err != nil { - return 0, err - } - var total uint64 - for _, k := range keys { - n := v.Counts[skuKey(k)].Lifetime - if n > math.MaxInt32-total { - return 0, fmt.Errorf("commerce: bonus purchase count overflow") - } - total += n - } - return total, nil -} -func (s *Service) HasPurchased(k gamedata.CashProductKey) bool { - s.mu.Lock() - defer s.mu.Unlock() - v, e := s.load() - if e != nil { - return false - } - return v.Counts[skuKey(k)].Count > 0 -} - -// ConsumeEntitlement is an authorization callback within the caller's account -// transaction. Each successful cash activation consumes one lifetime purchase. -// It must not be called recursively from the purchase hook or delegate. -func (s *Service) ConsumeEntitlement(k gamedata.CashProductKey) bool { - s.mu.Lock() - defer s.mu.Unlock() - v, err := s.load() - if err != nil { - return false - } - n := v.Counts[skuKey(k)].Lifetime - used := map[string]uint64{} - raw, err := s.store.Load("commerce_pass_receipts") - if err != nil { - return false - } - if raw != nil { - if err = json.Unmarshal(raw, &used); err != nil || used == nil { - return false - } - } - key := skuKey(k) - if used[key] >= n { - return false - } - used[key]++ - raw, err = json.Marshal(used) - if err != nil { - return false - } - return s.store.Save("commerce_pass_receipts", raw) == nil -} diff --git a/go/internal/server/commerce/service_test.go b/go/internal/server/commerce/service_test.go deleted file mode 100644 index 75f424c..0000000 --- a/go/internal/server/commerce/service_test.go +++ /dev/null @@ -1,320 +0,0 @@ -package commerce - -import ( - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gameconfig" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/readonly" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "encoding/json" - "errors" - "fmt" - "path/filepath" - "testing" - "time" -) - -type purchaseEconomy struct { - calls int - costs, rewards []gamedata.Reward - fail bool -} - -func (e *purchaseEconomy) Apply(_ string, c, r []gamedata.Reward) ([]byte, error) { - e.calls++ - e.costs = c - e.rewards = r - if e.fail { - return nil, fmt.Errorf("insufficient balance") - } - return []byte{8, 1}, nil -} -func serviceFixture(t *testing.T, limit uint64) (*Service, *purchaseEconomy, *stateio.Memory) { - t.Helper() - design := &gamedata.CashCatalog{Products: []gamedata.CashProductDesign{ - {Key: gamedata.CashProductKey{GroupID: 1, ProductID: 1}, GoogleSKU: "topup", PriceType: 1, PriceCount: 2200, RandomBoxID: 10, Recharge: true, NominalPaidDiamonds: 100}, - {Key: gamedata.CashProductKey{GroupID: 2, ProductID: 1}, GoogleSKU: "goods", PriceType: 1, PriceCount: 22000, RandomBoxID: 20, PurchaseLimitType: limit, PurchaseLimitCount: 1}, - {Key: gamedata.CashProductKey{GroupID: 2, ProductID: 1, SaleGroup: 1}, GoogleSKU: "goods-sale", PriceType: 1, PriceCount: 11000, RandomBoxID: 21, PurchaseLimitType: 4, PurchaseLimitCount: 1}, - }} - c, err := NewCatalog("test", design, gameconfig.Default().Purchases) - if err != nil { - t.Fatal(err) - } - eco := &purchaseEconomy{} - store := stateio.NewMemory() - s, err := NewService(c, store, eco) - if err != nil { - t.Fatal(err) - } - return s, eco, store -} -func buyRequest(seq, group, id, sale uint64, billing string) []byte { - b := wire.AppendVarint(nil, 1, seq) - b = wire.AppendVarint(b, 3, group) - line := wire.AppendVarint(nil, 1, id) - line = wire.AppendVarint(line, 2, sale) - line = wire.AppendVarint(line, 3, 1) - b = wire.AppendBytes(b, 4, line) - if billing == "" { - billing = fmt.Sprint(seq) - } - if billing == "pay-1" { - billing = "1" - } - typ, amount := uint64(2), uint64(1000) - if sale == 1 { - amount = 500 - } - if group == 1 { - typ = 0 - amount = 0 - } - info := wire.AppendBytes(nil, 1, []byte(billing)) - info = wire.AppendBytes(info, 2, []byte(fmt.Sprintf("bd2-local-commerce-v1:%s:%d:%d", billing, typ, amount))) - b = wire.AppendBytes(b, 7, info) - return b -} -func TestPurchaseReplayPersistsAndRejectsCollisions(t *testing.T) { - s, eco, store := serviceFixture(t, 0) - req := buyRequest(1, 2, 1, 0, "pay-1") - code, out, ok, err := s.HandleSession("/CashShopBuy", req, "session") - if err != nil || !ok || code != 61 { - t.Fatal(code, ok, err) - } - if eco.calls != 1 || len(eco.costs) != 1 || eco.costs[0].Type != 2 || eco.costs[0].Count != 1000 || eco.rewards[0].ID != 20 { - t.Fatal(eco) - } - s2, err := NewService(s.catalog, store, eco) - if err != nil { - t.Fatal(err) - } - _, replayed, _, err := s2.HandleSession("/CashShopBuy", req, "session") - if err != nil || !bytes.Equal(out, replayed) || eco.calls != 1 { - t.Fatal(err, eco.calls) - } - changed := buyRequest(1, 2, 1, 1, "") - if _, _, _, err = s2.HandleSession("/CashShopBuy", changed, "session"); err == nil { - t.Fatal("changed sequence accepted") - } - if _, _, _, err = s2.HandleSession("/CashShopBuy", req, "other-session"); err == nil { - t.Fatal("billing identity reused") - } -} -func TestSaleGroupsLimitsAndReset(t *testing.T) { - s, eco, _ := serviceFixture(t, 1) - now := time.Date(2026, 10, 5, 1, 0, 0, 0, time.UTC) - s.SetClock(func() time.Time { return now }, 0) - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(2, 2, 1, 0, ""), "s"); err == nil { - t.Fatal("daily limit ignored") - } - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(3, 2, 1, 1, ""), "s"); err != nil { - t.Fatal("sale group should be independent", err) - } - now = now.Add(24 * time.Hour) - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(4, 2, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - if eco.calls != 3 { - t.Fatal(eco.calls) - } -} -func TestFailedEconomyDoesNotConsumeLimitAndFreeRecharge(t *testing.T) { - s, eco, _ := serviceFixture(t, 4) - eco.fail = true - req := buyRequest(1, 2, 1, 0, "") - if _, _, _, err := s.HandleSession("/CashShopBuy", req, "s"); err == nil { - t.Fatal("failure accepted") - } - if s.HasPurchased(gamedata.CashProductKey{GroupID: 2, ProductID: 1}) { - t.Fatal("failed buy persisted") - } - eco.fail = false - if _, _, _, err := s.HandleSession("/CashShopBuy", req, "s"); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(2, 1, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - if len(eco.costs) != 0 || eco.rewards[0].ID != 10 { - t.Fatal("free recharge wrong", eco) - } -} -func TestAuthenticatedSessionRequired(t *testing.T) { - s, eco, _ := serviceFixture(t, 0) - if _, _, _, err := s.Handle("/CashShopBuy", buyRequest(1, 2, 1, 0, "")); err == nil || eco.calls != 0 { - t.Fatal("unauthenticated buy accepted") - } -} -func TestPassEntitlementConsumedOncePerPurchase(t *testing.T) { - s, _, _ := serviceFixture(t, 0) - k := gamedata.CashProductKey{GroupID: 2, ProductID: 1} - if s.ConsumeEntitlement(k) { - t.Fatal("unpaid entitlement authorized") - } - for i := uint64(1); i <= 2; i++ { - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(i, 2, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - if !s.ConsumeEntitlement(k) || s.ConsumeEntitlement(k) { - t.Fatal("entitlement count not enforced") - } - } -} -func TestRelayRequiresPreviousProduct(t *testing.T) { - s, eco, _ := serviceFixture(t, 0) - if err := s.AttachPackageRules([]gamedata.CashPackageDesign{{GroupID: 2, ID: 1, PackageType: 8, ContentsGroupID: 7, ContentsSortID: 1}, {GroupID: 2, ID: 1, SaleGroup: 1, PackageType: 8, ContentsGroupID: 7, ContentsSortID: 2}}); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 1, ""), "s"); err == nil || eco.calls != 0 { - t.Fatal("relay step skipped") - } - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(2, 2, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(3, 2, 1, 1, ""), "s"); err != nil { - t.Fatal(err) - } -} - -func TestStaleClientQuoteRejectedBeforeEconomy(t *testing.T) { - s, eco, _ := serviceFixture(t, 0) - req := buyRequest(1, 2, 1, 0, "") - req = bytes.ReplaceAll(req, []byte(":2:1000"), []byte(":3:1000")) - if _, _, _, err := s.HandleSession("/CashShopBuy", req, "s"); err == nil || eco.calls != 0 { - t.Fatal("stale currency accepted") - } - req = buyRequest(2, 2, 1, 0, "") - req = bytes.ReplaceAll(req, []byte(":2:1000"), []byte(":2:9999")) - if _, _, _, err := s.HandleSession("/CashShopBuy", req, "s"); err == nil || eco.calls != 0 { - t.Fatal("stale amount accepted") - } -} -func TestShopPublishesAllWindowsAndAuthorizesActiveProducts(t *testing.T) { - s, eco, _ := serviceFixture(t, 0) - now := time.Date(2026, 10, 5, 0, 0, 0, 0, time.UTC) - s.SetClock(func() time.Time { return now }, 0) - seed := &readonly.Seed{Responses: map[string]readonly.Response{"/CashShopInfo": {PacketCode: 60, Fields: []readonly.Field{{Number: 1, Type: 2, Fields: []readonly.Field{{Number: 1, Type: 0, Varint: 2}, {Number: 2, Type: 0, Varint: 1}, {Number: 4, Type: 0, Varint: uint64(now.Add(-time.Hour).UnixMilli())}, {Number: 5, Type: 0, Varint: uint64(now.Add(time.Hour).UnixMilli())}, {Number: 8, Type: 0, Varint: 123}}}}}}} - if err := s.AttachShopSeed(seed); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 0, ""), "s"); err != nil { - t.Fatal(err) - } - now = now.Add(2 * time.Hour) - if _, _, _, err := s.HandleSession("/CashShopBuy", buyRequest(2, 2, 1, 0, ""), "s"); err == nil { - t.Fatal("expired product accepted") - } - code, out, ok, err := s.HandleSession("/CashShopInfo", wire.AppendVarint(nil, 1, 1), "s") - if err != nil || !ok || code != 60 { - t.Fatal(err) - } - products := 0 - _ = wire.Walk(out, func(f wire.Field) error { - if f.Number == 1 { - products++ - } - return nil - }) - if products != 1 || eco.calls != 1 { - t.Fatal(products, eco.calls) - } -} - -type walletPurchaseEconomy struct { - wallet *player.Wallet - store stateio.Store -} - -func (e walletPurchaseEconomy) Apply(identity string, costs, rewards []gamedata.Reward) ([]byte, error) { - if err := e.wallet.ExchangeOnce(identity, costs, nil); err != nil { - return nil, err - } - raw, err := json.Marshal(rewards) - if err != nil { - return nil, err - } - if err = e.store.Save("commerce_test_rewards", raw); err != nil { - return nil, err - } - return []byte{8, 1}, nil -} -func TestSQLiteAccountTransactionRollsBackCostRewardAndReceipt(t *testing.T) { - s, _, _ := serviceFixture(t, 4) - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - t.Cleanup(func() { _ = repo.Close() }) - wallet, err := player.OpenWallet(repo, player.Currency{Jewelry: 1000}) - if err != nil { - t.Fatal(err) - } - if err = wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - s, err = NewService(s.catalog, repo, walletPurchaseEconomy{wallet, repo}) - if err != nil { - t.Fatal(err) - } - s.AttachPurchaseHook(func(string, gamedata.CashProductDesign, uint64) error { return fmt.Errorf("entitlement failure") }) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 0, ""), "s"); err == nil { - t.Fatal("hook failure lost") - } - if err = op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - if err = repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - wallet, err = player.OpenWallet(repo, player.Currency{}) - if err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Jewelry != 1000 { - t.Fatal("paid diamonds survived rollback") - } - for _, name := range []string{"commerce", "commerce_test_rewards"} { - raw, err := repo.Load(name) - if err != nil || raw != nil { - t.Fatal("purchase state survived rollback", name, err) - } - } - s, err = NewService(s.catalog, repo, walletPurchaseEconomy{wallet, repo}) - if err != nil { - t.Fatal(err) - } - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.HandleSession("/CashShopBuy", buyRequest(1, 2, 1, 0, ""), "s"); err != nil { - _ = op.Rollback() - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Jewelry != 0 || !s.HasPurchased(gamedata.CashProductKey{GroupID: 2, ProductID: 1}) { - t.Fatal("retry did not commit") - } -} diff --git a/go/internal/server/cryptox/cryptox_test.go b/go/internal/server/cryptox/cryptox_test.go deleted file mode 100644 index 2d55f76..0000000 --- a/go/internal/server/cryptox/cryptox_test.go +++ /dev/null @@ -1,37 +0,0 @@ -package cryptox - -import ( - "bytes" - "testing" -) - -func TestAESRoundTripAndFixedKey(t *testing.T) { - plain := []byte("BD2 AES fixture: blocks and PKCS7") - encoded, err := EncryptBase64(plain, Key()) - if err != nil { - t.Fatal(err) - } - got, err := DecryptBase64(encoded, Key()) - if err != nil { - t.Fatal(err) - } - if !bytes.Equal(got, plain) { - t.Fatalf("round trip = %q, want %q", got, plain) - } -} - -func TestSessionKeyValidation(t *testing.T) { - key, err := SessionKey("7eeb5c2480c298f6fc4c00d22b1b0b15") - if err != nil || string(key) != "7eeb5c2480c298f6fc4c00d22b1b0b15" { - t.Fatalf("SessionKey = %q, %v", key, err) - } - if _, err := SessionKey("not-a-key"); err == nil { - t.Fatal("SessionKey accepted invalid key") - } -} - -func TestRejectBadPadding(t *testing.T) { - if _, err := Decrypt(make([]byte, 16), Key()); err == nil { - t.Fatal("Decrypt accepted bad padding") - } -} diff --git a/go/internal/server/dbcrypt/pages_test.go b/go/internal/server/dbcrypt/pages_test.go deleted file mode 100644 index c49bb19..0000000 --- a/go/internal/server/dbcrypt/pages_test.go +++ /dev/null @@ -1,26 +0,0 @@ -package dbcrypt - -import ( - "bytes" - "testing" -) - -func TestRoundTrip(t *testing.T) { - plain := make([]byte, PageSize*2) - copy(plain, Header) - copy(plain[PageSize:], []byte("second independent page")) - encrypted, err := EncryptPages(plain) - if err != nil { - t.Fatal(err) - } - if bytes.Equal(encrypted, plain) { - t.Fatal("encryption did not change data") - } - decrypted, err := DecryptPages(encrypted) - if err != nil { - t.Fatal(err) - } - if !bytes.Equal(decrypted, plain) { - t.Fatal("round trip changed pages") - } -} diff --git a/go/internal/server/deck/auto_recovery.go b/go/internal/server/deck/auto_recovery.go deleted file mode 100644 index 8bfd1e4..0000000 --- a/go/internal/server/deck/auto_recovery.go +++ /dev/null @@ -1,242 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "crypto/sha256" - "encoding/json" - "fmt" - "sort" -) - -func (s *Store) AttachAutoRecovery(f func(uint64, uint64, []uint64) (player.AutoRecoveryResult, error)) { - s.mu.Lock() - defer s.mu.Unlock() - s.autoRecovery = f -} -func (s *Store) AttachAutoRecoveryAllowed(f func() (bool, error)) { - s.mu.Lock() - defer s.mu.Unlock() - s.autoRecoveryAllowed = f -} - -type autoRecoveryReceipt struct { - Digest string - Body []byte -} - -func (s *Store) handleAutoRecovery(req []byte) (int, []byte, bool, error) { - fail := func(e error) (int, []byte, bool, error) { return 373, nil, true, e } - if e := checkSeq(req); e != nil { - return fail(e) - } - caster, _, e := wire.Varint(req, 2) - if e != nil { - return fail(e) - } - seq, _, _ := wire.Varint(req, 1) - s.mu.Lock() - defer s.mu.Unlock() - key := s.sessionID + ":" + fmt.Sprint(seq) - digest := fmt.Sprintf("%x", sha256.Sum256(req)) - if s.storage != nil { - raw, ok, err := s.storage.LoadEntry("deck", "auto_recovery", key) - if err != nil { - return fail(err) - } - if ok { - var r autoRecoveryReceipt - if err = json.Unmarshal(raw, &r); err != nil { - return fail(err) - } - if r.Digest != digest { - return fail(fmt.Errorf("deck: changed automatic recovery replay")) - } - return 373, r.Body, true, nil - } - } - n := clone(s.state) - if len(n.Deck) == 0 { - return fail(fmt.Errorf("deck: automatic recovery has no battle deck")) - } - result := player.AutoRecoveryResult{Caster: caster, Catalyst: n.AutoReviveCatalyst} - if s.wallet != nil { - result.Catalyst = s.wallet.CatalystBalance() - } - mode := uint64(2) - var settings fieldSettings - if s.characters != nil { - settings, e = s.loadFieldSettings() - if e != nil { - return fail(e) - } - if caster != settings.Caster { - return fail(fmt.Errorf("deck: automatic recovery caster differs from saved setting")) - } - if n.FieldCharControlDeckType == 2 { - allExhausted := true - for _, v := range n.Deck { - hp, err := s.characters.CurrentHealth(v.CharacterInvenIndex) - if err != nil { - return fail(err) - } - if hp > 0 { - allExhausted = false - } - } - if allExhausted { - mode = 3 - } - } - field := s.visibleFieldDeckLocked() - var targets []uint64 - seen := map[uint64]bool{} - add := func(id uint64) error { - if seen[id] { - return nil - } - seen[id] = true - c, ok := s.characters.Find(id) - if !ok || player.IsStoryCharacter(c) || player.IsCharmCharacter(c) { - return nil - } - hp, err := s.characters.CurrentHealth(id) - if err != nil { - return err - } - if hp == 0 { - targets = append(targets, id) - } - return nil - } - if n.FieldCharControlDeckType == 0 { //nolint:staticcheck // QF1003 - for _, v := range n.Deck { - if e = add(v.CharacterInvenIndex); e != nil { - return fail(e) - } - } - } else if n.FieldCharControlDeckType == 1 { - for _, v := range field { - if e = add(v.CharacterInvenIndex); e != nil { - return fail(e) - } - } - } - allowed := true - if s.autoRecoveryAllowed != nil { - allowed, e = s.autoRecoveryAllowed() - if e != nil { - return fail(e) - } - } - if settings.AutoRevive && allowed && len(targets) > 0 { - if s.autoRecovery == nil { - return fail(fmt.Errorf("deck: automatic recovery executor unavailable")) - } - result, e = s.autoRecovery(seq, caster, targets) - if e != nil { - return fail(e) - } - settings.Caster = result.Caster - if len(result.Characters) > 0 { - mode = 1 - } - } - // On recovery failure, replace fatigued party members with living, - // permanent owned characters in inventory order. Keep the fatigued - // member when no replacement exists so the client can show exhaustion. - if mode != 1 && len(targets) > 0 { - all := s.characters.RawAll() - sort.Slice(all, func(i, j int) bool { return all[i].InvenIndex < all[j].InvenIndex }) - used := map[uint64]bool{} - for _, v := range n.Deck { - used[v.CharacterInvenIndex] = true - } - for _, v := range field { - used[v.CharacterInvenIndex] = true - } - replacements := map[uint64]player.Character{} - for _, id := range targets { - for _, c := range all { - if used[c.InvenIndex] || player.IsStoryCharacter(c) || player.IsCharmCharacter(c) { - continue - } - hp, err := s.characters.CurrentHealth(c.InvenIndex) - if err != nil { - return fail(err) - } - if hp == 0 { - continue - } - replacements[id] = c - used[c.InvenIndex] = true - break - } - if _, ok := replacements[id]; !ok { - mode = 3 - } - } - for i, v := range n.Deck { - if c, ok := replacements[v.CharacterInvenIndex]; ok { - n.Deck[i].CharacterInvenIndex = c.InvenIndex - } - } - for i, v := range n.FieldDeck { - if c, ok := replacements[v.CharacterInvenIndex]; ok { - n.FieldDeck[i].CharacterInvenIndex = c.InvenIndex - n.FieldDeck[i].CostumeInvenIndex = c.UseCostume - } - } - } - } else if caster != 0 { - return fail(fmt.Errorf("deck: automatic recovery character provider unavailable")) - } - var out []byte - for _, v := range n.Deck { - b := wire.AppendVarint(nil, 1, v.CharacterInvenIndex) - b = wire.AppendVarint(b, 2, v.CostumeInvenIndex) - b = wire.AppendVarint(b, 3, v.Slot) - out = wire.AppendBytes(out, 1, b) - } - for _, v := range n.FieldDeck { - b := wire.AppendVarint(nil, 1, v.Slot) - b = wire.AppendVarint(b, 2, v.CharacterInvenIndex) - b = wire.AppendVarint(b, 3, v.CostumeInvenIndex) - out = wire.AppendBytes(out, 2, b) - } - for _, c := range result.Characters { - out = wire.AppendVarint(out, 3, c.InvenIndex) - out = wire.AppendBytes(out, 5, player.CharacterWire(c)) - } - out = wire.AppendVarint(out, 4, mode) - if result.Caster != 0 { - out = wire.AppendVarint(out, 9, result.Caster) - } - out = wire.AppendVarint(out, 6, result.Experience) - out = wire.AppendVarint(out, 7, result.Catalyst) - out = wire.AppendVarint(out, 8, result.Disabled) - if s.storage != nil { - core, err := json.Marshal(n) - if err != nil { - return fail(err) - } - raw, err := json.Marshal(autoRecoveryReceipt{digest, out}) - if err != nil { - return fail(err) - } - changes := []stateio.EntryMutation{{Bucket: "auto_recovery", Key: key, Payload: raw}} - if s.characters != nil { - raw, err = json.Marshal(settings) - if err != nil { - return fail(err) - } - changes = append(changes, stateio.EntryMutation{Bucket: "field_settings", Key: "state", Payload: raw}) - } - if err = s.storage.SaveWithEntries("deck", core, changes); err != nil { - return fail(err) - } - } - s.state = n - return 373, out, true, nil -} diff --git a/go/internal/server/deck/auto_recovery_test.go b/go/internal/server/deck/auto_recovery_test.go deleted file mode 100644 index 1f97abd..0000000 --- a/go/internal/server/deck/auto_recovery_test.go +++ /dev/null @@ -1,114 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" - "bytes" - "testing" -) - -func recoveryFixture(t *testing.T) *presetFixture { - f := newPresetFixture(t) - f.deck.state.FieldCharControlDeckType = 0 - if e := f.deck.AttachFieldSettings(&gamedata.FieldSettingsDesign{TalentSlots: 1, CharacterTalentClass: map[uint64]uint64{350: 10, 360: 14}}); e != nil { - t.Fatal(e) - } - if err := f.characters.AttachMaxHealth(func(player.Character) (uint64, error) { return 100, nil }); err != nil { - t.Fatal(err) - } - if err := f.characters.SetCurrentHealth(100, 100); err != nil { - t.Fatal(err) - } - if err := f.characters.SetCurrentHealth(200, 0); err != nil { - t.Fatal(err) - } - if _, _, _, e := f.deck.Handle("/DeckSave", req(1, triple(200, 1, 1))); e != nil { - t.Fatal(e) - } - if _, _, _, e := f.deck.Handle("/CharAutoReviveSet", req(2, wire.AppendVarint(nil, 2, 1), wire.AppendVarint(nil, 3, 100))); e != nil { - t.Fatal(e) - } - return f -} - -func TestAutoRecoveryPersistsResponseAndDisabledSettingNeverExecutes(t *testing.T) { - f := recoveryFixture(t) - calls := 0 - f.deck.AttachAutoRecovery(func(seq, caster uint64, targets []uint64) (player.AutoRecoveryResult, error) { - calls++ - if caster != 100 || len(targets) != 1 || targets[0] != 200 { - t.Fatal("wrong actual fatigue targets") - } - if err := f.characters.SetCurrentHealth(200, 25); err != nil { - return player.AutoRecoveryResult{}, err - } - c, _ := f.characters.Find(200) - c.HP = 25 - return player.AutoRecoveryResult{Caster: 100, Characters: []player.Character{c}, Experience: 2, Catalyst: 10}, nil - }) - b := req(3, wire.AppendVarint(nil, 2, 100)) - code, out, _, e := f.deck.Handle("/DeckCharAutoRevive", b) - if e != nil || code != 373 { - t.Fatal(e) - } - mode, _, _ := wire.Varint(out, 4) - caster, _, _ := wire.Varint(out, 9) - if mode != 1 || caster != 100 { - t.Fatalf("response%x", out) - } - reopened, e := OpenStore(f.storage, f.seed, testPresetDesign) - if e != nil { - t.Fatal(e) - } - reopened.BeginSession("preset-test") - _, again, _, e := reopened.Handle("/DeckCharAutoRevive", b) - if e != nil || !bytes.Equal(out, again) || calls != 1 { - t.Fatal("automatic recovery receipt replay") - } - if _, _, _, e = f.deck.Handle("/CharAutoReviveSet", req(4, wire.AppendVarint(nil, 3, 100))); e != nil { - t.Fatal(e) - } - if err := f.characters.SetCurrentHealth(200, 0); err != nil { - t.Fatal(err) - } - _, _, _, e = f.deck.Handle("/DeckCharAutoRevive", req(5, wire.AppendVarint(nil, 2, 100))) - if e != nil || calls != 1 { - t.Fatal("disabled automatic recovery executed", e) - } - if got := f.deck.CurrentDeck()[0].CharacterInvenIndex; got != 100 { - t.Fatalf("fatigued member not replaced: %d", got) - } - if _, _, _, e = f.deck.Handle("/CharAutoReviveSet", req(6, wire.AppendVarint(nil, 3, 200))); e == nil { - t.Fatal("disabled setting accepted Immortal class14 caster") - } -} - -func TestAutoRecoveryFailureReportsExhaustionAndStoryCannotRecover(t *testing.T) { - f := recoveryFixture(t) - if _, _, _, e := f.deck.Handle("/DeckSave", req(3, triple(100, 1, 1), triple(200, 2, 2))); e != nil { - t.Fatal(e) - } - f.deck.AttachAutoRecovery(func(uint64, uint64, []uint64) (player.AutoRecoveryResult, error) { - return player.AutoRecoveryResult{Caster: 100, Catalyst: 0, Disabled: 2}, nil - }) - _, out, _, e := f.deck.Handle("/DeckCharAutoRevive", req(4, wire.AppendVarint(nil, 2, 100))) - mode, _, _ := wire.Varint(out, 4) - disabled, _, _ := wire.Varint(out, 8) - if e != nil || mode != 3 || disabled != 2 { - t.Fatalf("failure mode%d disabled%d %v", mode, disabled, e) - } - if err := f.characters.SetCurrentHealth(100, 0); err != nil { - t.Fatal(err) - } - f.deck.state.FieldCharControlDeckType = 2 - f.deck.AttachAutoRecovery(func(uint64, uint64, []uint64) (player.AutoRecoveryResult, error) { - t.Fatal("story mode must not recover temporary party") - return player.AutoRecoveryResult{}, nil - }) - _, out, _, e = f.deck.Handle("/DeckCharAutoRevive", req(5, wire.AppendVarint(nil, 2, 100))) - mode, _, _ = wire.Varint(out, 4) - if e != nil || mode != 3 { - t.Fatalf("story fatigue mode%d %v", mode, e) - } -} diff --git a/go/internal/server/deck/costume_use.go b/go/internal/server/deck/costume_use.go deleted file mode 100644 index 6194ad9..0000000 --- a/go/internal/server/deck/costume_use.go +++ /dev/null @@ -1,65 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/wire" - "fmt" - "maps" - "math" -) - -func (s *Store) handleCostumeUse(req []byte) (int, []byte, bool, error) { - fail := func(e error) (int, []byte, bool, error) { return 41, nil, true, e } - if e := checkSeq(req); e != nil { - return fail(e) - } - assignments := map[uint64]uint64{} - e := wire.Walk(req, func(f wire.Field) error { - if f.Number != 2 { - return nil - } - if f.Type != 2 { - return fmt.Errorf("deck: invalid costume use entry") - } - cost, _, e := wire.Varint(f.Value, 1) - if e != nil || cost == 0 || cost > math.MaxInt64 { - return fmt.Errorf("deck: invalid costume index") - } - char, _, e := wire.Varint(f.Value, 2) - if e != nil || char == 0 || char > math.MaxInt64 { - return fmt.Errorf("deck: invalid costume character") - } - if _, ok := assignments[char]; ok { - return fmt.Errorf("deck: repeated costume character") - } - assignments[char] = cost - return nil - }) - if e != nil { - return fail(e) - } - if len(assignments) == 0 { - return fail(fmt.Errorf("deck: missing costume assignments")) - } - s.mu.Lock() - defer s.mu.Unlock() - if s.characters != nil { - if s.collection == nil { - return fail(fmt.Errorf("deck: costume collection unavailable")) - } - for char, cost := range assignments { - if _, ok := s.characters.Find(char); !ok { - return fail(fmt.Errorf("deck: unknown costume character")) - } - c, ok := s.collection.CostumeByIndex(cost) - if !ok || c.UseChar != char { - return fail(fmt.Errorf("deck: costume not owned by character")) - } - } - if _, e = s.characters.ApplyPresetCostumes(assignments); e != nil { - return fail(e) - } - } - n := clone(s.state) - maps.Copy(n.Costumes, assignments) - return 41, nil, true, s.commit(n) -} diff --git a/go/internal/server/deck/costume_use_test.go b/go/internal/server/deck/costume_use_test.go deleted file mode 100644 index bd32fd8..0000000 --- a/go/internal/server/deck/costume_use_test.go +++ /dev/null @@ -1,93 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/accountstate" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" - "path/filepath" - "testing" -) - -func TestBatchCostumeUseRestoresCharacterSelectionsFromSQLite(t *testing.T) { - p := filepath.Join(t.TempDir(), "state.db") - r, e := accountstate.Open(p) - if e != nil { - t.Fatal(e) - } - seed, e := LoadSeed("../../../seed/v2_35_10/decks.json") - if e != nil { - t.Fatal(e) - } - initial := []player.Character{{InvenIndex: 100, ID: 350, HP: 80, Level: 1, UseCostume: 1001, CostumeID: 3501, ConnectPotentialCostume: 3501}, {InvenIndex: 200, ID: 360, HP: 60, Level: 1, UseCostume: 2001, CostumeID: 3601, ConnectPotentialCostume: 3601}} - inv, e := player.OpenInventory(r, &player.Starter{Version: "2.35.10"}) - if e != nil { - t.Fatal(e) - } - chars, e := player.OpenCharacterStore(r, initial, inv, "", "") - if e != nil { - t.Fatal(e) - } - coll, e := player.OpenCollectionStore(r, []player.Costume{{InvenIndex: 1001, ID: 3501, UseChar: 100}, {InvenIndex: 1002, ID: 3502, UseChar: 100}, {InvenIndex: 2001, ID: 3601, UseChar: 200}, {InvenIndex: 2002, ID: 3602, UseChar: 200}}) - if e != nil { - t.Fatal(e) - } - if e = chars.AttachCollection(coll); e != nil { - t.Fatal(e) - } - if e = coll.EnsurePersisted(); e != nil { - t.Fatal(e) - } - if e = chars.EnsurePersisted(); e != nil { - t.Fatal(e) - } - d, e := OpenStore(r, seed) - if e != nil { - t.Fatal(e) - } - d.characters = chars - d.collection = coll - invalid := req(1, wire.AppendBytes(nil, 2, wire.AppendVarint(wire.AppendVarint(nil, 1, 1002), 2, 100)), wire.AppendBytes(nil, 2, wire.AppendVarint(wire.AppendVarint(nil, 1, 1001), 2, 200))) - if _, _, _, err := d.Handle("/CostumeUse", invalid); err == nil { - t.Fatal("accepted another character's costume") - } - if c, _ := chars.Find(100); c.UseCostume != 1001 { - t.Fatal("invalid batch partially changed first character") - } - request := req(1, wire.AppendBytes(nil, 2, wire.AppendVarint(wire.AppendVarint(nil, 1, 1002), 2, 100)), wire.AppendBytes(nil, 2, wire.AppendVarint(wire.AppendVarint(nil, 1, 2002), 2, 200))) - if _, body, _, e := d.Handle("/CostumeUse", request); e != nil || len(body) != 0 { - t.Fatal("empty response protocol", e) - } - if e = r.Close(); e != nil { - t.Fatal(e) - } - r, e = accountstate.Open(p) - if e != nil { - t.Fatal(e) - } - defer func() { - if err := r.Close(); err != nil { - t.Error(err) - } - }() - inv, e = player.OpenInventory(r, &player.Starter{Version: "2.35.10"}) - if e != nil { - t.Fatal(e) - } - coll, e = player.OpenCollectionStore(r, nil) - if e != nil { - t.Fatal(e) - } - chars, e = player.OpenCharacterStore(r, initial, inv, "", "") - if e != nil { - t.Fatal(e) - } - if e = chars.AttachCollection(coll); e != nil { - t.Fatal(e) - } - for _, v := range []struct{ index, cost, id, hp, connect uint64 }{{100, 1002, 3502, 80, 3501}, {200, 2002, 3602, 60, 3601}} { - c, ok := chars.Find(v.index) - if !ok || c.UseCostume != v.cost || c.CostumeID != v.id || c.HP != v.hp || c.ConnectPotentialCostume != v.connect { - t.Fatalf("reconnected character%+v", c) - } - } -} diff --git a/go/internal/server/deck/deck.go b/go/internal/server/deck/deck.go deleted file mode 100644 index f08c79e..0000000 --- a/go/internal/server/deck/deck.go +++ /dev/null @@ -1,535 +0,0 @@ -// Package deck owns local deck, field-party, waypoint, and selected-costume -// state. It stores typed JSON, never captured protobuf/base64 envelopes. -package deck - -import ( - "encoding/json" - "errors" - "fmt" - "maps" - "os" - "sync" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" -) - -type DeckEntry struct { - CharacterInvenIndex uint64 `json:"character_inven_index"` - // CostumeInvenIndex is retained as the persisted Go/JSON name for the - // development save format. On the wire DeckDBInfo field 2 is Position: a - // zero-based battle-grid cell (or -1 while unassigned), not a costume - // inventory index. - CostumeInvenIndex uint64 `json:"costume_inven_index"` - Slot uint64 `json:"slot"` -} -type FieldEntry struct { - Slot uint64 `json:"slot"` - CharacterInvenIndex uint64 `json:"character_inven_index"` - CostumeInvenIndex uint64 `json:"costume_inven_index"` -} -type Seed struct { - Version string `json:"version"` - FieldDeck []FieldEntry `json:"field_deck"` - FieldCharControlDeckType uint64 `json:"field_char_control_deck_type"` - AutoReviveCatalyst uint64 `json:"auto_revive_catalyst,omitempty"` -} -type state struct { - Version string `json:"version"` - Deck []DeckEntry `json:"deck"` - FieldDeck []FieldEntry `json:"field_deck"` - FieldCharControlDeckType uint64 `json:"field_char_control_deck_type"` - Waypoints map[uint64][]uint64 `json:"waypoints"` - Costumes map[uint64]uint64 `json:"costumes"` - Packs map[uint64]uint64 `json:"packs"` - HighestTotalBattlePower uint64 `json:"highest_total_battle_power"` - PortraitCostumeID uint64 `json:"portrait_costume_id"` - AutoReviveCatalyst uint64 `json:"auto_revive_catalyst"` -} - -// PortraitCostume exposes the current portrait without changing the frozen deck schema. -func (s *Store) PortraitCostume() uint64 { - s.mu.RLock() - defer s.mu.RUnlock() - return s.state.PortraitCostumeID -} - -type Store struct { - autoRecovery func(uint64, uint64, []uint64) (player.AutoRecoveryResult, error) - autoRecoveryAllowed func() (bool, error) - fieldSettingsDesign *gamedata.FieldSettingsDesign - fieldSettingsPack func() (int, error) - mu sync.RWMutex - storage stateio.AtomicEntryStore - state state - presets map[uint64]Preset - presetSlots uint64 - presetDesign gamedata.PresetDesign - costumeSettings map[uint64]CostumeSetting - wallet *player.Wallet - characters *player.CharacterStore - equipment *player.EquipmentInventory - collection *player.CollectionStore - sessionID string - replies map[string]deckReply - waypointDesign func(uint64) (gamedata.WaypointPack, error) - waypointPack func(uint64, bool) error -} - -type deckReply struct { - code int - body []byte -} - -func (s *Store) CurrentDeck() []DeckEntry { - s.mu.RLock() - defer s.mu.RUnlock() - return append([]DeckEntry(nil), s.state.Deck...) -} - -func LoadSeed(path string) (Seed, error) { - b, e := os.ReadFile(path) - if e != nil { - return Seed{}, fmt.Errorf("deck: read seed: %w", e) - } - var s Seed - if e = json.Unmarshal(b, &s); e != nil { - return Seed{}, fmt.Errorf("deck: decode seed: %w", e) - } - if e = s.validate(); e != nil { - return Seed{}, e - } - return s, nil -} -func (s Seed) validate() error { - if s.Version != versionconfig.State() { - return errors.New("deck: wrong seed version") - } - return validField(s.FieldDeck) -} -func validField(entries []FieldEntry) error { - if len(entries) == 0 || len(entries) > 5 { - return errors.New("deck: invalid field deck size") - } - characters := map[uint64]bool{} - costumes := map[uint64]bool{} - sequences := map[uint64]bool{} - for _, e := range entries { - if e.Slot == 0 || e.Slot > 5 || e.CharacterInvenIndex == 0 || - characters[e.CharacterInvenIndex] || sequences[e.Slot] || - (e.CostumeInvenIndex != 0 && costumes[e.CostumeInvenIndex]) { - return errors.New("deck: invalid field deck") - } - characters[e.CharacterInvenIndex] = true - sequences[e.Slot] = true - if e.CostumeInvenIndex != 0 { - costumes[e.CostumeInvenIndex] = true - } - } - for slot := uint64(1); slot <= uint64(len(entries)); slot++ { - if !sequences[slot] { - return errors.New("deck: field deck has a missing sequence") - } - } - return nil -} - -func validDeck(entries []DeckEntry) error { - if len(entries) == 0 || len(entries) > 5 { - return errors.New("deck: invalid battle deck size") - } - characters := map[uint64]bool{} - positions := map[uint64]bool{} - sequences := map[uint64]bool{} - for _, entry := range entries { - position := entry.CostumeInvenIndex - unassigned := position == ^uint64(0) // int32 -1 sign-extends in protobuf varints. - if entry.CharacterInvenIndex == 0 || (!unassigned && position > 11) || entry.Slot == 0 || entry.Slot > 5 || - characters[entry.CharacterInvenIndex] || (!unassigned && positions[position]) || sequences[entry.Slot] { - return errors.New("deck: invalid battle deck") - } - characters[entry.CharacterInvenIndex] = true - if !unassigned { - positions[position] = true - } - sequences[entry.Slot] = true - } - return nil -} -func NewStore(seed Seed, designs ...gamedata.PresetDesign) (*Store, error) { - if e := seed.validate(); e != nil { - return nil, e - } - var design gamedata.PresetDesign - if len(designs) > 1 { - return nil, errors.New("deck: multiple preset designs") - } - if len(designs) == 1 { - design = designs[0] - if err := design.Validate(); err != nil { - return nil, err - } - } - return &Store{presetDesign: design, state: state{Version: versionconfig.State(), FieldDeck: append([]FieldEntry(nil), seed.FieldDeck...), FieldCharControlDeckType: seed.FieldCharControlDeckType, AutoReviveCatalyst: seed.AutoReviveCatalyst, Waypoints: map[uint64][]uint64{}, Costumes: map[uint64]uint64{}, Packs: map[uint64]uint64{}}, presets: map[uint64]Preset{}, presetSlots: design.BaseCount, costumeSettings: map[uint64]CostumeSetting{}, replies: map[string]deckReply{}}, nil -} -func OpenStore(storage stateio.Store, seed Seed, designs ...gamedata.PresetDesign) (*Store, error) { - s, e := NewStore(seed, designs...) - if e != nil { - return nil, e - } - entries, ok := storage.(stateio.AtomicEntryStore) - if storage == nil || !ok { - return nil, errors.New("deck: nil storage") - } - s.storage = entries - b, e := storage.Load("deck") - if e != nil { - return nil, fmt.Errorf("deck: load state: %w", e) - } - if b == nil { - if e = stateio.RequireNoEntries(entries, "deck", "presets", "preset_config", "costume_settings", "field_settings"); e != nil { - return nil, fmt.Errorf("deck: invalid entry storage: %w", e) - } - return s, nil - } - if e = stateio.RequireExactJSONObject(b, "version", "deck", "field_deck", "field_char_control_deck_type", "waypoints", "costumes", "packs", "highest_total_battle_power", "portrait_costume_id", "auto_revive_catalyst"); e != nil { - return nil, fmt.Errorf("deck: incompatible state layout: %w", e) - } - var loaded state - if e = json.Unmarshal(b, &loaded); e != nil { - return nil, fmt.Errorf("deck: malformed state: %w", e) - } - if loaded.Version != versionconfig.State() || (len(loaded.Deck) != 0 && validDeck(loaded.Deck) != nil) || validField(loaded.FieldDeck) != nil || loaded.Waypoints == nil || loaded.Costumes == nil || loaded.Packs == nil { - return nil, errors.New("deck: invalid saved state") - } - if e = validWaypointState(loaded.Waypoints); e != nil { - return nil, e - } - s.state = loaded - if e = s.loadPresetEntries(); e != nil { - return nil, e - } - return s, nil -} -func (s *Store) EnsurePersisted() error { - s.mu.Lock() - defer s.mu.Unlock() - b, e := s.storage.Load("deck") - if e != nil { - return e - } - if b != nil { - return nil - } - return s.commit(clone(s.state)) -} -func (s *Store) commit(next state) error { - if s.storage != nil { - b, e := json.MarshalIndent(next, "", " ") - if e != nil { - return e - } - if e = s.storage.Save("deck", append(b, '\n')); e != nil { - return e - } - } - s.state = next - return nil -} -func clone(x state) state { - y := x - y.Deck = append([]DeckEntry(nil), x.Deck...) - y.FieldDeck = append([]FieldEntry(nil), x.FieldDeck...) - y.Waypoints = map[uint64][]uint64{} - for k, v := range x.Waypoints { - y.Waypoints[k] = append([]uint64(nil), v...) - } - y.Costumes = map[uint64]uint64{} - maps.Copy(y.Costumes, x.Costumes) - y.Packs = map[uint64]uint64{} - maps.Copy(y.Packs, x.Packs) - return y -} -func checkSeq(req []byte) error { - v, ok, e := wire.Varint(req, 1) - if e != nil || !ok || v == 0 || v > 2147483647 { - return errors.New("deck: invalid request sequence") - } - return nil -} -func triples(req []byte) ([]DeckEntry, error) { - var out []DeckEntry - e := wire.Walk(req, func(f wire.Field) error { - if f.Number != 2 { - return nil - } - if f.Type != 2 { - return errors.New("deck: deck field") - } - a, aok, e := wire.Varint(f.Value, 1) - if e != nil || !aok || a == 0 { - return errors.New("deck: deck character") - } - position, _, e := wire.Varint(f.Value, 2) - if e != nil || (position > 11 && position != ^uint64(0)) { - return errors.New("deck: invalid deck position") - } - sequence, sequenceOK, e := wire.Varint(f.Value, 3) - if e != nil || !sequenceOK || sequence == 0 || sequence > 5 { - return errors.New("deck: invalid deck sequence") - } - out = append(out, DeckEntry{CharacterInvenIndex: a, CostumeInvenIndex: position, Slot: sequence}) - return nil - }) - if e != nil { - return nil, e - } - if len(out) == 0 { - return nil, errors.New("deck: empty deck") - } - if e := validDeck(out); e != nil { - return nil, e - } - return out, nil -} -func fieldEntries(req []byte) ([]FieldEntry, error) { - var out []FieldEntry - e := wire.Walk(req, func(f wire.Field) error { - if f.Number != 2 { - return nil - } - if f.Type != 2 { - return errors.New("deck: field deck entry is not a message") - } - slot, slotOK, err := wire.Varint(f.Value, 1) - if err != nil || !slotOK || slot == 0 { - return errors.New("deck: invalid field deck slot") - } - character, characterOK, err := wire.Varint(f.Value, 2) - if err != nil || !characterOK || character == 0 { - return errors.New("deck: invalid field deck character") - } - costume, _, err := wire.Varint(f.Value, 3) - if err != nil { - return errors.New("deck: invalid field deck costume") - } - out = append(out, FieldEntry{Slot: slot, CharacterInvenIndex: character, CostumeInvenIndex: costume}) - return nil - }) - if e != nil { - return nil, e - } - if len(out) == 0 { - return nil, errors.New("deck: empty field deck") - } - return out, validField(out) -} - -func (s *Store) validateOwnedDeckLocked(entries []DeckEntry) error { - if s.characters == nil { - return nil - } - for _, entry := range entries { - if _, found := s.characters.Find(entry.CharacterInvenIndex); !found { - return fmt.Errorf("deck: battle deck references unknown character %d", entry.CharacterInvenIndex) - } - } - return nil -} - -func (s *Store) validateOwnedFieldDeckLocked(entries []FieldEntry) error { - if s.characters == nil || s.collection == nil { - return nil - } - for _, entry := range entries { - character, found := s.characters.Find(entry.CharacterInvenIndex) - if !found { - return fmt.Errorf("deck: field deck references unknown character %d", entry.CharacterInvenIndex) - } - if player.IsStoryCharacter(character) && !s.temporaryAllowed(character) { - return fmt.Errorf("deck: field character unavailable in this pack") - } - if entry.CostumeInvenIndex == 0 { - continue - } - if (player.IsStoryCharacter(character) || player.IsCharmCharacter(character)) && character.UseCostume == entry.CostumeInvenIndex { - continue - } - costume, found := s.collection.CostumeByIndex(entry.CostumeInvenIndex) - if !found { - return fmt.Errorf("deck: field deck references unknown costume %d", entry.CostumeInvenIndex) - } - if costume.UseChar != entry.CharacterInvenIndex { - return fmt.Errorf("deck: costume %d does not belong to character %d", entry.CostumeInvenIndex, entry.CharacterInvenIndex) - } - } - return nil -} - -// Handle implements session.Handler. Every mutation validates its complete -// typed request before committing a replacement JSON state. -func (s *Store) Handle(path string, req []byte) (int, []byte, bool, error) { - switch path { - case "/TalentSlotSave", "/CharAutoReviveSet": - return s.handleFieldSettings(path, req) - case "/PresetInfo": - return s.handlePresetInfo(req) - case "/PresetSave": - return s.handlePresetSave(req) - case "/PresetAddSlot": - return s.handlePresetAddSlot(req) - case "/PresetInfoChange": - return s.handlePresetInfoChange(req) - case "/PresetDelete": - return s.handlePresetDelete(req) - case "/PresetUse": - return s.handlePresetUse(req) - case "/DeckCostumeSettingInfo": - return s.handleCostumeSettingInfo(req) - case "/DeckCostumeSettingSave": - return s.handleCostumeSettingSave(req) - case "/DeckInfo": - if e := checkSeq(req); e != nil { - return 0, nil, true, e - } - s.mu.RLock() - defer s.mu.RUnlock() - out := encodeDeck(s.state.Deck) - if s.fieldSettingsDesign != nil { - v, e := s.loadFieldSettings() - if e != nil { - return 0, nil, true, e - } - for _, id := range s.projectTalentSlots(v.TalentIDs) { - out = wire.AppendVarint(out, 2, id) - } - } - return 8, out, true, nil - case "/FieldDeckInfo": - if e := checkSeq(req); e != nil { - return 0, nil, true, e - } - s.mu.RLock() - defer s.mu.RUnlock() - return 273, encodeField(s.visibleFieldDeckLocked()), true, nil - case "/DeckCharAutoRevive": - return s.handleAutoRecovery(req) - case "/WaypointInfo": - return s.handleWaypoint(path, req) - case "/DeckSave": - if e := checkSeq(req); e != nil { - return 0, nil, true, e - } - x, e := triples(req) - if e != nil { - return 0, nil, true, e - } - s.mu.Lock() - defer s.mu.Unlock() - if e = s.validateOwnedDeckLocked(x); e != nil { - return 0, nil, true, e - } - n := clone(s.state) - n.Deck = x - e = s.commit(n) - return 10, nil, true, e - case "/FieldDeckSave": - if e := checkSeq(req); e != nil { - return 0, nil, true, e - } - x, e := fieldEntries(req) - if e != nil { - return 0, nil, true, e - } - s.mu.Lock() - defer s.mu.Unlock() - if e = s.validateOwnedFieldDeckLocked(x); e != nil { - return 0, nil, true, e - } - n := clone(s.state) - n.FieldDeck = x - e = s.commit(n) - return 274, nil, true, e - case "/SaveFieldCharControlDeckType": - // Define_FieldCharControllDeckType is a proto3 enum whose valid values - // are BATTLE=0, FIELD=1 and STORY=2. BATTLE is the protobuf default, so - // the generated client deliberately omits field 2 when it switches out - // of story mode after the final quest. An absent field is therefore a - // real value 0, not a malformed request. - v, _, e := wire.Varint(req, 2) - if e != nil || v > 2 { - return 0, nil, true, errors.New("deck: invalid field control type") - } - if e = checkSeq(req); e != nil { - return 0, nil, true, e - } - s.mu.Lock() - defer s.mu.Unlock() - n := clone(s.state) - n.FieldCharControlDeckType = v - e = s.commit(n) - return 288, nil, true, e - case "/WaypointSave", "/WaypointUse": - return s.handleWaypoint(path, req) - case "/CostumeUse": - return s.handleCostumeUse(req) - case "/SaveTotalBattlePower": - power, ok, e := wire.Varint(req, 2) - if e != nil || !ok || power == 0 { - return 0, nil, true, errors.New("deck: invalid total battle power") - } - if e = checkSeq(req); e != nil { - return 0, nil, true, e - } - s.mu.Lock() - defer s.mu.Unlock() - n := clone(s.state) - if power > n.HighestTotalBattlePower { - n.HighestTotalBattlePower = power - } - if e = s.commit(n); e != nil { - return 0, nil, true, e - } - return 258, wire.AppendVarint(nil, 1, n.HighestTotalBattlePower), true, nil - case "/UserPortraitChange": - costumeID, ok, e := wire.Varint(req, 2) - if e != nil || !ok || costumeID == 0 { - return 0, nil, true, errors.New("deck: invalid portrait costume") - } - if e = checkSeq(req); e != nil { - return 0, nil, true, e - } - s.mu.Lock() - defer s.mu.Unlock() - n := clone(s.state) - n.PortraitCostumeID = costumeID - if e = s.commit(n); e != nil { - return 0, nil, true, e - } - return 75, wire.AppendVarint(nil, 1, costumeID), true, nil - } - return 0, nil, false, nil -} -func encodeDeck(xs []DeckEntry) []byte { - var b []byte - for _, x := range xs { - v := wire.AppendVarint(nil, 1, x.CharacterInvenIndex) - v = wire.AppendVarint(v, 2, x.CostumeInvenIndex) - v = wire.AppendVarint(v, 3, x.Slot) - b = wire.AppendBytes(b, 1, v) - } - return b -} -func encodeField(xs []FieldEntry) []byte { - var b []byte - for _, x := range xs { - v := wire.AppendVarint(nil, 1, x.Slot) - v = wire.AppendVarint(v, 2, x.CharacterInvenIndex) - v = wire.AppendVarint(v, 3, x.CostumeInvenIndex) - b = wire.AppendBytes(b, 1, v) - } - return b -} diff --git a/go/internal/server/deck/deck_test.go b/go/internal/server/deck/deck_test.go deleted file mode 100644 index 352d910..0000000 --- a/go/internal/server/deck/deck_test.go +++ /dev/null @@ -1,377 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" - "path/filepath" - "testing" -) - -func seeded(t *testing.T) *Store { - t.Helper() - x, e := LoadSeed(filepath.Join("..", "..", "..", "seed", "v2_35_10", "decks.json")) - if e != nil { - t.Fatal(e) - } - s, e := NewStore(x, testPresetDesign) - if e != nil { - t.Fatal(e) - } - return s -} -func req(seq uint64, fields ...[]byte) []byte { - b := wire.AppendVarint(nil, 1, seq) - for _, f := range fields { - b = append(b, f...) - } - return b -} -func triple(a, b, c uint64) []byte { - v := wire.AppendVarint(nil, 1, a) - v = wire.AppendVarint(v, 2, b) - v = wire.AppendVarint(v, 3, c) - return wire.AppendBytes(nil, 2, v) -} - -func attachFormationOwnership(t *testing.T, store *Store) { - t.Helper() - starter := &player.Starter{ - Version: versionconfig.State(), - Characters: []player.Character{ - {InvenIndex: 101, ID: 350, Level: 1}, - {InvenIndex: 102, ID: 351, Level: 1}, - {InvenIndex: 103, ID: 352, Level: 1}, - {InvenIndex: 104, ID: 353, Level: 1}, - {InvenIndex: 105, ID: 354, Level: 1}, - }, - Costumes: []player.Costume{ - {InvenIndex: 201, ID: 60101, UseChar: 101}, - {InvenIndex: 202, ID: 60201, UseChar: 102}, - {InvenIndex: 203, ID: 60301, UseChar: 103}, - {InvenIndex: 204, ID: 60401, UseChar: 104}, - {InvenIndex: 205, ID: 60501, UseChar: 105}, - }, - } - inventory, err := player.OpenInventory(stateio.NewMemory(), starter) - if err != nil { - t.Fatal(err) - } - characters, err := player.OpenCharacterStore(stateio.NewMemory(), starter.Characters, inventory, "", "") - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(stateio.NewMemory(), starter.Costumes) - if err != nil { - t.Fatal(err) - } - store.characters = characters - store.collection = collection -} -func TestFieldDeckSeedAndSave(t *testing.T) { - s := seeded(t) - code, b, ok, e := s.Handle("/FieldDeckInfo", req(1)) - if e != nil || !ok || code != 273 { - t.Fatalf("info %d %t %v", code, ok, e) - } - n := 0 - if e = wire.Walk(b, func(f wire.Field) error { - if f.Number == 1 { - n++ - } - return nil - }); e != nil || n != 5 { - t.Fatalf("seed field deck: %d %v", n, e) - } - // FieldDeckDBInfo is sequence#1, character#2, costume#3. - field := triple(1, 99, 199) - code, _, ok, e = s.Handle("/FieldDeckSave", req(2, field)) - if e != nil || !ok || code != 274 { - t.Fatalf("save: %d %t %v", code, ok, e) - } - _, b, _, _ = s.Handle("/FieldDeckInfo", req(3)) - var got uint64 - _ = wire.Walk(b, func(f wire.Field) error { - if f.Number == 1 { - got, _, _ = wire.Varint(f.Value, 2) - } - return nil - }) - if got != 99 { - t.Fatalf("saved char=%d", got) - } -} -func TestDeckPersistenceAndCommands(t *testing.T) { - seed, e := LoadSeed(filepath.Join("..", "..", "..", "seed", "v2_35_10", "decks.json")) - if e != nil { - t.Fatal(e) - } - storage := stateio.NewMemory() - s, e := OpenStore(storage, seed) - if e != nil { - t.Fatal(e) - } - code, _, _, e := s.Handle("/DeckSave", req(1, triple(100, 2, 1))) - if e != nil || code != 10 { - t.Fatalf("deck save: %d %v", code, e) - } - way := wire.AppendVarint(nil, 2, 21) - if e = s.ConfigureWaypoints(func(uint64) (gamedata.WaypointPack, error) { - return gamedata.WaypointPack{Points: map[uint64]gamedata.Waypoint{1: {ID: 1, MapID: 212}}}, nil - }, func(uint64, bool) error { return nil }); e != nil { - t.Fatal(e) - } - way = wire.AppendVarint(way, 3, 1) - if code, _, _, e = s.Handle("/WaypointSave", req(2, way)); e != nil || code != 32 { - t.Fatalf("way: %d %v", code, e) - } - use := wire.AppendVarint(nil, 1, 200) - use = wire.AppendVarint(use, 2, 100) - if code, _, _, e = s.Handle("/CostumeUse", req(3, wire.AppendBytes(nil, 2, use))); e != nil || code != 41 { - t.Fatalf("use: %d %v", code, e) - } - if _, _, handled, err := s.Handle("/PackBuy", req(4, wire.AppendVarint(nil, 2, 21))); handled || err != nil { - t.Fatal("pack purchase must be handled by world") - } - var b []byte - reopened, e := OpenStore(storage, seed) - if e != nil { - t.Fatal(e) - } - _, b, _, e = reopened.Handle("/DeckInfo", req(5)) - if e != nil { - t.Fatal(e) - } - entry, _, _ := wire.Bytes(b, 1) - id, _, _ := wire.Varint(entry, 1) - if id != 100 { - t.Fatalf("persist deck=%d", id) - } -} -func TestRejectsInvalidMutations(t *testing.T) { - s := seeded(t) - if _, _, _, e := s.Handle("/DeckSave", req(1, triple(1, 2, 1), triple(3, 4, 1))); e == nil { - t.Fatal("duplicate slots accepted") - } - if _, _, _, e := s.Handle("/WaypointSave", req(1, wire.AppendVarint(nil, 2, 21))); e == nil { - t.Fatal("waypoint missing id accepted") - } -} - -func TestDeckSaveValidatesFormationShapeAndOwnership(t *testing.T) { - valid := [][]byte{ - triple(101, 0, 1), triple(102, 1, 2), triple(103, 2, 3), - triple(104, 3, 4), triple(105, 4, 5), - } - tests := []struct { - name string - entries [][]byte - }{ - {name: "six characters", entries: append(append([][]byte{}, valid...), triple(101, 5, 1))}, - {name: "duplicate character", entries: [][]byte{triple(101, 0, 1), triple(101, 1, 2)}}, - {name: "duplicate position", entries: [][]byte{triple(101, 0, 1), triple(102, 0, 2)}}, - {name: "duplicate sequence", entries: [][]byte{triple(101, 0, 1), triple(102, 1, 1)}}, - {name: "position outside twelve cells", entries: [][]byte{triple(101, 12, 1)}}, - {name: "sequence outside formation", entries: [][]byte{triple(101, 0, 6)}}, - {name: "unknown character", entries: [][]byte{triple(999, 0, 1)}}, - } - for _, test := range tests { - t.Run(test.name, func(t *testing.T) { - s := seeded(t) - attachFormationOwnership(t, s) - if _, _, handled, err := s.Handle("/DeckSave", req(1, test.entries...)); err == nil || !handled { - t.Fatalf("invalid deck accepted handled=%v err=%v", handled, err) - } - if len(s.state.Deck) != 0 { - t.Fatalf("invalid deck mutated state: %+v", s.state.Deck) - } - }) - } - s := seeded(t) - attachFormationOwnership(t, s) - if code, _, handled, err := s.Handle("/DeckSave", req(2, valid...)); err != nil || !handled || code != 10 || len(s.state.Deck) != 5 { - t.Fatalf("valid deck rejected code=%d handled=%v deck=%+v err=%v", code, handled, s.state.Deck, err) - } - unassigned := [][]byte{ - triple(101, ^uint64(0), 1), triple(102, ^uint64(0), 2), triple(103, ^uint64(0), 3), - triple(104, ^uint64(0), 4), triple(105, ^uint64(0), 5), - } - if code, _, handled, err := s.Handle("/DeckSave", req(3, unassigned...)); err != nil || !handled || code != 10 { - t.Fatalf("official unassigned positions rejected code=%d handled=%v err=%v", code, handled, err) - } -} - -func TestFieldDeckSaveValidatesFormationShapeAndOwnership(t *testing.T) { - valid := [][]byte{ - triple(1, 101, 201), triple(2, 102, 202), triple(3, 103, 203), - triple(4, 104, 204), triple(5, 105, 205), - } - tests := []struct { - name string - entries [][]byte - }{ - {name: "six characters", entries: append(append([][]byte{}, valid...), triple(1, 101, 201))}, - {name: "duplicate character", entries: [][]byte{triple(1, 101, 201), triple(2, 101, 202)}}, - {name: "duplicate costume", entries: [][]byte{triple(1, 101, 201), triple(2, 102, 201)}}, - {name: "duplicate sequence", entries: [][]byte{triple(1, 101, 201), triple(1, 102, 202)}}, - {name: "sequence outside formation", entries: [][]byte{triple(6, 101, 201)}}, - {name: "unknown character", entries: [][]byte{triple(1, 999, 201)}}, - {name: "unknown costume", entries: [][]byte{triple(1, 101, 999)}}, - {name: "costume owned by another character", entries: [][]byte{triple(1, 101, 202)}}, - } - for _, test := range tests { - t.Run(test.name, func(t *testing.T) { - s := seeded(t) - attachFormationOwnership(t, s) - before := append([]FieldEntry(nil), s.state.FieldDeck...) - if _, _, handled, err := s.Handle("/FieldDeckSave", req(1, test.entries...)); err == nil || !handled { - t.Fatalf("invalid field deck accepted handled=%v err=%v", handled, err) - } - if len(s.state.FieldDeck) != len(before) || s.state.FieldDeck[0] != before[0] { - t.Fatalf("invalid field deck mutated state: %+v", s.state.FieldDeck) - } - }) - } - s := seeded(t) - attachFormationOwnership(t, s) - if code, _, handled, err := s.Handle("/FieldDeckSave", req(2, valid...)); err != nil || !handled || code != 274 || len(s.state.FieldDeck) != 5 { - t.Fatalf("valid field deck rejected code=%d handled=%v deck=%+v err=%v", code, handled, s.state.FieldDeck, err) - } - // A zero costume index means to use the character's current costume; the - // client explicitly supports this fallback when resolving its field leader. - if code, _, handled, err := s.Handle("/FieldDeckSave", req(3, triple(1, 101, 0))); err != nil || !handled || code != 274 { - t.Fatalf("current-costume fallback rejected code=%d handled=%v err=%v", code, handled, err) - } -} - -func TestDeckSavesRequireRequestSequence(t *testing.T) { - s := seeded(t) - if _, _, handled, err := s.Handle("/DeckSave", triple(101, 0, 1)); err == nil || !handled { - t.Fatalf("DeckSave without seq accepted handled=%v err=%v", handled, err) - } - if _, _, handled, err := s.Handle("/FieldDeckSave", triple(1, 101, 201)); err == nil || !handled { - t.Fatalf("FieldDeckSave without seq accepted handled=%v err=%v", handled, err) - } -} - -func TestSaveFieldCharControlDeckTypeAcceptsProtoDefaultBattle(t *testing.T) { - s := seeded(t) - // FCCD_BATTLE=0 is omitted by proto3 serialization. This is the exact - // shape sent when the client leaves story control after the final quest. - code, _, handled, err := s.Handle("/SaveFieldCharControlDeckType", req(1)) - if err != nil || !handled || code != 288 || s.state.FieldCharControlDeckType != 0 { - t.Fatalf("battle save code=%d handled=%v type=%d err=%v", code, handled, s.state.FieldCharControlDeckType, err) - } - for seq, value := range []uint64{1, 2} { - request := req(uint64(seq+2), wire.AppendVarint(nil, 2, value)) - if code, _, handled, err = s.Handle("/SaveFieldCharControlDeckType", request); err != nil || !handled || code != 288 || s.state.FieldCharControlDeckType != value { - t.Fatalf("type %d save code=%d handled=%v stored=%d err=%v", value, code, handled, s.state.FieldCharControlDeckType, err) - } - } - if _, _, handled, err = s.Handle("/SaveFieldCharControlDeckType", req(9, wire.AppendVarint(nil, 2, 3))); !handled || err == nil { - t.Fatalf("unknown enum accepted handled=%v err=%v", handled, err) - } -} - -func TestTotalBattlePowerKeepsHighest(t *testing.T) { - s := seeded(t) - for seq, power := range []uint64{1300, 900} { - request := req(uint64(seq+1), wire.AppendVarint(nil, 2, power)) - code, response, ok, err := s.Handle("/SaveTotalBattlePower", request) - if err != nil || !ok || code != 258 { - t.Fatalf("save power: code=%d ok=%v err=%v", code, ok, err) - } - highest, found, err := wire.Varint(response, 1) - if err != nil || !found || highest != 1300 { - t.Fatalf("highest=%d found=%v err=%v", highest, found, err) - } - } -} - -func TestPortraitChangeEchoesAndStoresCostume(t *testing.T) { - s := seeded(t) - request := req(1, wire.AppendVarint(nil, 2, 3501)) - code, response, ok, err := s.Handle("/UserPortraitChange", request) - if err != nil || !ok || code != 75 { - t.Fatalf("portrait: code=%d ok=%v err=%v", code, ok, err) - } - id, found, err := wire.Varint(response, 1) - if err != nil || !found || id != 3501 || s.state.PortraitCostumeID != 3501 { - t.Fatalf("portrait response=%d found=%v stored=%d err=%v", id, found, s.state.PortraitCostumeID, err) - } -} - -func TestDeckCharAutoReviveUsesCurrentFormationWithoutInventingRevives(t *testing.T) { - s := seeded(t) - if code, _, handled, err := s.Handle("/DeckSave", req(1, triple(535607162, 10, 1), triple(535604120, 1, 2))); code != 10 || !handled || err != nil { - t.Fatalf("save code=%d handled=%v err=%v", code, handled, err) - } - code, response, handled, err := s.Handle("/DeckCharAutoRevive", req(2)) - if err != nil || !handled || code != 373 { - t.Fatalf("auto revive code=%d handled=%v err=%v", code, handled, err) - } - var battle, field, revive int - if err := wire.Walk(response, func(f wire.Field) error { - switch f.Number { - case 1: - battle++ - if battle == 1 { - index, _, _ := wire.Varint(f.Value, 1) - if index != 535607162 { - t.Errorf("first battle character %d", index) - } - } - case 2: - field++ - case 3, 5: - revive++ - } - return nil - }); err != nil || battle != 2 || field != 5 || revive != 0 { - t.Fatalf("formation battle=%d field=%d invented revive=%d err=%v", battle, field, revive, err) - } - mode, _, _ := wire.Varint(response, 4) - catalyst, _, _ := wire.Varint(response, 7) - if mode != 2 || catalyst != 200 { - t.Fatalf("official formation mode=%d catalyst=%d", mode, catalyst) - } - if _, _, handled, err := s.Handle("/DeckCharAutoRevive", req(3, wire.AppendVarint(nil, 2, 535604120))); !handled || err == nil { - t.Fatalf("unverified caster incorrectly accepted handled=%v err=%v", handled, err) - } - if _, _, handled, err := s.Handle("/DeckCharAutoRevive", nil); !handled || err == nil { - t.Fatalf("missing sequence incorrectly accepted handled=%v err=%v", handled, err) - } -} - -func TestDeckCharAutoRevivePreservesExplicitZeroCatalyst(t *testing.T) { - seed, err := LoadSeed(filepath.Join("..", "..", "..", "seed", "v2_35_10", "decks.json")) - if err != nil { - t.Fatal(err) - } - storage := stateio.NewMemory() - s, err := OpenStore(storage, seed) - if err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/DeckSave", req(1, triple(88, 11, 1))); err != nil { - t.Fatal(err) - } - s.state.AutoReviveCatalyst = 0 - if err := s.commit(s.state); err != nil { - t.Fatal(err) - } - reopened, err := OpenStore(storage, seed) - if err != nil || reopened.state.AutoReviveCatalyst != 0 || reopened.state.Deck[0].CharacterInvenIndex != 88 { - t.Fatalf("reloaded state=%+v err=%v", reopened.state, err) - } -} - -var testPresetDesign = func() gamedata.PresetDesign { - d := gamedata.PresetDesign{BaseCount: 5, Maximum: 12, PriceType: 4, Price: 2000, Icons: map[uint64]bool{}} - for id := uint64(1); id <= 21; id++ { - d.Icons[id] = true - } - return d -}() diff --git a/go/internal/server/deck/field_settings_test.go b/go/internal/server/deck/field_settings_test.go deleted file mode 100644 index 1f23473..0000000 --- a/go/internal/server/deck/field_settings_test.go +++ /dev/null @@ -1,171 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "testing" - "time" -) - -func TestTalentSlotsRestoreOwnershipAndClearSlots(t *testing.T) { - f := newPresetFixture(t) - design := &gamedata.FieldSettingsDesign{TalentSlots: 3, CharacterTalentClass: map[uint64]uint64{350: 10, 360: 9}} - if e := f.deck.AttachFieldSettings(design); e != nil { - t.Fatal(e) - } - save := req(1, wire.AppendVarint(nil, 2, 360), wire.AppendVarint(nil, 2, 0), wire.AppendVarint(nil, 2, 350)) - if _, _, _, e := f.deck.Handle("/TalentSlotSave", save); e != nil { - t.Fatal(e) - } - for _, bad := range [][]uint64{{350, 350, 0}, {999, 0, 0}, {100, 0, 0}, {350}} { - var r []byte - for _, id := range bad { - r = wire.AppendVarint(r, 2, id) - } - if _, _, _, e := f.deck.Handle("/TalentSlotSave", req(2, r)); e == nil { - t.Fatal("invalid slots accepted", bad) - } - } - if _, _, _, e := f.deck.Handle("/CharAutoReviveSet", req(3, wire.AppendVarint(nil, 2, 1), wire.AppendVarint(nil, 3, 200))); e == nil { - t.Fatal("wrong talent caster accepted") - } - if _, _, _, e := f.deck.Handle("/CharAutoReviveSet", req(4, wire.AppendVarint(nil, 2, 1), wire.AppendVarint(nil, 3, 100))); e != nil { - t.Fatal(e) - } - reopened, e := OpenStore(f.storage, f.seed, f.deck.presetDesign) - if e != nil { - t.Fatal(e) - } - reopened.characters = f.characters - if e = reopened.AttachFieldSettings(design); e != nil { - t.Fatal(e) - } - ids, e := reopened.TalentCharacterIDs() - if e != nil || ids[0] != 360 || ids[1] != 0 || ids[2] != 350 { - t.Fatal(ids, e) - } - on, caster, e := reopened.AutoReviveSettings() - if e != nil || !on || caster != 100 { - t.Fatal(on, caster, e) - } - _, b, _, e := reopened.Handle("/DeckInfo", req(5)) - if e != nil { - t.Fatal(e) - } - n := 0 - if err := wire.Walk(b, func(f wire.Field) error { - if f.Number == 2 { - n++ - } - return nil - }); err != nil { - t.Fatal(err) - } - if n != 3 { - t.Fatal("DeckInfo missing empty quick slot", n) - } -} - -func TestCharmFieldAndTalentSlotsExpireWithoutPermanentOwnership(t *testing.T) { - f := newPresetFixture(t) - base := f.characters.RawAll() - index := player.CharmCharacterIndexBase + 11 - install := func(expiry uint64) { - m := stateio.NewMemory() - inv, e := player.OpenInventory(m, &player.Starter{Version: f.seed.Version}) - if e != nil { - t.Fatal(e) - } - chars := append(append([]player.Character(nil), base...), player.Character{InvenIndex: index, ID: 9010, Level: 1, CostumeID: 90101, UseCostume: 0, ExpiryTime: expiry}) - s, e := player.OpenCharacterStore(m, chars, inv, "", "") - if e != nil { - t.Fatal(e) - } - f.deck.characters = s - } - install(uint64(time.Now().Add(time.Hour).UnixMilli())) - d := &gamedata.FieldSettingsDesign{TalentSlots: 1, CharacterTalentClass: map[uint64]uint64{9010: 10}, CharacterTemporaryPack: map[uint64]int{9010: 99}} - if err := f.deck.AttachFieldSettingsPack(func() (int, error) { return 4, nil }); err != nil { - t.Fatal(err) - } - if e := f.deck.AttachFieldSettings(d); e != nil { - t.Fatal(e) - } - if _, _, _, e := f.deck.Handle("/TalentSlotSave", req(1, wire.AppendVarint(nil, 2, 9010))); e != nil { - t.Fatal(e) - } - if _, _, _, e := f.deck.Handle("/FieldDeckSave", req(2, triple(1, index, 0))); e != nil { - t.Fatal(e) - } - if _, _, _, e := f.deck.Handle("/CharAutoReviveSet", req(3, wire.AppendVarint(nil, 2, 1), wire.AppendVarint(nil, 3, index))); e == nil { - t.Fatal("charm recovery caster accepted") - } - install(uint64(time.Now().Add(-time.Second).UnixMilli())) - ids, e := f.deck.TalentCharacterIDs() - if e != nil || ids[0] != 0 { - t.Fatal("expired charm quick slot leaked", ids, e) - } - if len(f.deck.CurrentFieldDeck()) != 0 { - t.Fatal("expired charm field party leaked") - } -} - -func TestTemporaryFieldAndTalentSlotsFollowPack(t *testing.T) { - f := newPresetFixture(t) - c := player.Character{InvenIndex: player.StoryCharacterIndexBase + 99, ID: 9001, Level: 1, CostumeID: 90011, UseCostume: 9999} - if e := f.characters.EnsureStoryCharacters([]player.Character{c}); e != nil { - t.Fatal(e) - } - pack := 4 - d := &gamedata.FieldSettingsDesign{TalentSlots: 1, CharacterTalentClass: map[uint64]uint64{9001: 9}, CharacterTemporaryPack: map[uint64]int{9001: 4}} - if err := f.deck.AttachFieldSettingsPack(func() (int, error) { return pack, nil }); err != nil { - t.Fatal(err) - } - if e := f.deck.AttachFieldSettings(d); e != nil { - t.Fatal(e) - } - if _, _, _, e := f.deck.Handle("/TalentSlotSave", req(1, wire.AppendVarint(nil, 2, c.ID))); e != nil { - t.Fatal(e) - } - if _, _, _, e := f.deck.Handle("/FieldDeckSave", req(2, triple(1, c.InvenIndex, c.UseCostume))); e != nil { - t.Fatal(e) - } - pack = 5 - ids, e := f.deck.TalentCharacterIDs() - if e != nil || ids[0] != 0 { - t.Fatal("temporary quick slot leaked", ids, e) - } - if len(f.deck.CurrentFieldDeck()) != 0 { - t.Fatal("temporary field party leaked") - } - if _, _, _, e := f.deck.Handle("/FieldDeckSave", req(3, triple(1, c.InvenIndex, c.UseCostume))); e == nil { - t.Fatal("foreign-pack temporary party accepted") - } - if _, _, _, e := f.deck.Handle("/CharAutoReviveSet", req(4)); e != nil { - t.Fatal("inactive quick slots blocked unrelated setting", e) - } -} - -func TestMalformedSettingsDoNotPersist(t *testing.T) { - f := newPresetFixture(t) - d := &gamedata.FieldSettingsDesign{TalentSlots: 2, CharacterTalentClass: map[uint64]uint64{350: 10, 360: 9}} - if e := f.deck.AttachFieldSettings(d); e != nil { - t.Fatal(e) - } - for _, bad := range [][]byte{req(0, wire.AppendVarint(nil, 2, 350)), req(1, wire.AppendBytes(nil, 2, []byte{0x80})), req(2, wire.AppendFixed64(nil, 2, 350)), req(^uint64(0), wire.AppendVarint(nil, 2, 350)), req(3, wire.AppendVarint(nil, 2, ^uint64(0)), wire.AppendVarint(nil, 2, 0))} { - if _, _, _, e := f.deck.Handle("/TalentSlotSave", bad); e == nil { - t.Fatal("malformed talent request accepted") - } - if _, ok, e := f.storage.LoadEntry("deck", "field_settings", "state"); e != nil || ok { - t.Fatal("invalid talent request wrote settings", e) - } - } - if _, _, _, e := f.deck.Handle("/CharAutoReviveSet", req(4, wire.AppendVarint(nil, 2, 2))); e == nil { - t.Fatal("invalid protobuf bool accepted") - } - if _, _, _, e := f.deck.Handle("/FieldDeckSave", req(5, triple(2, 100, 1001))); e == nil { - t.Fatal("field gap accepted") - } -} diff --git a/go/internal/server/deck/preset.go b/go/internal/server/deck/preset.go deleted file mode 100644 index 5cd592c..0000000 --- a/go/internal/server/deck/preset.go +++ /dev/null @@ -1,888 +0,0 @@ -package deck - -import ( - "encoding/binary" - "encoding/json" - "errors" - "fmt" - "maps" - "slices" - "sort" - "strconv" - "strings" - "unicode" - "unicode/utf8" - - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -type Preset struct { - Name string `json:"name"` - ResourceID uint64 `json:"resource_id"` - ResourceColor uint64 `json:"resource_color"` - Slot uint64 `json:"slot"` - Decks []PresetDeck `json:"decks"` - Blesses []PresetBless `json:"blesses"` -} - -type PresetDeck struct { - Deck DeckEntry `json:"deck"` - CostumeIndex uint64 `json:"costume_index"` - Equipment []PresetEquipmentItem `json:"equipment"` - Team uint64 `json:"team"` -} - -type PresetEquipmentItem struct { - Type uint64 `json:"type"` - Index uint64 `json:"index"` -} - -type PresetBless struct { - DeckType uint64 `json:"deck_type"` - IDs []uint64 `json:"ids"` -} - -type CostumeSetting struct { - CharacterIndex uint64 `json:"character_index"` - Sequence []CostumeSettingItem `json:"sequence"` - BattleMode uint64 `json:"battle_mode"` - MonsterID uint64 `json:"monster_id"` -} - -type CostumeSettingItem struct { - CostumeIndex int64 `json:"costume_index"` - BurstLevel uint64 `json:"burst_level"` -} - -func (s *Store) AttachPresetRuntime(wallet *player.Wallet, characters *player.CharacterStore, equipment *player.EquipmentInventory, collection *player.CollectionStore) error { - if wallet == nil || characters == nil || equipment == nil || collection == nil { - return errors.New("deck: incomplete preset runtime") - } - s.mu.Lock() - defer s.mu.Unlock() - s.wallet, s.characters, s.equipment, s.collection = wallet, characters, equipment, collection - return s.validatePresetOwnershipLocked() -} - -func (s *Store) BeginSession(id string) { - s.mu.Lock() - defer s.mu.Unlock() - if id == "" || id == s.sessionID { - return - } - s.sessionID = id - s.replies = map[string]deckReply{} -} - -func (s *Store) PresetSlotCount() uint64 { - s.mu.RLock() - defer s.mu.RUnlock() - return s.presetSlots -} - -func (s *Store) loadPresetEntries() error { - rawConfig, found, err := s.storage.LoadEntry("deck", "preset_config", "slots") - if err != nil { - return err - } - if found { - if err := json.Unmarshal(rawConfig, &s.presetSlots); err != nil || s.presetSlots < s.presetDesign.BaseCount || s.presetSlots > s.presetDesign.Maximum { - return errors.New("deck: invalid preset slot configuration") - } - } - raw, err := s.storage.ListEntries("deck", "presets") - if err != nil { - return err - } - for key, payload := range raw { - slot, err := strconv.ParseUint(key, 10, 64) - if err != nil || key != strconv.FormatUint(slot, 10) { - return fmt.Errorf("deck: invalid preset key %q", key) - } - var preset Preset - if err := json.Unmarshal(payload, &preset); err != nil || preset.Slot != slot { - return fmt.Errorf("deck: invalid preset %q", key) - } - if err := s.validatePresetShape(preset, s.presetSlots); err != nil { - return fmt.Errorf("deck: invalid preset %q: %w", key, err) - } - s.presets[slot] = preset - } - rawSettings, err := s.storage.ListEntries("deck", "costume_settings") - if err != nil { - return err - } - for key, payload := range rawSettings { - index, err := strconv.ParseUint(key, 10, 64) - if err != nil || index == 0 || key != strconv.FormatUint(index, 10) { - return fmt.Errorf("deck: invalid costume setting key %q", key) - } - var setting CostumeSetting - if err := json.Unmarshal(payload, &setting); err != nil || setting.CharacterIndex != index { - return fmt.Errorf("deck: invalid costume setting %q", key) - } - s.costumeSettings[index] = setting - } - return nil -} - -func (s *Store) validatePresetOwnershipLocked() error { - if s.characters == nil || s.collection == nil || s.equipment == nil { - return nil - } - for _, preset := range s.presets { - if err := s.validatePresetOwnedLocked(preset); err != nil { - return fmt.Errorf("deck: saved preset %d: %w", preset.Slot, err) - } - } - for _, setting := range s.costumeSettings { - if _, found := s.characters.Find(setting.CharacterIndex); !found { - return fmt.Errorf("deck: costume setting references unknown character %d", setting.CharacterIndex) - } - for _, item := range setting.Sequence { - if item.CostumeIndex > 0 { - if _, found := s.collection.CostumeByIndex(uint64(item.CostumeIndex)); !found { - return fmt.Errorf("deck: costume setting references unknown costume %d", item.CostumeIndex) - } - } - } - } - return nil -} - -func (s *Store) validatePresetShape(p Preset, slotCount uint64) error { - if p.Slot >= slotCount || (p.ResourceID != 0 && !s.presetDesign.Icons[p.ResourceID]) || p.ResourceColor > 5 || !validPresetName(p.Name) || len(p.Decks) > 5 { - return errors.New("invalid metadata or deck count") - } - characters, positions, sequences := map[uint64]bool{}, map[uint64]bool{}, map[uint64]bool{} - for _, deck := range p.Decks { - if deck.Deck.CharacterInvenIndex == 0 || deck.Deck.CostumeInvenIndex > 11 || deck.Deck.Slot == 0 || deck.Deck.Slot > 5 || deck.Team != 0 || - characters[deck.Deck.CharacterInvenIndex] || positions[deck.Deck.CostumeInvenIndex] || sequences[deck.Deck.Slot] { - return errors.New("invalid deck entry") - } - characters[deck.Deck.CharacterInvenIndex] = true - positions[deck.Deck.CostumeInvenIndex] = true - sequences[deck.Deck.Slot] = true - if len(deck.Equipment) != 5 { - return errors.New("preset deck requires five equipment slots") - } - seen := map[uint64]bool{} - for _, item := range deck.Equipment { - if item.Type >= 5 || seen[item.Type] { - return errors.New("invalid preset equipment slot") - } - seen[item.Type] = true - } - } - return nil -} - -func validPresetName(value string) bool { - if !utf8.ValidString(value) || utf8.RuneCountInString(value) < 1 || utf8.RuneCountInString(value) > 16 || strings.Contains(value, "<") || strings.Contains(value, ">") { - return false - } - for _, r := range value { - if unicode.IsControl(r) { - return false - } - } - return true -} - -func (s *Store) validatePresetOwnedLocked(p Preset) error { - if err := s.validatePresetShape(p, s.presetSlots); err != nil { - return err - } - ownedEquipment := make(map[uint64]player.Equipment) - for _, item := range s.equipment.All() { - ownedEquipment[item.InvenIndex] = item - } - seenEquipment := map[uint64]bool{} - for _, deck := range p.Decks { - if _, found := s.characters.Find(deck.Deck.CharacterInvenIndex); !found { - return fmt.Errorf("unknown character %d", deck.Deck.CharacterInvenIndex) - } - if deck.CostumeIndex != 0 { - costume, found := s.collection.CostumeByIndex(deck.CostumeIndex) - if !found || costume.UseChar != deck.Deck.CharacterInvenIndex { - return fmt.Errorf("unknown costume %d", deck.CostumeIndex) - } - } - for _, reference := range deck.Equipment { - if reference.Index == 0 { - continue - } - if seenEquipment[reference.Index] || ownedEquipment[reference.Index].InvenIndex == 0 { - return fmt.Errorf("invalid or repeated equipment %d", reference.Index) - } - seenEquipment[reference.Index] = true - } - binding := player.PresetEquipmentBinding{CharacterIndex: deck.Deck.CharacterInvenIndex, Equipment: make([]uint64, 5)} - for _, reference := range deck.Equipment { - binding.Equipment[reference.Type] = reference.Index - } - if err := s.equipment.ValidatePresetEquipment([]player.PresetEquipmentBinding{binding}); err != nil { - return err - } - } - return nil -} - -func decodePreset(data []byte) (Preset, error) { - var p Preset - name, _, err := wire.Bytes(data, 1) - if err != nil { - return p, err - } - p.Name = string(name) - p.ResourceID, _, err = wire.Varint(data, 2) - if err != nil { - return p, err - } - p.ResourceColor, _, err = wire.Varint(data, 3) - if err != nil { - return p, err - } - p.Slot, _, err = wire.Varint(data, 4) - if err != nil { - return p, err - } - err = wire.Walk(data, func(field wire.Field) error { - if field.Type != 2 { - return nil - } - switch field.Number { - case 5: - deck, err := decodePresetDeck(field.Value) - if err != nil { - return err - } - p.Decks = append(p.Decks, deck) - case 6: - bless, err := decodePresetBless(field.Value) - if err != nil { - return err - } - p.Blesses = append(p.Blesses, bless) - } - return nil - }) - return p, err -} - -func decodePresetDeck(data []byte) (PresetDeck, error) { - var result PresetDeck - base, found, err := wire.Bytes(data, 1) - if err != nil || !found { - return result, errors.New("deck: preset missing deck base") - } - character, ok, err := wire.Varint(base, 1) - if err != nil || !ok || character == 0 { - return result, errors.New("deck: invalid preset character") - } - position, _, err := wire.Varint(base, 2) - if err != nil { - return result, err - } - sequence, ok, err := wire.Varint(base, 3) - if err != nil || !ok || sequence == 0 { - return result, errors.New("deck: invalid preset sequence") - } - result.Deck = DeckEntry{CharacterInvenIndex: character, CostumeInvenIndex: position, Slot: sequence} - result.CostumeIndex, _, err = wire.Varint(data, 2) - if err != nil { - return result, err - } - result.Team, _, err = wire.Varint(data, 4) - if err != nil { - return result, err - } - err = wire.Walk(data, func(field wire.Field) error { - if field.Number != 3 { - return nil - } - if field.Type != 2 { - return errors.New("deck: invalid preset equipment") - } - typeID, _, err := wire.Varint(field.Value, 1) - if err != nil { - return err - } - index, _, err := wire.Varint(field.Value, 2) - if err != nil { - return err - } - result.Equipment = append(result.Equipment, PresetEquipmentItem{Type: typeID, Index: index}) - return nil - }) - sort.Slice(result.Equipment, func(i, j int) bool { return result.Equipment[i].Type < result.Equipment[j].Type }) - return result, err -} - -func decodePresetBless(data []byte) (PresetBless, error) { - var result PresetBless - result.DeckType, _, _ = wire.Varint(data, 1) - err := wire.Walk(data, func(field wire.Field) error { - if field.Number != 2 { - return nil - } - values, err := repeatedUint64(field) - if err != nil { - return err - } - result.IDs = append(result.IDs, values...) - return nil - }) - return result, err -} - -func repeatedUint64(field wire.Field) ([]uint64, error) { - if field.Type == 0 { - value, count := binary.Uvarint(field.Value) - if count <= 0 { - return nil, wire.ErrMalformed - } - return []uint64{value}, nil - } - if field.Type != 2 { - return nil, wire.ErrMalformed - } - var out []uint64 - for offset := 0; offset < len(field.Value); { - value, count := binary.Uvarint(field.Value[offset:]) - if count <= 0 { - return nil, wire.ErrMalformed - } - out = append(out, value) - offset += count - } - return out, nil -} - -func presetWire(p Preset) []byte { - var out []byte - if p.Name != "" { - out = wire.AppendString(out, 1, p.Name) - } - if p.ResourceID != 0 { - out = wire.AppendVarint(out, 2, p.ResourceID) - } - if p.ResourceColor != 0 { - out = wire.AppendVarint(out, 3, p.ResourceColor) - } - if p.Slot != 0 { - out = wire.AppendVarint(out, 4, p.Slot) - } - for _, deck := range p.Decks { - out = wire.AppendBytes(out, 5, presetDeckWire(deck)) - } - for _, bless := range p.Blesses { - var b []byte - if bless.DeckType != 0 { - b = wire.AppendVarint(b, 1, bless.DeckType) - } - for _, id := range bless.IDs { - b = wire.AppendVarint(b, 2, id) - } - out = wire.AppendBytes(out, 6, b) - } - return out -} - -func presetDeckWire(deck PresetDeck) []byte { - base := wire.AppendVarint(nil, 1, deck.Deck.CharacterInvenIndex) - if deck.Deck.CostumeInvenIndex != 0 { - base = wire.AppendVarint(base, 2, deck.Deck.CostumeInvenIndex) - } - base = wire.AppendVarint(base, 3, deck.Deck.Slot) - out := wire.AppendBytes(nil, 1, base) - if deck.CostumeIndex != 0 { - out = wire.AppendVarint(out, 2, deck.CostumeIndex) - } - for _, item := range deck.Equipment { - var b []byte - if item.Type != 0 { - b = wire.AppendVarint(b, 1, item.Type) - } - if item.Index != 0 { - b = wire.AppendVarint(b, 2, item.Index) - } - out = wire.AppendBytes(out, 3, b) - } - if deck.Team != 0 { - out = wire.AppendVarint(out, 4, deck.Team) - } - return out -} - -func (s *Store) presetCacheKey(kind string, seq uint64) string { - return kind + ":" + s.sessionID + ":" + strconv.FormatUint(seq, 10) -} - -func (s *Store) persistPresetLocked(p Preset) error { - payload, err := json.Marshal(p) - if err != nil { - return err - } - core, err := s.corePayloadLocked() - if err != nil { - return err - } - return s.storage.SaveWithEntries("deck", core, []stateio.EntryMutation{{Bucket: "presets", Key: strconv.FormatUint(p.Slot, 10), Payload: payload}}) -} - -func (s *Store) corePayloadLocked() ([]byte, error) { - payload, err := json.MarshalIndent(s.state, "", " ") - if err != nil { - return nil, err - } - return append(payload, '\n'), nil -} - -func (s *Store) cachedReplyLocked(kind string, seq uint64) (deckReply, bool) { - reply, found := s.replies[s.presetCacheKey(kind, seq)] - if found { - reply.body = append([]byte(nil), reply.body...) - } - return reply, found -} - -func (s *Store) rememberReplyLocked(kind string, seq uint64, code int, body []byte) { - s.replies[s.presetCacheKey(kind, seq)] = deckReply{code: code, body: append([]byte(nil), body...)} -} - -func requestSequence(request []byte) (uint64, error) { - seq, found, err := wire.Varint(request, 1) - if err != nil || !found || seq == 0 { - return 0, errors.New("deck: invalid request sequence") - } - return seq, nil -} - -func (s *Store) handlePresetInfo(request []byte) (int, []byte, bool, error) { - if _, err := requestSequence(request); err != nil { - return 0, nil, true, err - } - s.mu.RLock() - defer s.mu.RUnlock() - slots := make([]uint64, 0, len(s.presets)) - for slot := range s.presets { - if slot < s.presetSlots { - slots = append(slots, slot) - } - } - slices.Sort(slots) - var response []byte - for _, slot := range slots { - response = wire.AppendBytes(response, 1, presetWire(s.presets[slot])) - } - return 178, response, true, nil -} - -func (s *Store) handlePresetSave(request []byte) (int, []byte, bool, error) { - seq, err := requestSequence(request) - if err != nil { - return 0, nil, true, err - } - raw, found, err := wire.Bytes(request, 2) - if err != nil || !found { - return 0, nil, true, errors.New("deck: PresetSave missing preset") - } - preset, err := decodePreset(raw) - if err != nil { - return 0, nil, true, fmt.Errorf("deck: decode preset: %w", err) - } - s.mu.Lock() - defer s.mu.Unlock() - if reply, found := s.cachedReplyLocked("save", seq); found { - return reply.code, reply.body, true, nil - } - if s.characters == nil || s.collection == nil || s.equipment == nil { - return 0, nil, true, errors.New("deck: preset runtime unavailable") - } - if err := s.validatePresetOwnedLocked(preset); err != nil { - return 0, nil, true, fmt.Errorf("deck: invalid preset: %w", err) - } - if err := s.persistPresetLocked(preset); err != nil { - return 0, nil, true, fmt.Errorf("deck: persist preset: %w", err) - } - s.presets[preset.Slot] = preset - s.rememberReplyLocked("save", seq, 179, nil) - return 179, nil, true, nil -} - -func (s *Store) handlePresetAddSlot(request []byte) (int, []byte, bool, error) { - seq, err := requestSequence(request) - if err != nil { - return 0, nil, true, err - } - count, found, err := wire.Varint(request, 2) - if err != nil || !found || count == 0 { - return 0, nil, true, errors.New("deck: PresetAddSlot invalid count") - } - s.mu.Lock() - defer s.mu.Unlock() - if reply, found := s.cachedReplyLocked("add-slot", seq); found { - return reply.code, reply.body, true, nil - } - if s.wallet == nil || s.presetDesign.Validate() != nil { - return 0, nil, true, errors.New("deck: preset wallet/design unavailable") - } - if s.presetSlots > s.presetDesign.Maximum || count > s.presetDesign.Maximum-s.presetSlots { - return 0, nil, true, errors.New("deck: preset slot limit exceeded") - } - if count > ^uint64(0)/s.presetDesign.Price { - return 0, nil, true, errors.New("deck: preset slot price overflow") - } - identity := "preset-slot:" + s.sessionID + ":" + strconv.FormatUint(seq, 10) - var spendErr error - switch s.presetDesign.PriceType { - case 4: - _, spendErr = s.wallet.SpendGoldOnce(identity, count*s.presetDesign.Price) - case 3: - _, spendErr = s.wallet.SpendFreeJewelryOnce(identity, count*s.presetDesign.Price) - case 2: - _, spendErr = s.wallet.SpendJewelryOnce(identity, count*s.presetDesign.Price) - case 12: - _, spendErr = s.wallet.SpendCatalystOnce(identity, count*s.presetDesign.Price) - } - if err := spendErr; err != nil { - return 0, nil, true, fmt.Errorf("deck: buy preset slot: %w", err) - } - next := s.presetSlots + count - payload, err := json.Marshal(next) - if err != nil { - return 0, nil, true, err - } - core, err := s.corePayloadLocked() - if err != nil { - return 0, nil, true, err - } - if err := s.storage.SaveWithEntries("deck", core, []stateio.EntryMutation{{Bucket: "preset_config", Key: "slots", Payload: payload}}); err != nil { - return 0, nil, true, fmt.Errorf("deck: persist preset slots: %w", err) - } - s.presetSlots = next - s.rememberReplyLocked("add-slot", seq, 180, nil) - return 180, nil, true, nil -} - -func (s *Store) handlePresetInfoChange(request []byte) (int, []byte, bool, error) { - seq, err := requestSequence(request) - if err != nil { - return 0, nil, true, err - } - nameBytes, _, err := wire.Bytes(request, 2) - if err != nil { - return 0, nil, true, errors.New("deck: invalid preset name") - } - resourceID, _, err := wire.Varint(request, 3) - if err != nil { - return 0, nil, true, errors.New("deck: invalid preset icon") - } - color, _, err := wire.Varint(request, 4) - if err != nil { - return 0, nil, true, errors.New("deck: invalid preset color") - } - // Ordinary preset slots are zero-based. Proto3 omits slot=0, so an absent - // field 5 is the first slot rather than a malformed request. - slot, _, err := wire.Varint(request, 5) - if err != nil { - return 0, nil, true, errors.New("deck: invalid preset slot") - } - s.mu.Lock() - defer s.mu.Unlock() - if reply, found := s.cachedReplyLocked("info-change", seq); found { - return reply.code, reply.body, true, nil - } - preset, exists := s.presets[slot] - if !exists { - preset = Preset{Slot: slot, Decks: []PresetDeck{}, Blesses: []PresetBless{}} - } - preset.Name, preset.ResourceID, preset.ResourceColor = string(nameBytes), resourceID, color - if err := s.validatePresetShape(preset, s.presetSlots); err != nil { - return 0, nil, true, fmt.Errorf("deck: invalid preset metadata: %w", err) - } - if err := s.persistPresetLocked(preset); err != nil { - return 0, nil, true, err - } - s.presets[slot] = preset - s.rememberReplyLocked("info-change", seq, 278, nil) - return 278, nil, true, nil -} - -func (s *Store) handlePresetDelete(request []byte) (int, []byte, bool, error) { - seq, err := requestSequence(request) - if err != nil { - return 0, nil, true, err - } - var slots []uint64 - if err := wire.Walk(request, func(field wire.Field) error { - if field.Number != 2 { - return nil - } - values, err := repeatedUint64(field) - if err != nil { - return err - } - slots = append(slots, values...) - return nil - }); err != nil || len(slots) == 0 { - return 0, nil, true, errors.New("deck: invalid preset delete slots") - } - s.mu.Lock() - defer s.mu.Unlock() - if reply, found := s.cachedReplyLocked("delete", seq); found { - return reply.code, reply.body, true, nil - } - seen := make(map[uint64]bool, len(slots)) - changes := make([]stateio.EntryMutation, 0, len(slots)) - for _, slot := range slots { - if slot >= s.presetSlots || seen[slot] { - return 0, nil, true, errors.New("deck: invalid or duplicate preset delete slot") - } - seen[slot] = true - changes = append(changes, stateio.EntryMutation{Bucket: "presets", Key: strconv.FormatUint(slot, 10), Delete: true}) - } - core, err := s.corePayloadLocked() - if err != nil { - return 0, nil, true, err - } - if err := s.storage.SaveWithEntries("deck", core, changes); err != nil { - return 0, nil, true, err - } - for slot := range seen { - delete(s.presets, slot) - } - // 2.35.10 contains the request/response classes but no PacketCode enum - // member. Code zero is the same compatibility fallback used for other - // unnumbered local endpoints; it must not be treated as an official value. - s.rememberReplyLocked("delete", seq, 0, nil) - return 0, nil, true, nil -} - -func (s *Store) handlePresetUse(request []byte) (int, []byte, bool, error) { - seq, err := requestSequence(request) - if err != nil { - return 0, nil, true, err - } - // PresetUse is also zero-based and slot=0 is omitted by proto3. - slot, _, err := wire.Varint(request, 2) - if err != nil { - return 0, nil, true, errors.New("deck: PresetUse missing slot") - } - s.mu.Lock() - defer s.mu.Unlock() - if reply, found := s.cachedReplyLocked("use", seq); found { - return reply.code, reply.body, true, nil - } - preset, found := s.presets[slot] - if !found || len(preset.Decks) == 0 { - return 0, nil, true, errors.New("deck: PresetUse references an empty slot") - } - if s.characters == nil || s.collection == nil || s.equipment == nil { - return 0, nil, true, errors.New("deck: preset runtime unavailable") - } - if err := s.validatePresetOwnedLocked(preset); err != nil { - return 0, nil, true, fmt.Errorf("deck: stale preset: %w", err) - } - deckEntries := make([]DeckEntry, 0, len(preset.Decks)) - assignments := make(map[uint64]uint64, len(preset.Decks)) - bindings := make([]player.PresetEquipmentBinding, 0, len(preset.Decks)) - for _, entry := range preset.Decks { - deckEntries = append(deckEntries, entry.Deck) - assignments[entry.Deck.CharacterInvenIndex] = entry.CostumeIndex - equipment := make([]uint64, 5) - for _, item := range entry.Equipment { - equipment[item.Type] = item.Index - } - bindings = append(bindings, player.PresetEquipmentBinding{CharacterIndex: entry.Deck.CharacterInvenIndex, Equipment: equipment}) - } - if _, err := s.characters.ApplyPresetCostumes(assignments); err != nil { - return 0, nil, true, fmt.Errorf("deck: apply preset costumes: %w", err) - } - affected, err := s.equipment.ApplyPresetEquipment(bindings) - if err != nil { - return 0, nil, true, fmt.Errorf("deck: apply preset equipment: %w", err) - } - next := clone(s.state) - next.Deck = append([]DeckEntry(nil), deckEntries...) - maps.Copy(next.Costumes, assignments) - if err := s.commit(next); err != nil { - return 0, nil, true, fmt.Errorf("deck: persist applied preset: %w", err) - } - var response []byte - for _, entry := range deckEntries { - base := wire.AppendVarint(nil, 1, entry.CharacterInvenIndex) - if entry.CostumeInvenIndex != 0 { - base = wire.AppendVarint(base, 2, entry.CostumeInvenIndex) - } - base = wire.AppendVarint(base, 3, entry.Slot) - response = wire.AppendBytes(response, 1, base) - } - characterSet := make(map[uint64]bool, len(assignments)+len(affected)) - for index := range assignments { - characterSet[index] = true - } - for _, character := range affected { - characterSet[character.InvenIndex] = true - } - indices := make([]uint64, 0, len(characterSet)) - for index := range characterSet { - indices = append(indices, index) - } - slices.Sort(indices) - for _, index := range indices { - if character, found := s.characters.Find(index); found { - response = wire.AppendBytes(response, 2, player.CharacterWire(character)) - } - } - for _, binding := range bindings { - info := wire.AppendVarint(nil, 1, binding.CharacterIndex) - for _, index := range binding.Equipment { - info = wire.AppendVarint(info, 2, index) - } - response = wire.AppendBytes(response, 3, info) - } - s.rememberReplyLocked("use", seq, 409, response) - return 409, response, true, nil -} - -func costumeSettingWire(setting CostumeSetting) []byte { - out := wire.AppendVarint(nil, 1, setting.CharacterIndex) - for _, item := range setting.Sequence { - entry := wire.AppendVarint(nil, 1, uint64(item.CostumeIndex)) - if item.BurstLevel != 0 { - entry = wire.AppendVarint(entry, 2, item.BurstLevel) - } - out = wire.AppendBytes(out, 2, entry) - } - if setting.BattleMode != 0 { - out = wire.AppendVarint(out, 3, setting.BattleMode) - } - if setting.MonsterID != 0 { - out = wire.AppendVarint(out, 4, setting.MonsterID) - } - return out -} - -func decodeCostumeSetting(data []byte) (CostumeSetting, error) { - var result CostumeSetting - character, found, err := wire.Varint(data, 1) - if err != nil || !found || character == 0 { - return result, errors.New("deck: costume setting missing character") - } - result.CharacterIndex = character - result.BattleMode, _, err = wire.Varint(data, 3) - if err != nil { - return result, err - } - result.MonsterID, _, err = wire.Varint(data, 4) - if err != nil { - return result, err - } - err = wire.Walk(data, func(field wire.Field) error { - if field.Number != 2 { - return nil - } - if field.Type != 2 { - return errors.New("deck: invalid costume setting sequence") - } - raw, found, err := wire.Varint(field.Value, 1) - if err != nil || !found { - return errors.New("deck: costume setting entry missing costume") - } - burst, _, err := wire.Varint(field.Value, 2) - if err != nil { - return err - } - result.Sequence = append(result.Sequence, CostumeSettingItem{CostumeIndex: int64(raw), BurstLevel: burst}) - return nil - }) - return result, err -} - -func (s *Store) validateCostumeSettingLocked(setting CostumeSetting) error { - if s.characters == nil || s.collection == nil { - return errors.New("deck: costume setting runtime unavailable") - } - if setting.BattleMode != 0 || setting.MonsterID != 0 { - return errors.New("deck: ordinary costume setting requires normal battle mode") - } - if _, found := s.characters.Find(setting.CharacterIndex); !found { - return fmt.Errorf("deck: unknown costume setting character %d", setting.CharacterIndex) - } - for _, item := range setting.Sequence { - if item.CostumeIndex <= 0 { - if item.CostumeIndex != 0 && item.CostumeIndex != -1 { - return fmt.Errorf("deck: invalid costume setting sentinel %d", item.CostumeIndex) - } - continue - } - costume, found := s.collection.CostumeByIndex(uint64(item.CostumeIndex)) - if !found || costume.UseChar != setting.CharacterIndex { - return fmt.Errorf("deck: costume %d does not belong to character %d", item.CostumeIndex, setting.CharacterIndex) - } - if item.BurstLevel > costume.BurstLevel { - return fmt.Errorf("deck: costume %d burst level exceeds owned level", item.CostumeIndex) - } - } - return nil -} - -func (s *Store) handleCostumeSettingInfo(request []byte) (int, []byte, bool, error) { - if _, err := requestSequence(request); err != nil { - return 0, nil, true, err - } - s.mu.RLock() - defer s.mu.RUnlock() - indices := make([]uint64, 0, len(s.costumeSettings)) - for index := range s.costumeSettings { - indices = append(indices, index) - } - slices.Sort(indices) - var response []byte - for _, index := range indices { - response = wire.AppendBytes(response, 1, costumeSettingWire(s.costumeSettings[index])) - } - return 397, response, true, nil -} - -func (s *Store) handleCostumeSettingSave(request []byte) (int, []byte, bool, error) { - seq, err := requestSequence(request) - if err != nil { - return 0, nil, true, err - } - raw, found, err := wire.Bytes(request, 2) - if err != nil || !found { - return 0, nil, true, errors.New("deck: DeckCostumeSettingSave missing setting") - } - setting, err := decodeCostumeSetting(raw) - if err != nil { - return 0, nil, true, err - } - s.mu.Lock() - defer s.mu.Unlock() - if reply, found := s.cachedReplyLocked("costume-setting-save", seq); found { - return reply.code, reply.body, true, nil - } - if err := s.validateCostumeSettingLocked(setting); err != nil { - return 0, nil, true, err - } - payload, err := json.Marshal(setting) - if err != nil { - return 0, nil, true, err - } - change := stateio.EntryMutation{Bucket: "costume_settings", Key: strconv.FormatUint(setting.CharacterIndex, 10), Payload: payload} - core, err := s.corePayloadLocked() - if err != nil { - return 0, nil, true, err - } - if err := s.storage.SaveWithEntries("deck", core, []stateio.EntryMutation{change}); err != nil { - return 0, nil, true, err - } - s.costumeSettings[setting.CharacterIndex] = setting - s.rememberReplyLocked("costume-setting-save", seq, 398, nil) - return 398, nil, true, nil -} diff --git a/go/internal/server/deck/preset_test.go b/go/internal/server/deck/preset_test.go deleted file mode 100644 index 280182b..0000000 --- a/go/internal/server/deck/preset_test.go +++ /dev/null @@ -1,380 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/gamedata" - "reflect" - "sort" - "testing" - - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func TestPresetChangedDesignControlsBaseLimitAndCurrency(t *testing.T) { - f := newPresetFixture(t) - design := gamedata.PresetDesign{BaseCount: 2, Maximum: 3, PriceType: 3, Price: 17} - d, err := OpenStore(f.storage, f.seed, design) - if err != nil { - t.Fatal(err) - } - w, err := player.OpenWallet(stateio.NewMemory(), player.Currency{FreeJewelry: 100}) - if err != nil { - t.Fatal(err) - } - d.wallet = w - d.BeginSession("changed-design") - if d.PresetSlotCount() != 2 { - t.Fatal("base count ignored") - } - request := wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 1) - if _, _, _, err := d.Handle("/PresetAddSlot", request); err != nil { - t.Fatal(err) - } - if d.PresetSlotCount() != 3 || w.Snapshot().FreeJewelry != 83 || w.Snapshot().Gold != 0 { - t.Fatal("changed price/currency ignored") - } - request = wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 1) - if _, _, _, err := d.Handle("/PresetAddSlot", request); err == nil { - t.Fatal("changed limit ignored") - } - if w.Snapshot().FreeJewelry != 83 { - t.Fatal("invalid purchase charged") - } -} - -type presetFixture struct { - storage *stateio.Memory - seed Seed - deck *Store - wallet *player.Wallet - characters *player.CharacterStore - equipment *player.EquipmentInventory - collection *player.CollectionStore - weapon player.Equipment -} - -func newPresetFixture(t *testing.T) *presetFixture { - t.Helper() - storage := stateio.NewMemory() - seed, err := LoadSeed("../../../seed/v2_35_10/decks.json") - if err != nil { - t.Fatal(err) - } - inventory, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(storage, []player.Costume{ - {InvenIndex: 1001, ID: 3501, UseChar: 100, BurstLevel: 3}, - {InvenIndex: 2001, ID: 3601, UseChar: 200, BurstLevel: 1}, - }) - if err != nil { - t.Fatal(err) - } - characters, err := player.OpenCharacterStore(storage, []player.Character{ - {InvenIndex: 100, ID: 350, Level: 20, UseCostume: 1001, CostumeID: 3501}, - {InvenIndex: 200, ID: 360, Level: 20, UseCostume: 2001, CostumeID: 3601}, - }, inventory, "", "") - if err != nil { - t.Fatal(err) - } - if err := characters.AttachCollection(collection); err != nil { - t.Fatal(err) - } - equipment, err := player.OpenEquipmentInventory(storage) - if err != nil { - t.Fatal(err) - } - if err := equipment.AttachSlots(map[uint64]uint64{10: 0, 11: 1, 12: 2, 13: 3, 14: 4}); err != nil { - t.Fatal(err) - } - if err := equipment.AttachCharacters(characters); err != nil { - t.Fatal(err) - } - weapon, err := equipment.GrantOnce("preset-test-weapon", 10) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{Gold: 20000}) - if err != nil { - t.Fatal(err) - } - decks, err := OpenStore(storage, seed, testPresetDesign) - if err != nil { - t.Fatal(err) - } - if err := decks.AttachPresetRuntime(wallet, characters, equipment, collection); err != nil { - t.Fatal(err) - } - decks.BeginSession("preset-test") - return &presetFixture{storage: storage, seed: seed, deck: decks, wallet: wallet, characters: characters, equipment: equipment, collection: collection, weapon: weapon} -} - -func presetRequest(slot, character, costume, equipment uint64, name string) []byte { - base := wire.AppendVarint(nil, 1, character) - base = wire.AppendVarint(base, 2, 0) - base = wire.AppendVarint(base, 3, 1) - deck := wire.AppendBytes(nil, 1, base) - deck = wire.AppendVarint(deck, 2, costume) - for equipmentType := range uint64(5) { - entry := wire.AppendVarint(nil, 1, equipmentType) - if equipmentType == 0 { - entry = wire.AppendVarint(entry, 2, equipment) - } - deck = wire.AppendBytes(deck, 3, entry) - } - preset := wire.AppendString(nil, 1, name) - preset = wire.AppendVarint(preset, 2, 1) - preset = wire.AppendVarint(preset, 4, slot) - preset = wire.AppendBytes(preset, 5, deck) - return wire.AppendBytes(nil, 2, preset) -} - -func TestPresetSaveInfoMetadataDeleteAndRestart(t *testing.T) { - f := newPresetFixture(t) - for seq, slot := range []uint64{2, 0} { - request := req(uint64(seq+1), presetRequest(slot, 100, 1001, f.weapon.InvenIndex, "编队")) - if code, _, handled, err := f.deck.Handle("/PresetSave", request); err != nil || !handled || code != 179 { - t.Fatalf("save slot %d code=%d handled=%v err=%v", slot, code, handled, err) - } - } - code, response, handled, err := f.deck.Handle("/PresetInfo", req(10)) - if err != nil || !handled || code != 178 { - t.Fatalf("info code=%d handled=%v err=%v", code, handled, err) - } - var slots []uint64 - if err := wire.Walk(response, func(field wire.Field) error { - if field.Number == 1 { - slot, _, err := wire.Varint(field.Value, 4) - if err != nil { - return err - } - slots = append(slots, slot) - } - return nil - }); err != nil || !reflect.DeepEqual(slots, []uint64{0, 2}) { - t.Fatalf("ordered slots=%v err=%v", slots, err) - } - - change := wire.AppendString(nil, 2, "主力队") - change = wire.AppendVarint(change, 3, 21) - change = wire.AppendVarint(change, 4, 5) - // slot=0 is absent on the real proto3 wire. - if code, _, _, err := f.deck.Handle("/PresetInfoChange", req(11, change)); err != nil || code != 278 { - t.Fatalf("metadata change code=%d err=%v", code, err) - } - if got := f.deck.presets[0]; got.Name != "主力队" || got.ResourceID != 21 || got.ResourceColor != 5 || len(got.Decks) != 1 { - t.Fatalf("metadata change lost content: %+v", got) - } - - restarted, err := OpenStore(f.storage, f.seed, testPresetDesign) - if err != nil { - t.Fatal(err) - } - if len(restarted.presets) != 2 || restarted.presets[0].Name != "主力队" { - t.Fatalf("restarted presets=%+v", restarted.presets) - } - restarted.BeginSession("restart") - deleteRequest := req(12, wire.AppendVarint(nil, 2, 0), wire.AppendVarint(nil, 2, 2)) - if code, _, handled, err := restarted.Handle("/PresetDelete", deleteRequest); err != nil || !handled || code != 0 { - t.Fatalf("delete code=%d handled=%v err=%v", code, handled, err) - } - if len(restarted.presets) != 0 || restarted.presetSlots != 5 { - t.Fatalf("delete changed slots/content: slots=%d presets=%+v", restarted.presetSlots, restarted.presets) - } -} - -func TestPresetMetadataCanCreateEmptySlot(t *testing.T) { - f := newPresetFixture(t) - change := wire.AppendString(nil, 2, "备用") - change = wire.AppendVarint(change, 3, 2) - change = wire.AppendVarint(change, 4, 1) - change = wire.AppendVarint(change, 5, 4) - if code, _, _, err := f.deck.Handle("/PresetInfoChange", req(1, change)); err != nil || code != 278 { - t.Fatalf("upsert code=%d err=%v", code, err) - } - if got := f.deck.presets[4]; got.Name != "备用" || len(got.Decks) != 0 { - t.Fatalf("empty metadata preset=%+v", got) - } -} - -func TestPresetAddSlotChargesOncePersistsAndCaps(t *testing.T) { - f := newPresetFixture(t) - add := req(20, wire.AppendVarint(nil, 2, 1)) - for attempt := range 2 { - if code, _, _, err := f.deck.Handle("/PresetAddSlot", add); err != nil || code != 180 { - t.Fatalf("add attempt %d code=%d err=%v", attempt, code, err) - } - } - if f.deck.PresetSlotCount() != 6 || f.wallet.Snapshot().Gold != 18000 { - t.Fatalf("slots=%d gold=%d", f.deck.PresetSlotCount(), f.wallet.Snapshot().Gold) - } - restarted, err := OpenStore(f.storage, f.seed, testPresetDesign) - if err != nil { - t.Fatal(err) - } - if restarted.PresetSlotCount() != 6 { - t.Fatalf("restarted slots=%d", restarted.PresetSlotCount()) - } - if err := restarted.AttachPresetRuntime(f.wallet, f.characters, f.equipment, f.collection); err != nil { - t.Fatal(err) - } - restarted.BeginSession("cap") - if _, _, _, err := restarted.Handle("/PresetAddSlot", req(21, wire.AppendVarint(nil, 2, 6))); err != nil { - t.Fatalf("buy remaining slots: %v", err) - } - if _, _, _, err := restarted.Handle("/PresetAddSlot", req(22, wire.AppendVarint(nil, 2, 1))); err == nil { - t.Fatal("accepted slot beyond maximum") - } -} - -func TestPresetUseAppliesDeckCostumeEquipmentWithoutChangingFieldDeck(t *testing.T) { - f := newPresetFixture(t) - wantField := append([]FieldEntry(nil), f.deck.state.FieldDeck...) - if _, _, _, err := f.deck.Handle("/PresetSave", req(1, presetRequest(0, 100, 1001, f.weapon.InvenIndex, "应用"))); err != nil { - t.Fatal(err) - } - code, response, handled, err := f.deck.Handle("/PresetUse", req(2, wire.AppendVarint(nil, 3, 999999999))) - if err != nil || !handled || code != 409 { - t.Fatalf("use code=%d handled=%v err=%v", code, handled, err) - } - if len(f.deck.state.Deck) != 1 || f.deck.state.Deck[0].CharacterInvenIndex != 100 || !reflect.DeepEqual(f.deck.state.FieldDeck, wantField) { - t.Fatalf("deck=%+v field changed=%v", f.deck.state.Deck, !reflect.DeepEqual(f.deck.state.FieldDeck, wantField)) - } - owned := f.equipment.All() - if len(owned) != 1 || owned[0].UseChar != 100 { - t.Fatalf("equipment=%+v", owned) - } - var deckCount, characterCount int - var equipmentValues []uint64 - if err := wire.Walk(response, func(field wire.Field) error { - switch field.Number { - case 1: - deckCount++ - case 2: - characterCount++ - case 3: - return wire.Walk(field.Value, func(nested wire.Field) error { - if nested.Number == 2 { - values, err := repeatedUint64(nested) - equipmentValues = append(equipmentValues, values...) - return err - } - return nil - }) - } - return nil - }); err != nil { - t.Fatal(err) - } - if deckCount != 1 || characterCount == 0 || !reflect.DeepEqual(equipmentValues, []uint64{f.weapon.InvenIndex, 0, 0, 0, 0}) { - t.Fatalf("response deck=%d chars=%d equipment=%v", deckCount, characterCount, equipmentValues) - } - // The client-supplied power is deliberately ignored; replay returns the - // exact same response and does not repeat cross-domain state transitions. - if replayCode, replay, _, replayErr := f.deck.Handle("/PresetUse", req(2)); replayErr != nil || replayCode != 409 || !reflect.DeepEqual(replay, response) { - t.Fatalf("replay code=%d equal=%v err=%v", replayCode, reflect.DeepEqual(replay, response), replayErr) - } -} - -func TestPresetSaveRejectsForgedOwnershipAndEquipmentSlot(t *testing.T) { - f := newPresetFixture(t) - tests := []struct { - name string - character uint64 - costume uint64 - equipment uint64 - }{ - {name: "character", character: 999, costume: 1001, equipment: f.weapon.InvenIndex}, - {name: "costume owner", character: 100, costume: 2001, equipment: f.weapon.InvenIndex}, - {name: "equipment", character: 100, costume: 1001, equipment: 999999}, - } - for seq, test := range tests { - t.Run(test.name, func(t *testing.T) { - if _, _, _, err := f.deck.Handle("/PresetSave", req(uint64(seq+1), presetRequest(0, test.character, test.costume, test.equipment, "伪造"))); err == nil { - t.Fatal("forged preset accepted") - } - }) - } - if len(f.deck.presets) != 0 { - t.Fatalf("rejected save mutated presets=%+v", f.deck.presets) - } -} - -func TestDeckCostumeSettingSentinelsClearAndRestart(t *testing.T) { - f := newPresetFixture(t) - setting := wire.AppendVarint(nil, 1, 100) - for _, item := range []struct { - index int64 - burst uint64 - }{{1001, 3}, {-1, 0}, {0, 0}} { - entry := wire.AppendVarint(nil, 1, uint64(item.index)) - entry = wire.AppendVarint(entry, 2, item.burst) - setting = wire.AppendBytes(setting, 2, entry) - } - if code, _, _, err := f.deck.Handle("/DeckCostumeSettingSave", req(1, wire.AppendBytes(nil, 2, setting))); err != nil || code != 398 { - t.Fatalf("save setting code=%d err=%v", code, err) - } - restarted, err := OpenStore(f.storage, f.seed, testPresetDesign) - if err != nil { - t.Fatal(err) - } - if err := restarted.AttachPresetRuntime(f.wallet, f.characters, f.equipment, f.collection); err != nil { - t.Fatal(err) - } - restarted.BeginSession("costume-restart") - code, response, _, err := restarted.Handle("/DeckCostumeSettingInfo", req(2)) - if err != nil || code != 397 { - t.Fatalf("info setting code=%d err=%v", code, err) - } - var got []int64 - if err := wire.Walk(response, func(field wire.Field) error { - if field.Number != 1 { - return nil - } - return wire.Walk(field.Value, func(nested wire.Field) error { - if nested.Number == 2 { - value, _, err := wire.Varint(nested.Value, 1) - got = append(got, int64(value)) - return err - } - return nil - }) - }); err != nil || !reflect.DeepEqual(got, []int64{1001, -1, 0}) { - t.Fatalf("setting sequence=%v err=%v", got, err) - } - - clear := wire.AppendVarint(nil, 1, 100) - if _, _, _, err := restarted.Handle("/DeckCostumeSettingSave", req(3, wire.AppendBytes(nil, 2, clear))); err != nil { - t.Fatal(err) - } - if len(restarted.costumeSettings[100].Sequence) != 0 { - t.Fatalf("setting was not cleared: %+v", restarted.costumeSettings[100]) - } -} - -func TestCostumeSettingInfoIsSorted(t *testing.T) { - f := newPresetFixture(t) - for seq, character := range []uint64{200, 100} { - setting := wire.AppendVarint(nil, 1, character) - if _, _, _, err := f.deck.Handle("/DeckCostumeSettingSave", req(uint64(seq+1), wire.AppendBytes(nil, 2, setting))); err != nil { - t.Fatal(err) - } - } - _, response, _, err := f.deck.Handle("/DeckCostumeSettingInfo", req(8)) - if err != nil { - t.Fatal(err) - } - var indices []uint64 - _ = wire.Walk(response, func(field wire.Field) error { - if field.Number == 1 { - index, _, _ := wire.Varint(field.Value, 1) - indices = append(indices, index) - } - return nil - }) - if !sort.SliceIsSorted(indices, func(i, j int) bool { return indices[i] < indices[j] }) || !reflect.DeepEqual(indices, []uint64{100, 200}) { - t.Fatalf("setting order=%v", indices) - } -} diff --git a/go/internal/server/deck/story_party.go b/go/internal/server/deck/story_party.go deleted file mode 100644 index b8b34c9..0000000 --- a/go/internal/server/deck/story_party.go +++ /dev/null @@ -1,24 +0,0 @@ -package deck - -import "bd2server/internal/server/player" - -func (s *Store) CurrentFieldDeck() []FieldEntry { - s.mu.RLock() - defer s.mu.RUnlock() - return s.visibleFieldDeckLocked() -} - -func (s *Store) visibleFieldDeckLocked() []FieldEntry { - var out []FieldEntry - for _, entry := range s.state.FieldDeck { - if s.characters != nil { - c, ok := s.characters.Find(entry.CharacterInvenIndex) - if !ok || player.IsStoryCharacter(c) && !s.temporaryAllowed(c) { - continue - } - } - entry.Slot = uint64(len(out) + 1) - out = append(out, entry) - } - return out -} diff --git a/go/internal/server/deck/waypoint.go b/go/internal/server/deck/waypoint.go deleted file mode 100644 index 6dbd28f..0000000 --- a/go/internal/server/deck/waypoint.go +++ /dev/null @@ -1,139 +0,0 @@ -package deck - -import ( - "encoding/binary" - "errors" - "fmt" - "math" - "slices" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -func (s *Store) ConfigureWaypoints(load func(uint64) (gamedata.WaypointPack, error), validate func(uint64, bool) error) error { - if load == nil || validate == nil { - return errors.New("deck: incomplete waypoint runtime") - } - s.mu.Lock() - defer s.mu.Unlock() - s.waypointDesign = load - s.waypointPack = validate - return nil -} - -func (s *Store) ActivatedWaypoint(pack, id uint64) bool { - s.mu.RLock() - defer s.mu.RUnlock() - return hasWaypoint(s.state.Waypoints[pack], id) -} -func hasWaypoint(points []uint64, id uint64) bool { - return slices.Contains(points, id) -} - -func validWaypointState(packs map[uint64][]uint64) error { - for pack, points := range packs { - if pack == 0 || pack > math.MaxInt32 || points == nil { - return errors.New("deck: invalid saved waypoint pack") - } - seen := map[uint64]bool{} - for _, id := range points { - if id == 0 || id > math.MaxInt32 || seen[id] { - return errors.New("deck: invalid saved waypoint activation") - } - seen[id] = true - } - } - return nil -} - -func (s *Store) handleWaypoint(path string, req []byte) (int, []byte, bool, error) { - seq, err := requestSequence(req) - if err != nil { - return 0, nil, true, err - } - pack, found, err := wire.Varint(req, 2) - if err != nil || !found || pack == 0 || pack > math.MaxInt32 { - return 0, nil, true, errors.New("deck: invalid waypoint pack") - } - s.mu.Lock() - defer s.mu.Unlock() - if s.waypointDesign == nil || s.waypointPack == nil { - return 0, nil, true, errors.New("deck: waypoint runtime unavailable") - } - if err = s.waypointPack(pack, path == "/WaypointUse"); err != nil { - return 0, nil, true, err - } - design, err := s.waypointDesign(pack) - if err != nil { - return 0, nil, true, err - } - if path == "/WaypointInfo" { - ids := append([]uint64(nil), s.state.Waypoints[pack]...) - slices.Sort(ids) - var packed []byte - for _, id := range ids { - if _, known := design.Points[id]; !known { - return 0, nil, true, fmt.Errorf("deck: saved waypoint %d absent from pack%d", id, pack) - } - packed = binary.AppendUvarint(packed, id) - } - if len(packed) == 0 { - return 31, nil, true, nil - } - return 31, wire.AppendBytes(nil, 1, packed), true, nil - } - id, found, err := wire.Varint(req, 3) - if err != nil || !found || id == 0 || id > math.MaxInt32 { - return 0, nil, true, errors.New("deck: invalid waypoint") - } - if _, known := design.Points[id]; !known { - return 0, nil, true, errors.New("deck: unknown waypoint") - } - if path == "/WaypointSave" { - if hasWaypoint(s.state.Waypoints[pack], id) { - return 32, nil, true, nil - } - next := clone(s.state) - next.Waypoints[pack] = append(next.Waypoints[pack], id) - err = s.commit(next) - return 32, nil, true, err - } - end, found, err := wire.Varint(req, 4) - if err != nil || !found || end == 0 || end > math.MaxInt32 || end == id { - return 0, nil, true, errors.New("deck: invalid waypoint destination") - } - target, known := design.Points[end] - if !known || target.MapID == 0 || !hasWaypoint(s.state.Waypoints[pack], end) || !hasWaypoint(s.state.Waypoints[pack], id) { - return 0, nil, true, errors.New("deck: waypoint is not activated") - } - moves, found, err := wire.Varint(req, 5) - if err != nil || !found || moves != 1 { - return 0, nil, true, errors.New("deck: invalid waypoint move count") - } - if reply, ok := s.cachedReplyLocked("waypoint-use", seq); ok { - return reply.code, reply.body, true, nil - } - if design.PriceUnit != 0 { - if s.wallet == nil || s.sessionID == "" { - return 0, nil, true, errors.New("deck: waypoint wallet session unavailable") - } - identity := fmt.Sprintf("waypoint:%s:%d", s.sessionID, seq) - switch design.PriceType { - case 4: - _, err = s.wallet.SpendGoldOnce(identity, design.PriceUnit) - case 3: - _, err = s.wallet.SpendFreeJewelryOnce(identity, design.PriceUnit) - case 2: - _, err = s.wallet.SpendJewelryOnce(identity, design.PriceUnit) - default: - err = errors.New("deck: unsupported waypoint currency") - } - if err != nil { - return 0, nil, true, err - } - } - // The client performs its warp and sends SaveUserPosition with scene coordinates. - s.rememberReplyLocked("waypoint-use", seq, 33, nil) - return 33, nil, true, nil -} diff --git a/go/internal/server/deck/waypoint_test.go b/go/internal/server/deck/waypoint_test.go deleted file mode 100644 index d544f7c..0000000 --- a/go/internal/server/deck/waypoint_test.go +++ /dev/null @@ -1,127 +0,0 @@ -package deck - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "errors" - "testing" -) - -func waypointFixture(t *testing.T) (*Store, *stateio.Memory) { - t.Helper() - memory := stateio.NewMemory() - seededStore := seeded(t) - s, err := OpenStore(memory, Seed{Version: seededStore.state.Version, FieldDeck: seededStore.state.FieldDeck, FieldCharControlDeckType: seededStore.state.FieldCharControlDeckType}) - if err != nil { - t.Fatal(err) - } - configureWaypointFixture(t, s) - return s, memory -} -func configureWaypointFixture(t *testing.T, s *Store) { - t.Helper() - if err := s.ConfigureWaypoints(func(uint64) (gamedata.WaypointPack, error) { - return gamedata.WaypointPack{Points: map[uint64]gamedata.Waypoint{1: {ID: 1, MapID: 10}, 2: {ID: 2, MapID: 20}}, PriceType: 4, PriceUnit: 7}, nil - }, func(pack uint64, _ bool) error { - if pack == 3 { - return errors.New("locked pack") - } - return nil - }); err != nil { - t.Fatal(err) - } -} -func waypointReq(seq, pack, start, end uint64) []byte { - body := wire.AppendVarint(nil, 2, pack) - if start != 0 { - body = wire.AppendVarint(body, 3, start) - } - if end != 0 { - body = wire.AppendVarint(body, 4, end) - body = wire.AppendVarint(body, 5, 1) - } - return req(seq, body) -} - -func TestWaypointActivationPersistsAllIDsAndInfoScopesPack(t *testing.T) { - s, memory := waypointFixture(t) - for _, r := range []struct{ seq, pack, id uint64 }{{1, 1, 2}, {2, 1, 1}, {3, 2, 2}, {4, 1, 1}} { - if _, _, _, err := s.Handle("/WaypointSave", waypointReq(r.seq, r.pack, r.id, 0)); err != nil { - t.Fatal(err) - } - } - reopened, err := OpenStore(memory, Seed{Version: s.state.Version, FieldDeck: s.state.FieldDeck, FieldCharControlDeckType: s.state.FieldCharControlDeckType}) - if err != nil { - t.Fatal(err) - } - configureWaypointFixture(t, reopened) - code, body, _, err := reopened.Handle("/WaypointInfo", waypointReq(5, 1, 0, 0)) - if err != nil || code != 31 || !bytes.Equal(body, []byte{10, 2, 1, 2}) { - t.Fatalf("pack1 info %d %x %v", code, body, err) - } - _, body, _, err = reopened.Handle("/WaypointInfo", waypointReq(6, 2, 0, 0)) - if err != nil || !bytes.Equal(body, []byte{10, 1, 2}) { - t.Fatalf("pack2 info %x %v", body, err) - } - if len(reopened.state.Waypoints[1]) != 2 { - t.Fatal("duplicate activation persisted") - } -} - -func TestWaypointTravelChargesOnceAndRejectsInvalidBeforeSpending(t *testing.T) { - s, memory := waypointFixture(t) - wallet, err := player.OpenWallet(memory, player.Currency{Gold: 14}) - if err != nil { - t.Fatal(err) - } - s.wallet = wallet - s.BeginSession("test-session") - for i := uint64(1); i <= 2; i++ { - if _, _, _, err = s.Handle("/WaypointSave", waypointReq(i, 1, i, 0)); err != nil { - t.Fatal(err) - } - } - for _, request := range [][]byte{waypointReq(3, 3, 1, 2), waypointReq(4, 1, 1, 99), waypointReq(5, 2, 1, 2)} { - if _, _, _, err = s.Handle("/WaypointUse", request); err == nil { - t.Fatal("invalid travel accepted") - } - if wallet.Snapshot().Gold != 14 { - t.Fatal("invalid travel charged") - } - } - valid := waypointReq(6, 1, 1, 2) - for range 2 { - code, _, _, err := s.Handle("/WaypointUse", valid) - if err != nil || code != 33 { - t.Fatalf("travel %d %v", code, err) - } - } - if wallet.Snapshot().Gold != 7 { - t.Fatal("repeat travel double charged") - } - reopened, err := OpenStore(memory, Seed{Version: s.state.Version, FieldDeck: s.state.FieldDeck, FieldCharControlDeckType: s.state.FieldCharControlDeckType}) - if err != nil { - t.Fatal(err) - } - configureWaypointFixture(t, reopened) - reopened.wallet = wallet - reopened.BeginSession("test-session") - if _, _, _, err = reopened.Handle("/WaypointUse", valid); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Gold != 7 { - t.Fatal("restart retry charged") - } - if _, _, _, err = reopened.Handle("/WaypointUse", waypointReq(7, 1, 2, 1)); err != nil { - t.Fatal(err) - } - if _, _, _, err = reopened.Handle("/WaypointUse", waypointReq(8, 1, 1, 2)); err == nil { - t.Fatal("insufficient balance accepted") - } - if wallet.Snapshot().Gold != 0 { - t.Fatal("failed debit changed wallet") - } -} diff --git a/go/internal/server/dbcrypt/pages.go b/go/internal/server/design/dbcrypt/pages.go similarity index 100% rename from go/internal/server/dbcrypt/pages.go rename to go/internal/server/design/dbcrypt/pages.go diff --git a/go/internal/server/gameconfig/config.go b/go/internal/server/design/gameconfig/config.go similarity index 98% rename from go/internal/server/gameconfig/config.go rename to go/internal/server/design/gameconfig/config.go index 06290bd..be64e3b 100644 --- a/go/internal/server/gameconfig/config.go +++ b/go/internal/server/design/gameconfig/config.go @@ -11,7 +11,7 @@ import ( "path/filepath" "slices" - "bd2server/internal/server/configfile" + "bd2server/internal/server/platform/configfile" ) const FileName = "game.json" diff --git a/go/internal/server/gamedata/achievement.go b/go/internal/server/design/gamedata/achievement.go similarity index 100% rename from go/internal/server/gamedata/achievement.go rename to go/internal/server/design/gamedata/achievement.go diff --git a/go/internal/server/gamedata/achievement_gameplay.go b/go/internal/server/design/gamedata/achievement_gameplay.go similarity index 100% rename from go/internal/server/gamedata/achievement_gameplay.go rename to go/internal/server/design/gamedata/achievement_gameplay.go diff --git a/go/internal/server/gamedata/achievement_level.go b/go/internal/server/design/gamedata/achievement_level.go similarity index 100% rename from go/internal/server/gamedata/achievement_level.go rename to go/internal/server/design/gamedata/achievement_level.go diff --git a/go/internal/server/gamedata/attributes.go b/go/internal/server/design/gamedata/attributes.go similarity index 100% rename from go/internal/server/gamedata/attributes.go rename to go/internal/server/design/gamedata/attributes.go diff --git a/go/internal/server/gamedata/avatar_rewards.go b/go/internal/server/design/gamedata/avatar_rewards.go similarity index 100% rename from go/internal/server/gamedata/avatar_rewards.go rename to go/internal/server/design/gamedata/avatar_rewards.go diff --git a/go/internal/server/gamedata/battle.go b/go/internal/server/design/gamedata/battle.go similarity index 94% rename from go/internal/server/gamedata/battle.go rename to go/internal/server/design/gamedata/battle.go index b519764..0201b2a 100644 --- a/go/internal/server/gamedata/battle.go +++ b/go/internal/server/design/gamedata/battle.go @@ -4,7 +4,7 @@ import ( "encoding/binary" "fmt" - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" _ "modernc.org/sqlite" ) @@ -72,7 +72,12 @@ func readBattleRewards(root, version string, packID, deckID, monsterID int) ([]B } rewards := make([]BattleReward, len(types)) for i := range rewards { - if types[i] == 0 || ids[i] == 0 || counts[i] == 0 { + validID := ids[i] != 0 + switch types[i] { + case 2, 3, 4, 12, 20, 22, 68: + validID = ids[i] == 0 + } + if types[i] == 0 || !validID || counts[i] == 0 { return nil, fmt.Errorf("gamedata: invalid battle deck %d reward", deckID) } rewards[i] = BattleReward{Type: types[i], ID: ids[i], Count: counts[i]} diff --git a/go/internal/server/design/gamedata/battle_difficulty.go b/go/internal/server/design/gamedata/battle_difficulty.go new file mode 100644 index 0000000..85b78bc --- /dev/null +++ b/go/internal/server/design/gamedata/battle_difficulty.go @@ -0,0 +1,126 @@ +package gamedata + +import ( + "database/sql" + "fmt" + "slices" +) + +type QuestBattleDeck struct { + DeckID uint64 + QuestIDs []uint64 +} + +// ResolveQuestBattleDeck applies story difficulty only to monsters bound to +// main quests. Side quests use their authored deck regardless of story level. +func ResolveQuestBattleDeck(root, version string, packID int, monsterID, deckID, difficulty uint64) (QuestBattleDeck, error) { + if monsterID == 0 || monsterID > 2147483647 || deckID == 0 || deckID > 2147483647 || difficulty > 2 { + return QuestBattleDeck{}, fmt.Errorf("gamedata: invalid quest monster/deck/difficulty %d/%d/%d", monsterID, deckID, difficulty) + } + db, cleanup, err := openPackDatabase(root, version, packID) + if err != nil { + return QuestBattleDeck{}, err + } + defer cleanup() + var raw []byte + if err := db.QueryRow("SELECT ProtoBuf FROM FieldMonsterTable WHERE id=?", monsterID).Scan(&raw); err != nil { + return QuestBattleDeck{}, fmt.Errorf("gamedata: quest monster %d: %w", monsterID, err) + } + quests, err := packedInts(raw, 22) + if err != nil { + return QuestBattleDeck{}, err + } + if len(quests) == 0 { + return QuestBattleDeck{}, fmt.Errorf("gamedata: quest monster %d has no quest range", monsterID) + } + decks, err := packedInts(raw, 2) + if err != nil { + return QuestBattleDeck{}, err + } + group, err := phaseScalar(raw, 21) + if err != nil { + return QuestBattleDeck{}, err + } + common, release, err := OpenDatabase(root, version, "common") + if err != nil { + return QuestBattleDeck{}, err + } + defer release() + var kind uint64 + for i, quest := range quests { + if quest == 0 || quest > 2147483647 { + return QuestBattleDeck{}, fmt.Errorf("gamedata: quest monster %d has invalid quest %d", monsterID, quest) + } + if err := common.QueryRow(fmt.Sprintf("SELECT ProtoBuf FROM QuestTable%d WHERE id=?", packID), quest).Scan(&raw); err != nil { + return QuestBattleDeck{}, fmt.Errorf("gamedata: monster %d quest %d: %w", monsterID, quest, err) + } + questKind, err := phaseScalar(raw, 63) + if err != nil { + return QuestBattleDeck{}, err + } + if questKind > 1 || (i > 0 && questKind != kind) { + return QuestBattleDeck{}, fmt.Errorf("gamedata: quest monster %d has unsupported or mixed quest types", monsterID) + } + kind = questKind + } + if kind == 1 { + difficulty = 0 + } + selected, err := battleDeckForDifficultyFromDB(db, deckID, difficulty) + if err != nil { + return QuestBattleDeck{}, err + } + if group == 0 { + allowed := slices.Contains(decks, deckID) + if kind == 0 && selected > difficulty*100000 { + allowed = allowed || slices.Contains(decks, selected-difficulty*100000) + } + if !allowed { + return QuestBattleDeck{}, fmt.Errorf("gamedata: quest monster %d does not contain deck %d", monsterID, deckID) + } + } else if _, err := battleDeckPhasesFromDB(db, monsterID, selected); err != nil { + return QuestBattleDeck{}, err + } + return QuestBattleDeck{DeckID: selected, QuestIDs: quests}, nil +} + +func openPackDatabase(root, version string, packID int) (*sql.DB, func(), error) { + if packID <= 0 { + return nil, nil, fmt.Errorf("gamedata: invalid battle pack %d", packID) + } + return OpenDatabase(root, version, fmt.Sprintf("pack%d", packID)) +} + +func battleDeckForDifficultyFromDB(db *sql.DB, deckID, difficulty uint64) (uint64, error) { + if deckID == 0 || deckID > 2147483647 || difficulty > 2 { + return 0, fmt.Errorf("gamedata: unsupported battle deck/difficulty %d/%d", deckID, difficulty) + } + var data []byte + if err := db.QueryRow("SELECT ProtoBuf FROM BattleDeckTable WHERE id=?", deckID).Scan(&data); err != nil { + return 0, fmt.Errorf("gamedata: battle deck %d: %w", deckID, err) + } + actual, err := phaseScalar(data, 24) + if err != nil { + return 0, err + } + if actual == difficulty { + return deckID, nil + } + // Only an ordinary deck can be promoted. Changing a previously selected + // difficulty deck would conceal stale client quest state. + if actual != 0 || difficulty == 0 { + return 0, fmt.Errorf("gamedata: deck %d difficulty %d differs from selected %d", deckID, actual, difficulty) + } + selected := deckID + difficulty*100000 + if err := db.QueryRow("SELECT ProtoBuf FROM BattleDeckTable WHERE id=?", selected).Scan(&data); err != nil { + return 0, fmt.Errorf("gamedata: difficulty deck %d: %w", selected, err) + } + actual, err = phaseScalar(data, 24) + if err != nil { + return 0, err + } + if actual != difficulty { + return 0, fmt.Errorf("gamedata: difficulty deck %d has difficulty %d, want %d", selected, actual, difficulty) + } + return selected, nil +} diff --git a/go/internal/server/gamedata/battle_phase.go b/go/internal/server/design/gamedata/battle_phase.go similarity index 98% rename from go/internal/server/gamedata/battle_phase.go rename to go/internal/server/design/gamedata/battle_phase.go index 45bc15a..3aeecf6 100644 --- a/go/internal/server/gamedata/battle_phase.go +++ b/go/internal/server/design/gamedata/battle_phase.go @@ -4,7 +4,7 @@ import ( "database/sql" "fmt" - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" ) // BattlePhase identifies a row in the pack's PhaseBattleTable. ID is the diff --git a/go/internal/server/gamedata/buff_reward_design.go b/go/internal/server/design/gamedata/buff_reward_design.go similarity index 100% rename from go/internal/server/gamedata/buff_reward_design.go rename to go/internal/server/design/gamedata/buff_reward_design.go diff --git a/go/internal/server/gamedata/calendar_ids.go b/go/internal/server/design/gamedata/calendar_ids.go similarity index 100% rename from go/internal/server/gamedata/calendar_ids.go rename to go/internal/server/design/gamedata/calendar_ids.go diff --git a/go/internal/server/gamedata/cash_bonuses.go b/go/internal/server/design/gamedata/cash_bonuses.go similarity index 100% rename from go/internal/server/gamedata/cash_bonuses.go rename to go/internal/server/design/gamedata/cash_bonuses.go diff --git a/go/internal/server/gamedata/cash_entitlements.go b/go/internal/server/design/gamedata/cash_entitlements.go similarity index 100% rename from go/internal/server/gamedata/cash_entitlements.go rename to go/internal/server/design/gamedata/cash_entitlements.go diff --git a/go/internal/server/gamedata/cash_mail.go b/go/internal/server/design/gamedata/cash_mail.go similarity index 100% rename from go/internal/server/gamedata/cash_mail.go rename to go/internal/server/design/gamedata/cash_mail.go diff --git a/go/internal/server/gamedata/cash_products.go b/go/internal/server/design/gamedata/cash_products.go similarity index 99% rename from go/internal/server/gamedata/cash_products.go rename to go/internal/server/design/gamedata/cash_products.go index 06dea77..a410133 100644 --- a/go/internal/server/gamedata/cash_products.go +++ b/go/internal/server/design/gamedata/cash_products.go @@ -5,7 +5,7 @@ import ( "fmt" "math" - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" ) // CashProductKey is the official three-part identity; an SDK SKU can be shared diff --git a/go/internal/server/gamedata/cash_rewards.go b/go/internal/server/design/gamedata/cash_rewards.go similarity index 100% rename from go/internal/server/gamedata/cash_rewards.go rename to go/internal/server/design/gamedata/cash_rewards.go diff --git a/go/internal/server/gamedata/char_awake.go b/go/internal/server/design/gamedata/char_awake.go similarity index 100% rename from go/internal/server/gamedata/char_awake.go rename to go/internal/server/design/gamedata/char_awake.go diff --git a/go/internal/server/gamedata/clear_packages.go b/go/internal/server/design/gamedata/clear_packages.go similarity index 100% rename from go/internal/server/gamedata/clear_packages.go rename to go/internal/server/design/gamedata/clear_packages.go diff --git a/go/internal/server/gamedata/content_open.go b/go/internal/server/design/gamedata/content_open.go similarity index 100% rename from go/internal/server/gamedata/content_open.go rename to go/internal/server/design/gamedata/content_open.go diff --git a/go/internal/server/gamedata/content_ticket.go b/go/internal/server/design/gamedata/content_ticket.go similarity index 100% rename from go/internal/server/gamedata/content_ticket.go rename to go/internal/server/design/gamedata/content_ticket.go diff --git a/go/internal/server/gamedata/cooking_recipes.go b/go/internal/server/design/gamedata/cooking_recipes.go similarity index 100% rename from go/internal/server/gamedata/cooking_recipes.go rename to go/internal/server/design/gamedata/cooking_recipes.go diff --git a/go/internal/server/gamedata/costume_burst.go b/go/internal/server/design/gamedata/costume_burst.go similarity index 100% rename from go/internal/server/gamedata/costume_burst.go rename to go/internal/server/design/gamedata/costume_burst.go diff --git a/go/internal/server/gamedata/costume_potential.go b/go/internal/server/design/gamedata/costume_potential.go similarity index 100% rename from go/internal/server/gamedata/costume_potential.go rename to go/internal/server/design/gamedata/costume_potential.go diff --git a/go/internal/server/gamedata/database.go b/go/internal/server/design/gamedata/database.go similarity index 98% rename from go/internal/server/gamedata/database.go rename to go/internal/server/design/gamedata/database.go index 319c8fc..5b7c0d9 100644 --- a/go/internal/server/gamedata/database.go +++ b/go/internal/server/design/gamedata/database.go @@ -9,7 +9,7 @@ import ( "path/filepath" "strings" - "bd2server/internal/server/dbcrypt" + "bd2server/internal/server/design/dbcrypt" ) // DatabaseName maps a logical client DB name to its GameData archive entry. diff --git a/go/internal/server/gamedata/database_cache.go b/go/internal/server/design/gamedata/database_cache.go similarity index 100% rename from go/internal/server/gamedata/database_cache.go rename to go/internal/server/design/gamedata/database_cache.go diff --git a/go/internal/server/gamedata/dispatch.go b/go/internal/server/design/gamedata/dispatch.go similarity index 100% rename from go/internal/server/gamedata/dispatch.go rename to go/internal/server/design/gamedata/dispatch.go diff --git a/go/internal/server/gamedata/equipment_collaboration.go b/go/internal/server/design/gamedata/equipment_collaboration.go similarity index 100% rename from go/internal/server/gamedata/equipment_collaboration.go rename to go/internal/server/design/gamedata/equipment_collaboration.go diff --git a/go/internal/server/gamedata/equipment_craft.go b/go/internal/server/design/gamedata/equipment_craft.go similarity index 100% rename from go/internal/server/gamedata/equipment_craft.go rename to go/internal/server/design/gamedata/equipment_craft.go diff --git a/go/internal/server/gamedata/equipment_gacha.go b/go/internal/server/design/gamedata/equipment_gacha.go similarity index 100% rename from go/internal/server/gamedata/equipment_gacha.go rename to go/internal/server/design/gamedata/equipment_gacha.go diff --git a/go/internal/server/gamedata/equipment_grades.go b/go/internal/server/design/gamedata/equipment_grades.go similarity index 100% rename from go/internal/server/gamedata/equipment_grades.go rename to go/internal/server/design/gamedata/equipment_grades.go diff --git a/go/internal/server/gamedata/equipment_option_reroll.go b/go/internal/server/design/gamedata/equipment_option_reroll.go similarity index 100% rename from go/internal/server/gamedata/equipment_option_reroll.go rename to go/internal/server/design/gamedata/equipment_option_reroll.go diff --git a/go/internal/server/gamedata/equipment_slots.go b/go/internal/server/design/gamedata/equipment_slots.go similarity index 100% rename from go/internal/server/gamedata/equipment_slots.go rename to go/internal/server/design/gamedata/equipment_slots.go diff --git a/go/internal/server/gamedata/equipment_smelting.go b/go/internal/server/design/gamedata/equipment_smelting.go similarity index 100% rename from go/internal/server/gamedata/equipment_smelting.go rename to go/internal/server/design/gamedata/equipment_smelting.go diff --git a/go/internal/server/gamedata/equipment_stats.go b/go/internal/server/design/gamedata/equipment_stats.go similarity index 100% rename from go/internal/server/gamedata/equipment_stats.go rename to go/internal/server/design/gamedata/equipment_stats.go diff --git a/go/internal/server/gamedata/equipment_upgrade.go b/go/internal/server/design/gamedata/equipment_upgrade.go similarity index 100% rename from go/internal/server/gamedata/equipment_upgrade.go rename to go/internal/server/design/gamedata/equipment_upgrade.go diff --git a/go/internal/server/gamedata/event_actions.go b/go/internal/server/design/gamedata/event_actions.go similarity index 98% rename from go/internal/server/gamedata/event_actions.go rename to go/internal/server/design/gamedata/event_actions.go index a5ab718..3d3131c 100644 --- a/go/internal/server/gamedata/event_actions.go +++ b/go/internal/server/design/gamedata/event_actions.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "fmt" "sort" ) diff --git a/go/internal/server/gamedata/event_ap.go b/go/internal/server/design/gamedata/event_ap.go similarity index 100% rename from go/internal/server/gamedata/event_ap.go rename to go/internal/server/design/gamedata/event_ap.go diff --git a/go/internal/server/gamedata/event_battle_challenge.go b/go/internal/server/design/gamedata/event_battle_challenge.go similarity index 98% rename from go/internal/server/gamedata/event_battle_challenge.go rename to go/internal/server/design/gamedata/event_battle_challenge.go index 10521cd..34fba5c 100644 --- a/go/internal/server/gamedata/event_battle_challenge.go +++ b/go/internal/server/design/gamedata/event_battle_challenge.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "encoding/binary" "fmt" ) diff --git a/go/internal/server/gamedata/event_exchange.go b/go/internal/server/design/gamedata/event_exchange.go similarity index 100% rename from go/internal/server/gamedata/event_exchange.go rename to go/internal/server/design/gamedata/event_exchange.go diff --git a/go/internal/server/gamedata/event_field_pack.go b/go/internal/server/design/gamedata/event_field_pack.go similarity index 100% rename from go/internal/server/gamedata/event_field_pack.go rename to go/internal/server/design/gamedata/event_field_pack.go diff --git a/go/internal/server/gamedata/event_games.go b/go/internal/server/design/gamedata/event_games.go similarity index 100% rename from go/internal/server/gamedata/event_games.go rename to go/internal/server/design/gamedata/event_games.go diff --git a/go/internal/server/gamedata/event_owned_item.go b/go/internal/server/design/gamedata/event_owned_item.go similarity index 100% rename from go/internal/server/gamedata/event_owned_item.go rename to go/internal/server/design/gamedata/event_owned_item.go diff --git a/go/internal/server/gamedata/event_play.go b/go/internal/server/design/gamedata/event_play.go similarity index 100% rename from go/internal/server/gamedata/event_play.go rename to go/internal/server/design/gamedata/event_play.go diff --git a/go/internal/server/gamedata/event_play_field.go b/go/internal/server/design/gamedata/event_play_field.go similarity index 100% rename from go/internal/server/gamedata/event_play_field.go rename to go/internal/server/design/gamedata/event_play_field.go diff --git a/go/internal/server/gamedata/event_tasks.go b/go/internal/server/design/gamedata/event_tasks.go similarity index 100% rename from go/internal/server/gamedata/event_tasks.go rename to go/internal/server/design/gamedata/event_tasks.go diff --git a/go/internal/server/gamedata/field_action_objects.go b/go/internal/server/design/gamedata/field_action_objects.go similarity index 100% rename from go/internal/server/gamedata/field_action_objects.go rename to go/internal/server/design/gamedata/field_action_objects.go diff --git a/go/internal/server/gamedata/field_buffs.go b/go/internal/server/design/gamedata/field_buffs.go similarity index 97% rename from go/internal/server/gamedata/field_buffs.go rename to go/internal/server/design/gamedata/field_buffs.go index b277fba..a1a6bd7 100644 --- a/go/internal/server/gamedata/field_buffs.go +++ b/go/internal/server/design/gamedata/field_buffs.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "encoding/binary" "fmt" "math" diff --git a/go/internal/server/gamedata/field_monster_maps.go b/go/internal/server/design/gamedata/field_monster_maps.go similarity index 97% rename from go/internal/server/gamedata/field_monster_maps.go rename to go/internal/server/design/gamedata/field_monster_maps.go index 10f24af..d4afde0 100644 --- a/go/internal/server/gamedata/field_monster_maps.go +++ b/go/internal/server/design/gamedata/field_monster_maps.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "fmt" ) diff --git a/go/internal/server/gamedata/field_monsters.go b/go/internal/server/design/gamedata/field_monsters.go similarity index 83% rename from go/internal/server/gamedata/field_monsters.go rename to go/internal/server/design/gamedata/field_monsters.go index c646ae1..9fac58b 100644 --- a/go/internal/server/gamedata/field_monsters.go +++ b/go/internal/server/design/gamedata/field_monsters.go @@ -9,12 +9,12 @@ import ( // FieldMonsterDesign keeps the regeneration group requested by MonsterInfo, // not the unrelated event monster group used by FieldEventSpawnReward. type FieldMonsterDesign struct { - ID, GroupID, QuestID int - BattleDeck uint64 - BattleDecks []uint64 - LifeSeconds uint64 - RegenSeconds, ResetType, Type, FieldBuff, UseBattleSkip uint64 - Reward Reward + ID, GroupID, QuestID int + BattleDeck uint64 + BattleDecks []uint64 + LifeSeconds uint64 + RegenSeconds, ResetType, Type, FieldBuff, UseBattleSkip, CrashType uint64 + Reward Reward } func LoadFieldMonsters(root, version string, pack int) ([]FieldMonsterDesign, error) { @@ -100,7 +100,7 @@ func loadFieldMonsters(db *sql.DB) ([]FieldMonsterDesign, error) { } } g.ID = id - for f, dst := range map[int]*uint64{30: &g.Type, 6: &g.FieldBuff, 31: &g.UseBattleSkip, 27: &g.Reward.Count, 28: &g.Reward.ID, 29: &g.Reward.Type} { + for f, dst := range map[int]*uint64{30: &g.Type, 6: &g.FieldBuff, 15: &g.CrashType, 31: &g.UseBattleSkip, 27: &g.Reward.Count, 28: &g.Reward.ID, 29: &g.Reward.Type} { v, e := optionalScalar(raw, f) if e != nil { return nil, e diff --git a/go/internal/server/gamedata/field_objects.go b/go/internal/server/design/gamedata/field_objects.go similarity index 100% rename from go/internal/server/gamedata/field_objects.go rename to go/internal/server/design/gamedata/field_objects.go diff --git a/go/internal/server/gamedata/field_pack.go b/go/internal/server/design/gamedata/field_pack.go similarity index 100% rename from go/internal/server/gamedata/field_pack.go rename to go/internal/server/design/gamedata/field_pack.go diff --git a/go/internal/server/gamedata/field_research.go b/go/internal/server/design/gamedata/field_research.go similarity index 99% rename from go/internal/server/gamedata/field_research.go rename to go/internal/server/design/gamedata/field_research.go index e661b70..6a9ef90 100644 --- a/go/internal/server/gamedata/field_research.go +++ b/go/internal/server/design/gamedata/field_research.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "database/sql" "fmt" ) diff --git a/go/internal/server/gamedata/field_reset.go b/go/internal/server/design/gamedata/field_reset.go similarity index 98% rename from go/internal/server/gamedata/field_reset.go rename to go/internal/server/design/gamedata/field_reset.go index 04ed632..be6b866 100644 --- a/go/internal/server/gamedata/field_reset.go +++ b/go/internal/server/design/gamedata/field_reset.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "fmt" "strconv" "strings" diff --git a/go/internal/server/gamedata/field_settings.go b/go/internal/server/design/gamedata/field_settings.go similarity index 100% rename from go/internal/server/gamedata/field_settings.go rename to go/internal/server/design/gamedata/field_settings.go diff --git a/go/internal/server/design/gamedata/field_traps.go b/go/internal/server/design/gamedata/field_traps.go new file mode 100644 index 0000000..aa9b5a0 --- /dev/null +++ b/go/internal/server/design/gamedata/field_traps.go @@ -0,0 +1,175 @@ +package gamedata + +import ( + "fmt" + "slices" +) + +type FieldTrap struct { + ID, MapID, SwitchID, ResetType, Type int + FieldBuff, CoolSeconds, RegenSeconds uint64 + DefaultEnabled bool + QuestRange []uint64 + Maps []int +} +type FieldTrapTrigger struct { + ID, Type, AutoInteraction int + QuestRange []uint64 + Maps []int +} +type FieldTrapSwitch struct { + ID, ObjectType, OrderType int + Objects []uint64 +} +type FieldTrapDesign struct { + Traps map[int]FieldTrap + Triggers map[int]FieldTrapTrigger + Switches map[int]FieldTrapSwitch + MapIDs []int + StoryModeImmune bool +} + +func LoadFieldTraps(root, version string, pack int) (FieldTrapDesign, error) { + d := FieldTrapDesign{Traps: map[int]FieldTrap{}, Triggers: map[int]FieldTrapTrigger{}, Switches: map[int]FieldTrapSwitch{}} + db, done, err := OpenDatabase(root, version, fmt.Sprintf("pack%d", pack)) + if err != nil { + return d, err + } + defer done() + for _, table := range []string{"FieldTrapTable", "TriggerTable", "FieldObjectSwitchTable"} { + rows, e := db.Query("SELECT id,ProtoBuf FROM " + table + " ORDER BY id") + if e != nil { + return d, e + } + for rows.Next() { + var id int + var raw []byte + if e = rows.Scan(&id, &raw); e != nil { + _ = rows.Close() + return d, e + } + values := map[int]uint64{} + for _, field := range []int{2, 3, 4, 5, 8, 9, 11, 13, 14, 17, 18} { + if table == "TriggerTable" && (field == 2 || field == 4 || field == 13) || table == "FieldObjectSwitchTable" && field == 4 || table == "FieldTrapTable" && field == 11 { + continue + } + values[field], e = optionalScalar(raw, field) + if e != nil { + _ = rows.Close() + return d, e + } + } + switch table { + case "FieldTrapTable": + q, e := packedInts(raw, 11) + if e != nil { + _ = rows.Close() + return d, e + } + if values[3] > 1 || values[14] > 1 || values[4] > 0x7fffffff || values[13] > 0x7fffffff { + _ = rows.Close() + return d, fmt.Errorf("gamedata: invalid trap %d pack %d", id, pack) + } + d.Traps[id] = FieldTrap{ID: id, MapID: int(values[8]), SwitchID: int(values[17]), ResetType: int(values[14]), Type: int(values[18]), FieldBuff: values[5], CoolSeconds: values[4], RegenSeconds: values[13], DefaultEnabled: values[3] == 1, QuestRange: q} + case "TriggerTable": + q, e := packedInts(raw, 12) + if e != nil { + _ = rows.Close() + return d, e + } + d.Triggers[id] = FieldTrapTrigger{ID: id, Type: int(values[11]), AutoInteraction: int(values[9]), QuestRange: q} + case "FieldObjectSwitchTable": + objects, e := packedInts(raw, 4) + if e != nil { + _ = rows.Close() + return d, e + } + d.Switches[id] = FieldTrapSwitch{ID: id, ObjectType: int(values[2]), OrderType: int(values[5]), Objects: objects} + } + } + e = rows.Err() + _ = rows.Close() + if e != nil { + return d, e + } + } + common, closeCommon, err := openStatDatabase(root, version) + if err != nil { + return d, err + } + defer closeCommon() + var packRaw []byte + if err := common.QueryRow("SELECT ProtoBuf FROM PackTable WHERE id=?", pack).Scan(&packRaw); err != nil { + return d, err + } + packType, err := optionalScalar(packRaw, 55) + if err != nil { + return d, err + } + // FieldDeckPacket.IsUsingStoryMode includes PackInfo.IsNewFieldSpecPack. + d.StoryModeImmune = packType == 0 || packType == 1000 || packType == 1 && pack >= 1010 || packType == 6 && pack >= 2009 + mapScenes := map[string]int{} + rows, err := common.Query("SELECT id,mapScenePath FROM MapTable WHERE packId=? ORDER BY id", pack) + if err != nil { + return d, err + } + for rows.Next() { + var id int + var scene string + if err = rows.Scan(&id, &scene); err != nil { + _ = rows.Close() + return d, err + } + mapScenes[scene] = id + d.MapIDs = append(d.MapIDs, id) + } + err = rows.Err() + _ = rows.Close() + if err != nil { + return d, err + } + scenes, closeScenes, err := OpenDatabase(root, version, "FieldObjectSceneData") + if err != nil { + return d, err + } + defer closeScenes() + rows, err = scenes.Query(fmt.Sprintf("SELECT tableId,sceneName,objectType FROM FieldObjectSceneData%d WHERE objectType IN (7,15,16)", pack)) + if err != nil { + return d, err + } + defer func() { _ = rows.Close() }() + for rows.Next() { + var id, kind int + var scene string + if err = rows.Scan(&id, &scene, &kind); err != nil { + return d, err + } + if id == 0 { + continue + } + mapID, found := mapScenes[scene] + if !found { + continue + } + if kind == 15 { + trigger, found := d.Triggers[id] + if !found { + return d, fmt.Errorf("gamedata: missing trigger %d pack %d", id, pack) + } + if !slices.Contains(trigger.Maps, mapID) { + trigger.Maps = append(trigger.Maps, mapID) + } + d.Triggers[id] = trigger + } else { + trap, found := d.Traps[id] + if !found { + return d, fmt.Errorf("gamedata: missing trap %d pack %d", id, pack) + } + if !slices.Contains(trap.Maps, mapID) { + trap.Maps = append(trap.Maps, mapID) + } + d.Traps[id] = trap + } + } + return d, rows.Err() +} diff --git a/go/internal/server/gamedata/first_gacha.go b/go/internal/server/design/gamedata/first_gacha.go similarity index 100% rename from go/internal/server/gamedata/first_gacha.go rename to go/internal/server/design/gamedata/first_gacha.go diff --git a/go/internal/server/gamedata/food.go b/go/internal/server/design/gamedata/food.go similarity index 100% rename from go/internal/server/gamedata/food.go rename to go/internal/server/design/gamedata/food.go diff --git a/go/internal/server/gamedata/friendship.go b/go/internal/server/design/gamedata/friendship.go similarity index 100% rename from go/internal/server/gamedata/friendship.go rename to go/internal/server/design/gamedata/friendship.go diff --git a/go/internal/server/gamedata/gacha.go b/go/internal/server/design/gamedata/gacha.go similarity index 99% rename from go/internal/server/gamedata/gacha.go rename to go/internal/server/design/gamedata/gacha.go index 187e27d..9247b60 100644 --- a/go/internal/server/gamedata/gacha.go +++ b/go/internal/server/design/gamedata/gacha.go @@ -14,7 +14,7 @@ import ( _ "modernc.org/sqlite" - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" ) const ( diff --git a/go/internal/server/gamedata/gacha_cash.go b/go/internal/server/design/gamedata/gacha_cash.go similarity index 100% rename from go/internal/server/gamedata/gacha_cash.go rename to go/internal/server/design/gamedata/gacha_cash.go diff --git a/go/internal/server/gamedata/gacha_character.go b/go/internal/server/design/gamedata/gacha_character.go similarity index 100% rename from go/internal/server/gamedata/gacha_character.go rename to go/internal/server/design/gamedata/gacha_character.go diff --git a/go/internal/server/gamedata/gacha_grades.go b/go/internal/server/design/gamedata/gacha_grades.go similarity index 100% rename from go/internal/server/gamedata/gacha_grades.go rename to go/internal/server/design/gamedata/gacha_grades.go diff --git a/go/internal/server/gamedata/gacha_infinite_metadata.go b/go/internal/server/design/gamedata/gacha_infinite_metadata.go similarity index 100% rename from go/internal/server/gamedata/gacha_infinite_metadata.go rename to go/internal/server/design/gamedata/gacha_infinite_metadata.go diff --git a/go/internal/server/gamedata/growth.go b/go/internal/server/design/gamedata/growth.go similarity index 100% rename from go/internal/server/gamedata/growth.go rename to go/internal/server/design/gamedata/growth.go diff --git a/go/internal/server/gamedata/hunting.go b/go/internal/server/design/gamedata/hunting.go similarity index 100% rename from go/internal/server/gamedata/hunting.go rename to go/internal/server/design/gamedata/hunting.go diff --git a/go/internal/server/gamedata/hunting_ap.go b/go/internal/server/design/gamedata/hunting_ap.go similarity index 96% rename from go/internal/server/gamedata/hunting_ap.go rename to go/internal/server/design/gamedata/hunting_ap.go index c9024ca..55e43a5 100644 --- a/go/internal/server/gamedata/hunting_ap.go +++ b/go/internal/server/design/gamedata/hunting_ap.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "fmt" "strconv" "strings" diff --git a/go/internal/server/gamedata/immortal.go b/go/internal/server/design/gamedata/immortal.go similarity index 100% rename from go/internal/server/gamedata/immortal.go rename to go/internal/server/design/gamedata/immortal.go diff --git a/go/internal/server/gamedata/inn.go b/go/internal/server/design/gamedata/inn.go similarity index 100% rename from go/internal/server/gamedata/inn.go rename to go/internal/server/design/gamedata/inn.go diff --git a/go/internal/server/gamedata/inventory_slots.go b/go/internal/server/design/gamedata/inventory_slots.go similarity index 100% rename from go/internal/server/gamedata/inventory_slots.go rename to go/internal/server/design/gamedata/inventory_slots.go diff --git a/go/internal/server/gamedata/item_craft.go b/go/internal/server/design/gamedata/item_craft.go similarity index 100% rename from go/internal/server/gamedata/item_craft.go rename to go/internal/server/design/gamedata/item_craft.go diff --git a/go/internal/server/design/gamedata/item_stacks.go b/go/internal/server/design/gamedata/item_stacks.go new file mode 100644 index 0000000..f2fea71 --- /dev/null +++ b/go/internal/server/design/gamedata/item_stacks.go @@ -0,0 +1,60 @@ +package gamedata + +import ( + "fmt" + "math" +) + +type ItemStackDesign struct { + Limits map[[2]uint64]uint64 +} + +func LoadItemStackDesign(root, version string) (*ItemStackDesign, error) { + db, release, err := OpenDatabase(root, version, "common") + if err != nil { + return nil, err + } + defer release() + design := &ItemStackDesign{Limits: map[[2]uint64]uint64{}} + for _, source := range []struct { + name string + kind, idField, limitField uint64 + }{ + {"FoodTable", 5, 7, 16}, {"ResourceTable", 8, 4, 12}, + {"RandomBoxTable", 9, 4, 12}, {"UseItemTable", 14, 3, 10}, + } { + rows, err := db.Query("SELECT id,ProtoBuf FROM " + source.name + " ORDER BY id") + if err != nil { + return nil, err + } + for rows.Next() { + var id uint64 + var raw []byte + if err := rows.Scan(&id, &raw); err != nil { + _ = rows.Close() + return nil, err + } + protoID, err := optionalScalar(raw, int(source.idField)) + if err != nil || id == 0 || protoID != id { + _ = rows.Close() + return nil, fmt.Errorf("gamedata: invalid %s stack identity %d", source.name, id) + } + limit, err := optionalScalar(raw, int(source.limitField)) + if err != nil || limit > math.MaxInt32 { + _ = rows.Close() + return nil, fmt.Errorf("gamedata: invalid %s stack limit %d", source.name, id) + } + if limit > 0 { + design.Limits[[2]uint64{source.kind, id}] = limit + } + } + if err := rows.Err(); err != nil { + _ = rows.Close() + return nil, err + } + if err := rows.Close(); err != nil { + return nil, err + } + } + return design, nil +} diff --git a/go/internal/server/gamedata/limited_costume.go b/go/internal/server/design/gamedata/limited_costume.go similarity index 100% rename from go/internal/server/gamedata/limited_costume.go rename to go/internal/server/design/gamedata/limited_costume.go diff --git a/go/internal/server/gamedata/login_passes.go b/go/internal/server/design/gamedata/login_passes.go similarity index 100% rename from go/internal/server/gamedata/login_passes.go rename to go/internal/server/design/gamedata/login_passes.go diff --git a/go/internal/server/gamedata/mini_content.go b/go/internal/server/design/gamedata/mini_content.go similarity index 100% rename from go/internal/server/gamedata/mini_content.go rename to go/internal/server/design/gamedata/mini_content.go diff --git a/go/internal/server/gamedata/mini_hub.go b/go/internal/server/design/gamedata/mini_hub.go similarity index 100% rename from go/internal/server/gamedata/mini_hub.go rename to go/internal/server/design/gamedata/mini_hub.go diff --git a/go/internal/server/gamedata/missions.go b/go/internal/server/design/gamedata/missions.go similarity index 100% rename from go/internal/server/gamedata/missions.go rename to go/internal/server/design/gamedata/missions.go diff --git a/go/internal/server/gamedata/monster_hunt.go b/go/internal/server/design/gamedata/monster_hunt.go similarity index 100% rename from go/internal/server/gamedata/monster_hunt.go rename to go/internal/server/design/gamedata/monster_hunt.go diff --git a/go/internal/server/gamedata/npc_reputation.go b/go/internal/server/design/gamedata/npc_reputation.go similarity index 100% rename from go/internal/server/gamedata/npc_reputation.go rename to go/internal/server/design/gamedata/npc_reputation.go diff --git a/go/internal/server/gamedata/npc_shop.go b/go/internal/server/design/gamedata/npc_shop.go similarity index 100% rename from go/internal/server/gamedata/npc_shop.go rename to go/internal/server/design/gamedata/npc_shop.go diff --git a/go/internal/server/gamedata/overwhelm.go b/go/internal/server/design/gamedata/overwhelm.go similarity index 100% rename from go/internal/server/gamedata/overwhelm.go rename to go/internal/server/design/gamedata/overwhelm.go diff --git a/go/internal/server/gamedata/pack_detail.go b/go/internal/server/design/gamedata/pack_detail.go similarity index 59% rename from go/internal/server/gamedata/pack_detail.go rename to go/internal/server/design/gamedata/pack_detail.go index dfca2cc..4bd54bc 100644 --- a/go/internal/server/gamedata/pack_detail.go +++ b/go/internal/server/design/gamedata/pack_detail.go @@ -7,7 +7,7 @@ import ( // PackDetailDesign identifies the monsters whose state is consumed by the // client's PackMapRewardInfo. Ordinary story battle rewards are excluded. -type PackDetailDesign struct{ RegenMonsterIDs []int } +type PackDetailDesign struct{ RegenMonsterIDs, MapIDs []int } func LoadPackDetailDesign(root, version string, packID int) (PackDetailDesign, error) { if packID <= 0 { @@ -18,7 +18,42 @@ func LoadPackDetailDesign(root, version string, packID int) (PackDetailDesign, e return PackDetailDesign{}, err } defer release() - return loadPackDetailDesign(db) + design, err := loadPackDetailDesign(db) + if err != nil { + return design, err + } + placements, err := LoadFieldMonsterMaps(root, version, packID) + if err != nil { + return design, err + } + placed := design.RegenMonsterIDs[:0] + for _, id := range design.RegenMonsterIDs { + if len(placements[id]) > 0 { + placed = append(placed, id) + } + } + design.RegenMonsterIDs = placed + common, closeCommon, err := openStatDatabase(root, version) + if err != nil { + return design, err + } + defer closeCommon() + rows, err := common.Query("SELECT id FROM MapTable WHERE packId=? ORDER BY id", packID) + if err != nil { + return design, err + } + defer func() { _ = rows.Close() }() + for rows.Next() { + var mapID int + if err := rows.Scan(&mapID); err != nil { + return design, err + } + design.MapIDs = append(design.MapIDs, mapID) + } + if err := rows.Err(); err != nil { + return design, err + } + return design, nil } func loadPackDetailDesign(db *sql.DB) (PackDetailDesign, error) { diff --git a/go/internal/server/gamedata/pack_jam.go b/go/internal/server/design/gamedata/pack_jam.go similarity index 100% rename from go/internal/server/gamedata/pack_jam.go rename to go/internal/server/design/gamedata/pack_jam.go diff --git a/go/internal/server/gamedata/pack_recovery.go b/go/internal/server/design/gamedata/pack_recovery.go similarity index 100% rename from go/internal/server/gamedata/pack_recovery.go rename to go/internal/server/design/gamedata/pack_recovery.go diff --git a/go/internal/server/gamedata/pack_summary.go b/go/internal/server/design/gamedata/pack_summary.go similarity index 100% rename from go/internal/server/gamedata/pack_summary.go rename to go/internal/server/design/gamedata/pack_summary.go diff --git a/go/internal/server/gamedata/pack_transition.go b/go/internal/server/design/gamedata/pack_transition.go similarity index 100% rename from go/internal/server/gamedata/pack_transition.go rename to go/internal/server/design/gamedata/pack_transition.go diff --git a/go/internal/server/gamedata/pictorial.go b/go/internal/server/design/gamedata/pictorial.go similarity index 100% rename from go/internal/server/gamedata/pictorial.go rename to go/internal/server/design/gamedata/pictorial.go diff --git a/go/internal/server/gamedata/preset.go b/go/internal/server/design/gamedata/preset.go similarity index 100% rename from go/internal/server/gamedata/preset.go rename to go/internal/server/design/gamedata/preset.go diff --git a/go/internal/server/gamedata/promotion.go b/go/internal/server/design/gamedata/promotion.go similarity index 100% rename from go/internal/server/gamedata/promotion.go rename to go/internal/server/design/gamedata/promotion.go diff --git a/go/internal/server/gamedata/quest_costume.go b/go/internal/server/design/gamedata/quest_costume.go similarity index 100% rename from go/internal/server/gamedata/quest_costume.go rename to go/internal/server/design/gamedata/quest_costume.go diff --git a/go/internal/server/gamedata/quest_difficulty.go b/go/internal/server/design/gamedata/quest_difficulty.go similarity index 100% rename from go/internal/server/gamedata/quest_difficulty.go rename to go/internal/server/design/gamedata/quest_difficulty.go diff --git a/go/internal/server/gamedata/quest_formation.go b/go/internal/server/design/gamedata/quest_formation.go similarity index 100% rename from go/internal/server/gamedata/quest_formation.go rename to go/internal/server/design/gamedata/quest_formation.go diff --git a/go/internal/server/gamedata/quests.go b/go/internal/server/design/gamedata/quests.go similarity index 100% rename from go/internal/server/gamedata/quests.go rename to go/internal/server/design/gamedata/quests.go diff --git a/go/internal/server/gamedata/random_box.go b/go/internal/server/design/gamedata/random_box.go similarity index 100% rename from go/internal/server/gamedata/random_box.go rename to go/internal/server/design/gamedata/random_box.go diff --git a/go/internal/server/gamedata/recruit.go b/go/internal/server/design/gamedata/recruit.go similarity index 100% rename from go/internal/server/gamedata/recruit.go rename to go/internal/server/design/gamedata/recruit.go diff --git a/go/internal/server/gamedata/recruit_npc.go b/go/internal/server/design/gamedata/recruit_npc.go similarity index 100% rename from go/internal/server/gamedata/recruit_npc.go rename to go/internal/server/design/gamedata/recruit_npc.go diff --git a/go/internal/server/gamedata/reward_costume.go b/go/internal/server/design/gamedata/reward_costume.go similarity index 100% rename from go/internal/server/gamedata/reward_costume.go rename to go/internal/server/design/gamedata/reward_costume.go diff --git a/go/internal/server/gamedata/reward_equipment.go b/go/internal/server/design/gamedata/reward_equipment.go similarity index 100% rename from go/internal/server/gamedata/reward_equipment.go rename to go/internal/server/design/gamedata/reward_equipment.go diff --git a/go/internal/server/gamedata/reward_graph.go b/go/internal/server/design/gamedata/reward_graph.go similarity index 100% rename from go/internal/server/gamedata/reward_graph.go rename to go/internal/server/design/gamedata/reward_graph.go diff --git a/go/internal/server/gamedata/reward_skin.go b/go/internal/server/design/gamedata/reward_skin.go similarity index 100% rename from go/internal/server/gamedata/reward_skin.go rename to go/internal/server/design/gamedata/reward_skin.go diff --git a/go/internal/server/design/gamedata/source.go b/go/internal/server/design/gamedata/source.go new file mode 100644 index 0000000..bc0224d --- /dev/null +++ b/go/internal/server/design/gamedata/source.go @@ -0,0 +1,115 @@ +package gamedata + +import "sync" + +type designEntry[V any] struct { + once sync.Once + value V + err error +} +type designIndex[K comparable, V any] struct { + mu sync.Mutex + entries map[K]*designEntry[V] +} + +func (i *designIndex[K, V]) get(key K, load func() (V, error)) (V, error) { + i.mu.Lock() + if i.entries == nil { + i.entries = make(map[K]*designEntry[V]) + } + entry := i.entries[key] + if entry == nil { + entry = &designEntry[V]{} + i.entries[key] = entry + } + i.mu.Unlock() + entry.once.Do(func() { entry.value, entry.err = load() }) + if entry.err != nil { + i.mu.Lock() + if i.entries[key] == entry { + delete(i.entries, key) + } + i.mu.Unlock() + } + return entry.value, entry.err +} + +// Source owns version-scoped immutable rules shared by all player instances. +// Callers must copy a rule before changing any map, slice or pointed-to value. +type Source struct { + root, version string + fieldObjects designIndex[int, FieldObjectDesign] + fieldResearch designIndex[int, FieldResearchDesign] + packDetail designIndex[int, PackDetailDesign] + fieldMonsters designIndex[int, []FieldMonsterDesign] + fieldMonsterMaps designIndex[int, map[int][]int] + fieldTraps designIndex[int, FieldTrapDesign] + nPCReputation designIndex[int, NPCReputationDesign] + inns designIndex[int, []InnRule] + waypoint designIndex[uint64, WaypointPack] + huntingPack designIndex[int, *HuntingPack] + monsterHunt designIndex[uint64, *MonsterHunt] + dispatch designIndex[[2]uint64, *DispatchDesign] + eventGames designIndex[[2]uint64, *EventGame] + eventFields designIndex[uint64, *EventField] + overwhelm designIndex[[2]int, OverwhelmQuestRule] + recruit designIndex[[2]uint64, RecruitNPC] + battleRewards designIndex[[2]uint64, []BattleReward] +} + +func NewSource(root, version string) *Source { return &Source{root: root, version: version} } +func (s *Source) FieldObjects(key int) (FieldObjectDesign, error) { + return s.fieldObjects.get(key, func() (FieldObjectDesign, error) { return LoadFieldObjects(s.root, s.version, key) }) +} +func (s *Source) FieldResearch(key int) (FieldResearchDesign, error) { + return s.fieldResearch.get(key, func() (FieldResearchDesign, error) { return LoadFieldResearch(s.root, s.version, key) }) +} +func (s *Source) PackDetail(key int) (PackDetailDesign, error) { + return s.packDetail.get(key, func() (PackDetailDesign, error) { return LoadPackDetailDesign(s.root, s.version, key) }) +} +func (s *Source) FieldMonsters(key int) ([]FieldMonsterDesign, error) { + return s.fieldMonsters.get(key, func() ([]FieldMonsterDesign, error) { return LoadFieldMonsters(s.root, s.version, key) }) +} +func (s *Source) FieldMonsterMaps(key int) (map[int][]int, error) { + return s.fieldMonsterMaps.get(key, func() (map[int][]int, error) { return LoadFieldMonsterMaps(s.root, s.version, key) }) +} +func (s *Source) FieldTraps(key int) (FieldTrapDesign, error) { + return s.fieldTraps.get(key, func() (FieldTrapDesign, error) { return LoadFieldTraps(s.root, s.version, key) }) +} +func (s *Source) NPCReputation(key int) (NPCReputationDesign, error) { + return s.nPCReputation.get(key, func() (NPCReputationDesign, error) { return LoadNPCReputation(s.root, s.version, key) }) +} +func (s *Source) Inns(key int) ([]InnRule, error) { + return s.inns.get(key, func() ([]InnRule, error) { return LoadInns(s.root, s.version, key) }) +} +func (s *Source) Waypoint(key uint64) (WaypointPack, error) { + return s.waypoint.get(key, func() (WaypointPack, error) { return LoadWaypointPack(s.root, s.version, key) }) +} +func (s *Source) HuntingPack(key int) (*HuntingPack, error) { + return s.huntingPack.get(key, func() (*HuntingPack, error) { return LoadHuntingPack(s.root, s.version, key) }) +} +func (s *Source) MonsterHunt(key uint64) (*MonsterHunt, error) { + return s.monsterHunt.get(key, func() (*MonsterHunt, error) { return LoadMonsterHunt(s.root, s.version, key) }) +} + +func (s *Source) DispatchDesign(group, id uint64) (*DispatchDesign, error) { + return s.dispatch.get([2]uint64{group, id}, func() (*DispatchDesign, error) { return LoadDispatchDesign(s.root, s.version, group, id) }) +} +func (s *Source) EventGame(kind, id uint64) (*EventGame, error) { + return s.eventGames.get([2]uint64{kind, id}, func() (*EventGame, error) { return LoadEventGame(s.root, s.version, kind, id) }) +} +func (s *Source) EventField(catalog *EventPlayCatalog, id uint64) (*EventField, error) { + return s.eventFields.get(id, func() (*EventField, error) { return catalog.Field(id) }) +} +func (s *Source) OverwhelmQuest(pack, quest int) (OverwhelmQuestRule, error) { + return s.overwhelm.get([2]int{pack, quest}, func() (OverwhelmQuestRule, error) { return LoadOverwhelmQuest(s.root, s.version, pack, quest) }) +} +func (s *Source) RecruitNPC(pack int, npc uint64) (RecruitNPC, error) { + return s.recruit.get([2]uint64{uint64(pack), npc}, func() (RecruitNPC, error) { return LoadRecruitNPC(s.root, s.version, pack, npc) }) +} +func (s *Source) BattleRewards(pack int, deck uint64) ([]BattleReward, error) { + return s.battleRewards.get([2]uint64{uint64(pack), deck}, func() ([]BattleReward, error) { return BattleDeckRewards(s.root, s.version, pack, deck) }) +} + +func (s *Source) Root() string { return s.root } +func (s *Source) Version() string { return s.version } diff --git a/go/internal/server/gamedata/stat_design.go b/go/internal/server/design/gamedata/stat_design.go similarity index 99% rename from go/internal/server/gamedata/stat_design.go rename to go/internal/server/design/gamedata/stat_design.go index e45ec23..f9a70a3 100644 --- a/go/internal/server/gamedata/stat_design.go +++ b/go/internal/server/design/gamedata/stat_design.go @@ -6,7 +6,7 @@ import ( "fmt" "math" - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" _ "modernc.org/sqlite" ) diff --git a/go/internal/server/gamedata/story_catalog.go b/go/internal/server/design/gamedata/story_catalog.go similarity index 100% rename from go/internal/server/gamedata/story_catalog.go rename to go/internal/server/design/gamedata/story_catalog.go diff --git a/go/internal/server/gamedata/story_characters.go b/go/internal/server/design/gamedata/story_characters.go similarity index 100% rename from go/internal/server/gamedata/story_characters.go rename to go/internal/server/design/gamedata/story_characters.go diff --git a/go/internal/server/gamedata/talent_dispatch.go b/go/internal/server/design/gamedata/talent_dispatch.go similarity index 98% rename from go/internal/server/gamedata/talent_dispatch.go rename to go/internal/server/design/gamedata/talent_dispatch.go index 7179c8d..38d79ea 100644 --- a/go/internal/server/gamedata/talent_dispatch.go +++ b/go/internal/server/design/gamedata/talent_dispatch.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "fmt" "time" ) diff --git a/go/internal/server/gamedata/talent_growth.go b/go/internal/server/design/gamedata/talent_growth.go similarity index 100% rename from go/internal/server/gamedata/talent_growth.go rename to go/internal/server/design/gamedata/talent_growth.go diff --git a/go/internal/server/gamedata/talent_use.go b/go/internal/server/design/gamedata/talent_use.go similarity index 99% rename from go/internal/server/gamedata/talent_use.go rename to go/internal/server/design/gamedata/talent_use.go index 726b571..fbf28d2 100644 --- a/go/internal/server/gamedata/talent_use.go +++ b/go/internal/server/design/gamedata/talent_use.go @@ -1,7 +1,7 @@ package gamedata import ( - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" "database/sql" "encoding/binary" "fmt" diff --git a/go/internal/server/gamedata/today_quests.go b/go/internal/server/design/gamedata/today_quests.go similarity index 100% rename from go/internal/server/gamedata/today_quests.go rename to go/internal/server/design/gamedata/today_quests.go diff --git a/go/internal/server/gamedata/validate.go b/go/internal/server/design/gamedata/validate.go similarity index 100% rename from go/internal/server/gamedata/validate.go rename to go/internal/server/design/gamedata/validate.go diff --git a/go/internal/server/gamedata/waypoint.go b/go/internal/server/design/gamedata/waypoint.go similarity index 100% rename from go/internal/server/gamedata/waypoint.go rename to go/internal/server/design/gamedata/waypoint.go diff --git a/go/internal/server/design/gamedata/world_rules.go b/go/internal/server/design/gamedata/world_rules.go new file mode 100644 index 0000000..f39edc9 --- /dev/null +++ b/go/internal/server/design/gamedata/world_rules.go @@ -0,0 +1,90 @@ +package gamedata + +import "fmt" + +type WorldRules struct { + Story *StoryCatalog + Packs map[int]map[int]QuestDesign + Transitions map[int]PackTransition + QuestCostumes *RegularGachaCatalog + FieldPacks map[int]FieldPack + SummaryTargets map[int]bool + Jam *PackJamDesign + Difficulties map[int]map[int]bool + Characters *StoryCharacterCatalog +} + +func LoadWorldRules(root, version string, placeholder uint64) (*WorldRules, error) { + story, err := LoadStoryCatalog(root, version) + if err != nil { + return nil, err + } + if err := orderMainQuests(story); err != nil { + return nil, err + } + rules := &WorldRules{Story: story, Packs: map[int]map[int]QuestDesign{}, Transitions: map[int]PackTransition{}} + var packs []int + for id, pack := range story.Packs { + rules.Packs[id] = pack.Quests + rules.Transitions[id] = PackTransition{PackID: id, NextPackID: pack.NextPackID} + packs = append(packs, id) + } + rules.QuestCostumes, err = LoadQuestCostumes(root, version, rules.Packs, story) + if err != nil { + return nil, err + } + rules.FieldPacks, err = LoadFieldPacks(root, version) + if err != nil { + return nil, err + } + rules.SummaryTargets, err = LoadPackSummaryTargets(root, version) + if err != nil { + return nil, err + } + rules.Jam, err = LoadPackJamDesign(root, version) + if err != nil { + return nil, err + } + rules.Difficulties, err = LoadQuestDifficulties(root, version) + if err != nil { + return nil, err + } + rules.Characters, err = LoadStoryCharacterCatalog(root, version, packs, placeholder) + if err != nil { + return nil, err + } + return rules, nil +} + +// orderMainQuests follows the actual QuestTable links. Subquests never become +// a predecessor merely because their numeric ID lies between two main quests. +func orderMainQuests(story *StoryCatalog) error { + for id, pack := range story.Packs { + start := 0 + for _, qid := range pack.MainQuestIDs { + if pack.Quests[qid].PriorQuestID == 0 { + if start != 0 { + return fmt.Errorf("world: pack %d has multiple main quest roots", id) + } + start = qid + } + } + var ordered []int + seen := map[int]bool{} + for qid := start; qid != 0; { + q, exists := pack.Quests[qid] + if !exists || q.Type != 0 || seen[qid] { + return fmt.Errorf("world: invalid main quest chain pack %d quest %d", id, qid) + } + seen[qid] = true + ordered = append(ordered, qid) + qid = q.NextQuestID + } + if len(ordered) != len(pack.MainQuestIDs) { + return fmt.Errorf("world: disconnected main quest chain pack %d", id) + } + pack.MainQuestIDs = ordered + story.Packs[id] = pack + } + return nil +} diff --git a/go/internal/server/schedule/schedule.go b/go/internal/server/design/schedule/schedule.go similarity index 58% rename from go/internal/server/schedule/schedule.go rename to go/internal/server/design/schedule/schedule.go index aa2bcfb..2fb7556 100644 --- a/go/internal/server/schedule/schedule.go +++ b/go/internal/server/design/schedule/schedule.go @@ -3,11 +3,9 @@ package schedule import ( - "bd2server/internal/server/versionconfig" + "bd2server/internal/server/platform/versionconfig" "errors" "math" - - "bd2server/internal/server/wire" ) // Season is the named semantic form of Proto.Net.SeasonInfo. @@ -33,7 +31,7 @@ type RegularSeason struct { // Service contains the configured regular-content calendar for this fixed // client/GameData version. It is server configuration, not a captured packet: -// Handle encodes every protobuf field from these named values. +// Values remain immutable after validation. type Service struct { Version string `json:"version"` CalculateMilliseconds uint64 `json:"calculate_milliseconds"` @@ -66,47 +64,3 @@ func (s *Service) Validate() error { } return nil } - -func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) { - if path != "/ScheduleInfo" { - return 0, nil, false, nil - } - seq, found, err := wire.Varint(request, 1) - if err != nil || !found || seq == 0 { - return 0, nil, true, errors.New("schedule: invalid request sequence") - } - if s == nil || s.CalculateMilliseconds == 0 || len(s.Contents) == 0 { - return 0, nil, true, errors.New("schedule: unavailable calendar") - } - response := wire.AppendVarint(nil, 1, s.CalculateMilliseconds) - for _, content := range s.Contents { - entry := wire.AppendVarint(nil, 1, content.ID) - entry = wire.AppendBytes(entry, 2, encodeSeason(content.Current)) - entry = wire.AppendBytes(entry, 3, encodeSeason(content.Next)) - response = wire.AppendBytes(response, 2, entry) - } - for _, regular := range s.Regular { - entry := wire.AppendVarint(nil, 1, regular.ContentID) - if regular.Season != 0 { - entry = wire.AppendVarint(entry, 2, regular.Season) - } - response = wire.AppendBytes(response, 3, entry) - } - return 117, response, true, nil -} - -func encodeSeason(season Season) []byte { - result := wire.AppendVarint(nil, 1, season.ID) - result = wire.AppendVarint(result, 2, season.StartMilliseconds) - result = wire.AppendVarint(result, 3, season.EndMilliseconds) - if season.Error { - result = wire.AppendVarint(result, 4, 1) - } - if season.Return { - result = wire.AppendVarint(result, 5, 1) - } - if season.RankRewardGroupID != 0 { - result = wire.AppendVarint(result, 6, season.RankRewardGroupID) - } - return result -} diff --git a/go/internal/server/domain/battle/battle.go b/go/internal/server/domain/battle/battle.go new file mode 100644 index 0000000..0733c03 --- /dev/null +++ b/go/internal/server/domain/battle/battle.go @@ -0,0 +1,220 @@ +// Package battle implements a deterministic local battle session. The client +// remains authoritative for turn simulation; the server validates sequencing +// and echoes the submitted combat state without capture replay. +package battle + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" +) + +type Service struct { + states map[string]*battleState + transientVersion uint64 + + gameDataRoot string + gameDataVersion string + currentPack func(command.Context) (int, error) + currentDifficulty func(command.Context) (uint64, error) + loadDifficultyDeck func(string, string, int, uint64, uint64, uint64) (gamedata.QuestBattleDeck, error) + validateQuest func(command.Context, int, []uint64) error + grantRewards func(command.Context, string, []gamedata.Reward) ([]byte, error) + loadRewards func(string, string, int, uint64) ([]gamedata.BattleReward, error) + loadPhases func(string, string, int, uint64, uint64) ([]gamedata.BattlePhase, error) + buffs func(command.Context) ([]gamedata.PictorialBuffStat, error) + onTutorialWin func(ctx command.Context) error + onMonsterWin func(ctx command.Context) error + commitHealth func(ctx command.Context, _ map[uint64]uint64) error + hunting HuntingRuntime + monsterHunt MonsterHuntRuntime + eventBattles []EventBattleRuntime + fieldMonsters FieldMonsterRuntime + consumeFieldBuff func(ctx command.Context, _ string) error +} +type FieldMonsterRuntime interface { + BeginFieldMonsterBattle(ctx command.Context, _ int, _ uint64, _ uint64) (string, bool, error) + CompleteFieldMonsterBattle(ctx command.Context, _ int, _ uint64, _ string) ([]byte, error) +} + +func (s *Service) AttachFieldMonsters(runtime FieldMonsterRuntime) { s.fieldMonsters = runtime } + +func (s *Service) AttachFieldBuffConsume(consume func(ctx command.Context, _ string) error) { + s.consumeFieldBuff = consume +} + +// EventBattleRuntime owns event stage eligibility, costs and settlement while +// the normal battle service transports the client's turn simulation. +type EventBattleRuntime interface { + HandlesBattle(mode uint64) bool + EnterBattle(ctx command.Context, request []byte, receipt string) ([]byte, error) + CompleteBattle(ctx command.Context, request []byte, receipt string) ([]byte, error) +} + +func (s *Service) AttachEventBattle(runtime EventBattleRuntime) { + if runtime != nil { + s.eventBattles = append(s.eventBattles, runtime) + } +} + +func (s *Service) eventBattle(mode uint64) EventBattleRuntime { + for _, runtime := range s.eventBattles { + if runtime.HandlesBattle(mode) { + return runtime + } + } + return nil +} + +type MonsterHuntRuntime interface { + EnterBattle(ctx command.Context, request []byte, receipt string) ([]byte, error) + CompleteBattle(ctx command.Context, request []byte, receipt string) ([]byte, error) +} + +func (s *Service) AttachMonsterHunt(runtime MonsterHuntRuntime) { + s.monsterHunt = runtime +} + +func isMonsterHunt(mode uint64) bool { return mode == 8 || mode == 24 } + +// HuntingRuntime validates the active hunting ground and settles each won +// encounter with the account's persistent AP, monsters and reward ledger. +type HuntingRuntime interface { + ValidateBattle(ctx command.Context, pack int, mode, monster, deck uint64) error + CompleteBattle(ctx command.Context, pack int, mode, monster, deck uint64, receipt string) ([]byte, [][]byte, error) +} + +const huntingGroundMode = 5 + +func (s *Service) AttachHunting(runtime HuntingRuntime) { + s.hunting = runtime +} + +type battleState struct { + entered bool + index uint64 + round uint64 + monster uint64 + deck uint64 + pack int + mode uint64 + enterReceipt string + fieldInstance string + initialBlue [][]byte + phases []gamedata.BattlePhase + phase int + phaseStarted bool + phaseSeq uint64 + phaseReply []byte + endSeq uint64 + endRequest []byte + endReply []byte +} + +// BeginLogin creates the transient battle state owned by this login identity. +// Persistent rewards require a successful BattleEnd command transaction. +func (s *Service) BeginLogin(ctx command.Context) { + id := ctx.SessionID + + if id == "" { + return + } + if s.states == nil { + s.states = make(map[string]*battleState) + } + if s.states[id] == nil { + s.transientVersion++ + if len(s.states) >= 1024 { + for key := range s.states { + if key != id { + delete(s.states, key) + break + } + } + } + s.states[id] = &battleState{} + } +} + +func (s *Service) TransientVersion() uint64 { + + return s.transientVersion +} + +func (s *Service) AttachMonsterWinMission(callback func(ctx command.Context) error) { + s.onMonsterWin = callback +} + +func (s *Service) AttachTutorialWin(callback func(ctx command.Context) error) { + s.onTutorialWin = callback +} + +// AttachCommittedHealth persists only completed battle results. Round state +// remains transient, so reconnect rolls back an unfinished battle. +// The callback may normalize values to field HP; the response uses those +// committed values rather than echoing battle-only HP buffs. +func (s *Service) AttachCommittedHealth(callback func(ctx command.Context, _ map[uint64]uint64) error) { + s.commitHealth = callback +} + +func (s *Service) Active(ctx command.Context) bool { + + return s.stateLocked(ctx).entered +} + +func NewService(gameDataRoot, gameDataVersion string, currentPack func(command.Context) (int, error)) *Service { + return &Service{ + gameDataRoot: gameDataRoot, gameDataVersion: gameDataVersion, + currentPack: currentPack, loadRewards: gamedata.BattleDeckRewards, + states: make(map[string]*battleState), + } +} + +func (s *Service) AttachRewards(grant func(command.Context, string, []gamedata.Reward) ([]byte, error)) { + s.grantRewards = grant +} + +func (s *Service) AttachQuestBattleValidation(validate func(command.Context, int, []uint64) error) { + s.validateQuest = validate +} + +// AttachCurrentDifficulty selects the GameData quest deck for the active pack. +func (s *Service) AttachCurrentDifficulty(resolve func(command.Context) (uint64, error)) { + s.currentDifficulty = resolve +} + +func (s *Service) AttachPictorialBuffs(buffs func(command.Context) ([]gamedata.PictorialBuffStat, error)) { + s.buffs = buffs +} + +// The local engine does not simulate combat. With verification disabled +// and no battle_result the client explicitly preserves its blue team; +// fabricating a verified result would overwrite HP, SP and action state. +// Dynamic official turn/SP values are not available in this request. + +// Packet code 142 follows BattleVerify(141). State 3 is the protocol's +// explicit SUCCESS value; PVE does not require authoritative team lists. + +// The local engine is the normal deterministic engine. + +// Retry requests carry the current deck; the response must restore +// the first phase's deck, as PhaseBattleManager.ApplyRetryResponse does. + +//nolint:staticcheck // QF1003 + +// Stable per-battle/round seed; reproducible across retries. + +// Monster Hunt owns its remaining HP, progression and daily/season +// rewards. Field HP and ordinary pack rewards must not settle here. + +func (s *Service) stateLocked(ctx command.Context) *battleState { + if s.states == nil { + s.states = make(map[string]*battleState) + } + key := ctx.SessionID + state := s.states[key] + if state == nil { + state = &battleState{} + s.states[key] = state + } + return state +} diff --git a/go/internal/server/hunting/ap_refresh.go b/go/internal/server/domain/battle/hunting/ap_refresh.go similarity index 58% rename from go/internal/server/hunting/ap_refresh.go rename to go/internal/server/domain/battle/hunting/ap_refresh.go index bf358a8..3dd26d6 100644 --- a/go/internal/server/hunting/ap_refresh.go +++ b/go/internal/server/domain/battle/hunting/ap_refresh.go @@ -1,10 +1,10 @@ package hunting import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/storage/stateio" "encoding/json" "fmt" "time" @@ -16,8 +16,7 @@ type apClock struct { } func (s *Service) AttachAPRefresh(design gamedata.HuntingAPDesign) error { - s.mu.Lock() - defer s.mu.Unlock() + if design.Max == 0 || design.Max > 2147483647 || design.ResetSeconds < 0 || design.ResetSeconds >= 86400 { return fmt.Errorf("hunting: invalid AP refresh design") } @@ -27,7 +26,7 @@ func (s *Service) AttachAPRefresh(design gamedata.HuntingAPDesign) error { } // The client clock is UTC+9. DailyResetTime is expressed in that clock. -func (s *Service) refreshAP() error { +func (s *Service) refreshAP(ctx command.Context) error { if s.apDesign == nil { return nil } @@ -35,7 +34,7 @@ func (s *Service) refreshAP() error { offset := (s.apDesign.ResetSeconds - 9*3600) * 1000 day := (now - offset) / 86400000 nextTime := (day+1)*86400000 + offset - b, e := s.storage.Load("huntingapclock") + b, e := s.storage.Load(ctx.State, "huntingapclock") if e != nil { return e } @@ -55,7 +54,7 @@ func (s *Service) refreshAP() error { } next := s.clone() next.Free = s.apDesign.Max - if e = s.persist(next); e != nil { + if e = s.persist(ctx, next); e != nil { return e } clock.Next = nextTime @@ -64,26 +63,5 @@ func (s *Service) refreshAP() error { if e != nil { return e } - return s.storage.Save("huntingapclock", raw) -} - -func (s *Service) APChargeInfo() ([]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() - if e := s.refreshAP(); e != nil { - return nil, e - } - raw, e := s.storage.Load("huntingapclock") - if e != nil || raw == nil { - return nil, e - } - var clock apClock - if e = json.Unmarshal(raw, &clock); e != nil { - return nil, e - } - b := wire.AppendVarint(nil, 1, uint64(clock.Next-86400000)) - item := wire.AppendVarint(nil, 3, 21) - item = wire.AppendVarint(item, 4, s.state.Free) - b = wire.AppendBytes(b, 2, item) - return wire.AppendBytes(nil, 1, b), nil + return s.storage.Save(ctx.State, "huntingapclock", raw) } diff --git a/go/internal/server/domain/battle/hunting/dispatch.go b/go/internal/server/domain/battle/hunting/dispatch.go new file mode 100644 index 0000000..e5181ca --- /dev/null +++ b/go/internal/server/domain/battle/hunting/dispatch.go @@ -0,0 +1,46 @@ +package hunting + +import ( + "bd2server/internal/server/design/gamedata" + "time" +) + +type dispatchReceipt struct { + Request, Response []byte + Code int +} +type dispatchJob struct { + Group, ID, Count, Start, End, Free, Bonus uint64 + Rewards []gamedata.BattleReward + Runs [][]gamedata.BattleReward +} +type dispatchState struct { + Version string `json:"version"` + Jobs map[string]dispatchJob `json:"jobs"` + Receipts map[string]dispatchReceipt `json:"receipts"` +} + +func (s *Service) AttachDispatchEligibility(check func(*gamedata.DispatchDesign) error) { + + s.dispatchEligibility = check +} + +func dispatchPlayed(j dispatchJob, seconds uint64) uint64 { + if seconds == 0 { + return j.Count + } + now := uint64(time.Now().UnixMilli()) + if now <= j.Start { + return 0 + } + n := min((now-j.Start)/(seconds*1000), j.Count) + return n +} + +func dispatchCompletedRewards(j dispatchJob, n uint64) []gamedata.BattleReward { + var out []gamedata.BattleReward + for i := uint64(0); i < n && i < uint64(len(j.Runs)); i++ { + out = append(out, j.Runs[i]...) + } + return out +} diff --git a/go/internal/server/domain/battle/hunting/hunting.go b/go/internal/server/domain/battle/hunting/hunting.go new file mode 100644 index 0000000..7ff040d --- /dev/null +++ b/go/internal/server/domain/battle/hunting/hunting.go @@ -0,0 +1,210 @@ +// Package hunting owns ordinary HuntingGround progress and repeatable rewards. +package hunting + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "fmt" + "maps" + "math" + "slices" + "strconv" + + "time" +) + +const BattleMode uint64 = 5 + +type packState struct { + Current, Highest uint64 + Auto bool + Defeated []uint64 +} +type snapshot struct { + Version string `json:"version"` + Free uint64 `json:"free"` + Bonus uint64 `json:"bonus"` + Packs map[string]packState `json:"packs"` + Receipts map[string]bool `json:"receipts"` +} + +// Rules supplies shared immutable hunting and dispatch designs. +type Rules interface { + HuntingPack(int) (*gamedata.HuntingPack, error) + DispatchDesign(uint64, uint64) (*gamedata.DispatchDesign, error) +} + +type Service struct { + storage stateio.Store + inventory *assets.Inventory + wallet *assets.Wallet + state snapshot + currentPack func(command.Context) (int, error) + load func(int) (*gamedata.HuntingPack, error) + dispatchLoad func(uint64, uint64) (*gamedata.DispatchDesign, error) + dispatchEligibility func(*gamedata.DispatchDesign) error + eligibility func(ctx command.Context, _ int, _ uint64) error + grant func(command.Context, string, []gamedata.Reward) ([]byte, error) + apDesign *gamedata.HuntingAPDesign + now func() time.Time +} + +func Open(ctx command.Context, store stateio.Store, rules Rules, inventory *assets.Inventory, wallet *assets.Wallet, currentPack func(command.Context) (int, error), free, bonus uint64) (*Service, error) { + if store == nil || rules == nil || inventory == nil || wallet == nil || currentPack == nil { + return nil, fmt.Errorf("hunting: invalid configuration") + } + s := &Service{storage: store, inventory: inventory, wallet: wallet, currentPack: currentPack, state: snapshot{versionconfig.State(), free, bonus, map[string]packState{}, map[string]bool{}}} + s.dispatchLoad = rules.DispatchDesign + s.load = rules.HuntingPack + b, err := store.Load(ctx.State, "hunting") + if err != nil { + return nil, err + } + if b != nil { + if err := stateio.RequireExactJSONObject(b, "version", "free", "bonus", "packs", "receipts"); err != nil { + return nil, err + } + if err := json.Unmarshal(b, &s.state); err != nil { + return nil, err + } + if s.state.Version != versionconfig.State() || s.state.Packs == nil || s.state.Receipts == nil { + return nil, fmt.Errorf("hunting: incompatible state") + } + } + if s.state.Free > math.MaxInt32 || s.state.Bonus > math.MaxInt32 { + return nil, fmt.Errorf("hunting: AP exceeds protocol range") + } + for key, st := range s.state.Packs { + pack, err := strconv.Atoi(key) + if err != nil || pack <= 0 || strconv.Itoa(pack) != key { + return nil, fmt.Errorf("hunting: invalid saved pack") + } + d, err := s.load(pack) + if err != nil { + return nil, err + } + if _, ok := ground(d, st.Current); !ok { + return nil, fmt.Errorf("hunting: saved current ground missing from GameData") + } + if st.Highest != 0 { + if _, ok := ground(d, st.Highest); !ok { + return nil, fmt.Errorf("hunting: saved highest ground missing from GameData") + } + } + g, _ := ground(d, st.Current) + seen := map[uint64]bool{} + for _, id := range st.Defeated { + member := id == g.BossID + for _, candidate := range g.Monsters { + member = member || id == candidate + } + if !member || seen[id] { + return nil, fmt.Errorf("hunting: invalid defeated monster") + } + seen[id] = true + } + } + for key, claimed := range s.state.Receipts { + if key == "" || !claimed { + return nil, fmt.Errorf("hunting: invalid receipt") + } + } + return s, nil +} +func (s *Service) HuntingAP(ctx command.Context) (uint64, uint64, error) { + + if e := s.refreshAP(ctx); e != nil { + return 0, 0, e + } + return s.state.Free, s.state.Bonus, nil +} + +func ground(d *gamedata.HuntingPack, id uint64) (gamedata.HuntingGround, bool) { + for _, g := range d.Grounds { + if g.ID == id { + return g, true + } + } + return gamedata.HuntingGround{}, false +} + +func (s *Service) ValidateBattle(ctx command.Context, pack int, mode, monster, deck uint64) error { + if mode != BattleMode { + return nil + } + + if e := s.refreshAP(ctx); e != nil { + return e + } + _, _, _, err := s.validate(pack, monster, deck) + return err +} +func (s *Service) validate(pack int, monster, deck uint64) (*gamedata.HuntingPack, gamedata.HuntingGround, gamedata.HuntingMonster, error) { + d, err := s.load(pack) + if err != nil { + return nil, gamedata.HuntingGround{}, gamedata.HuntingMonster{}, err + } + st := s.state.Packs[strconv.Itoa(pack)] + g, ok := ground(d, st.Current) + if !ok { + return nil, g, gamedata.HuntingMonster{}, fmt.Errorf("hunting: battle before ground enter") + } + member := monster == g.BossID + for _, id := range g.Monsters { + member = member || id == monster + } + m := d.Monsters[monster] + if monster == g.BossID { + for _, id := range g.Monsters { + done := false + for _, dead := range st.Defeated { + done = done || dead == id + } + if !done { + return nil, g, m, fmt.Errorf("hunting: boss locked until ordinary monsters are defeated") + } + } + } + if slices.Contains(st.Defeated, monster) { + return nil, g, m, fmt.Errorf("hunting: defeated monster requires reentry") + } + known := false + for _, id := range m.Decks { + known = known || id == deck + } + if !member || !known { + return nil, g, m, fmt.Errorf("hunting: monster/deck does not belong to selected ground") + } + cost := d.NormalAP + if m.Type == 1 { + cost = d.BossAP + } + if s.state.Free < cost && s.state.Bonus < cost-s.state.Free { + return nil, g, m, fmt.Errorf("hunting: insufficient hunting AP") + } + return d, g, m, nil +} + +func (s *Service) clone() snapshot { + next := s.state + next.Packs = map[string]packState{} + maps.Copy(next.Packs, s.state.Packs) + next.Receipts = map[string]bool{} + maps.Copy(next.Receipts, s.state.Receipts) + return next +} +func (s *Service) persist(ctx command.Context, next snapshot) error { + b, err := json.Marshal(next) + if err != nil { + return err + } + if err := s.storage.Save(ctx.State, "hunting", b); err != nil { + return err + } + s.state = next + return nil +} diff --git a/go/internal/server/domain/battle/hunting/progress.go b/go/internal/server/domain/battle/hunting/progress.go new file mode 100644 index 0000000..130967e --- /dev/null +++ b/go/internal/server/domain/battle/hunting/progress.go @@ -0,0 +1,157 @@ +package hunting + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/storage/stateio" + "bytes" + "encoding/json" + "fmt" +) + +type battleReceipt struct { + Fingerprint string + Bundle []byte + Monsters [][]byte +} +type battleLedger struct { + Version string `json:"version"` + Receipts map[string]battleReceipt `json:"receipts"` +} + +func (s *Service) loadBattleReceipts(ctx command.Context) (map[string]battleReceipt, error) { + b, e := s.storage.Load(ctx.State, "huntingbattle") + if e != nil { + return nil, e + } + if b == nil { + return map[string]battleReceipt{}, nil + } + if e = stateio.RequireExactJSONObject(b, "version", "receipts"); e != nil { + return nil, e + } + var l battleLedger + d := json.NewDecoder(bytes.NewReader(b)) + d.DisallowUnknownFields() + if e = d.Decode(&l); e != nil { + return nil, e + } + if l.Version != versionconfig.State() || l.Receipts == nil { + return nil, fmt.Errorf("hunting: invalid battle ledger") + } + return l.Receipts, nil +} +func (s *Service) saveBattleReceipts(ctx command.Context, rs map[string]battleReceipt) error { + b, e := json.Marshal(battleLedger{versionconfig.State(), rs}) + if e != nil { + return e + } + return s.storage.Save(ctx.State, "huntingbattle", b) +} +func (s *Service) AttachEligibility(check func(ctx command.Context, _ int, _ uint64) error) { + + s.eligibility = check +} +func (s *Service) AttachRewards(grant func(command.Context, string, []gamedata.Reward) ([]byte, error)) { + + s.grant = grant +} + +// EnsureForPack initializes the ordinary stage when the world loads a pack. + +// Ground mutations have their own persisted session+sequence replay ledger. + +func (s *Service) reload(ctx command.Context) error { + b, e := s.storage.Load(ctx.State, "hunting") + if e != nil { + return e + } + if b == nil { + return nil + } + var st snapshot + if e = json.Unmarshal(b, &st); e != nil { + return e + } + s.state = st + return nil +} +func (s *Service) CanExchangeAP(ctx command.Context, costs, rewards []gamedata.Reward) error { + + if e := s.reload(ctx); e != nil { + return e + } + if e := s.refreshAP(ctx); e != nil { + return e + } + _, e := s.exchangeAP(costs, rewards) + return e +} +func (s *Service) exchangeAP(costs, rewards []gamedata.Reward) (snapshot, error) { + st := s.clone() + apply := func(rs []gamedata.Reward, cost bool) error { + for _, r := range rs { + if (r.Type != 21 && r.Type != 23) || r.ID != 0 || r.Count == 0 { + return fmt.Errorf("hunting: invalid AP exchange") + } + p := &st.Free + if r.Type == 23 { + p = &st.Bonus + } + if cost { + if *p < r.Count { + return fmt.Errorf("hunting: insufficient AP") + } + *p -= r.Count + } else { + if r.Count > 2147483647-*p { + return fmt.Errorf("hunting: AP overflow") + } + *p += r.Count + } + } + return nil + } + if e := apply(costs, true); e != nil { + return st, e + } + return st, apply(rewards, false) +} +func (s *Service) ExchangeAPOnce(ctx command.Context, identity string, costs, rewards []gamedata.Reward) error { + + if identity == "" { + return fmt.Errorf("hunting: missing AP identity") + } + if e := s.reload(ctx); e != nil { + return e + } + key := "apexchange:" + identity + fpb, e := json.Marshal([][]gamedata.Reward{costs, rewards}) + if e != nil { + return e + } + rs, e := s.loadBattleReceipts(ctx) + if e != nil { + return e + } + fp := string(fpb) + if r, ok := rs[key]; ok { + if r.Fingerprint != fp { + return fmt.Errorf("hunting: AP identity conflict") + } + return nil + } + if e := s.refreshAP(ctx); e != nil { + return e + } + st, e := s.exchangeAP(costs, rewards) + if e != nil { + return e + } + if e = s.persist(ctx, st); e != nil { + return e + } + rs[key] = battleReceipt{Fingerprint: fp} + return s.saveBattleReceipts(ctx, rs) +} diff --git a/go/internal/server/domain/battle/hunting/protocol_adapter.go b/go/internal/server/domain/battle/hunting/protocol_adapter.go new file mode 100644 index 0000000..adf2859 --- /dev/null +++ b/go/internal/server/domain/battle/hunting/protocol_adapter.go @@ -0,0 +1,770 @@ +package hunting + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/protocol/wire" + "bd2server/internal/server/storage/stateio" + "bytes" + "crypto/rand" + "encoding/binary" + "encoding/json" + "fmt" + "maps" + "math" + "math/big" + "sort" + "strconv" + "strings" + "time" +) + +func (s *Service) EnsureForPack(ctx command.Context, pack int) ([]byte, error) { + + d, e := s.load(pack) + if e != nil { + return nil, e + } + if len(d.Grounds) == 0 { + return nil, nil + } + st := s.state.Packs[strconv.Itoa(pack)] + if st.Current == 0 { + st.Current = d.Grounds[0].ID + next := s.clone() + next.Packs[strconv.Itoa(pack)] = st + if e = s.persist(ctx, next); e != nil { + return nil, e + } + } + return s.info(pack, d) +} + +func (s *Service) SnapshotForPack(ctx command.Context, pack int) ([]byte, error) { + + d, e := s.load(pack) + if e != nil { + return nil, e + } + if len(d.Grounds) == 0 { + return nil, nil + } + return s.info(pack, d) +} + +func (s *Service) handleGroundSession(ctx command.Context, path string, req []byte, session string) (int, []byte, bool, error) { + if path != "/HuntingGroundEnter" { + return s.handleGroundBase(ctx, path, req) + } + seq, found, e := wire.Varint(req, 1) + if e != nil || !found || seq == 0 { + return 110, nil, true, fmt.Errorf("hunting: missing sequence") + } + key := "enter:" + session + ":" + strconv.FormatUint(seq, 10) + rs, e := s.loadBattleReceipts(ctx) + if e != nil { + return 110, nil, true, e + } + fp := fmt.Sprintf("%x", req) + if r, ok := rs[key]; ok { + if r.Fingerprint != fp { + return 110, nil, true, fmt.Errorf("hunting: enter sequence conflict") + } + return 110, r.Bundle, true, nil + } + code, out, ok, e := s.handleGroundBase(ctx, path, req) + if e != nil || !ok { + return code, out, ok, e + } + rs[key] = battleReceipt{Fingerprint: fp, Bundle: out} + if e = s.saveBattleReceipts(ctx, rs); e != nil { + return code, nil, true, e + } + return code, out, ok, nil +} + +// A catalog entry is not an account's active hunting run. The detail +// endpoint preserves an explicitly present empty message until entry. + +func (s *Service) info(pack int, d *gamedata.HuntingPack) ([]byte, error) { + st := s.state.Packs[strconv.Itoa(pack)] + initial := st.Current == 0 + if st.Current == 0 { + st.Current = d.Grounds[0].ID + } + g, ok := ground(d, st.Current) + if !ok { + return nil, fmt.Errorf("hunting: saved ground missing from GameData") + } + b := wire.AppendVarint(nil, 2, st.Current) + if st.Highest != 0 { + b = wire.AppendVarint(b, 3, st.Highest) + } + b = wire.AppendVarint(b, 5, uint64(pack)) + if st.Auto { + b = wire.AppendVarint(b, 1, 1) + } + var monsterInfos [][]byte + if initial { + // The initial catalog lists ordinary monsters only. A boss is not an + // active account encounter simply because its design row exists. + for _, id := range g.Monsters { + monsterInfos = append(monsterInfos, monsterWire(d.Monsters[id], true)) + } + } else { + monsterInfos = monstersWithState(d, g, st.Defeated) + } + for _, m := range monsterInfos { + b = wire.AppendBytes(b, 4, m) + } + return b, nil +} + +func monsterWire(m gamedata.HuntingMonster, active bool) []byte { + b := wire.AppendVarint(nil, 1, m.ID) + b = wire.AppendVarint(b, 2, m.Decks[0]) + if active { + b = wire.AppendVarint(b, 6, 1) + } + return b +} + +func monstersWithState(d *gamedata.HuntingPack, g gamedata.HuntingGround, defeated []uint64) [][]byte { + dead := map[uint64]bool{} + for _, id := range defeated { + dead[id] = true + } + boss := len(g.Monsters) == 0 + all := true + for _, id := range g.Monsters { + all = all && dead[id] + } + boss = boss || all + out := make([][]byte, 0, len(g.Monsters)+1) + for _, id := range g.Monsters { + out = append(out, monsterWire(d.Monsters[id], !dead[id])) + } + if boss { + out = append(out, monsterWire(d.Monsters[g.BossID], true)) + } + return out +} + +func (s *Service) CompleteBattle(ctx command.Context, pack int, mode, monster, deck uint64, receipt string) ([]byte, [][]byte, error) { + if mode != BattleMode { + return nil, nil, nil + } + + if receipt == "" { + return nil, nil, fmt.Errorf("hunting: missing battle receipt") + } + fingerprint := fmt.Sprintf("%d:%d:%d:%d", pack, mode, monster, deck) + ledger, err := s.loadBattleReceipts(ctx) + if err != nil { + return nil, nil, err + } + if saved, ok := ledger[receipt]; ok { + if saved.Fingerprint != fingerprint { + return nil, nil, fmt.Errorf("hunting: battle receipt conflict") + } + return saved.Bundle, saved.Monsters, nil + } + d, g, m, err := s.validate(pack, monster, deck) + if err != nil { + return nil, nil, err + } + rewards := m.Rewards[deck] + identity := "hunting:" + receipt + currency := make([]gamedata.Reward, 0) + stack := make([]gamedata.BattleReward, 0) + for _, r := range rewards { + if r.Type == 2 || r.Type == 3 || r.Type == 4 || r.Type == 12 || r.Type == 20 { + currency = append(currency, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 + } else { + stack = append(stack, r) + } + } + var grantedBundle []byte + var items []assets.Item + if s.grant != nil { + rs := make([]gamedata.Reward, len(rewards)) + for i, r := range rewards { + rs[i] = gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count} //nolint:staticcheck // S1016 + } + + grantedBundle, err = s.grant(ctx, identity, rs) + + if err != nil { + return nil, nil, err + } + } else { + if _, err := s.wallet.GrantQuestOnce(ctx, identity+":currency", currency); err != nil { + return nil, nil, err + } + items, err = s.inventory.GrantOnce(ctx, identity+":items", stack) + if err != nil { + return nil, nil, err + } + if len(items) == 0 { + items = s.inventory.GrantedItems(identity + ":items") + } + } + next := s.clone() + cost := d.NormalAP + if m.Type == 1 { + cost = d.BossAP + } + if next.Free >= cost { + next.Free -= cost + } else { + next.Bonus -= cost - next.Free + next.Free = 0 + } + st := next.Packs[strconv.Itoa(pack)] + if monster == g.BossID { + if st.Highest < g.ID { + st.Highest = g.ID + } + st.Defeated = nil + } else { + st.Defeated = append(append([]uint64(nil), st.Defeated...), monster) + } + next.Packs[strconv.Itoa(pack)] = st + next.Receipts[receipt] = true + var bundle []byte + if s.grant != nil { + bundle = grantedBundle + } + for _, item := range items { + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(item)) + view := wire.AppendVarint(nil, 2, item.ID) + view = wire.AppendVarint(view, 3, item.Type) + view = wire.AppendVarint(view, 4, item.Count) + bundle = wire.AppendBytes(bundle, 6, view) + } + for _, r := range currency { + if s.grant != nil { + break + } + b := wire.AppendVarint(nil, 3, r.Type) + b = wire.AppendVarint(b, 4, r.Count) + bundle = wire.AppendBytes(bundle, 1, b) + } + updates := [][]byte{monsterWire(m, false)} + if monster == g.BossID { + for _, id := range g.Monsters { + updates = append(updates, monsterWire(d.Monsters[id], true)) + } + } else if len(st.Defeated) == len(g.Monsters) { + updates = append(updates, monsterWire(d.Monsters[g.BossID], true)) + } + if err := s.persist(ctx, next); err != nil { + return nil, nil, err + } + latest, err := s.loadBattleReceipts(ctx) + if err != nil { + return nil, nil, err + } + maps.Copy(ledger, latest) + ledger[receipt] = battleReceipt{Fingerprint: fingerprint, Bundle: bundle, Monsters: updates} + if err := s.saveBattleReceipts(ctx, ledger); err != nil { + return nil, nil, err + } + return bundle, updates, nil +} + +func packed(req []byte, n int) ([]uint64, error) { + var out []uint64 + err := wire.Walk(req, func(f wire.Field) error { + if f.Number != n { + return nil + } + if f.Type == 0 { + v, _ := binary.Uvarint(f.Value) + out = append(out, v) + return nil + } + if f.Type != 2 { + return fmt.Errorf("hunting: invalid repeated integer") + } + for raw := f.Value; len(raw) > 0; { + v, n := binary.Uvarint(raw) + if n <= 0 { + return fmt.Errorf("hunting: invalid packed integer") + } + out = append(out, v) + raw = raw[n:] + } + return nil + }) + return out, err +} + +func (s *Service) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + if strings.HasPrefix(path, "/HuntDispatch") { + return s.handleDispatch(ctx, path, req, commandSession) + } + return s.handleGroundSession(ctx, path, req, commandSession) +} + +func (s *Service) handleDispatch(ctx command.Context, path string, req []byte, session string) (int, []byte, bool, error) { + code := map[string]int{"/HuntDispatchInfo": 189, "/HuntDispatchStart": 190, "/HuntDispatchEnd": 191, "/HuntDispatchRewardPreview": 194, "/HuntDispatch": 0}[path] + if path != "/HuntDispatchInfo" && path != "/HuntDispatchStart" && path != "/HuntDispatchEnd" && path != "/HuntDispatchRewardPreview" && path != "/HuntDispatch" { + return 0, nil, false, nil + } + + if e := s.refreshAP(ctx); e != nil { + return code, nil, true, e + } + seq, found, err := wire.Varint(req, 1) + if err != nil || !found || seq == 0 { + return code, nil, true, fmt.Errorf("hunting: missing sequence") + } + ds := dispatchState{Version: versionconfig.State(), Jobs: map[string]dispatchJob{}, Receipts: map[string]dispatchReceipt{}} + raw, err := s.storage.Load(ctx.State, "huntdispatch") + if err != nil { + return code, nil, true, err + } + if raw != nil { + if err = stateio.RequireExactJSONObject(raw, "version", "jobs", "receipts"); err != nil { + return code, nil, true, err + } + decoder := json.NewDecoder(bytes.NewReader(raw)) + decoder.DisallowUnknownFields() + if err = decoder.Decode(&ds); err != nil { + return code, nil, true, err + } + if ds.Version != versionconfig.State() || ds.Jobs == nil || ds.Receipts == nil { + return code, nil, true, fmt.Errorf("hunting: invalid dispatch state") + } + } + if path == "/HuntDispatchInfo" { + var out []byte + keys := make([]string, 0, len(ds.Jobs)) + for k := range ds.Jobs { + keys = append(keys, k) + } + sort.Strings(keys) + for _, k := range keys { + out = wire.AppendBytes(out, 1, dispatchJobWire(ds.Jobs[k])) + } + // A client without a remembered dispatch pack needs a default. This + // server selects the smallest cleared ordinary hunting pack. + var minimum uint64 + for key, state := range s.state.Packs { + if state.Highest == 0 { + continue + } + pack, err := strconv.ParseUint(key, 10, 32) + if err != nil || pack == 0 { + return code, nil, true, fmt.Errorf("hunting: invalid saved dispatch pack") + } + if minimum == 0 || pack < minimum { + minimum = pack + } + } + if minimum != 0 { + out = wire.AppendVarint(out, 2, minimum) + } + return code, out, true, nil + } + for k, j := range ds.Jobs { + if k != fmt.Sprintf("%d:%d", j.Group, j.ID) || j.Group == 0 || j.ID == 0 || j.Count == 0 || j.Count > 20 || j.Start == 0 || j.End < j.Start || len(j.Runs) != int(j.Count) || j.Free > 2147483647 || j.Bonus > 2147483647 { + return code, nil, true, fmt.Errorf("hunting: invalid saved dispatch job") + } + } + for k, r := range ds.Receipts { + if k == "" || len(r.Request) == 0 || r.Response == nil { + return code, nil, true, fmt.Errorf("hunting: invalid saved dispatch receipt") + } + } + key := session + ":" + strconv.FormatUint(seq, 10) + if r, ok := ds.Receipts[key]; ok { + if !bytes.Equal(r.Request, append([]byte(path), req...)) { + return code, nil, true, fmt.Errorf("hunting: sequence reused with different dispatch request") + } + return r.Code, r.Response, true, nil + } + group, _, e := wire.Varint(req, 2) + id, _, e2 := wire.Varint(req, 3) + if e != nil || e2 != nil || group == 0 || id == 0 { + return code, nil, true, fmt.Errorf("hunting: invalid dispatch ID") + } + jobKey := fmt.Sprintf("%d:%d", group, id) + job, exists := ds.Jobs[jobKey] + d, err := s.dispatchLoad(group, id) + if err != nil { + return code, nil, true, err + } + var out []byte + if path == "/HuntDispatchRewardPreview" { + if !exists { + return code, nil, true, fmt.Errorf("hunting: dispatch not started") + } + played := dispatchPlayed(job, d.ClearTime) + out = wire.AppendVarint(out, 1, played) + preview := dispatchCompletedRewards(job, played) + remaining := d.AP * (job.Count - played) + bonus := min(remaining, job.Bonus) + free := remaining - bonus + if free > 0 { + preview = append(preview, gamedata.BattleReward{Type: 21, Count: free}) + } + if bonus > 0 { + preview = append(preview, gamedata.BattleReward{Type: 23, Count: bonus}) + } + bundle := dispatchPreview(preview) + for _, r := range preview { + v := wire.AppendVarint(nil, 2, r.ID) + v = wire.AppendVarint(v, 3, r.Type) + v = wire.AppendVarint(v, 4, r.Count) + bundle = wire.AppendBytes(bundle, 1, v) + } + out = wire.AppendBytes(out, 2, bundle) + return code, out, true, nil + } + if path == "/HuntDispatchEnd" { + if !exists { + return code, nil, true, fmt.Errorf("hunting: dispatch not started") + } + played := dispatchPlayed(job, d.ClearTime) + completed := dispatchCompletedRewards(job, played) + remaining := d.AP * (job.Count - played) + refundBonus := min(remaining, job.Bonus) + refundFree := remaining - refundBonus + next := s.clone() + next.Free += refundFree + next.Bonus += refundBonus + if e := s.persist(ctx, next); e != nil { + return code, nil, true, e + } + if refundFree > 0 { + completed = append(completed, gamedata.BattleReward{Type: 21, Count: refundFree}) + } + if refundBonus > 0 { + completed = append(completed, gamedata.BattleReward{Type: 23, Count: refundBonus}) + } + bundle, e := s.dispatchGrant(ctx, "dispatch:"+key, completed) + if e != nil { + return code, nil, true, e + } + out = wire.AppendBytes(out, 1, bundle) + delete(ds.Jobs, jobKey) + } else { + count, _, e := wire.Varint(req, 4) + if e != nil || count == 0 || count > 20 { + return code, nil, true, fmt.Errorf("hunting: dispatch count must be 1..20") + } + if exists { + return code, nil, true, fmt.Errorf("hunting: dispatch already running") + } + for _, active := range ds.Jobs { + design, e := s.dispatchLoad(active.Group, active.ID) + if e != nil { + return code, nil, true, e + } + if design.TypeGroup == d.TypeGroup { + return code, nil, true, fmt.Errorf("hunting: dispatch category already running") + } + } + if s.dispatchEligibility != nil { + if e = s.dispatchEligibility(d); e != nil { + return code, nil, true, e + } + } else { + if d.TypeGroup != 0 { + return code, nil, true, fmt.Errorf("hunting: SkyWay dispatch eligibility unavailable") + } + st := s.state.Packs[strconv.Itoa(int(d.Pack))] + if st.Highest < d.GroundID { + return code, nil, true, fmt.Errorf("hunting: dispatch requires cleared difficulty") + } + } + if d.AP > uint64(2147483647)/count { + return code, nil, true, fmt.Errorf("hunting: dispatch AP overflow") + } + cost := d.AP * count + if s.state.Free < cost && s.state.Bonus < cost-s.state.Free { + return code, nil, true, fmt.Errorf("hunting: insufficient hunting AP") + } + rewards, e := d.Roll(count, func(n uint64) (uint64, error) { + v, e := rand.Int(rand.Reader, new(big.Int).SetUint64(n)) + if e != nil { + return 0, e + } + return v.Uint64(), nil + }) + if e != nil { + return code, nil, true, e + } + next := s.clone() + free := min(cost, next.Free) + bonus := cost - free + next.Free -= free + next.Bonus -= bonus + if path == "/HuntDispatchStart" { + now := uint64(time.Now().UnixMilli()) + runs := make([][]gamedata.BattleReward, count) + for i := range runs { + runs[i], e = d.Roll(1, func(n uint64) (uint64, error) { + v, e := rand.Int(rand.Reader, new(big.Int).SetUint64(n)) + if e != nil { + return 0, e + } + return v.Uint64(), nil + }) + if e != nil { + return code, nil, true, e + } + } + job = dispatchJob{Group: group, ID: id, Count: count, Start: now, End: now + d.ClearTime*count*1000, Free: free, Bonus: bonus, Rewards: rewards, Runs: runs} + ds.Jobs[jobKey] = job + out = wire.AppendBytes(out, 1, dispatchJobWire(job)) + } else { + beforeFree, beforeBonus := s.state.Free, s.state.Bonus + bundle, e := s.dispatchGrant(ctx, "dispatch:"+key, rewards) + if e != nil { + return code, nil, true, e + } + out = wire.AppendBytes(out, 1, bundle) + if s.state.Free >= beforeFree { + next.Free += s.state.Free - beforeFree + } else { + delta := beforeFree - s.state.Free + if next.Free < delta { + return code, nil, true, fmt.Errorf("hunting: AP grant conflict") + } + next.Free -= delta + } + if s.state.Bonus >= beforeBonus { + next.Bonus += s.state.Bonus - beforeBonus + } else { + delta := beforeBonus - s.state.Bonus + if next.Bonus < delta { + return code, nil, true, fmt.Errorf("hunting: AP grant conflict") + } + next.Bonus -= delta + } + } + if e = s.persist(ctx, next); e != nil { + return code, nil, true, e + } + } + ds.Receipts[key] = dispatchReceipt{append([]byte(path), req...), out, code} + b, e := json.Marshal(ds) + if e != nil { + return code, nil, true, e + } + if e = s.storage.Save(ctx.State, "huntdispatch", b); e != nil { + return code, nil, true, e + } + return code, out, true, nil +} + +func dispatchJobWire(j dispatchJob) []byte { + var out []byte + for n, v := range map[int]uint64{1: j.Group, 2: j.ID, 3: j.Count, 4: j.Start, 5: j.End, 6: j.Free, 7: j.Bonus} { + out = wire.AppendVarint(out, n, v) + } + return out +} + +func dispatchPreview(rs []gamedata.BattleReward) []byte { + var out []byte + for _, r := range rs { + v := wire.AppendVarint(nil, 2, r.ID) + v = wire.AppendVarint(v, 3, r.Type) + v = wire.AppendVarint(v, 4, r.Count) + out = wire.AppendBytes(out, 6, v) + } + return out +} + +func (s *Service) dispatchGrant(ctx command.Context, identity string, rs []gamedata.BattleReward) ([]byte, error) { + if s.grant != nil { + var rewards []gamedata.Reward + var ap []gamedata.BattleReward + for _, r := range rs { + if r.Type == 21 || r.Type == 23 { + ap = append(ap, r) + } else { + rewards = append(rewards, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 + } + } + + bundle, err := s.grant(ctx, identity, rewards) + + if err != nil { + return nil, err + } + bundle = append(bundle, dispatchPreview(ap)...) + for _, r := range ap { + v := wire.AppendVarint(nil, 3, r.Type) + v = wire.AppendVarint(v, 4, r.Count) + bundle = wire.AppendBytes(bundle, 1, v) + } + return bundle, nil + } + + var currency []gamedata.Reward + var items []gamedata.BattleReward + for _, r := range rs { + switch r.Type { + case 21, 23: + continue + case 2, 3, 4, 12, 20: + currency = append(currency, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 + default: + items = append(items, r) + } + } + if _, err := s.wallet.GrantQuestOnce(ctx, identity+":currency", currency); err != nil { + return nil, err + } + granted, err := s.inventory.GrantOnce(ctx, identity+":items", items) + if err != nil { + return nil, err + } + if len(granted) == 0 { + granted = s.inventory.GrantedItems(identity + ":items") + } + out := dispatchPreview(rs) + for _, r := range rs { + if r.Type == 21 || r.Type == 23 { + v := wire.AppendVarint(nil, 3, r.Type) + v = wire.AppendVarint(v, 4, r.Count) + out = wire.AppendBytes(out, 1, v) + } + } + for _, r := range currency { + v := wire.AppendVarint(nil, 2, r.ID) + v = wire.AppendVarint(v, 3, r.Type) + v = wire.AppendVarint(v, 4, r.Count) + out = wire.AppendBytes(out, 1, v) + } + for _, item := range granted { + out = wire.AppendBytes(out, 1, assets.ItemWire(item)) + } + return out, nil +} + +func (s *Service) APChargeInfo(ctx command.Context) ([]byte, error) { + + if e := s.refreshAP(ctx); e != nil { + return nil, e + } + raw, e := s.storage.Load(ctx.State, "huntingapclock") + if e != nil || raw == nil { + return nil, e + } + var clock apClock + if e = json.Unmarshal(raw, &clock); e != nil { + return nil, e + } + b := wire.AppendVarint(nil, 1, uint64(clock.Next-86400000)) + item := wire.AppendVarint(nil, 3, 21) + item = wire.AppendVarint(item, 4, s.state.Free) + b = wire.AppendBytes(b, 2, item) + return wire.AppendBytes(nil, 1, b), nil +} +func (s *Service) handleGroundBase(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + if path == "/HuntDispatch" || path == "/HuntDispatchInfo" || path == "/HuntDispatchStart" || path == "/HuntDispatchEnd" || path == "/HuntDispatchRewardPreview" { + return s.handleDispatch(ctx, path, req, ctx.SessionID) + } + if path != "/HuntingGroundInfo" && path != "/HuntingGroundInfoList" && path != "/HuntingGroundEnter" { + return 0, nil, false, nil + } + + seq, found, err := wire.Varint(req, 1) + if err != nil || !found || seq == 0 { + return 0, nil, true, fmt.Errorf("hunting: missing sequence") + } + if path == "/HuntingGroundInfoList" { + ids, err := packed(req, 2) + if err != nil { + return 387, nil, true, err + } + var response []byte + seen := map[uint64]bool{} + for _, id := range ids { + if id == 0 || id > math.MaxInt32 || seen[id] { + return 387, nil, true, fmt.Errorf("hunting: invalid pack list") + } + seen[id] = true + d, err := s.load(int(id)) + if err != nil { + return 387, nil, true, err + } + if len(d.Grounds) == 0 { + response = wire.AppendBytes(response, 1, wire.AppendVarint(nil, 5, id)) + continue + } + info, err := s.info(int(id), d) + if err != nil { + return 387, nil, true, err + } + response = wire.AppendBytes(response, 1, info) + } + return 387, response, true, nil + } + pack, found, err := wire.Varint(req, 2) + if err != nil || !found || pack == 0 || pack > math.MaxInt32 { + return 134, nil, true, fmt.Errorf("hunting: invalid pack") + } + d, err := s.load(int(pack)) + if err != nil { + return 134, nil, true, err + } + if path == "/HuntingGroundInfo" { + + if s.state.Packs[strconv.Itoa(int(pack))].Current == 0 { + return 134, wire.AppendBytes(nil, 1, nil), true, nil + } + b, err := s.info(int(pack), d) + return 134, wire.AppendBytes(nil, 1, b), true, err + } + if len(d.Grounds) == 0 { + return 110, nil, true, fmt.Errorf("hunting: pack has no hunting ground") + } + current, err := s.currentPack(ctx) + if err != nil || current != int(pack) { + return 110, nil, true, fmt.Errorf("hunting: enter pack is not current") + } + id, _, err := wire.Varint(req, 3) + if err != nil { + return 110, nil, true, err + } + auto, _, err := wire.Varint(req, 4) + if err != nil || auto > 1 { + return 110, nil, true, fmt.Errorf("hunting: invalid auto flag") + } + st := s.state.Packs[strconv.Itoa(int(pack))] + g, ok := ground(d, id) + if !ok { + return 110, nil, true, fmt.Errorf("hunting: unknown ground") + } + if s.eligibility != nil { + if err := s.eligibility(ctx, int(pack), g.Difficulty); err != nil { + return 110, nil, true, err + } + } else if id != d.Grounds[0].ID { + return 110, nil, true, fmt.Errorf("hunting: main quest difficulty eligibility unavailable") + } + if st.Current != id { + st.Defeated = nil + } + st.Current, st.Auto = id, auto != 0 + next := s.clone() + next.Packs[strconv.Itoa(int(pack))] = st + if err := s.persist(ctx, next); err != nil { + return 110, nil, true, err + } + var out []byte + for _, m := range monstersWithState(d, g, st.Defeated) { + out = wire.AppendBytes(out, 1, m) + } + return 110, out, true, nil +} diff --git a/go/internal/server/domain/battle/monsterhunt/monsterhunt.go b/go/internal/server/domain/battle/monsterhunt/monsterhunt.go new file mode 100644 index 0000000..56106cc --- /dev/null +++ b/go/internal/server/domain/battle/monsterhunt/monsterhunt.go @@ -0,0 +1,177 @@ +// Package monsterhunt owns the configured Fiend Hunt seasons and local player progress. +package monsterhunt + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "bd2server/internal/server/protocol/staticdata" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "fmt" + "strconv" + + "time" +) + +const BattleMode uint64 = 8 +const PracticeMode uint64 = 24 + +type Season struct { + ID, Hunt, Start, End, Calculate, RankGroup uint64 + Independent bool +} +type user struct { + Season, Hunt, Level, StartHP, HighestHP, CurrentDamage, DailyDamage, HighestDate, DailyLevel, DailyDate uint64 + Played, Claimed bool + ClearLevel uint64 +} +type reply struct{ Request, Response []byte } +type snapshot struct { + Version string `json:"version"` + Slots uint64 `json:"slots"` + Users map[string]user `json:"users"` + Decks [][]byte `json:"decks"` + Settings [][]byte `json:"settings"` + Presets map[string][]byte `json:"presets"` + Replies map[string]reply `json:"replies"` +} +type encounter struct{ Season, Hunt, Level, HP, Mode, BeforeLevel, BeforeHP, Team uint64 } +type Service struct { + storage stateio.Store + seed *readonly.Seed + seasons []Season + state snapshot + inventory *assets.Inventory + wallet *assets.Wallet + characters *roster.CharacterStore + equipment *assets.EquipmentInventory + collection *roster.CollectionStore + presets *gamedata.PresetDesign + baseSlots, maxSlots uint64 + now func() time.Time + active map[string]encounter + transientVersion uint64 + load func(uint64) (*gamedata.MonsterHunt, error) + rewardGrant func(ctx command.Context, _ string, _ []gamedata.Reward) ([]byte, error) +} + +// RuleLoader returns shared immutable hunt rules. +type RuleLoader func(uint64) (*gamedata.MonsterHunt, error) + +// Rules holds the shared season calendar and preset configuration. +type Rules struct { + seasons []Season + seed *readonly.Seed + presets *gamedata.PresetDesign + load RuleLoader +} + +func NewRules(seasons []Season, seed *readonly.Seed, presets *gamedata.PresetDesign, load RuleLoader) (*Rules, error) { + if len(seasons) == 0 || seed == nil || presets == nil || load == nil { + return nil, fmt.Errorf("monsterhunt: incomplete rules") + } + for _, c := range seasons { + if c.ID == 0 || c.Hunt == 0 || c.Start > c.End { + return nil, fmt.Errorf("monsterhunt: invalid configured season") + } + if _, err := load(c.Hunt); err != nil { + return nil, err + } + } + return &Rules{seasons: append([]Season(nil), seasons...), seed: seed, presets: presets, load: load}, nil +} + +func (s *Service) AttachRewards(grant func(ctx command.Context, _ string, _ []gamedata.Reward) ([]byte, error)) { + s.rewardGrant = grant +} +func (s *Service) AttachCharacters(ctx command.Context, c *roster.CharacterStore) { s.characters = c } + +func (s *Service) PresetSlotCount() uint64 { return s.state.Slots } + +func (s *Service) TransientVersion() uint64 { + + return s.transientVersion +} + +func (s *Service) current() Season { + // Future rows are public calendar data, not an instruction to switch the + // player's current hunt before its window begins. Regular client schedules + // have a single slot; independent hunts use their separate client list. + now := uint64(s.now().UnixMilli()) + var regular []Season + for _, c := range s.seasons { + if !c.Independent { + regular = append(regular, c) + } + } + if len(regular) == 0 { + regular = s.seasons + } + return selectSeason(regular, now) +} + +func selectSeason(rows []Season, now uint64) Season { + var active, begun, future Season + for _, c := range rows { + if c.Start <= now { + if begun.ID == 0 || c.Start > begun.Start || c.Start == begun.Start && c.ID > begun.ID { + begun = c + } + if now < c.End && (active.ID == 0 || c.Start > active.Start || c.Start == active.Start && c.ID > active.ID) { + active = c + } + } else if future.ID == 0 || c.Start < future.Start || c.Start == future.Start && c.ID < future.ID { + future = c + } + } + if active.ID != 0 { + return active + } + if begun.ID != 0 { + return begun + } + return future +} + +func (s *Service) playing(c Season) bool { + n := uint64(s.now().UnixMilli()) + return n >= c.Start && n < c.End +} +func (s *Service) getUser(c Season) (user, error) { + u, ok := s.state.Users[strconv.FormatUint(c.ID, 10)] + if ok { + if u.DailyDate != 0 && u.DailyDate/86400000 != uint64(s.now().UnixMilli())/86400000 { + u.DailyDamage = 0 + u.DailyLevel = 0 + } + return u, nil + } + d, e := s.load(c.Hunt) + if e != nil { + return u, e + } + hp, e := d.HP(1) + return user{Season: c.ID, Hunt: c.Hunt, Level: 1, StartHP: hp, HighestHP: hp}, e +} + +func (s *Service) save(ctx command.Context, next snapshot) error { + b, e := json.Marshal(next) + if e != nil { + return e + } + if e = s.storage.Save(ctx.State, "monsterhunt", b); e != nil { + return e + } + s.state = next + return nil +} +func (s *Service) clone() snapshot { + raw, _ := json.Marshal(s.state) + var n snapshot + _ = json.Unmarshal(raw, &n) + return n +} + +var codes = map[string]int{"/MonsterHuntScheduleInfo": 0, "/MonsterHuntUserInfo": 196, "/MonsterHuntRankInfo": 197, "/MonsterHuntDeckInfo": 263, "/MonsterHuntDeckSave": 264, "/MonsterHuntChangeTeam": 265, "/MonsterHuntQuickBattle": 266, "/MonsterHuntPresetSlotAdd": 400, "/MonsterHuntPresetInfo": 405, "/MonsterHuntPresetSave": 406, "/MonsterHuntPresetUse": 410, "/MonsterHuntPresetDelete": 0, "/MonsterHuntPresetInfoChange": 0, "/MonsterHuntSeasonReward": 0} diff --git a/go/internal/server/monsterhunt/monsterhunt.go b/go/internal/server/domain/battle/monsterhunt/protocol_adapter.go similarity index 66% rename from go/internal/server/monsterhunt/monsterhunt.go rename to go/internal/server/domain/battle/monsterhunt/protocol_adapter.go index 20b8244..ef67fde 100644 --- a/go/internal/server/monsterhunt/monsterhunt.go +++ b/go/internal/server/domain/battle/monsterhunt/protocol_adapter.go @@ -1,7 +1,14 @@ -// Package monsterhunt owns the configured Fiend Hunt seasons and local player progress. package monsterhunt import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/protocol/staticdata" + "bd2server/internal/server/protocol/wire" + "bd2server/internal/server/storage/stateio" "bytes" "encoding/binary" "encoding/json" @@ -10,122 +17,215 @@ import ( "sort" "strconv" "strings" - "sync" "time" "unicode/utf8" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/readonly" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" ) -const BattleMode uint64 = 8 -const PracticeMode uint64 = 24 - -type season struct { - ID, Hunt, Start, End, Calculate, RankGroup uint64 - Independent bool -} -type user struct { - Season, Hunt, Level, StartHP, HighestHP, CurrentDamage, DailyDamage, HighestDate, DailyLevel, DailyDate uint64 - Played, Claimed bool - ClearLevel uint64 -} -type reply struct{ Request, Response []byte } -type snapshot struct { - Version string `json:"version"` - Slots uint64 `json:"slots"` - Users map[string]user `json:"users"` - Decks [][]byte `json:"decks"` - Settings [][]byte `json:"settings"` - Presets map[string][]byte `json:"presets"` - Replies map[string]reply `json:"replies"` -} -type encounter struct{ Season, Hunt, Level, HP, Mode, BeforeLevel, BeforeHP, Team uint64 } -type Service struct { - mu sync.Mutex - storage stateio.Store - root, version string - seed *readonly.Seed - seasons []season - state snapshot - inventory *player.Inventory - wallet *player.Wallet - characters *player.CharacterStore - equipment *player.EquipmentInventory - collection *player.CollectionStore - presets *gamedata.PresetDesign - baseSlots, maxSlots uint64 - now func() time.Time - active map[string]encounter - load func(uint64) (*gamedata.MonsterHunt, error) - rewardGrant func(string, []gamedata.Reward) ([]byte, error) -} - -func Open(storage stateio.Store, root, version string, seed *readonly.Seed, inventory *player.Inventory, wallet *player.Wallet) (*Service, error) { - if storage == nil || seed == nil || inventory == nil || wallet == nil { - return nil, fmt.Errorf("monsterhunt: incomplete configuration") - } - s := &Service{storage: storage, root: root, version: version, seed: seed, inventory: inventory, wallet: wallet, now: time.Now, active: map[string]encounter{}} - designs := map[uint64]*gamedata.MonsterHunt{} - s.load = func(id uint64) (*gamedata.MonsterHunt, error) { - if d, ok := designs[id]; ok { - return d, nil - } - d, e := gamedata.LoadMonsterHunt(root, version, id) - if e == nil { - designs[id] = d - } - return d, e - } - for _, f := range seed.Responses["/MonsterHuntScheduleInfo"].Fields { - if f.Number != 1 || f.Type != 2 { - continue - } - var c season - for _, v := range f.Fields { - switch v.Number { - case 1: - for _, x := range v.Fields { - switch x.Number { - case 1: - c.ID = x.Varint - case 2: - c.Start = x.Varint - case 3: - c.End = x.Varint - } - } - case 2: - c.Hunt = v.Varint - case 4: - c.Calculate = v.Varint - case 6: - c.Independent = v.Varint != 0 - case 7: - c.RankGroup = v.Varint +func validateRequest(req []byte) error { + return wire.Walk(req, func(f wire.Field) error { + if f.Type == 0 { + v, _ := binary.Uvarint(f.Value) + if v > math.MaxInt32 { + return fmt.Errorf("monsterhunt: integer outside protocol range") } } - if c.ID == 0 || c.Hunt == 0 || c.Start > c.End { - return nil, fmt.Errorf("monsterhunt: invalid configured season") + return nil + }) +} + +func (s *Service) ValidatePack(ctx command.Context, pack int, req []byte) error { + + id, e := scalar(req, 6) + if e != nil { + return e + } + d, e := s.load(id) + if e != nil { + return e + } + if pack <= 0 || uint64(pack) != d.PackID { + return fmt.Errorf("monsterhunt: hunt does not belong to current pack") + } + return nil +} + +func (s *Service) rankWire(u user) []byte { + b := wire.AppendVarint(nil, 7, 1) + b = wire.AppendDouble(b, 8, s.score(u)) + return wire.AppendDouble(b, 10, 100) +} + +func (s *Service) grant(ctx command.Context, identity string, rewards []gamedata.BattleReward) ([]byte, error) { + if s.rewardGrant != nil { + rs := make([]gamedata.Reward, len(rewards)) + for i, r := range rewards { + rs[i] = gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count} //nolint:staticcheck // S1016 } - s.seasons = append(s.seasons, c) + return s.rewardGrant(ctx, identity, rs) } - if len(s.seasons) == 0 { - return nil, fmt.Errorf("monsterhunt: configured schedule missing") + var currency []gamedata.Reward + var stack []gamedata.BattleReward + for _, r := range rewards { + switch r.Type { + case 2, 3, 4, 12, 20: + currency = append(currency, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 + default: + stack = append(stack, r) + } } - d, e := gamedata.LoadMonsterHuntPresetDesign(root, version) + if _, e := s.wallet.GrantQuestOnce(ctx, identity+":currency", currency); e != nil { + return nil, e + } + items, e := s.inventory.GrantOnce(ctx, identity+":items", stack) if e != nil { return nil, e } - s.presets = d - s.baseSlots = d.BaseCount - s.maxSlots = d.Maximum + if len(items) == 0 { + items = s.inventory.GrantedItems(identity + ":items") + } + var bundle []byte + for _, item := range items { + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(item)) + v := wire.AppendVarint(nil, 2, item.ID) + v = wire.AppendVarint(v, 3, item.Type) + v = wire.AppendVarint(v, 4, item.Count) + bundle = wire.AppendBytes(bundle, 6, v) + } + for _, r := range currency { + v := wire.AppendVarint(nil, 3, r.Type) + v = wire.AppendVarint(v, 4, r.Count) + bundle = wire.AppendBytes(bundle, 1, v) + } + return bundle, nil +} + +func (s *Service) presetBindings(ctx command.Context, p []byte) (map[uint64]uint64, []assets.PresetEquipmentBinding, error) { + assignments := map[uint64]uint64{} + var bindings []assets.PresetEquipmentBinding + entries, e := messages(p, 5) + if e != nil { + return nil, nil, e + } + for _, entry := range entries { + base, _, e := wire.Bytes(entry, 1) + if e != nil { + return nil, nil, e + } + index, _ := scalar(base, 1) + costume, _ := scalar(entry, 2) + if _, duplicate := assignments[index]; duplicate { + return nil, nil, fmt.Errorf("monsterhunt: duplicate preset character") + } + assignments[index] = costume + if costume != 0 && s.collection != nil { + c, found := s.collection.CostumeByIndex(costume) + if !found || c.UseChar != index { + return nil, nil, fmt.Errorf("monsterhunt: preset costume not owned by character") + } + } + binding := assets.PresetEquipmentBinding{CharacterIndex: index, Equipment: make([]uint64, 5)} + equipment, e := messages(entry, 3) + if e != nil { + return nil, nil, e + } + seen := map[uint64]bool{} + for _, item := range equipment { + t, _ := scalar(item, 1) + id, _ := scalar(item, 2) + if t >= 5 || seen[t] { + return nil, nil, fmt.Errorf("monsterhunt: invalid preset equipment slot") + } + seen[t] = true + binding.Equipment[t] = id + } + bindings = append(bindings, binding) + } + if len(bindings) > 0 && s.equipment != nil { + if e = s.equipment.ValidatePresetEquipment(ctx, bindings); e != nil { + return nil, nil, e + } + } + return assignments, bindings, nil +} + +func (s *Service) applyPreset(ctx command.Context, p []byte) ([]byte, error) { + if s.characters == nil || s.equipment == nil || s.collection == nil { + return nil, fmt.Errorf("monsterhunt: preset ownership runtime unavailable") + } + assignments, bindings, e := s.presetBindings(ctx, p) + if e != nil { + return nil, e + } + if _, e = s.characters.ApplyPresetCostumes(ctx, assignments); e != nil { + return nil, e + } + if len(bindings) > 0 { + if _, e = s.equipment.ApplyPresetEquipment(ctx, bindings); e != nil { + return nil, e + } + } + var out []byte + for _, binding := range bindings { + c, ok := s.characters.Find(ctx, binding.CharacterIndex) + if ok { + out = wire.AppendBytes(out, 2, roster.CharacterWire(c)) + } + b := wire.AppendVarint(nil, 1, binding.CharacterIndex) + for _, id := range binding.Equipment { + b = wire.AppendVarint(b, 2, id) + } + out = wire.AppendBytes(out, 3, b) + } + return out, nil +} + +func (s *Service) validateSettings(ctx command.Context, settings [][]byte) error { + for _, setting := range settings { + index, e := scalar(setting, 1) + if e != nil || index == 0 { + return fmt.Errorf("monsterhunt: invalid costume setting character") + } + if s.characters != nil { + if _, ok := s.characters.Find(ctx, index); !ok { + return fmt.Errorf("monsterhunt: setting character not owned") + } + } + seq, e := messages(setting, 2) + if e != nil { + return e + } + mode, _ := scalar(setting, 3) + if mode != BattleMode && mode != PracticeMode { + return fmt.Errorf("monsterhunt: invalid setting battle mode") + } + for _, item := range seq { + costume, _, e := wire.Varint(item, 1) + if e != nil { + return e + } + if costume == 0 || costume == ^uint64(0) { + continue + } + if s.collection != nil { + c, ok := s.collection.CostumeByIndex(costume) + if !ok || c.UseChar != index { + return fmt.Errorf("monsterhunt: setting costume not owned") + } + } + } + } + return nil +} + +func Open(ctx command.Context, storage stateio.Store, rules *Rules, inventory *assets.Inventory, wallet *assets.Wallet) (*Service, error) { + if storage == nil || rules == nil || rules.seed == nil || rules.presets == nil || rules.load == nil || len(rules.seasons) == 0 || inventory == nil || wallet == nil { + return nil, fmt.Errorf("monsterhunt: incomplete configuration") + } + s := &Service{storage: storage, seed: rules.seed, seasons: rules.seasons, presets: rules.presets, load: rules.load, inventory: inventory, wallet: wallet, now: time.Now, active: map[string]encounter{}} + s.baseSlots = rules.presets.BaseCount + s.maxSlots = rules.presets.Maximum s.state = snapshot{Version: versionconfig.State(), Slots: s.baseSlots, Users: map[string]user{}, Presets: map[string][]byte{}, Replies: map[string]reply{}} - raw, e := storage.Load("monsterhunt") + raw, e := storage.Load(ctx.State, "monsterhunt") if e != nil { return nil, e } @@ -140,15 +240,10 @@ func Open(storage stateio.Store, root, version string, seed *readonly.Seed, inve return nil, fmt.Errorf("monsterhunt: incompatible state") } } - for _, c := range s.seasons { - if _, e = s.load(c.Hunt); e != nil { - return nil, e - } - } - if e = s.validateDecks(s.state.Decks); e != nil { + if e = s.validateDecks(ctx, s.state.Decks); e != nil { return nil, e } - if e = s.validateSettings(s.state.Settings); e != nil { + if e = s.validateSettings(ctx, s.state.Settings); e != nil { return nil, e } for key, u := range s.state.Users { @@ -169,85 +264,37 @@ func Open(storage stateio.Store, root, version string, seed *readonly.Seed, inve if key != strconv.FormatUint(slot, 10) { return nil, fmt.Errorf("monsterhunt: invalid saved preset key") } - if e = s.validatePreset(p); e != nil { + if e = s.validatePreset(ctx, p); e != nil { return nil, e } } return s, nil } -func (s *Service) AttachRewards(grant func(string, []gamedata.Reward) ([]byte, error)) { - s.rewardGrant = grant -} -func (s *Service) AttachCharacters(c *player.CharacterStore) { s.characters = c } -func (s *Service) AttachPresetRuntime(c *player.CharacterStore, e *player.EquipmentInventory, col *player.CollectionStore) error { + +func (s *Service) AttachPresetRuntime(ctx command.Context, c *roster.CharacterStore, e *assets.EquipmentInventory, col *roster.CollectionStore) error { if c == nil || e == nil || col == nil { return fmt.Errorf("monsterhunt: incomplete preset runtime") } s.characters = c s.equipment = e s.collection = col - if err := s.validateDecks(s.state.Decks); err != nil { + if err := s.validateDecks(ctx, s.state.Decks); err != nil { return err } - if err := s.validateSettings(s.state.Settings); err != nil { + if err := s.validateSettings(ctx, s.state.Settings); err != nil { return err } for _, p := range s.state.Presets { - if err := s.validatePreset(p); err != nil { + if err := s.validatePreset(ctx, p); err != nil { return err } } return nil } -func (s *Service) PresetSlotCount() uint64 { s.mu.Lock(); defer s.mu.Unlock(); return s.state.Slots } -func (s *Service) BeginSession(id string) { - s.mu.Lock() - defer s.mu.Unlock() - // Encounters are keyed by session-specific receipts and survive repeated activation. -} -func (s *Service) current() season { - // Future rows are public calendar data, not an instruction to switch the - // player's current hunt before its window begins. Regular client schedules - // have a single slot; independent hunts use their separate client list. - now := uint64(s.now().UnixMilli()) - var regular []season - for _, c := range s.seasons { - if !c.Independent { - regular = append(regular, c) - } - } - if len(regular) == 0 { - regular = s.seasons - } - return selectSeason(regular, now) -} -func selectSeason(rows []season, now uint64) season { - var active, begun, future season - for _, c := range rows { - if c.Start <= now { - if begun.ID == 0 || c.Start > begun.Start || c.Start == begun.Start && c.ID > begun.ID { - begun = c - } - if now < c.End && (active.ID == 0 || c.Start > active.Start || c.Start == active.Start && c.ID > active.ID) { - active = c - } - } else if future.ID == 0 || c.Start < future.Start || c.Start == future.Start && c.ID < future.ID { - future = c - } - } - if active.ID != 0 { - return active - } - if begun.ID != 0 { - return begun - } - return future -} - -func (s *Service) scheduleInfo(req []byte) (int, []byte, bool, error) { +func (s *Service) scheduleInfo(ctx command.Context, req []byte) (int, []byte, bool, error) { current := s.current() - var independent []season + var independent []Season for _, row := range s.seasons { if row.Independent { independent = append(independent, row) @@ -289,29 +336,10 @@ func (s *Service) scheduleInfo(req []byte) (int, []byte, bool, error) { projected := &readonly.Seed{Version: s.seed.Version, Responses: map[string]readonly.Response{ "/MonsterHuntScheduleInfo": {PacketCode: response.PacketCode, Fields: fields}, }} - return projected.Handle("/MonsterHuntScheduleInfo", req) + return projected.Handle(ctx, "/MonsterHuntScheduleInfo", req) } -func (s *Service) playing(c season) bool { - n := uint64(s.now().UnixMilli()) - return n >= c.Start && n < c.End -} -func (s *Service) getUser(c season) (user, error) { - u, ok := s.state.Users[strconv.FormatUint(c.ID, 10)] - if ok { - if u.DailyDate != 0 && u.DailyDate/86400000 != uint64(s.now().UnixMilli())/86400000 { - u.DailyDamage = 0 - u.DailyLevel = 0 - } - return u, nil - } - d, e := s.load(c.Hunt) - if e != nil { - return u, e - } - hp, e := d.HP(1) - return user{Season: c.ID, Hunt: c.Hunt, Level: 1, StartHP: hp, HighestHP: hp}, e -} -func (s *Service) encodeUser(u user, c season) []byte { + +func (s *Service) encodeUser(u user, c Season) []byte { var b []byte for n, v := range map[int]uint64{1: u.Season, 2: u.Hunt, 3: u.Level, 4: u.StartHP, 6: u.HighestHP, 7: u.HighestDate, 8: u.CurrentDamage, 9: u.DailyDamage, 11: u.DailyLevel, 12: u.DailyDate} { if n == 3 { @@ -329,23 +357,7 @@ func (s *Service) encodeUser(u user, c season) []byte { } return b } -func (s *Service) save(next snapshot) error { - b, e := json.Marshal(next) - if e != nil { - return e - } - if e = s.storage.Save("monsterhunt", b); e != nil { - return e - } - s.state = next - return nil -} -func (s *Service) clone() snapshot { - raw, _ := json.Marshal(s.state) - var n snapshot - _ = json.Unmarshal(raw, &n) - return n -} + func scalar(b []byte, n int) (uint64, error) { v, _, e := wire.Varint(b, n) if e != nil || v > math.MaxInt64 { @@ -353,6 +365,7 @@ func scalar(b []byte, n int) (uint64, error) { } return v, nil } + func messages(b []byte, n int) ([][]byte, error) { var out [][]byte e := wire.Walk(b, func(f wire.Field) error { @@ -366,6 +379,7 @@ func messages(b []byte, n int) ([][]byte, error) { }) return out, e } + func packed(b []byte, n int) ([]uint64, error) { var out []uint64 e := wire.Walk(b, func(f wire.Field) error { @@ -392,19 +406,14 @@ func packed(b []byte, n int) ([]uint64, error) { }) return out, e } -func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { - return s.HandleSession(path, req, "local") -} -var codes = map[string]int{"/MonsterHuntScheduleInfo": 0, "/MonsterHuntUserInfo": 196, "/MonsterHuntRankInfo": 197, "/MonsterHuntDeckInfo": 263, "/MonsterHuntDeckSave": 264, "/MonsterHuntChangeTeam": 265, "/MonsterHuntQuickBattle": 266, "/MonsterHuntPresetSlotAdd": 400, "/MonsterHuntPresetInfo": 405, "/MonsterHuntPresetSave": 406, "/MonsterHuntPresetUse": 410, "/MonsterHuntPresetDelete": 0, "/MonsterHuntPresetInfoChange": 0, "/MonsterHuntSeasonReward": 0} - -func (s *Service) HandleSession(path string, req []byte, session string) (int, []byte, bool, error) { +func (s *Service) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID code, ok := codes[path] if !ok { return 0, nil, false, nil } - s.mu.Lock() - defer s.mu.Unlock() + seq, e := scalar(req, 1) if e != nil || seq == 0 || seq > math.MaxInt32 { return code, nil, true, fmt.Errorf("monsterhunt: invalid sequence") @@ -412,7 +421,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ if e = validateRequest(req); e != nil { return code, nil, true, e } - key := session + ":" + path + ":" + strconv.FormatUint(seq, 10) + key := commandSession + ":" + path + ":" + strconv.FormatUint(seq, 10) if r, found := s.state.Replies[key]; found { if !bytes.Equal(req, r.Request) { return code, nil, true, fmt.Errorf("monsterhunt: sequence reused with different request") @@ -420,7 +429,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ return code, r.Response, true, nil } if path == "/MonsterHuntScheduleInfo" { - return s.scheduleInfo(req) + return s.scheduleInfo(ctx, req) } c := s.current() u, e := s.getUser(c) @@ -449,20 +458,20 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ case "/MonsterHuntDeckSave": next.Decks, e = messages(req, 2) if e == nil { - e = s.validateDecks(next.Decks) + e = s.validateDecks(ctx, next.Decks) } if e == nil { next.Settings, e = messages(req, 3) } if e == nil { - e = s.validateSettings(next.Settings) + e = s.validateSettings(ctx, next.Settings) } mutation = true case "/MonsterHuntChangeTeam": var d []byte d, _, e = wire.Bytes(req, 2) if e == nil { - e = s.validateDecks([][]byte{d}) + e = s.validateDecks(ctx, [][]byte{d}) } mode, _ := scalar(req, 4) if mode != BattleMode && mode != PracticeMode { @@ -472,7 +481,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ found := false team, _ := scalar(d, 1) for receipt, a := range s.active { - if !strings.HasPrefix(receipt, session+":") || a.Mode != mode { + if !strings.HasPrefix(receipt, commandSession+":") || a.Mode != mode { continue } found = true @@ -537,7 +546,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ var p []byte p, _, e = wire.Bytes(req, 2) if e == nil { - e = s.validatePreset(p) + e = s.validatePreset(ctx, p) } if e == nil { slot, _ := scalar(p, 4) @@ -562,7 +571,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ p, _, e = wire.ReplaceVarint(p, b, v) } if e == nil { - e = s.validatePreset(p) + e = s.validatePreset(ctx, p) } next.Presets[strconv.FormatUint(slot, 10)] = p mutation = true @@ -592,13 +601,13 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ cost := count * s.presets.Price switch s.presets.PriceType { case 4: - _, e = s.wallet.SpendGoldOnce("monsterhunt:"+key, cost) + _, e = s.wallet.SpendGoldOnce(ctx, "monsterhunt:"+key, cost) case 3: - _, e = s.wallet.SpendFreeJewelryOnce("monsterhunt:"+key, cost) + _, e = s.wallet.SpendFreeJewelryOnce(ctx, "monsterhunt:"+key, cost) case 2: - _, e = s.wallet.SpendJewelryOnce("monsterhunt:"+key, cost) + _, e = s.wallet.SpendJewelryOnce(ctx, "monsterhunt:"+key, cost) case 12: - _, e = s.wallet.SpendCatalystOnce("monsterhunt:"+key, cost) + _, e = s.wallet.SpendCatalystOnce(ctx, "monsterhunt:"+key, cost) default: e = fmt.Errorf("monsterhunt: unsupported preset currency") } @@ -615,7 +624,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ } next.Decks, e = s.presetDecks(p) if e == nil { - out, e = s.applyPreset(p) + out, e = s.applyPreset(ctx, p) } if e == nil { for _, v := range next.Decks { @@ -634,7 +643,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ e = loadErr break } - bundle, grantErr := s.grant("monsterhunt:"+key, dailyDifference(d, u.DailyLevel, u.ClearLevel)) + bundle, grantErr := s.grant(ctx, "monsterhunt:"+key, dailyDifference(d, u.DailyLevel, u.ClearLevel)) if grantErr != nil { e = grantErr break @@ -659,7 +668,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ break } rewards := s.rankRewards(d, c.RankGroup) - bundle, grantErr := s.grant("monsterhunt:"+key, rewards) + bundle, grantErr := s.grant(ctx, "monsterhunt:"+key, rewards) if grantErr != nil { e = grantErr break @@ -677,17 +686,18 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ } if mutation { next.Replies[key] = reply{append([]byte(nil), req...), out} - if e = s.save(next); e != nil { + if e = s.save(ctx, next); e != nil { return code, nil, true, e } if updateReceipt != "" { s.active[updateReceipt] = updateEncounter + s.transientVersion++ } } return code, out, true, nil } -func (s *Service) validateDecks(ds [][]byte) error { +func (s *Service) validateDecks(ctx command.Context, ds [][]byte) error { seen := map[uint64]bool{} for _, d := range ds { t, e := scalar(d, 1) @@ -709,7 +719,7 @@ func (s *Service) validateDecks(ds [][]byte) error { return fmt.Errorf("monsterhunt: invalid deck character") } if s.characters != nil { - if _, found := s.characters.Find(i); !found { + if _, found := s.characters.Find(ctx, i); !found { return fmt.Errorf("monsterhunt: character not owned") } } @@ -719,7 +729,8 @@ func (s *Service) validateDecks(ds [][]byte) error { } return nil } -func (s *Service) validatePreset(p []byte) error { + +func (s *Service) validatePreset(ctx command.Context, p []byte) error { slot, e := scalar(p, 4) if e != nil || slot >= s.state.Slots { return fmt.Errorf("monsterhunt: invalid preset slot") @@ -737,11 +748,12 @@ func (s *Service) validatePreset(p []byte) error { if e != nil { return e } - if _, _, e = s.presetBindings(p); e != nil { + if _, _, e = s.presetBindings(ctx, p); e != nil { return e } - return s.validateDecks(d) + return s.validateDecks(ctx, d) } + func (s *Service) presetDecks(p []byte) ([][]byte, error) { entries, e := messages(p, 5) if e != nil { @@ -768,9 +780,8 @@ func (s *Service) presetDecks(p []byte) ([][]byte, error) { return out, nil } -func (s *Service) EnterBattle(req []byte, receipt string) ([]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() +func (s *Service) EnterBattle(ctx command.Context, req []byte, receipt string) ([]byte, error) { + mode, e := scalar(req, 5) if e != nil || mode != BattleMode && mode != PracticeMode { return nil, fmt.Errorf("monsterhunt: invalid battle mode") @@ -816,15 +827,16 @@ func (s *Service) EnterBattle(req []byte, receipt string) ([]byte, error) { hp = u.StartHP } s.active[receipt] = encounter{c.ID, id, level, hp, mode, u.Level, u.StartHP, 1} + s.transientVersion++ if mode == PracticeMode { u.Level = level u.StartHP = hp } return wire.AppendBytes(nil, 5, s.encodeUser(u, c)), nil } -func (s *Service) CompleteBattle(req []byte, receipt string) ([]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() + +func (s *Service) CompleteBattle(ctx command.Context, req []byte, receipt string) ([]byte, error) { + key := "battle:" + receipt if r, ok := s.state.Replies[key]; ok { if !bytes.Equal(r.Request, req) { @@ -842,6 +854,7 @@ func (s *Service) CompleteBattle(req []byte, receipt string) ([]byte, error) { } if a.Mode == PracticeMode { delete(s.active, receipt) + s.transientVersion++ return nil, nil } c := s.current() @@ -897,7 +910,7 @@ func (s *Service) CompleteBattle(req []byte, receipt string) ([]byte, error) { out = wire.AppendVarint(out, 27, clear) var bundle []byte for level := previousClear + 1; level <= clear; level++ { - b, e := s.grant(fmt.Sprintf("monsterhunt:season:%d:clear:%d", c.ID, level), d.Rewards[level].Clear) + b, e := s.grant(ctx, fmt.Sprintf("monsterhunt:season:%d:clear:%d", c.ID, level), d.Rewards[level].Clear) if e != nil { return nil, e } @@ -909,7 +922,7 @@ func (s *Service) CompleteBattle(req []byte, receipt string) ([]byte, error) { } day := uint64(s.now().UnixMilli()) / 86400000 if damage > 0 && (u.DailyDate/86400000 != day || a.Level > u.DailyLevel) { - bundle, e := s.grant("monsterhunt:"+key+":daily", dailyDifference(d, u.DailyLevel, a.Level)) + bundle, e := s.grant(ctx, "monsterhunt:"+key+":daily", dailyDifference(d, u.DailyLevel, a.Level)) if e != nil { return nil, e } @@ -924,9 +937,10 @@ func (s *Service) CompleteBattle(req []byte, receipt string) ([]byte, error) { next := s.clone() next.Users[strconv.FormatUint(c.ID, 10)] = u next.Replies[key] = reply{append([]byte(nil), req...), out} - if e = s.save(next); e != nil { + if e = s.save(ctx, next); e != nil { return nil, e } delete(s.active, receipt) + s.transientVersion++ return out, nil } diff --git a/go/internal/server/domain/battle/monsterhunt/rewards.go b/go/internal/server/domain/battle/monsterhunt/rewards.go new file mode 100644 index 0000000..4a954f2 --- /dev/null +++ b/go/internal/server/domain/battle/monsterhunt/rewards.go @@ -0,0 +1,50 @@ +package monsterhunt + +import ( + "bd2server/internal/server/design/gamedata" +) + +// Daily payout is the best reward achieved that UTC day. Improvements pay +// only positive per-item differences from the already paid level. +func dailyDifference(d *gamedata.MonsterHunt, previous, next uint64) []gamedata.BattleReward { + paid := map[[2]uint64]uint64{} + for _, r := range d.Rewards[previous].Daily { + paid[[2]uint64{r.Type, r.ID}] += r.Count + } + var out []gamedata.BattleReward + for _, r := range d.Rewards[next].Daily { + n := paid[[2]uint64{r.Type, r.ID}] + if r.Count > n { + r.Count -= n + out = append(out, r) + } + } + return out +} + +// This server has one persistent account state: a submitted participant ranks +// first among the local participants. No official leaderboard is imported. +func (s *Service) score(u user) float64 { + progress := float64(0) + if u.HighestHP > 0 { + progress = float64(u.HighestHP-u.StartHP) / float64(u.HighestHP) * 10000 + } + return float64(u.ClearLevel*1000000) + progress +} + +func (s *Service) rankRewards(d *gamedata.MonsterHunt, group uint64) []gamedata.BattleReward { + var selected *gamedata.MonsterHuntRankReward + for _, r := range d.Ranks[group] { + threshold := float64(1) + if r.Type == 1 { + threshold = 100 + } + if r.Ranking >= threshold && (selected == nil || r.Type < selected.Type || r.Type == selected.Type && r.Ranking < selected.Ranking) { + selected = &r + } + } + if selected == nil { + return nil + } + return selected.Rewards +} diff --git a/go/internal/server/battle/battle.go b/go/internal/server/domain/battle/protocol_adapter.go similarity index 65% rename from go/internal/server/battle/battle.go rename to go/internal/server/domain/battle/protocol_adapter.go index b44daab..804a093 100644 --- a/go/internal/server/battle/battle.go +++ b/go/internal/server/domain/battle/protocol_adapter.go @@ -1,180 +1,15 @@ -// Package battle implements a deterministic local battle session. The client -// remains authoritative for turn simulation; the server validates sequencing -// and echoes the submitted combat state without capture replay. package battle import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/protocol/wire" "bytes" "errors" "fmt" "math" - "sync" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" ) -type Service struct { - mu sync.Mutex - states map[string]*battleState - activeSession string - gameDataRoot string - gameDataVersion string - inventory *player.Inventory - currentPack func() (int, error) - currentDifficulty func() (uint64, error) - loadDifficultyDeck func(string, string, int, uint64, uint64) (uint64, error) - loadRewards func(string, string, int, uint64) ([]gamedata.BattleReward, error) - loadPhases func(string, string, int, uint64, uint64) ([]gamedata.BattlePhase, error) - buffs func() ([]gamedata.PictorialBuffStat, error) - onTutorialWin func() error - onMonsterWin func() error - commitHealth func(map[uint64]uint64) error - hunting HuntingRuntime - monsterHunt MonsterHuntRuntime - eventBattles []EventBattleRuntime - fieldMonsters FieldMonsterRuntime - consumeFieldBuff func(string) error -} -type FieldMonsterRuntime interface { - BeginFieldMonsterBattle(int, uint64, uint64) (string, bool, error) - CompleteFieldMonsterBattle(int, uint64, string) ([]byte, error) -} - -func (s *Service) AttachFieldMonsters(runtime FieldMonsterRuntime) { s.fieldMonsters = runtime } - -func (s *Service) AttachFieldBuffConsume(consume func(string) error) { s.consumeFieldBuff = consume } - -// EventBattleRuntime owns event stage eligibility, costs and settlement while -// the normal battle service transports the client's turn simulation. -type EventBattleRuntime interface { - HandlesBattle(mode uint64) bool - EnterBattle(request []byte, receipt string) ([]byte, error) - CompleteBattle(request []byte, receipt string) ([]byte, error) -} - -func (s *Service) AttachEventBattle(runtime EventBattleRuntime) { - if runtime != nil { - s.eventBattles = append(s.eventBattles, runtime) - } -} - -func (s *Service) eventBattle(mode uint64) EventBattleRuntime { - for _, runtime := range s.eventBattles { - if runtime.HandlesBattle(mode) { - return runtime - } - } - return nil -} - -type MonsterHuntRuntime interface { - EnterBattle(request []byte, receipt string) ([]byte, error) - CompleteBattle(request []byte, receipt string) ([]byte, error) -} - -func (s *Service) AttachMonsterHunt(runtime MonsterHuntRuntime) { - s.monsterHunt = runtime -} - -func isMonsterHunt(mode uint64) bool { return mode == 8 || mode == 24 } - -// HuntingRuntime validates the active hunting ground and settles each won -// encounter with the account's persistent AP, monsters and reward ledger. -type HuntingRuntime interface { - ValidateBattle(pack int, mode, monster, deck uint64) error - CompleteBattle(pack int, mode, monster, deck uint64, receipt string) ([]byte, [][]byte, error) -} - -const huntingGroundMode = 5 - -func (s *Service) AttachHunting(runtime HuntingRuntime) { - s.hunting = runtime -} - -type battleState struct { - entered bool - index uint64 - round uint64 - monster uint64 - deck uint64 - pack int - mode uint64 - enterReceipt string - fieldInstance string - initialBlue [][]byte - phases []gamedata.BattlePhase - phase int - phaseStarted bool - phaseSeq uint64 - phaseReply []byte - endSeq uint64 - endRequest []byte - endReply []byte -} - -// BeginSession discards an unfinished battle when LoginUser creates a new -// game session. Persistent rewards are written only by a successful BattleEnd -// request transaction, so reconnect returns to the last pre-battle commit. -func (s *Service) BeginSession(id string) { - s.mu.Lock() - defer s.mu.Unlock() - if id == "" { - return - } - if s.states == nil { - s.states = make(map[string]*battleState) - } - if s.states[id] == nil { - if len(s.states) >= 1024 { - for key := range s.states { - if key != id { - delete(s.states, key) - break - } - } - } - s.states[id] = &battleState{} - } - s.activeSession = id -} - -func (s *Service) AttachMonsterWinMission(callback func() error) { s.onMonsterWin = callback } - -func (s *Service) AttachTutorialWin(callback func() error) { s.onTutorialWin = callback } - -// AttachCommittedHealth persists only completed battle results. Round state -// remains transient, so reconnect rolls back an unfinished battle. -// The callback may normalize values to field HP; the response uses those -// committed values rather than echoing battle-only HP buffs. -func (s *Service) AttachCommittedHealth(callback func(map[uint64]uint64) error) { - s.commitHealth = callback -} - -func (s *Service) Active() bool { - s.mu.Lock() - defer s.mu.Unlock() - return s.stateLocked().entered -} - -func NewService(gameDataRoot, gameDataVersion string, inventory *player.Inventory, currentPack func() (int, error)) *Service { - return &Service{ - gameDataRoot: gameDataRoot, gameDataVersion: gameDataVersion, - inventory: inventory, currentPack: currentPack, loadRewards: gamedata.BattleDeckRewards, - states: make(map[string]*battleState), - } -} - -// AttachCurrentDifficulty selects the GameData quest deck for the active pack. -func (s *Service) AttachCurrentDifficulty(resolve func() (uint64, error)) { - s.currentDifficulty = resolve -} - -func (s *Service) AttachPictorialBuffs(buffs func() ([]gamedata.PictorialBuffStat, error)) { - s.buffs = buffs -} - func checkSeq(request []byte) error { seq, found, err := wire.Varint(request, 1) if err != nil || !found || seq == 0 { @@ -183,16 +18,15 @@ func checkSeq(request []byte) error { return nil } -func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) { +func (s *Service) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { if path != "/BattleEnter" && path != "/BattleStart" && path != "/BattleRetry" && path != "/BattleVerifyState" && path != "/BattleEnd" && path != "/BattleExit" && path != "/BattlePhaseChange" { return 0, nil, false, nil } if err := checkSeq(request); err != nil { return 0, nil, true, err } - s.mu.Lock() - defer s.mu.Unlock() - state := s.stateLocked() + + state := s.stateLocked(ctx) switch path { case "/BattlePhaseChange": if !state.entered { @@ -222,6 +56,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) state.phase++ state.index, state.deck, state.phaseStarted = next.DeckID, next.DeckID, false state.phaseSeq, state.phaseReply = seq, append([]byte(nil), response...) + s.transientVersion++ return 632, response, true, nil case "/BattleVerifyState": if !state.entered { @@ -241,14 +76,14 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) } packID := 0 if s.currentPack != nil { - packID, err = s.currentPack() + packID, err = s.currentPack(ctx) if err != nil { return 0, nil, true, fmt.Errorf("battle: resolve current pack: %w", err) } if packID <= 0 { return 0, nil, true, errors.New("battle: current pack is invalid") } - } else if s.inventory != nil && s.gameDataRoot != "" { + } else if s.gameDataRoot != "" { return 0, nil, true, errors.New("battle: current pack resolver is unavailable") } monster, _, _ := wire.Varint(request, 3) @@ -256,7 +91,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) eventRuntime := s.eventBattle(mode) if eventRuntime != nil { seq, _, _ := wire.Varint(request, 1) - huntResponse, err = eventRuntime.EnterBattle(request, fmt.Sprintf("%s:%d", s.activeSession, seq)) + huntResponse, err = eventRuntime.EnterBattle(ctx, request, fmt.Sprintf("%s:%d", ctx.SessionID, seq)) if err != nil { return 0, nil, true, fmt.Errorf("battle: enter event: %w", err) } @@ -266,14 +101,14 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) return 0, nil, true, errors.New("battle: monster hunt runtime unavailable") } if validator, ok := s.monsterHunt.(interface { - ValidatePack(int, []byte) error + ValidatePack(ctx command.Context, _ int, _ []byte) error }); ok { - if err := validator.ValidatePack(packID, request); err != nil { + if err := validator.ValidatePack(ctx, packID, request); err != nil { return 0, nil, true, err } } seq, _, _ := wire.Varint(request, 1) - huntResponse, err = s.monsterHunt.EnterBattle(request, fmt.Sprintf("%s:%d", s.activeSession, seq)) + huntResponse, err = s.monsterHunt.EnterBattle(ctx, request, fmt.Sprintf("%s:%d", ctx.SessionID, seq)) if err != nil { return 0, nil, true, fmt.Errorf("battle: enter monster hunt: %w", err) } @@ -282,28 +117,35 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) if s.hunting == nil { return 0, nil, true, errors.New("battle: hunting runtime unavailable") } - if err := s.hunting.ValidateBattle(packID, mode, monster, deck); err != nil { + if err := s.hunting.ValidateBattle(ctx, packID, mode, monster, deck); err != nil { return 0, nil, true, err } } if mode == 1 && s.currentDifficulty != nil { - difficulty, resolveErr := s.currentDifficulty() + difficulty, resolveErr := s.currentDifficulty(ctx) if resolveErr != nil { return 0, nil, true, fmt.Errorf("battle: resolve difficulty: %w", resolveErr) } loader := s.loadDifficultyDeck if loader == nil { - loader = gamedata.BattleDeckForDifficulty + loader = gamedata.ResolveQuestBattleDeck } - deck, err = loader(s.gameDataRoot, s.gameDataVersion, packID, deck, difficulty) + selection, err := loader(s.gameDataRoot, s.gameDataVersion, packID, monster, deck, difficulty) if err != nil { return 0, nil, true, fmt.Errorf("battle: select difficulty deck: %w", err) } + if s.validateQuest == nil { + return 0, nil, true, errors.New("battle: quest ownership validator unavailable") + } + if err := s.validateQuest(ctx, packID, selection.QuestIDs); err != nil { + return 0, nil, true, fmt.Errorf("battle: quest ownership: %w", err) + } + deck = selection.DeckID } var phases []gamedata.BattlePhase fieldInstance := "" if mode == 2 && eventRuntime == nil && s.fieldMonsters != nil && monster != 0 { - instance, _, e := s.fieldMonsters.BeginFieldMonsterBattle(packID, monster, deck) + instance, _, e := s.fieldMonsters.BeginFieldMonsterBattle(ctx, packID, monster, deck) if e != nil { return 0, nil, true, e } @@ -325,7 +167,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) response := wire.AppendVarint(nil, 2, deck) response = append(response, huntResponse...) if s.buffs != nil { - buffs, err := s.buffs() + buffs, err := s.buffs(ctx) if err != nil { return 0, nil, true, fmt.Errorf("battle: account pictorial buffs: %w", err) } @@ -341,9 +183,9 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) // The local engine is the normal deterministic engine. response = wire.AppendVarint(response, 6, 1) seq, _, _ := wire.Varint(request, 1) - identity := fmt.Sprintf("%s:%d", s.activeSession, seq) + identity := fmt.Sprintf("%s:%d", ctx.SessionID, seq) if s.consumeFieldBuff != nil { - if err := s.consumeFieldBuff(identity); err != nil { + if err := s.consumeFieldBuff(ctx, identity); err != nil { return 0, nil, true, fmt.Errorf("battle: consume field buff: %w", err) } } @@ -353,6 +195,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) state.enterReceipt = identity state.fieldInstance = fieldInstance state.phases, state.phase, state.phaseStarted, state.phaseSeq, state.phaseReply = phases, 0, false, 0, nil + s.transientVersion++ return 52, response, true, nil case "/BattleRetry": if !state.entered { @@ -383,6 +226,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) state.deck = state.phases[0].DeckID } state.phase, state.phaseStarted, state.phaseSeq, state.phaseReply = 0, false, 0, nil + s.transientVersion++ return 58, response, true, nil case "/BattleStart": if !state.entered { @@ -422,6 +266,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) if nextRound == 1 { state.initialBlue = initialBlue } + s.transientVersion++ // Stable per-battle/round seed; reproducible across retries. seed := index*7919 + state.round*104729 response = wire.AppendVarint(response, 3, seed) @@ -446,25 +291,27 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) } response := wire.AppendVarint(nil, 1, result) if eventRuntime := s.eventBattle(state.mode); eventRuntime != nil { - extra, err := eventRuntime.CompleteBattle(request, state.enterReceipt) + extra, err := eventRuntime.CompleteBattle(ctx, request, state.enterReceipt) if err != nil { return 0, nil, true, fmt.Errorf("battle: settle event: %w", err) } response = append(response, extra...) state.rememberEnd(request, response) state.entered, state.deck, state.pack, state.initialBlue = false, 0, 0, nil + s.transientVersion++ return 15, response, true, nil } if isMonsterHunt(state.mode) { // Monster Hunt owns its remaining HP, progression and daily/season // rewards. Field HP and ordinary pack rewards must not settle here. - extra, err := s.monsterHunt.CompleteBattle(request, state.enterReceipt) + extra, err := s.monsterHunt.CompleteBattle(ctx, request, state.enterReceipt) if err != nil { return 0, nil, true, fmt.Errorf("battle: settle monster hunt: %w", err) } response = append(response, extra...) state.rememberEnd(request, response) state.entered, state.deck, state.pack, state.initialBlue = false, 0, 0, nil + s.transientVersion++ return 15, response, true, nil } var resultCharacters [][]byte @@ -505,7 +352,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) return 0, nil, true, err } if s.commitHealth != nil && len(finishedHealth) != 0 { - if err := s.commitHealth(finishedHealth); err != nil { + if err := s.commitHealth(ctx, finishedHealth); err != nil { return 0, nil, true, fmt.Errorf("battle: persist completed character health: %w", err) } } @@ -522,8 +369,8 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) rewardBundle := false if result == 1 && state.mode == huntingGroundMode { seq, _, _ := wire.Varint(request, 1) - bundle, monsters, err := s.hunting.CompleteBattle(state.pack, state.mode, state.monster, state.deck, - fmt.Sprintf("%s:%d", s.activeSession, seq)) + bundle, monsters, err := s.hunting.CompleteBattle(ctx, state.pack, state.mode, state.monster, state.deck, + fmt.Sprintf("%s:%d", ctx.SessionID, seq)) if err != nil { return 0, nil, true, fmt.Errorf("battle: settle hunting encounter: %w", err) } @@ -533,7 +380,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) response = wire.AppendBytes(response, 5, bundle) rewardBundle = true } - if result == 1 && state.mode != huntingGroundMode && s.inventory != nil && state.monster != 0 && s.gameDataRoot != "" { + if result == 1 && state.mode != huntingGroundMode && state.monster != 0 && s.gameDataRoot != "" { if state.pack <= 0 { return 0, nil, true, errors.New("battle: victory has no locked pack") } @@ -549,20 +396,23 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) if state.fieldInstance != "" { rewardIdentity = state.fieldInstance } - items, grantErr := s.inventory.GrantOnce(rewardIdentity, rewards) + if s.grantRewards == nil { + return 0, nil, true, errors.New("battle: reward runtime unavailable") + } + definitions := make([]gamedata.Reward, len(rewards)) + for i, reward := range rewards { + definitions[i] = gamedata.Reward(reward) + } + bundle, grantErr := s.grantRewards(ctx, rewardIdentity, definitions) if grantErr != nil { return 0, nil, true, grantErr } - var bundle []byte - for _, item := range items { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(item)) - } if len(bundle) != 0 { response = wire.AppendBytes(response, 5, bundle) rewardBundle = true } if state.fieldInstance != "" { - monsterRow, e := s.fieldMonsters.CompleteFieldMonsterBattle(state.pack, state.monster, state.fieldInstance) + monsterRow, e := s.fieldMonsters.CompleteFieldMonsterBattle(ctx, state.pack, state.monster, state.fieldInstance) if e != nil { return 0, nil, true, e } @@ -570,12 +420,12 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) } } if result == 1 && state.monster != 0 && s.onMonsterWin != nil { - if err := s.onMonsterWin(); err != nil { + if err := s.onMonsterWin(ctx); err != nil { return 0, nil, true, fmt.Errorf("battle: monster win mission: %w", err) } } if result == 1 && s.onTutorialWin != nil { - if err := s.onTutorialWin(); err != nil { + if err := s.onTutorialWin(ctx); err != nil { return 0, nil, true, fmt.Errorf("battle: update tutorial kill mission: %w", err) } } @@ -587,9 +437,11 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) } state.rememberEnd(request, response) state.entered, state.deck, state.pack, state.initialBlue = false, 0, 0, nil + s.transientVersion++ return 15, response, true, nil case "/BattleExit": state.entered, state.index, state.round, state.deck, state.pack, state.initialBlue = false, 0, 0, 0, 0, nil + s.transientVersion++ return 388, nil, true, nil } panic("unreachable") @@ -600,19 +452,3 @@ func (state *battleState) rememberEnd(request, response []byte) { state.endRequest = append([]byte(nil), request...) state.endReply = append([]byte(nil), response...) } - -func (s *Service) stateLocked() *battleState { - if s.states == nil { - s.states = make(map[string]*battleState) - } - key := s.activeSession - if key == "" { - key = "__direct_test__" - } - state := s.states[key] - if state == nil { - state = &battleState{} - s.states[key] = state - } - return state -} diff --git a/go/internal/server/domain/command/context.go b/go/internal/server/domain/command/context.go new file mode 100644 index 0000000..7d18b95 --- /dev/null +++ b/go/internal/server/domain/command/context.go @@ -0,0 +1,16 @@ +// Package command identifies an authorized player command independently of transport and runtime. +package command + +import ( + "bd2server/internal/server/storage/stateio" + "context" +) + +type Identity struct{ AccountID, SessionID, RequestID string } +type Context struct { + Identity + Cancellation context.Context + State stateio.AtomicEntryStore +} + +func (c Context) Cancelled() bool { return c.Cancellation != nil && c.Cancellation.Err() != nil } diff --git a/go/internal/server/domain/commerce/attendance.go b/go/internal/server/domain/commerce/attendance.go new file mode 100644 index 0000000..a8c7979 --- /dev/null +++ b/go/internal/server/domain/commerce/attendance.go @@ -0,0 +1,25 @@ +package commerce + +import ( + "bd2server/internal/server/domain/command" + "bd2server/internal/server/storage/stateio" +) + +type attendanceHandler interface { + Handle(ctx command.Context, _ string, _ []byte) (int, []byte, bool, error) +} + +// AttendanceHandler preserves the original event progress response and adds +// subscription grants in extension fields understood by the commerce plugin. +// All operations execute inside the transport's account transaction. +type AttendanceHandler struct { + Events attendanceHandler + Economy *EntitlementEconomy + LoginPasses *LoginPasses + Store stateio.Store +} + +type attendanceReceipt struct { + Digest string `json:"digest"` + Bundle []byte `json:"bundle"` +} diff --git a/go/internal/server/domain/commerce/bonuses.go b/go/internal/server/domain/commerce/bonuses.go new file mode 100644 index 0000000..ae41907 --- /dev/null +++ b/go/internal/server/domain/commerce/bonuses.go @@ -0,0 +1,99 @@ +package commerce + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/storage/stateio" + "crypto/sha256" + "encoding/hex" + "encoding/json" + "fmt" + "sort" +) + +type bonusGroup struct{ ProductGroup, ContentsGroup uint64 } +type bonusPurchases interface { + LifetimePurchaseTotal(ctx command.Context, _ []gamedata.CashProductKey) (uint64, error) +} +type bonusState struct { + Claims map[string]bool `json:"claims"` + Receipts map[string]purchaseReceipt `json:"receipts"` +} +type CashBonuses struct { + store stateio.Store + economy Economy + counts bonusPurchases + groups map[bonusGroup][]gamedata.CashProductKey + rewards map[bonusGroup][]gamedata.CashBonusReward + ordered []bonusGroup + claimIDs map[string]bool +} + +func bonusClaimID(group bonusGroup, id uint64) string { + return fmt.Sprintf("%d:%d:%d", group.ProductGroup, group.ContentsGroup, id) +} + +func NewCashBonuses(ctx command.Context, store stateio.Store, economy Economy, counts bonusPurchases, design *gamedata.CashBonusCatalog, packages []gamedata.CashPackageDesign) (*CashBonuses, error) { + if store == nil || economy == nil || counts == nil || design == nil { + return nil, fmt.Errorf("commerce: missing cash bonus dependency") + } + s := &CashBonuses{store: store, economy: economy, counts: counts, groups: map[bonusGroup][]gamedata.CashProductKey{}, rewards: map[bonusGroup][]gamedata.CashBonusReward{}, claimIDs: map[string]bool{}} + seen := map[gamedata.CashProductKey]bool{} + for _, p := range packages { + // EPackages.BonusBundleGroup=8; unrelated packages can reuse a contents ID. + if p.PackageType != 8 || len(design.Groups[p.ContentsGroupID]) == 0 { + continue + } + k := gamedata.CashProductKey{GroupID: p.GroupID, ProductID: p.ID, SaleGroup: p.SaleGroup} + if seen[k] { + return nil, fmt.Errorf("commerce: duplicate cash bonus product %+v", k) + } + seen[k] = true + group := bonusGroup{p.GroupID, p.ContentsGroupID} + s.groups[group] = append(s.groups[group], k) + s.rewards[group] = design.Groups[p.ContentsGroupID] + } + for group, rows := range s.rewards { + s.ordered = append(s.ordered, group) + for _, row := range rows { + s.claimIDs[bonusClaimID(group, row.ID)] = true + } + } + sort.Slice(s.ordered, func(i, j int) bool { + if s.ordered[i].ProductGroup != s.ordered[j].ProductGroup { + return s.ordered[i].ProductGroup < s.ordered[j].ProductGroup + } + return s.ordered[i].ContentsGroup < s.ordered[j].ContentsGroup + }) + _, err := s.load(ctx) + return s, err +} + +func (s *CashBonuses) load(ctx command.Context) (bonusState, error) { + v := bonusState{Claims: map[string]bool{}, Receipts: map[string]purchaseReceipt{}} + raw, err := s.store.Load(ctx.State, "commerce_cash_bonuses") + if err != nil || raw == nil { + return v, err + } + if err = stateio.RequireExactJSONObject(raw, "claims", "receipts"); err != nil { + return v, err + } + if err = json.Unmarshal(raw, &v); err != nil { + return v, err + } + if v.Claims == nil || v.Receipts == nil { + return v, fmt.Errorf("commerce: malformed cash bonus state") + } + for id, claimed := range v.Claims { + if !claimed || !s.claimIDs[id] { + return v, fmt.Errorf("commerce: invalid cash bonus claim %s", id) + } + } + for id, r := range v.Receipts { + digest, err := hex.DecodeString(r.Digest) + if id == "" || err != nil || len(digest) != sha256.Size || len(r.Response) == 0 { + return v, fmt.Errorf("commerce: invalid cash bonus receipt") + } + } + return v, nil +} diff --git a/go/internal/server/commerce/catalog.go b/go/internal/server/domain/commerce/catalog.go similarity index 98% rename from go/internal/server/commerce/catalog.go rename to go/internal/server/domain/commerce/catalog.go index 6835c73..e9a8811 100644 --- a/go/internal/server/commerce/catalog.go +++ b/go/internal/server/domain/commerce/catalog.go @@ -6,8 +6,8 @@ import ( "math" "sort" - "bd2server/internal/server/gameconfig" - "bd2server/internal/server/gamedata" + "bd2server/internal/server/design/gameconfig" + "bd2server/internal/server/design/gamedata" ) type Product struct { diff --git a/go/internal/server/domain/commerce/clear_packages.go b/go/internal/server/domain/commerce/clear_packages.go new file mode 100644 index 0000000..a947768 --- /dev/null +++ b/go/internal/server/domain/commerce/clear_packages.go @@ -0,0 +1,85 @@ +package commerce + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "fmt" + + "time" +) + +type ClearPackageInventory interface { + All(ctx command.Context) []assets.Item +} +type clearClaim struct{ Kind, GroupID, TicketID, TargetID, Level uint64 } +type clearClaimReceipt struct { + Claim clearClaim `json:"claim"` + Response []byte `json:"response"` +} +type ClearPackages struct { + store stateio.Store + economy Economy + items ClearPackageInventory + design map[clearClaim]gamedata.ClearPackageRewardDesign + packCleared func(uint64, uint64) bool + towerCleared func(uint64, uint64) bool + now func() time.Time +} + +func NewClearPackages(ctx command.Context, store stateio.Store, design *gamedata.ClearPackageCatalog, economy Economy, items ClearPackageInventory) (*ClearPackages, error) { + if store == nil || design == nil || economy == nil || items == nil { + return nil, fmt.Errorf("commerce: invalid clear package dependencies") + } + s := &ClearPackages{store: store, economy: economy, items: items, design: map[clearClaim]gamedata.ClearPackageRewardDesign{}, now: time.Now} + for _, r := range design.Rewards { + k := clearClaim{r.Kind, r.GroupID, r.TicketID, r.TargetID, r.Level} + if _, ok := s.design[k]; ok { + return nil, fmt.Errorf("commerce: duplicate clear reward") + } + s.design[k] = r + } + _, err := s.load(ctx) + return s, err +} +func (s *ClearPackages) AttachProgress(pack, tower func(uint64, uint64) bool) { + s.packCleared = pack + s.towerCleared = tower +} +func (s *ClearPackages) load(ctx command.Context) (map[string]clearClaimReceipt, error) { + v := map[string]clearClaimReceipt{} + raw, err := s.store.Load(ctx.State, "commerce_clear_claims") + if err != nil || raw == nil { + return v, err + } + err = json.Unmarshal(raw, &v) + if err == nil && v == nil { + err = fmt.Errorf("commerce: invalid clear claim state") + } + if err == nil { + for identity, receipt := range v { + if identity != clearClaimID(receipt.Claim) || len(receipt.Response) == 0 { + err = fmt.Errorf("commerce: invalid saved clear claim") + break + } + if _, ok := s.design[receipt.Claim]; !ok { + err = fmt.Errorf("commerce: unknown saved clear claim") + break + } + } + } + return v, err +} +func clearClaimID(c clearClaim) string { + return fmt.Sprintf("clear-package:%d:%d:%d:%d:%d", c.Kind, c.GroupID, c.TicketID, c.TargetID, c.Level) +} +func (s *ClearPackages) entitled(ctx command.Context, ticket uint64) bool { + for _, item := range s.items.All(ctx) { + if item.Type == 19 && item.ID == ticket && item.Count > 0 && (item.ExpiryTime == 0 || item.ExpiryTime > uint64(s.now().UnixMilli())) { + return true + } + } + return false +} diff --git a/go/internal/server/domain/commerce/delivery.go b/go/internal/server/domain/commerce/delivery.go new file mode 100644 index 0000000..87130e9 --- /dev/null +++ b/go/internal/server/domain/commerce/delivery.go @@ -0,0 +1,25 @@ +package commerce + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "fmt" + "time" +) + +type CashMailIssuer interface { + IssueCashOnce(ctx command.Context, _ string, _ uint64, _ []gamedata.Reward, _ time.Time) error +} + +func (e *EntitlementEconomy) AttachCashMail(issuer CashMailIssuer) error { + if issuer == nil { + return fmt.Errorf("commerce: cash mail issuer unavailable") + } + if _, ok := e.graph.(interface { + ResolveDelivery([]gamedata.BattleReward) (gamedata.CashDelivery, error) + }); !ok { + return fmt.Errorf("commerce: cash delivery resolver unavailable") + } + e.mail = issuer + return nil +} diff --git a/go/internal/server/domain/commerce/entitlements.go b/go/internal/server/domain/commerce/entitlements.go new file mode 100644 index 0000000..196b66b --- /dev/null +++ b/go/internal/server/domain/commerce/entitlements.go @@ -0,0 +1,92 @@ +package commerce + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/storage/stateio" + "crypto/sha256" + "encoding/json" + "fmt" + "strconv" + + "time" +) + +type grantedResolver interface { + ResolveGranted([]gamedata.BattleReward) ([]gamedata.BattleReward, error) +} +type entitlementReceipt struct { + Definition []byte `json:"definition"` + Bundle []byte `json:"bundle"` +} +type cashSubscription struct { + Start, Expiry int64 + Claimed uint64 + LastDay string +} +type entitlementState struct { + Receipts map[string]entitlementReceipt `json:"receipts"` + Subscriptions map[string]cashSubscription `json:"subscriptions"` +} +type EntitlementEconomy struct { + store stateio.Store + base Economy + graph grantedResolver + items *assets.Inventory + design *gamedata.CashEntitlementDesign + now func() time.Time + resetSeconds int64 + mail CashMailIssuer +} + +func NewEntitlementEconomy(ctx command.Context, store stateio.Store, base Economy, graph grantedResolver, items *assets.Inventory, design *gamedata.CashEntitlementDesign) (*EntitlementEconomy, error) { + if store == nil || base == nil || graph == nil || items == nil || design == nil { + return nil, fmt.Errorf("commerce: missing entitlement dependency") + } + e := &EntitlementEconomy{store: store, base: base, graph: graph, items: items, design: design, now: time.Now} + _, err := e.load(ctx) + return e, err +} +func (e *EntitlementEconomy) SetClock(now func() time.Time, resetSeconds int64) { + e.now = now + e.resetSeconds = resetSeconds +} +func (e *EntitlementEconomy) day() string { + return e.now().UTC().Add(-time.Duration(e.resetSeconds) * time.Second).Format("2006-01-02") +} +func (e *EntitlementEconomy) load(ctx command.Context) (entitlementState, error) { + s := entitlementState{Receipts: map[string]entitlementReceipt{}, Subscriptions: map[string]cashSubscription{}} + raw, err := e.store.Load(ctx.State, "commerce_entitlements") + if err != nil || raw == nil { + return s, err + } + if err = stateio.RequireExactJSONObject(raw, "receipts", "subscriptions"); err != nil { + return s, err + } + if err = json.Unmarshal(raw, &s); err != nil { + return s, err + } + if s.Receipts == nil || s.Subscriptions == nil { + return s, fmt.Errorf("commerce: malformed entitlement state") + } + for id, r := range s.Receipts { + if id == "" || len(r.Definition) != sha256.Size { + return s, fmt.Errorf("commerce: malformed entitlement receipt") + } + } + for id, sub := range s.Subscriptions { + ticket, err := strconv.ParseUint(id, 10, 64) + if err != nil || len(e.design.Attendance[ticket]) == 0 || sub.Start <= 0 || sub.Expiry != 0 && sub.Expiry < sub.Start || sub.Claimed == 0 || sub.LastDay == "" { + return s, fmt.Errorf("commerce: malformed subscription") + } + } + return s, nil +} +func (e *EntitlementEconomy) save(ctx command.Context, s entitlementState) error { + b, err := json.Marshal(s) + if err != nil { + return err + } + return e.store.Save(ctx.State, "commerce_entitlements", b) +} diff --git a/go/internal/server/domain/commerce/gacha/first_gacha.go b/go/internal/server/domain/commerce/gacha/first_gacha.go new file mode 100644 index 0000000..943d119 --- /dev/null +++ b/go/internal/server/domain/commerce/gacha/first_gacha.go @@ -0,0 +1,38 @@ +package gacha + +import ( + "bd2server/internal/server/design/gamedata" + + assets "bd2server/internal/server/domain/inventory" + "errors" +) + +const firstGachaResponseDigest = "first-gacha-confirm-v1" + +type firstGachaPreview struct { + sequence uint64 + rewards []gamedata.FirstGachaReward + equipment map[int]assets.Equipment + response []byte +} + +func firstGachaEquipment(catalog *gamedata.EquipmentGachaCatalog, id, sortID uint64) (assets.Equipment, error) { + if catalog == nil { + return assets.Equipment{}, errors.New("gacha: first gacha equipment catalog missing") + } + main, sub, private, err := catalog.RollOptions(id) + if err != nil { + return assets.Equipment{}, err + } + entry := assets.Equipment{ID: id, SortID: sortID, Rank: []uint64{0, 0, 0}} + for _, option := range main { + entry.MainOption = append(entry.MainOption, assets.EquipmentOption{GroupID: option.GroupID, ID: option.ID}) + } + for _, option := range sub { + entry.SubOption = append(entry.SubOption, assets.EquipmentOption{GroupID: option.GroupID, ID: option.ID}) + } + if private != nil { + entry.PrivateOption = &assets.EquipmentOption{GroupID: private.GroupID, ID: private.ID} + } + return entry, nil +} diff --git a/go/internal/server/domain/commerce/gacha/gacha.go b/go/internal/server/domain/commerce/gacha/gacha.go new file mode 100644 index 0000000..2769951 --- /dev/null +++ b/go/internal/server/domain/commerce/gacha/gacha.go @@ -0,0 +1,209 @@ +// Package gacha implements server-authoritative draws from verified GameData, +// including the starter reroll and scheduled daily-free batches. +package gacha + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "errors" + "fmt" + + "time" +) + +func (s *Service) infiniteGrant() string { + return fmt.Sprintf("cash-product:%d:%d", s.design.ProductGroupID, s.design.ProductID) +} + +type Service struct { + design *gamedata.InfiniteGachaDesign + first *gamedata.FirstGachaDesign + regular *gamedata.RegularGachaCatalog + collection *roster.CollectionStore + wallet *assets.Wallet + inventory *assets.Inventory + equipmentCatalog *gamedata.EquipmentGachaCatalog + equipmentInventory *assets.EquipmentInventory + onPreview func(ctx command.Context) error + onDraw func(ctx command.Context, _ uint64) error + schedule *ScheduleSeed + previewEventIndex uint64 + + firstPreviews map[string]firstGachaPreview + transientVersion uint64 + now func() time.Time +} + +func (s *Service) TransientVersion() uint64 { + + return s.transientVersion +} + +func (s *Service) AttachDrawMission(callback func(ctx command.Context, _ uint64) error) { + s.onDraw = callback +} + +func (s *Service) AttachPreviewMission(callback func(ctx command.Context) error) { + s.onPreview = callback +} +func (s *Service) AttachInventory(inventory *assets.Inventory) { s.inventory = inventory } +func (s *Service) AttachPreviewEventIndex(ctx command.Context, eventIndex uint64) error { + if eventIndex == 0 { + return errors.New("gacha: invalid preview event index") + } + s.previewEventIndex = eventIndex + return nil +} +func (s *Service) AttachSchedule(ctx command.Context, seed *ScheduleSeed) error { + if err := seed.Validate(seedVersion(seed)); err != nil { + return err + } + copy := *seed + copy.Schedules = append([]ScheduleWindow(nil), seed.Schedules...) + copy.StepUps = append([]ScheduleWindow(nil), seed.StepUps...) + s.schedule = © + return nil +} +func (s *Service) AttachEquipmentGacha(catalog *gamedata.EquipmentGachaCatalog, inventory *assets.EquipmentInventory) { + s.equipmentCatalog, s.equipmentInventory = catalog, inventory +} + +func (s *Service) AttachFirstGacha(ctx command.Context, design *gamedata.FirstGachaDesign) error { + if design == nil || design.GachaID == 0 || design.Count == 0 || design.Group.GachaSubType != 3 { + return errors.New("gacha: invalid first gacha design") + } + s.first = design + + s.firstPreviews = make(map[string]firstGachaPreview) + + return nil +} + +// FirstGachaCompleted reflects UserDBInfo.IsFirstGacha. The official symbol +// map names the client-side property IsDoneFirstGachaPick, and the client sets +// it after a GachaSubType=3 purchase succeeds. +func (s *Service) FirstGachaCompleted() bool { + return s.collection.FirstGachaCompleted() +} + +func NewService(design *gamedata.InfiniteGachaDesign, regular *gamedata.RegularGachaCatalog, collection *roster.CollectionStore, wallet *assets.Wallet) (*Service, error) { + if design == nil || regular == nil || collection == nil || wallet == nil { + return nil, errors.New("gacha: invalid service configuration") + } + return &Service{design: design, regular: regular, collection: collection, wallet: wallet, now: time.Now}, nil +} + +func (s *Service) dailyFreeAllowance(groupID, base uint64) (uint64, bool, error) { + if s.schedule == nil { + return 0, false, nil + } + now := s.now().UnixMilli() + for _, entry := range s.schedule.Schedules { + if entry.GroupID != groupID || now < 0 || uint64(now) < entry.StartTime || uint64(now) > entry.EndTime { + continue + } + if entry.FreeCountBonus { + bonus := s.regular.GachaEventAddFreeCount() + if ^uint64(0)-base < bonus { + return 0, false, errors.New("gacha: daily free allowance overflows") + } + base += bonus + } + return base, true, nil + } + return 0, false, nil +} + +func (s *Service) stepForGacha(gachaID uint64) (uint64, gamedata.GachaStepDesign, bool) { + step, ok := s.regular.StepForGacha(gachaID) + if !ok { + return 0, gamedata.GachaStepDesign{}, false + } + for _, group := range s.regular.StepUps() { + for _, candidate := range group.Steps { + if candidate.GachaID == gachaID && candidate == step { + return group.ID, step, true + } + } + } + return 0, gamedata.GachaStepDesign{}, false +} + +func costumeFixedStates(fixed gamedata.GachaFixedDesign, result gamedata.GachaFixedRoll) []roster.GachaFixedState { + states := make([]roster.GachaFixedState, 0, 2) + if fixed.CostumeGrade4Count != 0 { + states = append(states, roster.GachaFixedState{ + FixedID: fixed.ID, Type: 0, Count: result.CostumeGrade4Count, ApplySort: result.CostumeGrade4Sort, + }) + } + if fixed.CostumeGrade5Count != 0 { + states = append(states, roster.GachaFixedState{ + FixedID: fixed.ID, Type: 1, Count: result.CostumeGrade5Count, ApplySort: result.CostumeGrade5Sort, + }) + } + return states +} + +func (s *Service) completeSharedFixedStates(fixedID uint64, updates []roster.GachaFixedState) []roster.GachaFixedState { + if fixedID == 0 || s.equipmentCatalog == nil { + return updates + } + known := false + for _, fixed := range s.equipmentCatalog.FixedDesigns() { + if fixed.ID == fixedID { + known = true + break + } + } + if !known { + return updates + } + states := make([]roster.GachaFixedState, 4) + for fixedType := range states { + states[fixedType] = roster.GachaFixedState{ + FixedID: fixedID, Type: uint64(fixedType), + Count: s.collection.GachaFixedCount(fixedID, uint64(fixedType)), ApplySort: -1, + } + } + for _, update := range updates { + if update.FixedID != fixedID || update.Type >= uint64(len(states)) { + return updates + } + states[update.Type] = update + } + return states +} + +func (s *Service) loginIdentity(ctx command.Context) string { + return ctx.SessionID +} + +func (s *Service) requestIdentity(ctx command.Context, gachaID, seq uint64) string { + return fmt.Sprintf("regular-gacha:%d:session:%s:seq:%d", gachaID, ctx.SessionID, seq) +} + +func (s *Service) creditOverflow(ctx command.Context, identity string, grant roster.CollectionGrant) error { + var mileage uint64 + for _, exchange := range grant.Exchanges { + if exchange.ExchangeItemType != 20 { + return fmt.Errorf("gacha: unsupported costume overflow type %d", exchange.ExchangeItemType) + } + if ^uint64(0)-mileage < exchange.ExchangeCount { + return errors.New("gacha: costume overflow mileage overflow") + } + mileage += exchange.ExchangeCount + } + if mileage == 0 { + return nil + } + _, err := s.wallet.GrantMileageOnce(ctx, identity, mileage) + return err +} + +// The client converts Unix time to UTC+9 and applies GameData's +// 09:00:00 DailyResetTime. That boundary is exactly 00:00 UTC. Keeping the +// key independent of the host OS timezone prevents an early reset on a +// Chinese development machine. +func dailyResetKey(now time.Time) string { return now.UTC().Format("2006-01-02") } diff --git a/go/internal/server/gacha/gacha.go b/go/internal/server/domain/commerce/gacha/protocol_adapter.go similarity index 73% rename from go/internal/server/gacha/gacha.go rename to go/internal/server/domain/commerce/gacha/protocol_adapter.go index 48441b3..da3fb53 100644 --- a/go/internal/server/gacha/gacha.go +++ b/go/internal/server/domain/commerce/gacha/protocol_adapter.go @@ -1,112 +1,25 @@ -// Package gacha implements server-authoritative draws from verified GameData, -// including the starter reroll and scheduled daily-free batches. package gacha import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "bd2server/internal/server/protocol/wire" "crypto/sha256" "encoding/binary" "errors" "fmt" "slices" - "sync" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" ) -func (s *Service) infiniteGrant() string { - return fmt.Sprintf("cash-product:%d:%d", s.design.ProductGroupID, s.design.ProductID) -} - -type Service struct { - design *gamedata.InfiniteGachaDesign - first *gamedata.FirstGachaDesign - regular *gamedata.RegularGachaCatalog - collection *player.CollectionStore - wallet *player.Wallet - inventory *player.Inventory - equipmentCatalog *gamedata.EquipmentGachaCatalog - equipmentInventory *player.EquipmentInventory - onPreview func() error - onDraw func(uint64) error - schedule *ScheduleSeed - previewEventIndex uint64 - sessionMu sync.RWMutex - sessionID string - firstMu sync.Mutex - firstPreviews map[string]firstGachaPreview - now func() time.Time -} - -func (s *Service) AttachDrawMission(callback func(uint64) error) { s.onDraw = callback } - -func (s *Service) AttachPreviewMission(callback func() error) { s.onPreview = callback } -func (s *Service) AttachInventory(inventory *player.Inventory) { s.inventory = inventory } -func (s *Service) AttachPreviewEventIndex(eventIndex uint64) error { - if eventIndex == 0 { - return errors.New("gacha: invalid preview event index") - } - s.previewEventIndex = eventIndex - return nil -} -func (s *Service) AttachSchedule(seed *ScheduleSeed) error { - if err := seed.Validate(seedVersion(seed)); err != nil { - return err - } - copy := *seed - copy.Schedules = append([]ScheduleWindow(nil), seed.Schedules...) - copy.StepUps = append([]ScheduleWindow(nil), seed.StepUps...) - s.schedule = © - return nil -} -func (s *Service) AttachEquipmentGacha(catalog *gamedata.EquipmentGachaCatalog, inventory *player.EquipmentInventory) { - s.equipmentCatalog, s.equipmentInventory = catalog, inventory -} - -func (s *Service) AttachFirstGacha(design *gamedata.FirstGachaDesign) error { - if design == nil || design.GachaID == 0 || design.Count == 0 || design.Group.GachaSubType != 3 { - return errors.New("gacha: invalid first gacha design") - } - s.first = design - s.firstMu.Lock() - s.firstPreviews = make(map[string]firstGachaPreview) - s.firstMu.Unlock() - return nil -} - -// FirstGachaCompleted reflects UserDBInfo.IsFirstGacha. The official symbol -// map names the client-side property IsDoneFirstGachaPick, and the client sets -// it after a GachaSubType=3 purchase succeeds. -func (s *Service) FirstGachaCompleted() bool { - return s.collection.FirstGachaCompleted() -} - -// BeginSession activates the transport's current login. The transport calls -// it before every authenticated request, not only when the login is created. -// Request sequence numbers restart with the client, so the login ID remains -// part of durable idempotency keys. In-memory first-gacha previews are already -// keyed by that ID and must survive activation: preview and confirmation are -// separate requests, and clearing here would discard the result immediately -// before the client confirms it. -func (s *Service) BeginSession(id string) { - s.sessionMu.Lock() - s.sessionID = id - s.sessionMu.Unlock() -} - -func NewService(design *gamedata.InfiniteGachaDesign, regular *gamedata.RegularGachaCatalog, collection *player.CollectionStore, wallet *player.Wallet) (*Service, error) { - if design == nil || regular == nil || collection == nil || wallet == nil { - return nil, errors.New("gacha: invalid service configuration") - } - return &Service{design: design, regular: regular, collection: collection, wallet: wallet, now: time.Now}, nil -} - -func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) { +func (s *Service) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { if path != "/GachaInfo" && path != "/GachaBuy" && path != "/GachaMultiBuy" && path != "/GachaPointExchange" && path != "/GachaPointManualExchange" && path != "/GachaBuyPreview" && path != "/GachaBuyPreviewLock" && path != "/GachaSelectionSave" && path != "/CashShopPurchaseCountInfo" && path != "/CashShopBuy" { return 0, nil, false, nil } + if ctx.SessionID == "" && path != "/GachaInfo" && path != "/CashShopPurchaseCountInfo" { + return 0, nil, true, fmt.Errorf("gacha: %s requires an authenticated session", path) + } seq, found, err := wire.Varint(request, 1) if err != nil || !found || seq == 0 { return 0, nil, true, fmt.Errorf("gacha: %s invalid sequence", path) @@ -115,27 +28,27 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) case "/GachaInfo": return 145, s.gachaInfo(), true, nil case "/GachaBuyPreview": - return s.preview(request) + return s.preview(ctx, request) case "/GachaBuyPreviewLock": - return s.lockPreview(request) + return s.lockPreview(ctx, request) case "/GachaBuy": - return s.buy(request, seq) + return s.buy(ctx, request, seq) case "/GachaMultiBuy": - return s.multiBuy(request, seq) + return s.multiBuy(ctx, request, seq) case "/GachaPointExchange": - return s.pointExchange(request, seq) + return s.pointExchange(ctx, request, seq) case "/GachaPointManualExchange": - return s.manualPointExchange(request, seq) + return s.manualPointExchange(ctx, request, seq) case "/GachaSelectionSave": - return s.saveSelection(request) + return s.saveSelection(ctx, request) case "/CashShopPurchaseCountInfo": var response []byte - for _, count := range s.PurchaseCountDBInfos() { + for _, count := range s.PurchaseCountDBInfos(ctx) { response = wire.AppendBytes(response, 1, count) } return 432, response, true, nil case "/CashShopBuy": - return s.confirm(request) + return s.confirm(ctx, request) default: return 0, nil, false, nil } @@ -144,7 +57,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) // PurchaseCountDBInfos returns the same persisted cash-product purchase state // used by /CashShopPurchaseCountInfo. LoginUser consumes this method too, so // both protocol surfaces remain consistent after a restart. -func (s *Service) PurchaseCountDBInfos() [][]byte { +func (s *Service) PurchaseCountDBInfos(ctx command.Context) [][]byte { products := []struct { group, id uint64 bought bool @@ -177,8 +90,8 @@ func (s *Service) PurchaseCountDBInfos() [][]byte { return result } -func (s *Service) saveSelection(request []byte) (int, []byte, bool, error) { - var selections []player.GachaSelection +func (s *Service) saveSelection(ctx command.Context, request []byte) (int, []byte, bool, error) { + var selections []roster.GachaSelection if err := wire.Walk(request, func(field wire.Field) error { if field.Number != 2 { return nil @@ -198,7 +111,7 @@ func (s *Service) saveSelection(request []byte) (int, []byte, bool, error) { if err != nil || !itemFound { return errors.New("gacha: selection item missing") } - selections = append(selections, player.GachaSelection{GroupID: groupID, Slot: slot, ItemID: itemID}) + selections = append(selections, roster.GachaSelection{GroupID: groupID, Slot: slot, ItemID: itemID}) return nil }); err != nil { return 198, nil, true, err @@ -237,18 +150,18 @@ func (s *Service) saveSelection(request []byte) (int, []byte, bool, error) { seenSlots[selection.Slot] = true seenItems[selection.ItemID] = true } - if err := s.collection.SaveGachaSelections(groupID, selections, changeLimit); err != nil { + if err := s.collection.SaveGachaSelections(ctx, groupID, selections, changeLimit); err != nil { return 198, nil, true, err } return 198, nil, true, nil } -func (s *Service) lockPreview(request []byte) (int, []byte, bool, error) { +func (s *Service) lockPreview(ctx command.Context, request []byte) (int, []byte, bool, error) { eventIndex, found, err := wire.Varint(request, 2) if err != nil || !found || eventIndex == 0 { return 175, nil, true, errors.New("gacha: missing preview event index") } - if err := s.collection.LockPreview(eventIndex); err != nil { + if err := s.collection.LockPreview(ctx, eventIndex); err != nil { return 175, nil, true, err } // no separate PacketCodeTypeProto member for the lock RPC. @@ -257,7 +170,7 @@ func (s *Service) lockPreview(request []byte) (int, []byte, bool, error) { return 175, nil, true, nil } -func (s *Service) multiBuy(request []byte, seq uint64) (int, []byte, bool, error) { +func (s *Service) multiBuy(ctx command.Context, request []byte, seq uint64) (int, []byte, bool, error) { const maxBatchSize = 256 var ids []uint64 appendID := func(id uint64) error { @@ -299,13 +212,13 @@ func (s *Service) multiBuy(request []byte, seq uint64) (int, []byte, bool, error if len(ids) == 0 { return 146, nil, true, errors.New("gacha: empty daily free batch") } - identity := fmt.Sprintf("multi-gacha:session:%s:seq:%d", s.loginIdentity(), seq) + identity := fmt.Sprintf("multi-gacha:session:%s:seq:%d", s.loginIdentity(ctx), seq) var canonical []byte for _, id := range ids { canonical = binary.AppendUvarint(canonical, id) } digest := fmt.Sprintf("%x", sha256.Sum256(canonical)) - if response, found, err := s.collection.GachaBatchResponse(identity, digest); found || err != nil { + if response, found, err := s.collection.GachaBatchResponse(ctx, identity, digest); found || err != nil { return 146, response, true, err } // Validate the complete batch before a single allowance or reward changes. @@ -348,7 +261,7 @@ func (s *Service) multiBuy(request []byte, seq uint64) (int, []byte, bool, error for slot, id := range ids { subrequest := wire.AppendVarint(wire.AppendVarint(nil, 1, seq), 2, id) slotIdentity := fmt.Sprintf("%s:slot:%d:id:%d", identity, slot, id) - _, single, _, err := s.buyWithIdentity(subrequest, seq, slotIdentity) + _, single, _, err := s.buyWithIdentity(ctx, subrequest, seq, slotIdentity) if err != nil { return 146, nil, true, fmt.Errorf("gacha: daily free batch slot %d: %w", slot, err) } @@ -373,38 +286,17 @@ func (s *Service) multiBuy(request []byte, seq uint64) (int, []byte, bool, error } response = wire.AppendBytes(response, 1, result) } - if err := s.collection.RecordGachaBatch(identity, digest, response); err != nil { + if err := s.collection.RecordGachaBatch(ctx, identity, digest, response); err != nil { return 146, nil, true, err } return 146, response, true, nil } -func (s *Service) dailyFreeAllowance(groupID, base uint64) (uint64, bool, error) { - if s.schedule == nil { - return 0, false, nil - } - now := s.now().UnixMilli() - for _, entry := range s.schedule.Schedules { - if entry.GroupID != groupID || now < 0 || uint64(now) < entry.StartTime || uint64(now) > entry.EndTime { - continue - } - if entry.FreeCountBonus { - bonus := s.regular.GachaEventAddFreeCount() - if ^uint64(0)-base < bonus { - return 0, false, errors.New("gacha: daily free allowance overflows") - } - base += bonus - } - return base, true, nil - } - return 0, false, nil +func (s *Service) buy(ctx command.Context, request []byte, seq uint64) (int, []byte, bool, error) { + return s.buyWithIdentity(ctx, request, seq, "") } -func (s *Service) buy(request []byte, seq uint64) (int, []byte, bool, error) { - return s.buyWithIdentity(request, seq, "") -} - -func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) (int, []byte, bool, error) { +func (s *Service) buyWithIdentity(ctx command.Context, request []byte, seq uint64, identity string) (int, []byte, bool, error) { id, found, err := wire.Varint(request, 2) if err != nil || !found || id == 0 { return 146, nil, true, errors.New("gacha: missing gacha id") @@ -413,7 +305,7 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( if err != nil || buyType > 3 { return 146, nil, true, fmt.Errorf("gacha: unsupported buy type %d", buyType) } - var tickets []player.Item + var tickets []assets.Item if err := wire.Walk(request, func(field wire.Field) error { if field.Number != 4 { return nil @@ -421,7 +313,7 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( if field.Type != 2 { return errors.New("gacha: invalid ticket info") } - var item player.Item + var item assets.Item var found bool var parseErr error if item.InvenIndex, found, parseErr = wire.Varint(field.Value, 1); parseErr != nil || !found { @@ -448,12 +340,12 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( return 146, nil, true, err } if s.first != nil && id == s.first.GachaID { - return s.confirmFirstGacha(request, seq, buyType, tickets, identity) + return s.confirmFirstGacha(ctx, request, seq, buyType, tickets, identity) } design, ok := s.regular.Gacha(id) if !ok && s.equipmentCatalog != nil { if equipment, found := s.equipmentCatalog.Gacha(id); found { - return s.buyEquipmentWithIdentity(seq, buyType, tickets, equipment, identity) + return s.buyEquipmentWithIdentity(ctx, seq, buyType, tickets, equipment, identity) } } if !ok { @@ -494,7 +386,7 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( } } if identity == "" { - identity = s.requestIdentity(id, seq) + identity = s.requestIdentity(ctx, id, seq) } grant, already := s.collection.Grant(identity) if isMoonrise && !already { @@ -542,7 +434,7 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( return 146, nil, true, errors.New("gacha: invalid ticket count") } var roll []uint64 - var fixedStates []player.GachaFixedState + var fixedStates []roster.GachaFixedState var selectionApplySortIDs []uint64 if isMoonrise { roll, err = design.RollSpecialSelection(moonriseSelected) @@ -585,16 +477,16 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( if s.inventory == nil { return 146, nil, true, errors.New("gacha: inventory not attached") } - if err := s.inventory.CanConsume(tickets); err != nil { + if err := s.inventory.CanConsume(ctx, tickets); err != nil { return 146, nil, true, err } } - var moonriseCost []player.Item + var moonriseCost []assets.Item if isMoonrise { if s.inventory == nil { return 146, nil, true, errors.New("gacha: inventory not attached") } - moonriseCost, err = s.inventory.SelectMutable(design.PriceType, design.PriceID, design.Price) + moonriseCost, err = s.inventory.SelectMutable(ctx, design.PriceType, design.PriceID, design.Price) if err != nil { return 146, nil, true, err } @@ -610,7 +502,7 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( chargePaid = design.PriceType == 2 } if charge != 0 { - currency := s.wallet.Snapshot() + currency := s.wallet.Snapshot(ctx) if chargePaid && currency.Jewelry < charge { return 146, nil, true, errors.New("gacha: insufficient paid jewelry") } @@ -619,7 +511,7 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( } } if len(tickets) != 0 { - if err := s.inventory.Consume(tickets); err != nil { + if err := s.inventory.Consume(ctx, tickets); err != nil { return 146, nil, true, err } } @@ -627,27 +519,27 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( case isDailyFree: // The daily allowance is the complete cost. case isDailyPaid: - if _, err := s.wallet.SpendJewelryOnce(identity, charge); err != nil { + if _, err := s.wallet.SpendJewelryOnce(ctx, identity, charge); err != nil { return 146, nil, true, err } case isMoonrise: - if err := s.inventory.Consume(moonriseCost); err != nil { + if err := s.inventory.Consume(ctx, moonriseCost); err != nil { return 146, nil, true, err } default: if charge != 0 { if chargePaid { - if _, err := s.wallet.SpendJewelryOnce(identity, charge); err != nil { + if _, err := s.wallet.SpendJewelryOnce(ctx, identity, charge); err != nil { return 146, nil, true, err } } else { - if _, err := s.wallet.SpendFreeJewelryOnce(identity, charge); err != nil { + if _, err := s.wallet.SpendFreeJewelryOnce(ctx, identity, charge); err != nil { return 146, nil, true, err } } } } - purchase := player.GachaPurchase{Group: group, BuyType: buyType, Fixed: fixedStates, SelectionApplySortIDs: selectionApplySortIDs} + purchase := roster.GachaPurchase{Group: group, BuyType: buyType, Fixed: fixedStates, SelectionApplySortIDs: selectionApplySortIDs} if isMoonrise { purchase.CompletionGrant = completionGrant } @@ -659,12 +551,12 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( purchase.StepUpGroupID = stepUpGroupID purchase.StepUpStep = stepDesign.Step } - grant, err = s.collection.GrantRegularPurchase(identity, roll, s.regular, purchase) + grant, err = s.collection.GrantRegularPurchase(ctx, identity, roll, s.regular, purchase) if err != nil { return 146, nil, true, err } } - if err := s.creditOverflow(identity, grant); err != nil { + if err := s.creditOverflow(ctx, identity, grant); err != nil { return 146, nil, true, err } response := wire.AppendBytes(nil, 1, s.rewardBundle(grant)) @@ -676,32 +568,17 @@ func (s *Service) buyWithIdentity(request []byte, seq uint64, identity string) ( response = wire.AppendVarint(response, 4, sortID) } if !already && s.onDraw != nil { - if err := s.onDraw(uint64(design.Count)); err != nil { + if err := s.onDraw(ctx, uint64(design.Count)); err != nil { return 146, nil, true, fmt.Errorf("gacha: draw mission: %w", err) } } return 146, response, true, nil } -func (s *Service) stepForGacha(gachaID uint64) (uint64, gamedata.GachaStepDesign, bool) { - step, ok := s.regular.StepForGacha(gachaID) - if !ok { - return 0, gamedata.GachaStepDesign{}, false - } - for _, group := range s.regular.StepUps() { - for _, candidate := range group.Steps { - if candidate.GachaID == gachaID && candidate == step { - return group.ID, step, true - } - } - } - return 0, gamedata.GachaStepDesign{}, false -} - // Scheduled equipment draws update GachaUser/GachaFixed accounting. Standalone // ticket-only draws have no schedule group and persist only their idempotency // marker plus the generated equipment instances. -func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player.Item, design gamedata.EquipmentGacha, identity string) (int, []byte, bool, error) { +func (s *Service) buyEquipmentWithIdentity(ctx command.Context, seq, buyType uint64, tickets []assets.Item, design gamedata.EquipmentGacha, identity string) (int, []byte, bool, error) { if s.equipmentInventory == nil { return 146, nil, true, errors.New("gacha: equipment inventory not attached") } @@ -738,10 +615,10 @@ func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player } } if identity == "" { - identity = s.requestIdentity(design.ID, seq) + identity = s.requestIdentity(ctx, design.ID, seq) } grant, already := s.collection.Grant(identity) - var entries []player.Equipment + var entries []assets.Equipment if already { for sort := 0; sort < design.Count; sort++ { entry, found := s.equipmentInventory.Granted(fmt.Sprintf("%s:equip:%d", identity, sort)) @@ -765,7 +642,7 @@ func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player if s.inventory == nil { return 146, nil, true, errors.New("gacha: inventory not attached") } - if err := s.inventory.Consume(tickets); err != nil { + if err := s.inventory.Consume(ctx, tickets); err != nil { return 146, nil, true, err } } @@ -773,7 +650,7 @@ func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player if design.TicketOnly { return 146, nil, true, errors.New("gacha: ticket-only equipment draw cannot use diamonds") } - if _, err := s.wallet.SpendFreeJewelryOnce(identity, design.Price/uint64(design.Count)*remain); err != nil { + if _, err := s.wallet.SpendFreeJewelryOnce(ctx, identity, design.Price/uint64(design.Count)*remain); err != nil { return 146, nil, true, err } } @@ -795,9 +672,9 @@ func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player if err != nil { return 146, nil, true, err } - var fixedStates []player.GachaFixedState + var fixedStates []roster.GachaFixedState if !design.TicketOnly { - fixedStates = []player.GachaFixedState{{FixedID: fixed.ID, Type: 2, Count: state.SRCount, ApplySort: state.SRSort}, {FixedID: fixed.ID, Type: 3, Count: state.URCount, ApplySort: state.URSort}} + fixedStates = []roster.GachaFixedState{{FixedID: fixed.ID, Type: 2, Count: state.SRCount, ApplySort: state.SRSort}, {FixedID: fixed.ID, Type: 3, Count: state.URCount, ApplySort: state.URSort}} fixedStates = s.completeSharedFixedStates(fixed.ID, fixedStates) } for sort, equipmentID := range roll { @@ -805,30 +682,30 @@ func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player if err != nil { return 146, nil, true, err } - entry := player.Equipment{ID: equipmentID, SortID: uint64(sort), Rank: []uint64{0, 0, 0}} + entry := assets.Equipment{ID: equipmentID, SortID: uint64(sort), Rank: []uint64{0, 0, 0}} for _, o := range main { - entry.MainOption = append(entry.MainOption, player.EquipmentOption{GroupID: o.GroupID, ID: o.ID}) + entry.MainOption = append(entry.MainOption, assets.EquipmentOption{GroupID: o.GroupID, ID: o.ID}) } for _, o := range sub { - entry.SubOption = append(entry.SubOption, player.EquipmentOption{GroupID: o.GroupID, ID: o.ID}) + entry.SubOption = append(entry.SubOption, assets.EquipmentOption{GroupID: o.GroupID, ID: o.ID}) } if private != nil { - entry.PrivateOption = &player.EquipmentOption{GroupID: private.GroupID, ID: private.ID} + entry.PrivateOption = &assets.EquipmentOption{GroupID: private.GroupID, ID: private.ID} } - saved, err := s.equipmentInventory.GrantGeneratedOnce(fmt.Sprintf("%s:equip:%d", identity, sort), entry) + saved, err := s.equipmentInventory.GrantGeneratedOnce(ctx, fmt.Sprintf("%s:equip:%d", identity, sort), entry) if err != nil { return 146, nil, true, err } entries = append(entries, saved) } if design.TicketOnly { - grant, err = s.collection.GrantEquipmentDraw(identity) + grant, err = s.collection.GrantEquipmentDraw(ctx, identity) } else { - purchase := player.GachaPurchase{Group: gamedata.GachaGroupDesign{ID: group.ID, PointCount: group.PointCount}, BuyType: buyType, Fixed: fixedStates} + purchase := roster.GachaPurchase{Group: gamedata.GachaGroupDesign{ID: group.ID, PointCount: group.PointCount}, BuyType: buyType, Fixed: fixedStates} if isDailyFree { purchase.DailyKey, purchase.DailyLimit = dailyResetKey(s.now()), freeDailyLimit } - grant, err = s.collection.GrantEquipmentPurchase(identity, uint64(len(entries)), purchase) + grant, err = s.collection.GrantEquipmentPurchase(ctx, identity, uint64(len(entries)), purchase) } if err != nil { return 146, nil, true, err @@ -836,7 +713,7 @@ func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player } bundle := []byte{} for _, entry := range entries { - bundle = wire.AppendBytes(bundle, 4, player.EquipmentWire(entry)) + bundle = wire.AppendBytes(bundle, 4, assets.EquipmentWire(entry)) } response := wire.AppendBytes(nil, 1, bundle) response = wire.AppendVarint(response, 2, grant.GachaPoint) @@ -844,14 +721,14 @@ func (s *Service) buyEquipmentWithIdentity(seq, buyType uint64, tickets []player response = wire.AppendBytes(response, 3, gachaFixedWire(fixed)) } if !already && s.onDraw != nil { - if err := s.onDraw(uint64(design.Count)); err != nil { + if err := s.onDraw(ctx, uint64(design.Count)); err != nil { return 146, nil, true, fmt.Errorf("gacha: equipment draw mission: %w", err) } } return 146, response, true, nil } -func gachaFixedWire(fixed player.GachaFixedState) []byte { +func gachaFixedWire(fixed roster.GachaFixedState) []byte { info := wire.AppendVarint(nil, 1, fixed.FixedID) info = wire.AppendVarint(info, 2, fixed.Type) info = wire.AppendVarint(info, 3, fixed.Count) @@ -860,52 +737,7 @@ func gachaFixedWire(fixed player.GachaFixedState) []byte { return wire.AppendVarint(info, 4, uint64(int64(fixed.ApplySort))) } -func costumeFixedStates(fixed gamedata.GachaFixedDesign, result gamedata.GachaFixedRoll) []player.GachaFixedState { - states := make([]player.GachaFixedState, 0, 2) - if fixed.CostumeGrade4Count != 0 { - states = append(states, player.GachaFixedState{ - FixedID: fixed.ID, Type: 0, Count: result.CostumeGrade4Count, ApplySort: result.CostumeGrade4Sort, - }) - } - if fixed.CostumeGrade5Count != 0 { - states = append(states, player.GachaFixedState{ - FixedID: fixed.ID, Type: 1, Count: result.CostumeGrade5Count, ApplySort: result.CostumeGrade5Sort, - }) - } - return states -} - -func (s *Service) completeSharedFixedStates(fixedID uint64, updates []player.GachaFixedState) []player.GachaFixedState { - if fixedID == 0 || s.equipmentCatalog == nil { - return updates - } - known := false - for _, fixed := range s.equipmentCatalog.FixedDesigns() { - if fixed.ID == fixedID { - known = true - break - } - } - if !known { - return updates - } - states := make([]player.GachaFixedState, 4) - for fixedType := range states { - states[fixedType] = player.GachaFixedState{ - FixedID: fixedID, Type: uint64(fixedType), - Count: s.collection.GachaFixedCount(fixedID, uint64(fixedType)), ApplySort: -1, - } - } - for _, update := range updates { - if update.FixedID != fixedID || update.Type >= uint64(len(states)) { - return updates - } - states[update.Type] = update - } - return states -} - -func (s *Service) manualPointExchange(request []byte, seq uint64) (int, []byte, bool, error) { +func (s *Service) manualPointExchange(ctx command.Context, request []byte, seq uint64) (int, []byte, bool, error) { groupID, found, err := wire.Varint(request, 2) if err != nil || !found || groupID == 0 { return 188, nil, true, errors.New("gacha: missing point exchange group") @@ -918,12 +750,12 @@ func (s *Service) manualPointExchange(request []byte, seq uint64) (int, []byte, if !ok || group.PointCount == 0 { return 188, nil, true, fmt.Errorf("gacha: group %d does not support point exchange", groupID) } - identity := fmt.Sprintf("gacha-point-manual:%s:%d:seq:%d", s.loginIdentity(), groupID, seq) - exchange, err := s.collection.ExchangeGachaPoint(identity, groupID, count) + identity := fmt.Sprintf("gacha-point-manual:%s:%d:seq:%d", s.loginIdentity(ctx), groupID, seq) + exchange, err := s.collection.ExchangeGachaPoint(ctx, identity, groupID, count) if err != nil { return 188, nil, true, err } - if _, err := s.wallet.GrantHopePowderOnce(identity, exchange.Count); err != nil { + if _, err := s.wallet.GrantHopePowderOnce(ctx, identity, exchange.Count); err != nil { return 188, nil, true, err } // Manual conversion grants account currency type 22, not a stackable @@ -934,7 +766,7 @@ func (s *Service) manualPointExchange(request []byte, seq uint64) (int, []byte, return 188, wire.AppendBytes(nil, 1, bundle), true, nil } -func (s *Service) pointExchange(request []byte, seq uint64) (int, []byte, bool, error) { +func (s *Service) pointExchange(ctx command.Context, request []byte, seq uint64) (int, []byte, bool, error) { groupID, found, err := wire.Varint(request, 2) if err != nil || !found || groupID == 0 { return 147, nil, true, errors.New("gacha: missing point exchange group") @@ -945,7 +777,7 @@ func (s *Service) pointExchange(request []byte, seq uint64) (int, []byte, bool, } if s.equipmentCatalog != nil { if group, ok := s.equipmentCatalog.Group(groupID); ok { - return s.pointExchangeEquipment(group, selectedItemID, seq) + return s.pointExchangeEquipment(ctx, group, selectedItemID, seq) } } group, ok := s.regular.Group(groupID) @@ -966,18 +798,18 @@ func (s *Service) pointExchange(request []byte, seq uint64) (int, []byte, bool, if _, ok := s.regular.Character(costumeID); !ok { return 147, nil, true, fmt.Errorf("gacha: pickup costume %d has no character design", costumeID) } - identity := fmt.Sprintf("gacha-point-costume:%s:%d:%d:seq:%d", s.loginIdentity(), groupID, costumeID, seq) - grant, err := s.collection.GrantGachaPointCostume(identity, group, costumeID, s.regular) + identity := fmt.Sprintf("gacha-point-costume:%s:%d:%d:seq:%d", s.loginIdentity(ctx), groupID, costumeID, seq) + grant, err := s.collection.GrantGachaPointCostume(ctx, identity, group, costumeID, s.regular) if err != nil { return 147, nil, true, err } - if err := s.creditOverflow(identity, grant); err != nil { + if err := s.creditOverflow(ctx, identity, grant); err != nil { return 147, nil, true, err } return 147, wire.AppendBytes(nil, 1, s.rewardBundle(grant)), true, nil } -func (s *Service) pointExchangeEquipment(group gamedata.EquipmentGachaGroup, selectedItemID, seq uint64) (int, []byte, bool, error) { +func (s *Service) pointExchangeEquipment(ctx command.Context, group gamedata.EquipmentGachaGroup, selectedItemID, seq uint64) (int, []byte, bool, error) { if s.equipmentInventory == nil { return 147, nil, true, errors.New("gacha: equipment inventory not attached") } @@ -990,14 +822,14 @@ func (s *Service) pointExchangeEquipment(group gamedata.EquipmentGachaGroup, sel if selectedItemID != 0 { return 147, nil, true, fmt.Errorf("gacha: equipment group %d does not support pickup selection", group.ID) } - identity := fmt.Sprintf("gacha-point-equipment:%s:%d:%d:seq:%d", s.loginIdentity(), group.ID, group.PickUpItemID, seq) + identity := fmt.Sprintf("gacha-point-equipment:%s:%d:%d:seq:%d", s.loginIdentity(ctx), group.ID, group.PickUpItemID, seq) equipmentIdentity := identity + ":equip" if _, already := s.collection.Grant(identity); already { entry, found := s.equipmentInventory.Granted(equipmentIdentity) if !found { return 147, nil, true, fmt.Errorf("gacha: equipment exchange retry %s is missing its instance", identity) } - bundle := wire.AppendBytes(nil, 4, player.EquipmentWire(entry)) + bundle := wire.AppendBytes(nil, 4, assets.EquipmentWire(entry)) return 147, wire.AppendBytes(nil, 1, bundle), true, nil } if s.collection.GachaUser(group.ID).Point < group.PickUpExchangeCost { @@ -1007,55 +839,34 @@ func (s *Service) pointExchangeEquipment(group gamedata.EquipmentGachaGroup, sel if err != nil { return 147, nil, true, err } - entry := player.Equipment{ID: group.PickUpItemID, Rank: []uint64{0, 0, 0}} + entry := assets.Equipment{ID: group.PickUpItemID, Rank: []uint64{0, 0, 0}} for _, option := range main { - entry.MainOption = append(entry.MainOption, player.EquipmentOption{GroupID: option.GroupID, ID: option.ID}) + entry.MainOption = append(entry.MainOption, assets.EquipmentOption{GroupID: option.GroupID, ID: option.ID}) } for _, option := range sub { - entry.SubOption = append(entry.SubOption, player.EquipmentOption{GroupID: option.GroupID, ID: option.ID}) + entry.SubOption = append(entry.SubOption, assets.EquipmentOption{GroupID: option.GroupID, ID: option.ID}) } if private != nil { - entry.PrivateOption = &player.EquipmentOption{GroupID: private.GroupID, ID: private.ID} + entry.PrivateOption = &assets.EquipmentOption{GroupID: private.GroupID, ID: private.ID} } - saved, err := s.equipmentInventory.GrantGeneratedOnce(equipmentIdentity, entry) + saved, err := s.equipmentInventory.GrantGeneratedOnce(ctx, equipmentIdentity, entry) if err != nil { return 147, nil, true, err } - if _, err := s.collection.GrantGachaPointEquipment(identity, group.ID, group.PickUpExchangeCost); err != nil { + if _, err := s.collection.GrantGachaPointEquipment(ctx, identity, group.ID, group.PickUpExchangeCost); err != nil { return 147, nil, true, err } - bundle := wire.AppendBytes(nil, 4, player.EquipmentWire(saved)) + bundle := wire.AppendBytes(nil, 4, assets.EquipmentWire(saved)) return 147, wire.AppendBytes(nil, 1, bundle), true, nil } -func (s *Service) loginIdentity() string { - s.sessionMu.RLock() - defer s.sessionMu.RUnlock() - if s.sessionID == "" { - return "local" - } - return s.sessionID -} - -func (s *Service) requestIdentity(gachaID, seq uint64) string { - s.sessionMu.RLock() - sessionID := s.sessionID - s.sessionMu.RUnlock() - if sessionID == "" { - // Unit tests and embedders without a session coordinator retain the - // original stable key. Production always supplies BeginSession. - return fmt.Sprintf("regular-gacha:%d:seq:%d", gachaID, seq) - } - return fmt.Sprintf("regular-gacha:%d:session:%s:seq:%d", gachaID, sessionID, seq) -} - -func (s *Service) preview(request []byte) (int, []byte, bool, error) { +func (s *Service) preview(ctx command.Context, request []byte) (int, []byte, bool, error) { id, found, err := wire.Varint(request, 2) if err != nil || !found { return 0, nil, true, fmt.Errorf("gacha: unsupported preview id %d", id) } if s.first != nil && id == s.first.GachaID { - return s.previewFirstGacha(request) + return s.previewFirstGacha(ctx, request) } if _, bought := s.collection.Grant(s.infiniteGrant()); bought { return 175, nil, true, errors.New("gacha: infinite product already purchased") @@ -1078,11 +889,11 @@ func (s *Service) preview(request []byte) (int, []byte, bool, error) { if err != nil { return 0, nil, true, err } - if err := s.collection.SetPreview(s.previewEventIndex, roll); err != nil { + if err := s.collection.SetPreview(ctx, s.previewEventIndex, roll); err != nil { return 0, nil, true, err } if s.onPreview != nil { - if err := s.onPreview(); err != nil { + if err := s.onPreview(ctx); err != nil { return 0, nil, true, fmt.Errorf("gacha: update preview mission: %w", err) } } @@ -1097,7 +908,7 @@ func (s *Service) preview(request []byte) (int, []byte, bool, error) { return 175, wire.AppendBytes(nil, 1, bundle), true, nil } -func (s *Service) confirm(request []byte) (int, []byte, bool, error) { +func (s *Service) confirm(ctx command.Context, request []byte) (int, []byte, bool, error) { group, found, err := wire.Varint(request, 3) if err != nil || !found { return 0, nil, true, fmt.Errorf("gacha: unsupported cash product group %d", group) @@ -1143,11 +954,11 @@ func (s *Service) confirm(request []byte) (int, []byte, bool, error) { if s.inventory == nil { return 61, nil, true, errors.New("gacha: inventory not attached") } - items, err := s.inventory.GrantOnce(ticketGrant, cashGroup.CashRewards) + items, err := s.inventory.GrantOnce(ctx, ticketGrant, cashGroup.CashRewards) if err != nil { return 61, nil, true, err } - if err := s.collection.RecordGrantMarker(productGrant); err != nil { + if err := s.collection.RecordGrantMarker(ctx, productGrant); err != nil { return 61, nil, true, err } if len(items) == 0 { @@ -1155,53 +966,35 @@ func (s *Service) confirm(request []byte) (int, []byte, bool, error) { } var bundle []byte for _, item := range items { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(item)) + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(item)) } return 61, wire.AppendBytes(nil, 1, bundle), true, nil } if group != s.design.ProductGroupID || productID != s.design.ProductID || saleGroup != s.design.SaleGroup { return 0, nil, true, fmt.Errorf("gacha: unsupported cash product group=%d product=%d", group, productID) } - grant, err := s.collection.ConfirmInfinite(s.infiniteGrant(), s.design) + grant, err := s.collection.ConfirmInfinite(ctx, s.infiniteGrant(), s.design) if err != nil { return 0, nil, true, err } - if err := s.creditOverflow(s.infiniteGrant(), grant); err != nil { + if err := s.creditOverflow(ctx, s.infiniteGrant(), grant); err != nil { return 61, nil, true, err } return 61, wire.AppendBytes(nil, 1, s.rewardBundle(grant)), true, nil } -func (s *Service) creditOverflow(identity string, grant player.CollectionGrant) error { - var mileage uint64 - for _, exchange := range grant.Exchanges { - if exchange.ExchangeItemType != 20 { - return fmt.Errorf("gacha: unsupported costume overflow type %d", exchange.ExchangeItemType) - } - if ^uint64(0)-mileage < exchange.ExchangeCount { - return errors.New("gacha: costume overflow mileage overflow") - } - mileage += exchange.ExchangeCount - } - if mileage == 0 { - return nil - } - _, err := s.wallet.GrantMileageOnce(identity, mileage) - return err -} - -func (s *Service) rewardBundle(grant player.CollectionGrant) []byte { +func (s *Service) rewardBundle(grant roster.CollectionGrant) []byte { var bundle []byte - newCharacters := make(map[uint64]player.Character, len(grant.CharacterIndices)) + newCharacters := make(map[uint64]roster.Character, len(grant.CharacterIndices)) for _, index := range grant.CharacterIndices { if character, found := s.collection.CharacterByIndex(index); found { newCharacters[character.ConnectPotentialCostume] = character - bundle = wire.AppendBytes(bundle, 2, player.CharacterWire(character)) + bundle = wire.AppendBytes(bundle, 2, roster.CharacterWire(character)) } } for _, index := range grant.CostumeIndices { if costume, found := s.collection.CostumeByIndex(index); found { - bundle = wire.AppendBytes(bundle, 3, player.CostumeWire(costume)) + bundle = wire.AppendBytes(bundle, 3, roster.CostumeWire(costume)) } } for position, costumeID := range grant.ViewCostumeIDs { @@ -1354,8 +1147,133 @@ func (s *Service) gachaInfo() []byte { return out } -// The client converts Unix time to UTC+9 and applies GameData's -// 09:00:00 DailyResetTime. That boundary is exactly 00:00 UTC. Keeping the -// key independent of the host OS timezone prevents an early reset on a -// Chinese development machine. -func dailyResetKey(now time.Time) string { return now.UTC().Format("2006-01-02") } +func (s *Service) previewFirstGacha(ctx command.Context, request []byte) (int, []byte, bool, error) { + if s.first == nil { + return 175, nil, true, errors.New("gacha: first gacha is not configured") + } + if s.FirstGachaCompleted() { + return 175, nil, true, errors.New("gacha: first gacha already completed") + } + for _, field := range []int{3, 4} { + value, _, err := wire.Varint(request, field) + if err != nil || value != 0 { + return 175, nil, true, errors.New("gacha: first gacha preview cannot use a cash product") + } + } + seq, found, err := wire.Varint(request, 1) + if err != nil || !found || seq == 0 { + return 175, nil, true, errors.New("gacha: invalid first gacha preview sequence") + } + session := s.loginIdentity(ctx) + + if previous, found := s.firstPreviews[session]; found && previous.sequence == seq { + return 175, append([]byte(nil), previous.response...), true, nil + } + rewards, err := s.first.Roll() + if err != nil { + return 175, nil, true, err + } + preview := firstGachaPreview{sequence: seq, rewards: append([]gamedata.FirstGachaReward(nil), rewards...), equipment: make(map[int]assets.Equipment)} + var bundle []byte + for sortID, reward := range rewards { + switch reward.Type { + case 11: + costume := wire.AppendVarint(nil, 2, reward.ID) + if sortID != 0 { + costume = wire.AppendVarint(costume, 6, uint64(sortID)) + } + bundle = wire.AppendBytes(bundle, 3, costume) + case 10: + entry, err := firstGachaEquipment(s.first.EquipmentCatalog(), reward.ID, uint64(sortID)) + if err != nil { + return 175, nil, true, err + } + preview.equipment[sortID] = entry + bundle = wire.AppendBytes(bundle, 4, assets.EquipmentWire(entry)) + default: + return 175, nil, true, fmt.Errorf("gacha: unsupported first reward type %d", reward.Type) + } + } + preview.response = wire.AppendBytes(nil, 1, bundle) + s.firstPreviews[session] = preview + s.transientVersion++ + if s.onPreview != nil { + if err := s.onPreview(ctx); err != nil { + delete(s.firstPreviews, session) + s.transientVersion++ + return 175, nil, true, fmt.Errorf("gacha: update first preview mission: %w", err) + } + } + return 175, append([]byte(nil), preview.response...), true, nil +} + +func (s *Service) confirmFirstGacha(ctx command.Context, _ []byte, seq, buyType uint64, tickets []assets.Item, explicitIdentity string) (int, []byte, bool, error) { + if s.first == nil || s.equipmentInventory == nil { + return 146, nil, true, errors.New("gacha: first gacha runtime is not attached") + } + if buyType != 1 || len(tickets) != 0 { + return 146, nil, true, errors.New("gacha: first gacha requires normal free confirmation") + } + identity := explicitIdentity + if identity == "" { + identity = s.requestIdentity(ctx, s.first.GachaID, seq) + } + responseIdentity := identity + ":response" + if response, found, err := s.collection.GachaBatchResponse(ctx, responseIdentity, firstGachaResponseDigest); found || err != nil { + return 146, response, true, err + } + if s.FirstGachaCompleted() { + return 146, nil, true, errors.New("gacha: first gacha already completed") + } + session := s.loginIdentity(ctx) + + preview, found := s.firstPreviews[session] + + if !found || len(preview.rewards) != s.first.Count { + return 146, nil, true, errors.New("gacha: first gacha confirmation requires a preview") + } + var costumeIDs, costumeSortIDs []uint64 + var equipment []assets.Equipment + for sortID, reward := range preview.rewards { + switch reward.Type { + case 11: + costumeIDs = append(costumeIDs, reward.ID) + costumeSortIDs = append(costumeSortIDs, uint64(sortID)) + case 10: + candidate, ok := preview.equipment[sortID] + if !ok || candidate.ID != reward.ID { + return 146, nil, true, fmt.Errorf("gacha: first preview missing equipment slot %d", sortID) + } + saved, err := s.equipmentInventory.GrantGeneratedOnce(ctx, fmt.Sprintf("%s:first-equip:%d", identity, sortID), candidate) + if err != nil { + return 146, nil, true, err + } + equipment = append(equipment, saved) + default: + return 146, nil, true, fmt.Errorf("gacha: unsupported first reward type %d", reward.Type) + } + } + grant, err := s.collection.GrantRegularPurchase(ctx, identity, costumeIDs, s.first.CostumeCatalog(), roster.GachaPurchase{ + Group: s.first.Group, BuyType: 1, RewardCount: uint64(s.first.Count), RewardSortIDs: costumeSortIDs, + }) + if err != nil { + return 146, nil, true, err + } + if err := s.creditOverflow(ctx, identity, grant); err != nil { + return 146, nil, true, err + } + bundle := s.rewardBundle(grant) + for _, entry := range equipment { + bundle = wire.AppendBytes(bundle, 4, assets.EquipmentWire(entry)) + } + response := wire.AppendBytes(nil, 1, bundle) + if err := s.collection.RecordGachaBatch(ctx, responseIdentity, firstGachaResponseDigest, response); err != nil { + return 146, nil, true, err + } + if s.onPreview != nil { + if err := s.onPreview(ctx); err != nil { + return 146, nil, true, fmt.Errorf("gacha: update first gacha mission: %w", err) + } + } + return 146, response, true, nil +} diff --git a/go/internal/server/gacha/schedule.go b/go/internal/server/domain/commerce/gacha/schedule.go similarity index 98% rename from go/internal/server/gacha/schedule.go rename to go/internal/server/domain/commerce/gacha/schedule.go index d01c33f..049a386 100644 --- a/go/internal/server/gacha/schedule.go +++ b/go/internal/server/domain/commerce/gacha/schedule.go @@ -4,7 +4,7 @@ import ( "fmt" "math" - "bd2server/internal/server/gamedata" + "bd2server/internal/server/design/gamedata" ) // ScheduleSeed contains server-owned, versioned dynamic gacha facts. GameData diff --git a/go/internal/server/domain/commerce/login_pass.go b/go/internal/server/domain/commerce/login_pass.go new file mode 100644 index 0000000..12ce34c --- /dev/null +++ b/go/internal/server/domain/commerce/login_pass.go @@ -0,0 +1,74 @@ +package commerce + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "fmt" + + "time" +) + +type loginPassProgress struct { + Free, Premium uint64 + LastDay string +} +type loginPassReceipt struct { + Bundle []byte + Rewarded map[uint64]bool +} +type loginPassState struct { + Progress map[uint64]loginPassProgress + Receipts map[string]loginPassReceipt +} +type LoginPasses struct { + store stateio.Store + design *gamedata.LoginPassCatalog + economy Economy + items ClearPackageInventory + available func(ctx command.Context, _ uint64) bool + now func() time.Time + resetSeconds int64 +} + +func NewLoginPasses(ctx command.Context, store stateio.Store, design *gamedata.LoginPassCatalog, economy Economy, items ClearPackageInventory, available func(ctx command.Context, _ uint64) bool) (*LoginPasses, error) { + if store == nil || design == nil || economy == nil || items == nil || available == nil { + return nil, fmt.Errorf("commerce: invalid login-pass dependencies") + } + s := &LoginPasses{store: store, design: design, economy: economy, items: items, available: available, now: time.Now} + _, err := s.load(ctx) + return s, err +} +func (s *LoginPasses) SetClock(now func() time.Time, resetSeconds int64) { + s.now = now + s.resetSeconds = resetSeconds +} +func (s *LoginPasses) load(ctx command.Context) (loginPassState, error) { + v := loginPassState{Progress: map[uint64]loginPassProgress{}, Receipts: map[string]loginPassReceipt{}} + raw, err := s.store.Load(ctx.State, "commerce_login_passes") + if err != nil || raw == nil { + return v, err + } + if err = json.Unmarshal(raw, &v); err != nil { + return v, err + } + if v.Progress == nil || v.Receipts == nil { + return v, fmt.Errorf("commerce: invalid login-pass state") + } + for group, p := range v.Progress { + rows, ok := s.design.Groups[group] + if !ok || p.Free > uint64(len(rows)) || p.Premium > p.Free || p.Free > 0 && p.LastDay == "" { + return v, fmt.Errorf("commerce: invalid login-pass progress") + } + } + return v, nil +} +func (s *LoginPasses) paid(ctx command.Context, ticket uint64) bool { + for _, i := range s.items.All(ctx) { + if i.Type == 19 && i.ID == ticket && i.Count > 0 && (i.ExpiryTime == 0 || i.ExpiryTime > uint64(s.now().UnixMilli())) { + return true + } + } + return false +} diff --git a/go/internal/server/commerce/native_products.go b/go/internal/server/domain/commerce/native_products.go similarity index 76% rename from go/internal/server/commerce/native_products.go rename to go/internal/server/domain/commerce/native_products.go index c9fac79..8a6a229 100644 --- a/go/internal/server/commerce/native_products.go +++ b/go/internal/server/domain/commerce/native_products.go @@ -4,7 +4,8 @@ import ( "fmt" "math" - "bd2server/internal/server/gamedata" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" ) // Native prices belong to GameData; cash quotes remain server policy. @@ -22,7 +23,7 @@ func nativePrice(d gamedata.CashProductDesign, count uint64) (uint64, error) { } type nativeCostResolver interface { - NativePurchaseCosts(gamedata.Reward) ([]gamedata.Reward, error) + NativePurchaseCosts(ctx command.Context, _ gamedata.Reward) ([]gamedata.Reward, error) } // AttachSpecialProducts reserves exact keys whose rewards depend on a prior @@ -39,9 +40,9 @@ func (s *Service) AttachSpecialProducts(keys []gamedata.CashProductKey) error { return nil } -func (e *EntitlementEconomy) NativePurchaseCosts(cost gamedata.Reward) ([]gamedata.Reward, error) { +func (e *EntitlementEconomy) NativePurchaseCosts(ctx command.Context, cost gamedata.Reward) ([]gamedata.Reward, error) { if resolver, ok := e.base.(nativeCostResolver); ok { - return resolver.NativePurchaseCosts(cost) + return resolver.NativePurchaseCosts(ctx, cost) } return []gamedata.Reward{cost}, nil } diff --git a/go/internal/server/domain/commerce/npcinn/inn.go b/go/internal/server/domain/commerce/npcinn/inn.go new file mode 100644 index 0000000..ce3f800 --- /dev/null +++ b/go/internal/server/domain/commerce/npcinn/inn.go @@ -0,0 +1,35 @@ +// Package npcinn implements motel opening and paid/free recovery. Each request +// runs within the account transaction, including currency, health and receipts. +package npcinn + +import "bd2server/internal/server/domain/command" + +import ( + "bd2server/internal/server/design/gamedata" + + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "bd2server/internal/server/storage/stateio" + "errors" +) + +type Context func(ctx command.Context, pack, npc uint64) (gamedata.InnRule, uint64, error) +type reply struct { + Digest string + Body []byte +} +type Service struct { + store stateio.Store + characters *roster.CharacterStore + wallet *assets.Wallet + context Context + level func(ctx command.Context) (uint64, error) + battle func(ctx command.Context) bool +} + +func New(store stateio.Store, characters *roster.CharacterStore, wallet *assets.Wallet, context Context, level func(ctx command.Context) (uint64, error), battle func(ctx command.Context) bool) (*Service, error) { + if store == nil || characters == nil || wallet == nil || context == nil || level == nil || battle == nil { + return nil, errors.New("npcinn: incomplete dependencies") + } + return &Service{store: store, characters: characters, wallet: wallet, context: context, level: level, battle: battle}, nil +} diff --git a/go/internal/server/npcinn/inn.go b/go/internal/server/domain/commerce/npcinn/protocol_adapter.go similarity index 64% rename from go/internal/server/npcinn/inn.go rename to go/internal/server/domain/commerce/npcinn/protocol_adapter.go index c851296..d738e73 100644 --- a/go/internal/server/npcinn/inn.go +++ b/go/internal/server/domain/commerce/npcinn/protocol_adapter.go @@ -1,8 +1,9 @@ -// Package npcinn implements motel opening and paid/free recovery. Each request -// runs within the account transaction, including currency, health and receipts. package npcinn import ( + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/roster" + "bd2server/internal/server/protocol/wire" "crypto/sha256" "encoding/binary" "encoding/hex" @@ -10,59 +11,29 @@ import ( "errors" "fmt" "math" - "sync" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" ) -type Context func(pack, npc uint64) (gamedata.InnRule, uint64, error) -type reply struct { - Digest string - Body []byte -} -type Service struct { - mu sync.Mutex - store stateio.Store - characters *player.CharacterStore - wallet *player.Wallet - context Context - level func() (uint64, error) - battle func() bool - session string -} - -func New(store stateio.Store, characters *player.CharacterStore, wallet *player.Wallet, context Context, level func() (uint64, error), battle func() bool) (*Service, error) { - if store == nil || characters == nil || wallet == nil || context == nil || level == nil || battle == nil { - return nil, errors.New("npcinn: incomplete dependencies") - } - return &Service{store: store, characters: characters, wallet: wallet, context: context, level: level, battle: battle}, nil -} -func (s *Service) BeginSession(id string) { s.mu.Lock(); defer s.mu.Unlock(); s.session = id } -func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) { +func (s *Service) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { code := 109 if path == "/CharAllRevival" { code = 11 } else if path != "/InnOpen" { return 0, nil, false, nil } - s.mu.Lock() - defer s.mu.Unlock() + fail := func(e error) (int, []byte, bool, error) { return code, nil, true, e } seq, ok, err := wire.Varint(request, 1) - if err != nil || !ok || seq == 0 || seq > math.MaxInt32 || s.session == "" { + if err != nil || !ok || seq == 0 || seq > math.MaxInt32 || ctx.SessionID == "" { return fail(errors.New("npcinn: invalid sequence or session")) } if err = wire.Walk(request, func(wire.Field) error { return nil }); err != nil { return fail(err) } - if s.battle() { + if s.battle(ctx) { return fail(errors.New("npcinn: unavailable during battle")) } if path == "/InnOpen" { - _, state, err := s.context(0, 0) + _, state, err := s.context(ctx, 0, 0) if err != nil { return fail(err) } @@ -76,12 +47,12 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) if err != nil || npc == 0 || npc > math.MaxInt32 { return fail(errors.New("npcinn: invalid NPC")) } - keyBytes := sha256.Sum256([]byte(fmt.Sprintf("%s:%s:%d", s.session, path, seq))) + keyBytes := sha256.Sum256([]byte(fmt.Sprintf("%s:%s:%d", ctx.SessionID, path, seq))) key := hex.EncodeToString(keyBytes[:]) hash := sha256.Sum256(request) digest := hex.EncodeToString(hash[:]) var replies map[string]reply - data, err := s.store.Load("npcinn") + data, err := s.store.Load(ctx.State, "npcinn") if err != nil { return fail(err) } @@ -97,11 +68,11 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) } return code, prior.Body, true, nil } - rule, state, err := s.context(pack, npc) + rule, state, err := s.context(ctx, pack, npc) if err != nil { return fail(err) } - level, err := s.level() + level, err := s.level(ctx) if err != nil { return fail(err) } @@ -140,19 +111,19 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) } rate := rule.ItemCount * (100 - discount) const divisor uint64 = 4000 - available := s.wallet.Snapshot().Gold + available := s.wallet.Snapshot(ctx).Gold var total uint64 - var targets []player.Character + var targets []roster.Character for _, idx := range indices { - c, found := s.characters.Find(idx) + c, found := s.characters.Find(ctx, idx) if !found { return fail(errors.New("npcinn: recovery character not owned")) } - maximum, err := s.characters.MaxHealth(idx) + maximum, err := s.characters.MaxHealth(ctx, idx) if err != nil { return fail(err) } - current, err := s.characters.CurrentHealth(idx) + current, err := s.characters.CurrentHealth(ctx, idx) if err != nil { return fail(err) } @@ -191,23 +162,23 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) targets = append(targets, c) } if total > 0 { - if _, err = s.wallet.SpendGoldOnce("npcinn:"+key, total); err != nil { + if _, err = s.wallet.SpendGoldOnce(ctx, "npcinn:"+key, total); err != nil { return fail(err) } } var out []byte for _, c := range targets { - if err = s.characters.SetCurrentHealth(c.InvenIndex, c.HP); err != nil { + if err = s.characters.SetCurrentHealth(ctx, c.InvenIndex, c.HP); err != nil { return fail(err) } - out = wire.AppendBytes(out, 1, player.CharacterWire(c)) + out = wire.AppendBytes(out, 1, roster.CharacterWire(c)) } replies[key] = reply{Digest: digest, Body: out} data, err = json.Marshal(replies) if err != nil { return fail(err) } - if err = s.store.Save("npcinn", data); err != nil { + if err = s.store.Save(ctx.State, "npcinn", data); err != nil { return fail(err) } return code, out, true, nil diff --git a/go/internal/server/npcshop/shop.go b/go/internal/server/domain/commerce/npcshop/protocol_adapter.go similarity index 57% rename from go/internal/server/npcshop/shop.go rename to go/internal/server/domain/commerce/npcshop/protocol_adapter.go index 923b95f..55687d1 100644 --- a/go/internal/server/npcshop/shop.go +++ b/go/internal/server/domain/commerce/npcshop/protocol_adapter.go @@ -1,155 +1,19 @@ -// Package npcshop implements the ordinary NPC buy/sell market from GameData. package npcshop import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/protocol/wire" "bytes" "crypto/sha256" "encoding/binary" - "encoding/json" "fmt" "math" "slices" "strconv" - "sync" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" ) -type Economy interface { - Apply(string, []gamedata.Reward, []gamedata.Reward) ([]byte, error) -} -type receipt struct { - Digest []byte - Response []byte -} -type count struct { - Period string - Bought uint64 -} -type snapshot struct { - Counts map[string]count `json:"counts"` - Receipts map[string]receipt `json:"receipts"` -} -type Service struct { - mu sync.Mutex - design gamedata.NPCShopDesign - store stateio.Store - economy Economy - items *player.Inventory - available func(uint64) bool - now func() time.Time - session func() string - reputation func(uint64) (uint64, uint64, error) - talentDiscount func(uint64, uint64) (uint64, error) - quotedSeed *uint64 -} - -func New(d gamedata.NPCShopDesign, store stateio.Store, e Economy, items *player.Inventory, available func(uint64) bool) (*Service, error) { - if len(d.Shops) == 0 || store == nil || e == nil || items == nil || available == nil { - return nil, fmt.Errorf("npcshop: invalid configuration") - } - s := &Service{design: d, store: store, economy: e, items: items, available: available, now: time.Now} - _, err := s.load() - return s, err -} -func (s *Service) SetSessionSource(source func() string) { s.session = source } -func (s *Service) BeginSession(id string) { - s.mu.Lock() - defer s.mu.Unlock() - if s.session != nil && s.session() == id { - return - } - s.session = func() string { return id } - s.quotedSeed = nil -} - -// SetReputationSource supplies the world's persisted pack reputation and -// GameData-derived shop discount. Normal reputation is the initial fallback. -func (s *Service) SetReputationSource(source func(uint64) (uint64, uint64, error)) { - s.reputation = source -} -func (s *Service) SetTalentDiscountSource(source func(uint64, uint64) (uint64, error)) { - s.talentDiscount = source -} -func (s *Service) talentDiscountFor(shop gamedata.NPCShop) (uint64, error) { - if s.talentDiscount == nil { - return 0, nil - } - var best uint64 - for _, npc := range s.design.ShopNPCs[shop.ID] { - n, e := s.talentDiscount(shop.PackID, npc) - if e != nil { - return 0, e - } - if n > 100 { - return 0, fmt.Errorf("npcshop: invalid talent discount") - } - if n > best { - best = n - } - } - return best, nil -} -func (s *Service) reputationFor(pack uint64) (uint64, uint64, error) { - if s.reputation != nil { - return s.reputation(pack) - } - return 1, 0, nil -} -func (s *Service) load() (snapshot, error) { - v := snapshot{Counts: map[string]count{}, Receipts: map[string]receipt{}} - b, e := s.store.Load("npc_shop") - if e != nil || len(b) == 0 { - return v, e - } - if e = stateio.RequireExactJSONObject(b, "counts", "receipts"); e != nil { - return v, e - } - if e = json.Unmarshal(b, &v); e != nil { - return v, e - } - if v.Counts == nil || v.Receipts == nil { - return v, fmt.Errorf("npcshop: invalid state") - } - return v, nil -} -func (s *Service) save(v snapshot) error { - b, e := json.Marshal(v) - if e != nil { - return e - } - return s.store.Save("npc_shop", b) -} -func key(shop, product uint64) string { return fmt.Sprintf("%d:%d", shop, product) } -func (s *Service) period(shop gamedata.NPCShop) (string, int64) { - t := s.now().UTC() - day := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, time.UTC) - switch shop.ResetType { - case 1: - return day.Format("2006-01-02"), day.AddDate(0, 0, 1).Unix() - t.Unix() - case 2: - day = day.AddDate(0, 0, -(int(day.Weekday())+6)%7) - return day.Format("2006-01-02"), day.AddDate(0, 0, 7).Unix() - t.Unix() - case 3: - day = time.Date(t.Year(), t.Month(), 1, 0, 0, 0, 0, time.UTC) - return day.Format("2006-01"), day.AddDate(0, 1, 0).Unix() - t.Unix() - default: - return "lifetime", 0 - } -} -func (s *Service) seed() uint64 { return uint64(s.now().UTC().Unix()/86400) % math.MaxInt32 } -func (s *Service) bought(v snapshot, shop, product uint64) uint64 { - period, _ := s.period(s.design.Shops[shop]) - c := v.Counts[key(shop, product)] - if c.Period != period { - return 0 - } - return c.Bought -} func (s *Service) shopWire(v snapshot, id uint64) []byte { _, remaining := s.period(s.design.Shops[id]) b := wire.AppendVarint(nil, 1, id) @@ -168,10 +32,11 @@ func (s *Service) shopWire(v snapshot, id uint64) []byte { } return b } -func (s *Service) allShops(v snapshot, field int) []byte { + +func (s *Service) allShops(ctx command.Context, v snapshot, field int) []byte { ids := make([]uint64, 0) for id, r := range s.design.Shops { - if s.available(r.PackID) { + if s.available(ctx, r.PackID) { ids = append(ids, id) } } @@ -182,7 +47,9 @@ func (s *Service) allShops(v snapshot, field int) []byte { } return b } + func scalar(b []byte, f int) (uint64, error) { v, _, e := wire.Varint(b, f); return v, e } + func messages(b []byte, f int) ([][]byte, error) { var out [][]byte e := wire.Walk(b, func(v wire.Field) error { @@ -197,7 +64,8 @@ func messages(b []byte, f int) ([][]byte, error) { }) return out, e } -func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) { + +func (s *Service) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { codes := map[string]int{"/ShopInfo": 53, "/ShopBuy": 54, "/ShopSell": 55, "/ShopOpen": 107, "/ShopReputationInfo": 520} code, ok := codes[path] if !ok { @@ -207,14 +75,13 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) if e != nil || seq == 0 || seq > math.MaxInt32 { return 0, nil, true, fmt.Errorf("npcshop: invalid sequence") } - s.mu.Lock() - defer s.mu.Unlock() - v, e := s.load() + + v, e := s.load(ctx) if e != nil { return 0, nil, true, e } if path == "/ShopInfo" { - b := s.allShops(v, 1) + b := s.allShops(ctx, v, 1) seed := s.seed() s.quotedSeed = &seed return code, wire.AppendVarint(b, 2, seed), true, nil @@ -224,16 +91,16 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) packs := map[uint64]bool{} isTalent := false for _, r := range s.design.Shops { - if s.available(r.PackID) { - discount, e := s.talentDiscountFor(r) + if s.available(ctx, r.PackID) { + discount, e := s.talentDiscountFor(ctx, r) if e != nil { return 0, nil, true, e } isTalent = isTalent || discount > 0 } - if !packs[r.PackID] && s.available(r.PackID) { + if !packs[r.PackID] && s.available(ctx, r.PackID) { packs[r.PackID] = true - state, _, err := s.reputationFor(r.PackID) + state, _, err := s.reputationFor(ctx, r.PackID) if err != nil { return 0, nil, true, err } @@ -249,19 +116,19 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) } if path == "/ShopReputationInfo" { pack, err := scalar(request, 2) - if err != nil || !s.available(pack) { + if err != nil || !s.available(ctx, pack) { return 0, nil, true, fmt.Errorf("npcshop: unavailable reputation pack") } - state, _, err := s.reputationFor(pack) + state, _, err := s.reputationFor(ctx, pack) if err != nil { return 0, nil, true, err } return code, wire.AppendVarint(nil, 1, state), true, nil } - if s.session == nil || s.session() == "" { + if ctx.SessionID == "" { return 0, nil, true, fmt.Errorf("npcshop: missing session identity") } - identity := "npcshop:" + s.session() + ":" + path + ":" + strconv.FormatUint(seq, 10) + identity := "npcshop:" + ctx.SessionID + ":" + path + ":" + strconv.FormatUint(seq, 10) digest := sha256.Sum256(request) if r, exists := v.Receipts[identity]; exists { if !bytes.Equal(r.Digest, digest[:]) { @@ -270,25 +137,25 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) return code, r.Response, true, nil } var costs, rewards []gamedata.Reward - var sold []player.Item + var sold []assets.Item changed := map[uint64]bool{} if path == "/ShopBuy" { - costs, rewards, changed, e = s.buy(request, &v) + costs, rewards, changed, e = s.buy(ctx, request, &v) if e == nil { costs, sold, e = s.purchaseMaterials(request, costs) } } else { - sold, rewards, e = s.sell(request) + sold, rewards, e = s.sell(ctx, request) } if e != nil { return 0, nil, true, e } if len(sold) > 0 { - if e = s.items.CanConsume(sold); e != nil { + if e = s.items.CanConsume(ctx, sold); e != nil { return 0, nil, true, e } } - bundle, e := s.economy.Apply(identity, costs, rewards) + bundle, e := s.economy.Apply(ctx, identity, costs, rewards) if e != nil { return 0, nil, true, e } @@ -296,7 +163,7 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) // Consume the concrete stacks the client removes, rather than letting a // type/id-only debit choose a different stack of the same resource. if len(sold) > 0 { - if e = s.items.Consume(sold); e != nil { + if e = s.items.Consume(ctx, sold); e != nil { return 0, nil, true, e } } @@ -310,12 +177,13 @@ func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) b = wire.AppendBytes(b, 2, s.shopWire(v, id)) } v.Receipts[identity] = receipt{Digest: digest[:], Response: b} - if e = s.save(v); e != nil { + if e = s.save(ctx, v); e != nil { return 0, nil, true, e } return code, b, true, nil } -func (s *Service) buy(request []byte, v *snapshot) ([]gamedata.Reward, []gamedata.Reward, map[uint64]bool, error) { + +func (s *Service) buy(ctx command.Context, request []byte, v *snapshot) ([]gamedata.Reward, []gamedata.Reward, map[uint64]bool, error) { groups, e := messages(request, 2) if e != nil || len(groups) == 0 { return nil, nil, nil, fmt.Errorf("npcshop: empty purchase") @@ -326,17 +194,17 @@ func (s *Service) buy(request []byte, v *snapshot) ([]gamedata.Reward, []gamedat for _, g := range groups { shop, e := scalar(g, 1) r, exists := s.design.Shops[shop] - if e != nil || !exists || !s.available(r.PackID) { + if e != nil || !exists || !s.available(ctx, r.PackID) { return nil, nil, nil, fmt.Errorf("npcshop: unavailable shop %d", shop) } - state, discount, err := s.reputationFor(r.PackID) + state, discount, err := s.reputationFor(ctx, r.PackID) if err != nil { return nil, nil, nil, err } if state == 0 || discount > 100 { return nil, nil, nil, fmt.Errorf("npcshop: unavailable reputation") } - talentDiscount, e := s.talentDiscountFor(r) + talentDiscount, e := s.talentDiscountFor(ctx, r) if e != nil { return nil, nil, nil, e } @@ -395,21 +263,22 @@ func (s *Service) buy(request []byte, v *snapshot) ([]gamedata.Reward, []gamedat } return costs, rewards, changed, nil } -func (s *Service) sell(request []byte) ([]player.Item, []gamedata.Reward, error) { + +func (s *Service) sell(ctx command.Context, request []byte) ([]assets.Item, []gamedata.Reward, error) { id, e := scalar(request, 2) shop, exists := s.design.Shops[id] - if e != nil || !exists || !s.available(shop.PackID) { + if e != nil || !exists || !s.available(ctx, shop.PackID) { return nil, nil, fmt.Errorf("npcshop: unavailable sell shop") } rows, e := messages(request, 3) if e != nil || len(rows) == 0 { return nil, nil, fmt.Errorf("npcshop: empty sale") } - inventory := map[uint64]player.Item{} - for _, item := range s.items.All() { + inventory := map[uint64]assets.Item{} + for _, item := range s.items.All(ctx) { inventory[item.InvenIndex] = item } - var costs []player.Item + var costs []assets.Item var rewards []gamedata.Reward seen := map[uint64]bool{} for _, b := range rows { @@ -458,7 +327,7 @@ func (s *Service) sell(request []byte) ([]player.Item, []gamedata.Reward, error) // Resource currency is removed from the exact inventory rows named by the // client after independently deriving the required type/id/count from design. -func (s *Service) purchaseMaterials(request []byte, costs []gamedata.Reward) ([]gamedata.Reward, []player.Item, error) { +func (s *Service) purchaseMaterials(request []byte, costs []gamedata.Reward) ([]gamedata.Reward, []assets.Item, error) { need := map[[2]uint64]uint64{} var currency []gamedata.Reward for _, c := range costs { @@ -472,7 +341,7 @@ func (s *Service) purchaseMaterials(request []byte, costs []gamedata.Reward) ([] if e != nil { return nil, nil, e } - var selected []player.Item + var selected []assets.Item got := map[[2]uint64]uint64{} seen := map[uint64]bool{} for _, b := range rows { @@ -486,7 +355,7 @@ func (s *Service) purchaseMaterials(request []byte, costs []gamedata.Reward) ([] seen[index] = true k := [2]uint64{typ, id} got[k] += n - selected = append(selected, player.Item{InvenIndex: index, ID: id, Type: typ, Count: n}) + selected = append(selected, assets.Item{InvenIndex: index, ID: id, Type: typ, Count: n}) } if len(got) != len(need) { return nil, nil, fmt.Errorf("npcshop: purchase material mismatch") @@ -498,40 +367,3 @@ func (s *Service) purchaseMaterials(request []byte, costs []gamedata.Reward) ([] } return currency, selected, nil } - -// Client ShopPacket.GetShopRandomValue uses the first WELL512 value for a -// versioned product and server-supplied daily seed. Client Rate is verified, -// never accepted as the authority for wallet arithmetic. -func (s *Service) rate(p gamedata.NPCProduct, shop, tab uint64) uint64 { - if p.PremiumPriceType == 1 && p.HighShop == shop && p.HighDay == uint64(s.now().UTC().Day()) { - return 100 + p.HighPremium - } - seed := s.marketSeed() + shop + tab + p.Reward.Type + p.Reward.ID - a := seed - b := seed + 13*90 - c := a ^ b ^ (a << 16) ^ (b << 15) - b = seed + 9*90 - b ^= b >> 11 - a = c ^ b - // WELL512's mask is the unsigned 32-bit pattern 0xDA442D24. The - // decompiler displays its signed int32 spelling, -633066204, inside - // an ulong cast; sign-extending that spelling changes the RNG result. - d := a ^ ((a << 5) & uint64(0xda442d24)) - a = seed + 15*90 - value := a ^ c ^ d ^ (a << 2) ^ (c << 18) ^ (b << 28) - lo := 100 - p.Discount - span := p.Discount + p.Premium - if span == 0 { - span = 1 - } - return lo + value%span -} - -// The client keeps ShopRandSeed until the next ShopInfo response. Advancing -// the server clock alone must not change a price already displayed in its UI. -func (s *Service) marketSeed() uint64 { - if s.quotedSeed != nil { - return *s.quotedSeed - } - return s.seed() -} diff --git a/go/internal/server/domain/commerce/npcshop/shop.go b/go/internal/server/domain/commerce/npcshop/shop.go new file mode 100644 index 0000000..bb20e7f --- /dev/null +++ b/go/internal/server/domain/commerce/npcshop/shop.go @@ -0,0 +1,177 @@ +// Package npcshop implements the ordinary NPC buy/sell market from GameData. +package npcshop + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "fmt" + "math" + + "time" +) + +type Economy interface { + Apply(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) +} +type receipt struct { + Digest []byte + Response []byte +} +type count struct { + Period string + Bought uint64 +} +type snapshot struct { + Counts map[string]count `json:"counts"` + Receipts map[string]receipt `json:"receipts"` +} +type Service struct { + design gamedata.NPCShopDesign + store stateio.Store + economy Economy + items *assets.Inventory + available func(ctx command.Context, _ uint64) bool + now func() time.Time + + reputation func(ctx command.Context, _ uint64) (uint64, uint64, error) + talentDiscount func(ctx command.Context, _ uint64, _ uint64) (uint64, error) + quotedSeed *uint64 +} + +func New(ctx command.Context, d gamedata.NPCShopDesign, store stateio.Store, e Economy, items *assets.Inventory, available func(ctx command.Context, _ uint64) bool) (*Service, error) { + if len(d.Shops) == 0 || store == nil || e == nil || items == nil || available == nil { + return nil, fmt.Errorf("npcshop: invalid configuration") + } + s := &Service{design: d, store: store, economy: e, items: items, available: available, now: time.Now} + _, err := s.load(ctx) + return s, err +} + +func (s *Service) BeginLogin(ctx command.Context) { + + s.quotedSeed = nil +} + +// SetReputationSource supplies the world's persisted pack reputation and +// GameData-derived shop discount. Normal reputation is the initial fallback. +func (s *Service) SetReputationSource(source func(ctx command.Context, _ uint64) (uint64, uint64, error)) { + s.reputation = source +} +func (s *Service) SetTalentDiscountSource(source func(ctx command.Context, _ uint64, _ uint64) (uint64, error)) { + s.talentDiscount = source +} +func (s *Service) talentDiscountFor(ctx command.Context, shop gamedata.NPCShop) (uint64, error) { + if s.talentDiscount == nil { + return 0, nil + } + var best uint64 + for _, npc := range s.design.ShopNPCs[shop.ID] { + n, e := s.talentDiscount(ctx, shop.PackID, npc) + if e != nil { + return 0, e + } + if n > 100 { + return 0, fmt.Errorf("npcshop: invalid talent discount") + } + if n > best { + best = n + } + } + return best, nil +} +func (s *Service) reputationFor(ctx command.Context, pack uint64) (uint64, uint64, error) { + if s.reputation != nil { + return s.reputation(ctx, pack) + } + return 1, 0, nil +} +func (s *Service) load(ctx command.Context) (snapshot, error) { + v := snapshot{Counts: map[string]count{}, Receipts: map[string]receipt{}} + b, e := s.store.Load(ctx.State, "npc_shop") + if e != nil || len(b) == 0 { + return v, e + } + if e = stateio.RequireExactJSONObject(b, "counts", "receipts"); e != nil { + return v, e + } + if e = json.Unmarshal(b, &v); e != nil { + return v, e + } + if v.Counts == nil || v.Receipts == nil { + return v, fmt.Errorf("npcshop: invalid state") + } + return v, nil +} +func (s *Service) save(ctx command.Context, v snapshot) error { + b, e := json.Marshal(v) + if e != nil { + return e + } + return s.store.Save(ctx.State, "npc_shop", b) +} +func key(shop, product uint64) string { return fmt.Sprintf("%d:%d", shop, product) } +func (s *Service) period(shop gamedata.NPCShop) (string, int64) { + t := s.now().UTC() + day := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, time.UTC) + switch shop.ResetType { + case 1: + return day.Format("2006-01-02"), day.AddDate(0, 0, 1).Unix() - t.Unix() + case 2: + day = day.AddDate(0, 0, -(int(day.Weekday())+6)%7) + return day.Format("2006-01-02"), day.AddDate(0, 0, 7).Unix() - t.Unix() + case 3: + day = time.Date(t.Year(), t.Month(), 1, 0, 0, 0, 0, time.UTC) + return day.Format("2006-01"), day.AddDate(0, 1, 0).Unix() - t.Unix() + default: + return "lifetime", 0 + } +} +func (s *Service) seed() uint64 { return uint64(s.now().UTC().Unix()/86400) % math.MaxInt32 } +func (s *Service) bought(v snapshot, shop, product uint64) uint64 { + period, _ := s.period(s.design.Shops[shop]) + c := v.Counts[key(shop, product)] + if c.Period != period { + return 0 + } + return c.Bought +} + +// Client ShopPacket.GetShopRandomValue uses the first WELL512 value for a +// versioned product and server-supplied daily seed. Client Rate is verified, +// never accepted as the authority for wallet arithmetic. +func (s *Service) rate(p gamedata.NPCProduct, shop, tab uint64) uint64 { + if p.PremiumPriceType == 1 && p.HighShop == shop && p.HighDay == uint64(s.now().UTC().Day()) { + return 100 + p.HighPremium + } + seed := s.marketSeed() + shop + tab + p.Reward.Type + p.Reward.ID + a := seed + b := seed + 13*90 + c := a ^ b ^ (a << 16) ^ (b << 15) + b = seed + 9*90 + b ^= b >> 11 + a = c ^ b + // WELL512's mask is the unsigned 32-bit pattern 0xDA442D24. The + // decompiler displays its signed int32 spelling, -633066204, inside + // an ulong cast; sign-extending that spelling changes the RNG result. + d := a ^ ((a << 5) & uint64(0xda442d24)) + a = seed + 15*90 + value := a ^ c ^ d ^ (a << 2) ^ (c << 18) ^ (b << 28) + lo := 100 - p.Discount + span := p.Discount + p.Premium + if span == 0 { + span = 1 + } + return lo + value%span +} + +// The client keeps ShopRandSeed until the next ShopInfo response. Advancing +// the server clock alone must not change a price already displayed in its UI. +func (s *Service) marketSeed() uint64 { + if s.quotedSeed != nil { + return *s.quotedSeed + } + return s.seed() +} diff --git a/go/internal/server/domain/commerce/pack_info.go b/go/internal/server/domain/commerce/pack_info.go new file mode 100644 index 0000000..f4ae5c0 --- /dev/null +++ b/go/internal/server/domain/commerce/pack_info.go @@ -0,0 +1,6 @@ +package commerce + +type PackInfoHandler struct { + World attendanceHandler + Claims *ClearPackages +} diff --git a/go/internal/server/domain/commerce/protocol_adapter.go b/go/internal/server/domain/commerce/protocol_adapter.go new file mode 100644 index 0000000..3e3ddaa --- /dev/null +++ b/go/internal/server/domain/commerce/protocol_adapter.go @@ -0,0 +1,1406 @@ +package commerce + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/protocol/staticdata" + "bd2server/internal/server/protocol/wire" + "bytes" + "crypto/sha256" + "encoding/binary" + "encoding/hex" + "encoding/json" + "fmt" + "math" + "slices" + "sort" + "strconv" + "strings" + "time" +) + +func (s *Service) predecessorBought(ctx command.Context, v purchaseState, key gamedata.CashProductKey) bool { + prior := s.predecessors[key] + if len(prior) == 0 { + return true + } + for _, p := range prior { + if v.Counts[skuKey(p)].Lifetime > 0 { + return true + } + } + if s.legacy != nil { + for _, raw := range s.legacy.PurchaseCountDBInfos(ctx) { + g, _, _ := wire.Varint(raw, 1) + id, _, _ := wire.Varint(raw, 2) + sale, _, _ := wire.Varint(raw, 3) + n, _, _ := wire.Varint(raw, 4) + if n > 0 { + for _, p := range prior { + if p.GroupID == g && p.ProductID == id && p.SaleGroup == sale { + return true + } + } + } + } + } + return false +} + +// AttachShopSeed retains versioned dynamic windows and event identities. It +// must run before serving sessions; absent windows never authorize timed goods. +func (s *Service) AttachShopSeed(ctx command.Context, seed *readonly.Seed) error { + if seed == nil { + return fmt.Errorf("commerce: missing shop seed") + } + _, raw, handled, err := seed.Handle(ctx, "/CashShopInfo", wire.AppendVarint(nil, 1, 1)) + if err != nil || !handled { + return fmt.Errorf("commerce: invalid shop seed: %w", err) + } + windows := map[gamedata.CashProductKey][2]uint64{} + var products [][]byte + err = wire.Walk(raw, func(f wire.Field) error { + if f.Number != 1 { + return nil + } + if f.Type != 2 { + return fmt.Errorf("commerce: invalid shop product") + } + g, _, e := wire.Varint(f.Value, 1) + if e != nil { + return e + } + id, _, e := wire.Varint(f.Value, 2) + if e != nil { + return e + } + sale, _, e := wire.Varint(f.Value, 3) + if e != nil { + return e + } + start, _, e := wire.Varint(f.Value, 4) + if e != nil { + return e + } + end, _, e := wire.Varint(f.Value, 5) + if e != nil { + return e + } + k := gamedata.CashProductKey{GroupID: g, ProductID: id, SaleGroup: sale} + if _, ok := windows[k]; ok { + return fmt.Errorf("commerce: duplicate shop schedule") + } + if end != 0 && start >= end { + return fmt.Errorf("commerce: invalid shop window") + } + windows[k] = [2]uint64{start, end} + products = append(products, append([]byte(nil), f.Value...)) + return nil + }) + if err != nil { + return err + } + s.shopProducts = products + s.shopWindows = windows + return nil +} + +func (s *Service) shopInfo() []byte { + var response []byte + for _, raw := range s.shopProducts { + g, _, _ := wire.Varint(raw, 1) + id, _, _ := wire.Varint(raw, 2) + sale, _, _ := wire.Varint(raw, 3) + _, ok := s.catalog.Design(gamedata.CashProductKey{GroupID: g, ProductID: id, SaleGroup: sale}) + if ok { + response = wire.AppendBytes(response, 1, raw) + } + } + t := s.now().UTC().Add(-time.Duration(s.resetSeconds) * time.Second) + day := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, time.UTC) + daily := day.AddDate(0, 0, 1) + days := (8 - int(day.Weekday())) % 7 + if days == 0 { + days = 7 + } + weekly := day.AddDate(0, 0, days) + monthly := time.Date(t.Year(), t.Month()+1, 1, 0, 0, 0, 0, time.UTC) + for i, reset := range []time.Time{daily, weekly, monthly} { + response = wire.AppendVarint(response, i+2, uint64(reset.Add(time.Duration(s.resetSeconds)*time.Second).UnixMilli())) + } + return response +} + +func (s *Service) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + switch path { + case "/CashShopInfo": + if s.shopWindows == nil { + return 0, nil, false, nil + } + if seq, ok, err := wire.Varint(request, 1); err != nil || !ok || seq == 0 || seq > math.MaxInt32 { + return 60, nil, true, fmt.Errorf("commerce: invalid shop sequence") + } + return 60, s.shopInfo(), true, nil + case "/CashShopPurchaseCountInfo": + if seq, ok, err := wire.Varint(request, 1); err != nil || !ok || seq == 0 || seq > math.MaxInt32 { + return 432, nil, true, fmt.Errorf("commerce: invalid count sequence") + } + infos, err := s.purchaseCountDBInfos(ctx) + if err != nil { + return 0, nil, true, err + } + var b []byte + for _, i := range infos { + b = wire.AppendBytes(b, 1, i) + } + return 432, b, true, nil + case "/CashShopBuy": + return s.buy(ctx, commandSession, request) + default: + return 0, nil, false, nil + } +} + +func parseBuy(request []byte) (uint64, []buyLine, []string, error) { + seq, found, e := wire.Varint(request, 1) + if e != nil || !found || seq == 0 || seq > math.MaxInt32 { + return 0, nil, nil, fmt.Errorf("commerce: invalid sequence") + } + group, found, e := wire.Varint(request, 3) + if e != nil || !found || group == 0 { + return 0, nil, nil, fmt.Errorf("commerce: invalid product group") + } + var lines []buyLine + var billing []string + seen := map[gamedata.CashProductKey]bool{} + e = wire.Walk(request, func(f wire.Field) error { + switch f.Number { + case 4: + if f.Type != 2 { + return fmt.Errorf("commerce: invalid buy info") + } + id, ok, err := wire.Varint(f.Value, 1) + if err != nil || !ok || id == 0 { + return fmt.Errorf("commerce: invalid product id") + } + sale, _, err := wire.Varint(f.Value, 2) + if err != nil { + return err + } + n, ok, err := wire.Varint(f.Value, 3) + if err != nil || !ok || n == 0 || n > math.MaxInt32 { + return fmt.Errorf("commerce: invalid buy count") + } + k := gamedata.CashProductKey{GroupID: group, ProductID: id, SaleGroup: sale} + if seen[k] { + return fmt.Errorf("commerce: duplicate buy info") + } + seen[k] = true + lines = append(lines, buyLine{k, n}) + case 7: + if f.Type != 2 { + return fmt.Errorf("commerce: invalid billing info") + } + return wire.Walk(f.Value, func(b wire.Field) error { + if b.Number == 1 || b.Number == 2 { + if b.Type != 2 { + return fmt.Errorf("commerce: invalid billing identity") + } + if len(b.Value) > 16384 { + return fmt.Errorf("commerce: excessive billing identity") + } + if len(b.Value) > 0 { + h := sha256.Sum256(b.Value) + billing = append(billing, fmt.Sprintf("%d:%x", b.Number, h)) + } + } + return nil + }) + } + return nil + }) + if e == nil && len(lines) == 0 { + e = fmt.Errorf("commerce: empty purchase") + } + return seq, lines, billing, e +} + +func (s *Service) buy(ctx command.Context, session string, request []byte) (int, []byte, bool, error) { + if session == "" { + return 61, nil, true, fmt.Errorf("commerce: authenticated session required") + } + seq, lines, billing, e := parseBuy(request) + if e != nil { + return 61, nil, true, e + } + var native, cash bool + for _, line := range lines { + if design, ok := s.catalog.Design(line.Key); ok { + if design.PriceType == 1 { + cash = true + } else { + native = true + } + } + } + if native && cash { + return 61, nil, true, fmt.Errorf("commerce: mixed cash and native purchase") + } + if !cash { + // Native BillingInfo contains display identifiers (often a product ID), + // not a unique payment receipt. Request identity provides replay safety. + billing = nil + } + h := sha256.Sum256(request) + digest := hex.EncodeToString(h[:]) + sessionHash := sha256.Sum256([]byte(session)) + identity := fmt.Sprintf("commerce:%x:%d", sessionHash, seq) + + v, e := s.load(ctx) + if e != nil { + return 61, nil, true, e + } + if r, ok := v.Receipts[identity]; ok { + if r.Digest != digest { + return 61, nil, true, fmt.Errorf("commerce: sequence reused with different purchase") + } + return 61, append([]byte(nil), r.Response...), true, nil + } + for _, b := range billing { + if _, ok := v.Billing[b]; ok { + return 61, nil, true, fmt.Errorf("commerce: billing receipt already used") + } + } + // Validate every line before charging the first one. + designs := make([]gamedata.CashProductDesign, len(lines)) + quotes := make([]Product, len(lines)) + for i, l := range lines { + if !s.predecessorBought(ctx, v, l.Key) { + return 61, nil, true, fmt.Errorf("commerce: preceding package purchase required") + } + d, ok := s.catalog.Design(l.Key) + if !ok { + return 61, nil, true, fmt.Errorf("commerce: unknown product") + } + if !s.available(d) { + return 61, nil, true, fmt.Errorf("commerce: product is not currently available") + } + if s.specialProducts[l.Key] && (len(lines) != 1 || l.Count != 1) { + return 61, nil, true, fmt.Errorf("commerce: special purchase requires one product") + } + var q Product + if d.PriceType == 1 { + var err error + q, err = s.catalog.Quote(l.Key, l.Count) + if err != nil { + return 61, nil, true, err + } + } else { + cost, err := nativePrice(d, l.Count) + if err != nil { + return 61, nil, true, err + } + q = Product{Enabled: true, ItemType: d.PriceType, Cost: cost} + } + if !q.Enabled { + return 61, nil, true, fmt.Errorf("commerce: product disabled") + } + if len(lines) > 1 && d.BulkOrderAvailability != 1 { + return 61, nil, true, fmt.Errorf("commerce: bulk purchase disabled") + } + n := s.count(v, d) + if s.legacy != nil { + for _, raw := range s.legacy.PurchaseCountDBInfos(ctx) { + g, _, _ := wire.Varint(raw, 1) + id, _, _ := wire.Varint(raw, 2) + sale, _, _ := wire.Varint(raw, 3) + legacyCount, _, _ := wire.Varint(raw, 4) + if g == l.Key.GroupID && id == l.Key.ProductID && sale == l.Key.SaleGroup && legacyCount > n { + n = legacyCount + } + } + } + if d.PurchaseLimitType > 4 { + return 61, nil, true, fmt.Errorf("commerce: unknown purchase limit") + } + if n > math.MaxInt32-l.Count { + return 61, nil, true, fmt.Errorf("commerce: purchase count overflow") + } + if d.PurchaseLimitType != 0 && (d.PurchaseLimitCount == 0 || n+l.Count > d.PurchaseLimitCount) { + return 61, nil, true, fmt.Errorf("commerce: purchase limit exceeded") + } + designs[i] = d + quotes[i] = q + } + if cash { + e = validateAcceptedQuote(request, quotes) + } + if e != nil { + return 61, nil, true, e + } + var bundle []byte + for i, l := range lines { + d, q := designs[i], quotes[i] + operation := fmt.Sprintf("%s:%d", identity, i) + var costs []gamedata.Reward + if q.Cost > 0 { + if d.PriceType != 1 { + costs = []gamedata.Reward{{Type: d.PriceType, ID: d.PriceID, Count: q.Cost}} + if resolver, ok := s.economy.(nativeCostResolver); ok { + var err error + costs, err = resolver.NativePurchaseCosts(ctx, costs[0]) + if err != nil { + return 61, nil, true, err + } + } + } else { + var typ uint64 + switch q.Currency { + case "paid_diamonds": + typ = 2 + case "diamonds": + typ = 3 + case "gold": + typ = 4 + default: + return 61, nil, true, fmt.Errorf("commerce: unknown currency %q", q.Currency) + } + costs = []gamedata.Reward{{Type: typ, Count: q.Cost}} + } + } + var special []byte + handled := false + // Delegate must only select known special products. Parent account transaction + // guarantees its grant and the subsequent debit commit together. + if s.delegate != nil && (d.PriceType == 1 || s.specialProducts[l.Key]) { + special, handled, e = s.delegate(ctx, l.Key, request) + if e != nil { + return 61, nil, true, e + } + if handled && (len(lines) != 1 || l.Count != 1) { + return 61, nil, true, fmt.Errorf("commerce: special purchase requires one product") + } + } + var rewards []gamedata.Reward + if !handled { + if d.RandomBoxID == 0 { + return 61, nil, true, fmt.Errorf("commerce: product reward missing") + } + rewards = append(rewards, gamedata.Reward{Type: 9, ID: d.RandomBoxID, Count: l.Count}) + if d.BonusRandomBoxID != 0 { + rewards = append(rewards, gamedata.Reward{Type: 9, ID: d.BonusRandomBoxID, Count: l.Count}) + } + } + var b []byte + var err error + if delivery, ok := s.economy.(interface { + ApplyPurchase(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) + }); ok { + b, err = delivery.ApplyPurchase(ctx, operation, costs, rewards) + } else { + b, err = s.economy.Apply(ctx, operation, costs, rewards) + } + if err != nil { + return 61, nil, true, err + } + bundle = append(bundle, b...) + bundle = append(bundle, special...) + if s.hook != nil { + if err = s.hook(ctx, operation, d, l.Count); err != nil { + return 61, nil, true, err + } + } + old := v.Counts[skuKey(l.Key)] + if old.Lifetime > math.MaxUint64-l.Count { + return 61, nil, true, fmt.Errorf("commerce: lifetime purchase count overflow") + } + v.Counts[skuKey(l.Key)] = purchaseCount{Count: s.count(v, d) + l.Count, Period: s.period(d.PurchaseLimitType), Lifetime: old.Lifetime + l.Count} + } + response := wire.AppendBytes(nil, 1, bundle) + v.Receipts[identity] = purchaseReceipt{digest, response} + for _, b := range billing { + v.Billing[b] = identity + } + encoded, e := json.Marshal(v) + if e == nil { + e = s.store.Save(ctx.State, "commerce", encoded) + } + return 61, response, true, e +} + +// The receipt repeats the client-visible quote for stale-catalog detection. +// Prices always come from Catalog, and a mismatch never changes the debit. +func validateAcceptedQuote(request []byte, quotes []Product) error { + var pay, receipt string + seen := false + err := wire.Walk(request, func(f wire.Field) error { + if f.Number != 7 { + return nil + } + if seen || f.Type != 2 { + return fmt.Errorf("commerce: invalid billing info") + } + seen = true + return wire.Walk(f.Value, func(b wire.Field) error { + if b.Number == 1 { + if b.Type != 2 || pay != "" { + return fmt.Errorf("commerce: invalid pay id") + } + pay = string(b.Value) + } + if b.Number == 2 { + if b.Type != 2 || receipt != "" { + return fmt.Errorf("commerce: invalid receipt") + } + receipt = string(b.Value) + } + return nil + }) + }) + if err != nil { + return err + } + parts := strings.Split(receipt, ":") + if len(parts) != 4 || parts[0] != "bd2-local-commerce-v1" || parts[1] != pay { + return fmt.Errorf("commerce: local billing quote required") + } + payID, err := strconv.ParseUint(pay, 10, 64) + if err != nil || payID == 0 || strconv.FormatUint(payID, 10) != pay { + return fmt.Errorf("commerce: invalid local pay id") + } + typ, err := strconv.ParseUint(parts[2], 10, 64) + if err != nil || strconv.FormatUint(typ, 10) != parts[2] { + return fmt.Errorf("commerce: invalid accepted quote currency") + } + amount, err := strconv.ParseUint(parts[3], 10, 64) + if err != nil || strconv.FormatUint(amount, 10) != parts[3] { + return fmt.Errorf("commerce: invalid accepted quote amount") + } + var expected uint64 + for _, q := range quotes { + if q.ItemType != typ || q.Cost > math.MaxInt32-expected { + return fmt.Errorf("commerce: accepted quote currency mismatch") + } + expected += q.Cost + } + if expected != amount { + return fmt.Errorf("commerce: purchase quote changed; refresh the shop") + } + return nil +} + +func (s *Service) purchaseCountDBInfos(ctx command.Context) ([][]byte, error) { + + v, e := s.load(ctx) + if e != nil { + return nil, e + } + counts := map[gamedata.CashProductKey]uint64{} + if s.legacy != nil { + for _, raw := range s.legacy.PurchaseCountDBInfos(ctx) { + g, _, _ := wire.Varint(raw, 1) + id, _, _ := wire.Varint(raw, 2) + sale, _, _ := wire.Varint(raw, 3) + n, _, _ := wire.Varint(raw, 4) + counts[gamedata.CashProductKey{GroupID: g, ProductID: id, SaleGroup: sale}] = n + } + } + for _, d := range s.catalog.Designs() { + if n := s.count(v, d); n > counts[d.Key] { + counts[d.Key] = n + } + } + keys := make([]gamedata.CashProductKey, 0, len(counts)) + for k := range counts { + keys = append(keys, k) + } + sort.Slice(keys, func(i, j int) bool { + a, b := keys[i], keys[j] + if a.GroupID != b.GroupID { + return a.GroupID < b.GroupID + } + if a.ProductID != b.ProductID { + return a.ProductID < b.ProductID + } + return a.SaleGroup < b.SaleGroup + }) + var out [][]byte + for _, k := range keys { + b := wire.AppendVarint(nil, 1, k.GroupID) + b = wire.AppendVarint(b, 2, k.ProductID) + b = wire.AppendVarint(b, 3, k.SaleGroup) + b = wire.AppendVarint(b, 4, counts[k]) + out = append(out, b) + } + return out, nil +} + +func (s *Service) PurchaseCountDBInfos(ctx command.Context) [][]byte { + v, _ := s.purchaseCountDBInfos(ctx) + return v +} + +func (h PackInfoHandler) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { + if path != "/PackInfo" { + return 0, nil, false, nil + } + code, response, handled, err := h.World.Handle(ctx, path, request) + if err != nil || !handled { + return code, response, handled, err + } + pack, evil, err := h.Claims.RewardDBInfos(ctx) + if err != nil { + return code, nil, true, err + } + var result []byte + err = wire.Walk(response, func(field wire.Field) error { + if field.Number != 3 && field.Number != 4 { + result = append(result, response[field.Start:field.End]...) + } + return nil + }) + if err != nil { + return code, nil, true, err + } + for _, info := range pack { + result = wire.AppendBytes(result, 3, info) + } + for _, info := range evil { + result = wire.AppendBytes(result, 4, info) + } + return code, result, true, nil +} + +// ClaimAndInfo is called by AttendanceHandler inside the account transaction. +// Free progression advances once per observed reset day, and buying premium +// catches up the already earned rows without advancing the login day count. +func (s *LoginPasses) ClaimAndInfo(ctx command.Context, identity string) ([]byte, [][]byte, error) { + + if identity == "" { + return nil, nil, fmt.Errorf("commerce: missing login-pass identity") + } + v, err := s.load(ctx) + if err != nil { + return nil, nil, err + } + keys := make([]uint64, 0, len(s.design.Groups)) + for group := range s.design.Groups { + keys = append(keys, group) + } + slices.Sort(keys) + r, seen := v.Receipts[identity] + if !seen { + r = loginPassReceipt{Rewarded: map[uint64]bool{}} + var rewards []gamedata.Reward + day := s.now().UTC().Add(-time.Duration(s.resetSeconds) * time.Second).Format("2006-01-02") + for _, group := range keys { + if !s.available(ctx, group) { + continue + } + rows := s.design.Groups[group] + if len(rows) == 0 { + continue + } + p := v.Progress[group] + if p.LastDay != day && p.Free < uint64(len(rows)) { + rewards = append(rewards, rows[p.Free].Free) + p.Free++ + p.LastDay = day + r.Rewarded[group] = true + } + if s.paid(ctx, rows[0].TicketID) { + for p.Premium < p.Free { + rewards = append(rewards, rows[p.Premium].Premium) + p.Premium++ + r.Rewarded[group] = true + } + } + v.Progress[group] = p + } + if len(rewards) > 0 { + r.Bundle, err = s.economy.Apply(ctx, "login-pass:"+identity, nil, rewards) + if err != nil { + return nil, nil, err + } + } + v.Receipts[identity] = r + raw, err := json.Marshal(v) + if err != nil { + return nil, nil, err + } + if err = s.store.Save(ctx.State, "commerce_login_passes", raw); err != nil { + return nil, nil, err + } + } + var infos [][]byte + for _, group := range keys { + if !s.available(ctx, group) { + continue + } + p := v.Progress[group] + if p.Free == 0 { + continue + } + b := wire.AppendVarint(nil, 1, group) + b = wire.AppendVarint(b, 2, p.Free) + if r.Rewarded[group] { + b = wire.AppendVarint(b, 3, 1) + } + infos = append(infos, b) + } + return append([]byte(nil), r.Bundle...), infos, nil +} + +// AttachEventShopSchedules joins event type 15 to EventShopTable and then to +// CashProductTable. Hub IDs are presentation identities, never product groups. +// All scheduled goods are published; purchase handlers enforce their windows. +// Call once before serving sessions. +func (s *Service) AttachEventShopSchedules(design *gamedata.CashCatalog, schedules []events.Schedule) error { + if design == nil || s.shopWindows == nil { + return fmt.Errorf("commerce: missing event shop catalog or shop seed") + } + shops := map[uint64]uint64{} + groups := map[uint64]bool{} + for _, shop := range design.EventShops { + if shop.ID == 0 || shop.ProductGroupID == 0 || shops[shop.ID] != 0 || groups[shop.ProductGroupID] { + return fmt.Errorf("commerce: invalid or duplicate event shop identity") + } + shops[shop.ID], groups[shop.ProductGroupID] = shop.ProductGroupID, true + } + windows := map[gamedata.CashProductKey][][2]uint64{} + var products [][]byte + for _, raw := range s.shopProducts { + group, _, _ := wire.Varint(raw, 1) + if !groups[group] { + products = append(products, raw) + } + } + rows := append([]events.Schedule(nil), schedules...) + sort.Slice(rows, func(i, j int) bool { return rows[i].UID < rows[j].UID }) + for _, row := range rows { + if row.Type != 15 { + continue + } + group := shops[row.ID] + if group == 0 || row.UID == 0 || row.Start <= 0 || row.Start >= row.End { + return fmt.Errorf("commerce: invalid event shop schedule %d", row.UID) + } + found := false + start, end := uint64(row.Start), uint64(row.End) + for _, product := range s.catalog.Designs() { + key := product.Key + if key.GroupID != group { + continue + } + found = true + for _, prior := range windows[key] { + if start < prior[1] && prior[0] < end { + return fmt.Errorf("commerce: overlapping event shop product windows") + } + } + windows[key] = append(windows[key], [2]uint64{start, end}) + raw := wire.AppendVarint(nil, 1, key.GroupID) + raw = wire.AppendVarint(raw, 2, key.ProductID) + raw = wire.AppendVarint(raw, 3, key.SaleGroup) + raw = wire.AppendVarint(raw, 4, start) + raw = wire.AppendVarint(raw, 5, end) + raw = wire.AppendVarint(raw, 8, row.UID) + products = append(products, raw) + } + if !found { + return fmt.Errorf("commerce: event shop %d has no products", row.ID) + } + } + s.shopProducts, s.eventShopWindows, s.eventShopGroups = products, windows, groups + return nil +} + +func (e *EntitlementEconomy) Apply(ctx command.Context, identity string, costs, rewards []gamedata.Reward) ([]byte, error) { + + return e.apply(ctx, identity, costs, rewards) +} + +func (e *EntitlementEconomy) ApplyResolved(ctx command.Context, identity string, costs, rewards []gamedata.Reward) ([]byte, error) { + + return e.applyPrepared(ctx, identity, costs, rewards, true, nil) +} + +func (e *EntitlementEconomy) apply(ctx command.Context, identity string, costs, rewards []gamedata.Reward) ([]byte, error) { + return e.applyPrepared(ctx, identity, costs, rewards, false, nil) +} + +func (e *EntitlementEconomy) applyPrepared(ctx command.Context, identity string, costs, rewards []gamedata.Reward, resolved bool, mailed []gamedata.BattleReward) ([]byte, error) { + if identity == "" { + return nil, fmt.Errorf("commerce: missing entitlement identity") + } + definition, _ := json.Marshal(struct{ Costs, Rewards []gamedata.Reward }{costs, rewards}) + digest := sha256.Sum256(definition) + s, err := e.load(ctx) + if err != nil { + return nil, err + } + if r, ok := s.Receipts[identity]; ok { + if !bytes.Equal(r.Definition, digest[:]) { + return nil, fmt.Errorf("commerce: entitlement identity reused") + } + return append([]byte(nil), r.Bundle...), nil + } + input := make([]gamedata.BattleReward, len(rewards)) + for i, r := range rewards { + input[i] = gamedata.BattleReward{Type: r.Type, ID: r.ID, Count: r.Count} //nolint:staticcheck // S1016 + } + leaves := input + if !resolved { + leaves, err = e.graph.ResolveGranted(input) + } + if err != nil { + return nil, err + } + // Some product boxes include the first attendance reward (draw-ticket + // subscriptions); others include only the upfront paid currency. Add only + // the missing first-day components on activation, then mark row one claimed. + initial := append([]gamedata.BattleReward(nil), leaves...) + for _, r := range initial { + rows := e.design.Attendance[r.ID] + if r.Type != 19 || len(rows) == 0 { + continue + } + sub := s.Subscriptions[strconv.FormatUint(r.ID, 10)] + if sub.Start != 0 && (sub.Expiry == 0 || sub.Expiry > e.now().UnixMilli()) { + continue + } + first, resolveErr := e.graph.ResolveGranted([]gamedata.BattleReward{rows[0].Reward}) + if resolveErr != nil { + return nil, resolveErr + } + available := map[[2]uint64]uint64{} + for _, leaf := range append(append([]gamedata.BattleReward(nil), leaves...), mailed...) { + k := [2]uint64{leaf.Type, leaf.ID} + available[k] += leaf.Count + } + for _, leaf := range first { + k := [2]uint64{leaf.Type, leaf.ID} + if available[k] < leaf.Count { + leaf.Count -= available[k] + leaves = append(leaves, leaf) + } + } + } + var regular []gamedata.Reward + var special []assets.Item + for _, r := range leaves { + if r.Count == 0 || r.Count > math.MaxInt32 { + return nil, fmt.Errorf("commerce: invalid entitlement quantity") + } + switch { + case r.Type == 62: + if !e.design.AvatarSets[r.ID] { + return nil, fmt.Errorf("commerce: unknown avatar set %d", r.ID) + } + // The shared gameplay economy expands AvatarSetTable members and + // emits real AvatarItem/AvatarMotion/AvatarChar ownership. + regular = append(regular, gamedata.Reward(r)) + case r.Type == 19 && e.design.TicketTypes[r.ID] == 2: + if len(e.design.Attendance[r.ID]) == 0 { + return nil, fmt.Errorf("commerce: subscription reward schedule missing") + } + now := e.now().UnixMilli() + expiry := max(int64(e.items.ContentTicketExpiry(r.ID)), now) + if r.Count > uint64((math.MaxInt64-expiry)/(30*86400000)) { + return nil, fmt.Errorf("commerce: subscription expiry overflow") + } + expiry += int64(r.Count) * 30 * 86400000 + special = append(special, assets.Item{Type: 19, ID: r.ID, Count: r.Count, ExpiryTime: uint64(expiry), TimeValue: uint64(e.now().UnixMilli())}) + key := strconv.FormatUint(r.ID, 10) + sub := s.Subscriptions[key] + if sub.Start == 0 || sub.Expiry <= now { + sub = cashSubscription{Start: now, Claimed: 1, LastDay: e.day()} + } + sub.Expiry = expiry + s.Subscriptions[key] = sub + default: + regular = append(regular, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 + if r.Type == 19 && e.design.TicketTypes[r.ID] == 3 && len(e.design.Attendance[r.ID]) > 0 { + key := strconv.FormatUint(r.ID, 10) + if _, exists := s.Subscriptions[key]; !exists { + s.Subscriptions[key] = cashSubscription{Start: e.now().UnixMilli(), Claimed: 1, LastDay: e.day()} + } + } + } + } + bundle, err := e.base.Apply(ctx, identity+":base", costs, regular) + if err != nil { + return nil, err + } + items, err := e.items.GrantCommerceOnce(ctx, identity+":special", special) + if err != nil { + return nil, err + } + for _, i := range items { + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(i)) + bundle = wire.AppendBytes(bundle, 6, assets.ItemWire(assets.Item{ID: i.ID, Type: i.Type, Count: i.Count})) + } + s.Receipts[identity] = entitlementReceipt{Definition: append([]byte(nil), digest[:]...), Bundle: append([]byte(nil), bundle...)} + if err = e.save(ctx, s); err != nil { + return nil, err + } + return bundle, nil +} + +// ClaimSubscriptions grants the next row once per server reset day. Missing +// login days are not retroactively claimed. First-row purchase rewards are +// already present in the product box, matching the client's first-row marker. +func (e *EntitlementEconomy) ClaimSubscriptions(ctx command.Context, identity string) ([]byte, error) { + + if identity == "" { + return nil, fmt.Errorf("commerce: missing attendance receipt identity") + } + s, err := e.load(ctx) + if err != nil { + return nil, err + } + receiptKey := "attendance-reply:" + identity + if receipt, ok := s.Receipts[receiptKey]; ok { + return append([]byte(nil), receipt.Bundle...), nil + } + keys := make([]string, 0, len(s.Subscriptions)) + for k := range s.Subscriptions { + keys = append(keys, k) + } + sort.Strings(keys) + var bundle []byte + for _, key := range keys { + sub := s.Subscriptions[key] + ticket, _ := strconv.ParseUint(key, 10, 64) + rows := e.design.Attendance[ticket] + if sub.LastDay == e.day() || sub.Expiry != 0 && sub.Expiry <= e.now().UnixMilli() || sub.Expiry == 0 && sub.Claimed >= uint64(len(rows)) { + continue + } + index := sub.Claimed % uint64(len(rows)) + r := rows[index] + grant, err := e.apply(ctx, fmt.Sprintf("commerce:attendance:%s:%s", key, e.day()), nil, []gamedata.Reward{{Type: r.Reward.Type, ID: r.Reward.ID, Count: r.Reward.Count}}) + if err != nil { + return nil, err + } + bundle = append(bundle, grant...) + // apply persists its receipt; reload before saving progress so it survives. + latest, err := e.load(ctx) + if err != nil { + return nil, err + } + sub.Claimed++ + sub.LastDay = e.day() + latest.Subscriptions[key] = sub + if err = e.save(ctx, latest); err != nil { + return nil, err + } + s = latest + } + latest, err := e.load(ctx) + if err != nil { + return nil, err + } + digest := sha256.Sum256([]byte(receiptKey)) + latest.Receipts[receiptKey] = entitlementReceipt{Definition: append([]byte(nil), digest[:]...), Bundle: append([]byte(nil), bundle...)} + if err = e.save(ctx, latest); err != nil { + return nil, err + } + return bundle, nil +} + +func (e *EntitlementEconomy) MergeAttendance(ctx command.Context, response []byte) ([]byte, error) { + + s, err := e.load(ctx) + if err != nil { + return nil, err + } + out := append([]byte(nil), response...) + keys := make([]string, 0, len(s.Subscriptions)) + for k := range s.Subscriptions { + keys = append(keys, k) + } + sort.Strings(keys) + for _, key := range keys { + sub := s.Subscriptions[key] + ticket, _ := strconv.ParseUint(key, 10, 64) + if sub.Expiry != 0 { + b := wire.AppendVarint(nil, 1, ticket) + b = wire.AppendVarint(b, 2, uint64(sub.Start)) + b = wire.AppendVarint(b, 3, uint64(sub.Expiry)) + out = wire.AppendBytes(out, 3, b) + } + if typ := e.design.AttendanceTypes[ticket]; typ != 0 { + rewards := []byte{} + n := min(sub.Claimed, uint64(len(e.design.Attendance[ticket]))) + for i := uint64(1); i <= n; i++ { + rewards = wire.AppendVarint(rewards, 1, i) + } + entry := wire.AppendVarint(nil, 1, typ) + entry = wire.AppendBytes(entry, 2, rewards) + out = wire.AppendBytes(out, 7, entry) + } else { + for i := uint64(1); i <= sub.Claimed; i++ { + b := wire.AppendVarint(nil, 1, ticket) + b = wire.AppendVarint(b, 2, i) + b = wire.AppendVarint(b, 3, 1) + out = wire.AppendBytes(out, 4, b) + } + } + } + return out, nil +} + +// ApplyPurchase selects rewards once, activates direct entitlements, and issues +// durable cash mail. All writes run in the enclosing account transaction. +func (e *EntitlementEconomy) ApplyPurchase(ctx command.Context, identity string, costs, rewards []gamedata.Reward) ([]byte, error) { + + if e.mail == nil { + return nil, fmt.Errorf("commerce: cash mail issuer unavailable") + } + definition, _ := json.Marshal(struct{ Costs, Rewards []gamedata.Reward }{costs, rewards}) + digest := sha256.Sum256(definition) + key := identity + ":delivery" + state, err := e.load(ctx) + if err != nil { + return nil, err + } + if receipt, ok := state.Receipts[key]; ok { + if !bytes.Equal(receipt.Definition, digest[:]) { + return nil, fmt.Errorf("commerce: delivery identity reused") + } + return append([]byte(nil), receipt.Bundle...), nil + } + resolver, ok := e.graph.(interface { + ResolveDelivery([]gamedata.BattleReward) (gamedata.CashDelivery, error) + }) + if !ok { + return nil, fmt.Errorf("commerce: missing cash delivery resolver") + } + input := make([]gamedata.BattleReward, len(rewards)) + for i, r := range rewards { + input[i] = gamedata.BattleReward(r) + } + plan, err := resolver.ResolveDelivery(input) + if err != nil { + return nil, err + } + direct := make([]gamedata.Reward, len(plan.Direct)) + for i, r := range plan.Direct { + direct[i] = gamedata.Reward(r) + } + var mailed []gamedata.BattleReward + for _, mail := range plan.Mail { + mailed = append(mailed, mail.Rewards...) + } + bundle, err := e.applyPrepared(ctx, identity+":direct", costs, direct, true, mailed) + if err != nil { + return nil, err + } + for i, mail := range plan.Mail { + attachments := make([]gamedata.Reward, len(mail.Rewards)) + for j, r := range mail.Rewards { + attachments[j] = gamedata.Reward(r) + } + if err := e.mail.IssueCashOnce(ctx, fmt.Sprintf("%s:%d", identity, i), mail.TemplateID, attachments, e.now()); err != nil { + return nil, err + } + } + state, err = e.load(ctx) + if err != nil { + return nil, err + } + state.Receipts[key] = entitlementReceipt{Definition: digest[:], Bundle: append([]byte(nil), bundle...)} + if err = e.save(ctx, state); err != nil { + return nil, err + } + return bundle, nil +} + +// Both clear-info fields are ordinary proto3 messages, not a oneof. The native +// client creates an empty message for the inactive kind (CommonPacket), and +// omits scalar zero values such as ClearPackagePack and the normal pack level. +func parseClearClaim(request []byte) (clearClaim, error) { + var scalar [2]uint64 + var rows [2][]byte + var seen [4]bool + err := wire.Walk(request, func(f wire.Field) error { + if f.Number < 1 || f.Number > 4 { + return nil + } + if seen[f.Number-1] { + return fmt.Errorf("commerce: duplicate clear claim field %d", f.Number) + } + seen[f.Number-1] = true + if f.Number <= 2 { + if f.Type != 0 { + return fmt.Errorf("commerce: invalid clear claim scalar %d", f.Number) + } + scalar[f.Number-1], _ = binary.Uvarint(f.Value) + if scalar[f.Number-1] > math.MaxInt32 { + return fmt.Errorf("commerce: clear claim scalar %d exceeds int32", f.Number) + } + } else { + if f.Type != 2 { + return fmt.Errorf("commerce: invalid clear claim row %d", f.Number) + } + rows[f.Number-3] = f.Value + } + return nil + }) + if err != nil { + return clearClaim{}, err + } + if scalar[0] == 0 { + return clearClaim{}, fmt.Errorf("commerce: invalid clear claim sequence") + } + kind := scalar[1] + if kind > 1 { + return clearClaim{}, fmt.Errorf("commerce: invalid clear claim type") + } + if !seen[kind+2] { + return clearClaim{}, fmt.Errorf("commerce: missing clear claim row") + } + var active [4]uint64 + for i, row := range rows { + var values [4]uint64 + var fields [4]bool + if err := wire.Walk(row, func(f wire.Field) error { + if f.Number < 1 || f.Number > 4 { + // Unknown active fields retain normal protobuf compatibility. An + // inactive row must contain only the schema's default scalars. + if uint64(i) != kind { + return fmt.Errorf("commerce: nonempty inactive clear claim row") + } + return nil + } + if fields[f.Number-1] || f.Type != 0 { + return fmt.Errorf("commerce: invalid clear claim row scalar %d", f.Number) + } + fields[f.Number-1] = true + v, _ := binary.Uvarint(f.Value) + if v > math.MaxInt32 { + return fmt.Errorf("commerce: clear claim row scalar %d exceeds int32", f.Number) + } + if uint64(i) != kind && v != 0 { + return fmt.Errorf("commerce: conflicting inactive clear claim row") + } + values[f.Number-1] = v + return nil + }); err != nil { + return clearClaim{}, err + } + if uint64(i) == kind { + active = values + } + } + if active[0] == 0 || active[1] == 0 { + return clearClaim{}, fmt.Errorf("commerce: invalid clear claim identity") + } + return clearClaim{kind, active[0], active[1], active[2], active[3]}, nil +} + +func (s *ClearPackages) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { + if path != "/ClearPackageReward" { + return 0, nil, false, nil + } + claim, err := parseClearClaim(request) + if err != nil { + return 286, nil, true, err + } + d, ok := s.design[claim] + if !ok { + return 286, nil, true, fmt.Errorf("commerce: unknown clear reward") + } + + v, err := s.load(ctx) + if err != nil { + return 286, nil, true, err + } + identity := clearClaimID(claim) + if receipt, ok := v[identity]; ok { + return 286, append([]byte(nil), receipt.Response...), true, nil + } + if d.Type == 1 && !s.entitled(ctx, d.TicketID) { + return 286, nil, true, fmt.Errorf("commerce: clear reward premium ticket required") + } + proof := s.packCleared + if claim.Kind == 1 { + proof = s.towerCleared + } + if proof == nil || !proof(d.TargetID, d.Level) { + return 286, nil, true, fmt.Errorf("commerce: clear reward progression incomplete") + } + rewards := []gamedata.Reward{{Type: 9, ID: d.RandomBoxID, Count: 1}} + var bundle []byte + if delivery, ok := s.economy.(interface { + ApplyPurchase(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) + }); ok { + // Clear reward groups carry MailId just like cash products. The + // delivery resolver selects the versioned mail template and keeps + // mailed contents out of the direct inventory response. + bundle, err = delivery.ApplyPurchase(ctx, identity, nil, rewards) + } else { + bundle, err = s.economy.Apply(ctx, identity, nil, rewards) + } + if err != nil { + return 286, nil, true, err + } + response := wire.AppendBytes(nil, 1, bundle) + v[identity] = clearClaimReceipt{Claim: claim, Response: response} + raw, err := json.Marshal(v) + if err == nil { + err = s.store.Save(ctx.State, "commerce_clear_claims", raw) + } + return 286, response, true, err +} + +// RewardDBInfos is attached to PackInfoResponse fields 3 and 4 so reconnects +// restore claimed reward buttons from server state. +func (s *ClearPackages) RewardDBInfos(ctx command.Context) (pack, evil [][]byte, err error) { + + v, err := s.load(ctx) + if err != nil { + return nil, nil, err + } + keys := make([]string, 0, len(v)) + for k := range v { + keys = append(keys, k) + } + sort.Strings(keys) + for _, k := range keys { + c := v[k].Claim + b := wire.AppendVarint(nil, 1, c.GroupID) + b = wire.AppendVarint(b, 2, c.TicketID) + b = wire.AppendVarint(b, 3, c.TargetID) + b = wire.AppendVarint(b, 4, c.Level) + if c.Kind == 0 { + pack = append(pack, b) + } else { + evil = append(evil, b) + } + } + return pack, evil, nil +} + +func (s *CashBonuses) rewarded(v bonusState, group bonusGroup, field int) []byte { + var out []byte + for _, row := range s.rewards[group] { + if v.Claims[bonusClaimID(group, row.ID)] { + out = wire.AppendVarint(out, field, row.ID) + } + } + return out +} + +func (s *CashBonuses) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + code := 588 + if path == "/CashBonusReward" { + code = 589 + } else if path != "/CashBonusInfo" { + return 0, nil, false, nil + } + seq, present, err := wire.Varint(request, 1) + if err != nil || !present || seq == 0 || seq > math.MaxInt32 { + return code, nil, true, fmt.Errorf("commerce: invalid cash bonus sequence") + } + if err = wire.Walk(request, func(wire.Field) error { return nil }); err != nil { + return code, nil, true, err + } + + v, err := s.load(ctx) + if err != nil { + return code, nil, true, err + } + if path == "/CashBonusInfo" { + var out []byte + for _, group := range s.ordered { + n, err := s.counts.LifetimePurchaseTotal(ctx, s.groups[group]) + if err != nil { + return code, nil, true, err + } + // Preserve the native empty response for groups with no purchases. + if n == 0 { + continue + } + b := wire.AppendVarint(nil, 1, group.ProductGroup) + b = wire.AppendVarint(b, 2, group.ContentsGroup) + b = wire.AppendVarint(b, 3, n) + b = append(b, s.rewarded(v, group, 4)...) + out = wire.AppendBytes(out, 1, b) + } + return code, out, true, nil + } + if commandSession == "" { + return code, nil, true, fmt.Errorf("commerce: cash bonus session unavailable") + } + identity := fmt.Sprintf("cash-bonus:%x:%d", sha256.Sum256([]byte(commandSession)), seq) + digest := fmt.Sprintf("%x", sha256.Sum256(request)) + if receipt, ok := v.Receipts[identity]; ok { + if receipt.Digest != digest { + return code, nil, true, fmt.Errorf("commerce: conflicting cash bonus replay") + } + return code, append([]byte(nil), receipt.Response...), true, nil + } + groupID, _, err := wire.Varint(request, 2) + if err != nil { + return code, nil, true, err + } + contents, _, err := wire.Varint(request, 3) + if err != nil { + return code, nil, true, err + } + id, _, err := wire.Varint(request, 4) + if err != nil { + return code, nil, true, err + } + group := bonusGroup{groupID, contents} + var chosen *gamedata.CashBonusReward + for _, row := range s.rewards[group] { + if row.ID == id { + copy := row + chosen = © + break + } + } + if chosen == nil { + return code, nil, true, fmt.Errorf("commerce: unknown cash bonus group=%d contents=%d bonus=%d", groupID, contents, id) + } + n, err := s.counts.LifetimePurchaseTotal(ctx, s.groups[group]) + if err != nil { + return code, nil, true, err + } + if n < chosen.RequireCount { + return code, nil, true, fmt.Errorf("commerce: cash bonus purchase threshold not reached group=%d contents=%d bonus=%d count=%d require=%d", groupID, contents, id, n, chosen.RequireCount) + } + claimID := bonusClaimID(group, id) + var bundle []byte + if !v.Claims[claimID] { + bundle, err = s.economy.Apply(ctx, "cash-bonus-grant:"+claimID, nil, []gamedata.Reward{chosen.Reward}) + if err != nil { + return code, nil, true, err + } + v.Claims[claimID] = true + } + response := wire.AppendBytes(nil, 1, bundle) + response = append(response, s.rewarded(v, group, 2)...) + v.Receipts[identity] = purchaseReceipt{Digest: digest, Response: response} + raw, err := json.Marshal(v) + if err == nil { + err = s.store.Save(ctx.State, "commerce_cash_bonuses", raw) + } + return code, response, true, err +} + +func (h AttendanceHandler) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + if path != "/Attendance" { + return 0, nil, false, nil + } + if commandSession == "" || h.Events == nil || h.Economy == nil || h.Store == nil { + return 0, nil, true, fmt.Errorf("commerce: attendance dependencies/session unavailable") + } + seq, found, err := wire.Varint(request, 1) + if err != nil || !found || seq == 0 { + return 0, nil, true, fmt.Errorf("commerce: missing attendance sequence") + } + code, response, handled, err := h.Events.Handle(ctx, path, request) + if err != nil || !handled { + return code, response, handled, err + } + // The event handler may already have granted ordinary attendance rewards. + // The client accepts exactly one reward envelope, so combine every grant in + // execution order under this wrapper's replay identity. + response, eventBundle, err := takeAttendanceRewardEnvelope(response) + if err != nil { + return code, nil, true, err + } + key := fmt.Sprintf("commerce_attendance:%x:%d", sha256.Sum256([]byte(commandSession)), seq) + digest := fmt.Sprintf("%x", sha256.Sum256(request)) + var receipt attendanceReceipt + previous, err := h.Store.Load(ctx.State, key) + if err != nil { + return code, nil, true, err + } + var loginBundle []byte + if h.LoginPasses != nil { + var infos [][]byte + loginBundle, infos, err = h.LoginPasses.ClaimAndInfo(ctx, key) + if err != nil { + return code, nil, true, err + } + for _, info := range infos { + response = wire.AppendBytes(response, 6, info) + } + } + if previous != nil { + if err = json.Unmarshal(previous, &receipt); err != nil || receipt.Digest != digest { + return code, nil, true, fmt.Errorf("commerce: conflicting attendance replay") + } + } else { + receipt.Digest = digest + subscriptionBundle, err := h.Economy.ClaimSubscriptions(ctx, key) + if err != nil { + return code, nil, true, err + } + // Preserve events -> login pass -> subscription execution order and all + // repeated reward entries in the single client envelope. + receipt.Bundle = append(receipt.Bundle, eventBundle...) + receipt.Bundle = append(receipt.Bundle, loginBundle...) + receipt.Bundle = append(receipt.Bundle, subscriptionBundle...) + raw, err := json.Marshal(receipt) + if err != nil { + return code, nil, true, err + } + if err = h.Store.Save(ctx.State, key, raw); err != nil { + return code, nil, true, err + } + } + response, err = h.Economy.MergeAttendance(ctx, response) + if err != nil { + return code, nil, true, err + } + if len(receipt.Bundle) != 0 { + response = wire.AppendBytes(response, 1001, receipt.Bundle) + response = wire.AppendString(response, 1002, key) + } + return code, response, true, nil +} + +// Strip the child envelope before adding the combined one. Keep native and +// unrelated unknown fields byte-for-byte; reject ambiguous child envelopes +// rather than returning a response the client would silently ignore. +func takeAttendanceRewardEnvelope(response []byte) ([]byte, []byte, error) { + var native, bundle []byte + var receipt string + var hasBundle, hasReceipt bool + err := wire.Walk(response, func(f wire.Field) error { + switch f.Number { + case 1001: + if f.Type != 2 || hasBundle { + return fmt.Errorf("commerce: ambiguous attendance reward bundle") + } + hasBundle = true + bundle = append([]byte(nil), f.Value...) + case 1002: + if f.Type != 2 || hasReceipt { + return fmt.Errorf("commerce: ambiguous attendance reward receipt") + } + hasReceipt = true + receipt = string(f.Value) + default: + native = append(native, response[f.Start:f.End]...) + } + return nil + }) + if err != nil { + return nil, nil, err + } + if hasBundle != hasReceipt || hasReceipt && (receipt == "" || len(receipt) > 1024) { + return nil, nil, fmt.Errorf("commerce: incomplete attendance reward envelope") + } + return native, bundle, nil +} diff --git a/go/internal/server/domain/commerce/service.go b/go/internal/server/domain/commerce/service.go new file mode 100644 index 0000000..fb39efd --- /dev/null +++ b/go/internal/server/domain/commerce/service.go @@ -0,0 +1,242 @@ +package commerce + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/storage/stateio" + "crypto/sha256" + "encoding/hex" + "encoding/json" + "fmt" + "math" + + "time" +) + +// Economy and the receipt store must participate in the caller's account +// transaction. A failed hook, delegate or save then rolls back all purchases. +type Economy interface { + Apply(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) +} +type CountProvider interface { + PurchaseCountDBInfos(ctx command.Context) [][]byte +} +type PurchaseDelegate func(command.Context, gamedata.CashProductKey, []byte) ([]byte, bool, error) +type PurchaseHook func(command.Context, string, gamedata.CashProductDesign, uint64) error +type purchaseReceipt struct { + Digest string `json:"digest"` + Response []byte `json:"response"` +} +type purchaseCount struct { + Count uint64 `json:"count"` + Period string `json:"period"` + Lifetime uint64 `json:"lifetime"` +} +type purchaseState struct { + Receipts map[string]purchaseReceipt `json:"receipts"` + Billing map[string]string `json:"billing"` + Counts map[string]purchaseCount `json:"counts"` +} +type Service struct { + catalog *Catalog + store stateio.Store + economy Economy + now func() time.Time + resetSeconds int64 + delegate PurchaseDelegate + specialProducts map[gamedata.CashProductKey]bool + hook PurchaseHook + legacy CountProvider + shopProducts [][]byte + shopWindows map[gamedata.CashProductKey][2]uint64 + eventShopWindows map[gamedata.CashProductKey][][2]uint64 + eventShopGroups map[uint64]bool + predecessors map[gamedata.CashProductKey][]gamedata.CashProductKey +} + +func NewService(ctx command.Context, catalog *Catalog, store stateio.Store, economy Economy) (*Service, error) { + if catalog == nil || store == nil || economy == nil { + return nil, fmt.Errorf("commerce: invalid service dependencies") + } + s := &Service{catalog: catalog, store: store, economy: economy, now: time.Now} + _, err := s.load(ctx) + return s, err +} +func (s *Service) AttachDelegate(d PurchaseDelegate) { s.delegate = d } +func (s *Service) AttachPurchaseHook(h PurchaseHook) { s.hook = h } +func (s *Service) AttachLegacyCounts(p CountProvider) { s.legacy = p } + +// AttachPackageRules enforces the client's type-2/type-8 step and relay +// ordering using versioned contentsGroupId/contentsSortId, never SKU numbers. +func (s *Service) AttachPackageRules(packages []gamedata.CashPackageDesign) error { + rules := map[gamedata.CashProductKey][]gamedata.CashProductKey{} + for _, p := range packages { + if (p.PackageType != 2 && p.PackageType != 8) || p.ContentsGroupID == 0 || p.ContentsSortID <= 1 { + continue + } + key := gamedata.CashProductKey{GroupID: p.GroupID, ProductID: p.ID, SaleGroup: p.SaleGroup} + var prior []gamedata.CashProductKey + for _, q := range packages { + if q.PackageType == p.PackageType && q.ContentsGroupID == p.ContentsGroupID && q.ContentsSortID == p.ContentsSortID-1 { + prior = append(prior, gamedata.CashProductKey{GroupID: q.GroupID, ProductID: q.ID, SaleGroup: q.SaleGroup}) + } + } + // Some current-version packages have a sort label above one but an + // independent contents group. Only a real same-group predecessor is a + // progression rule; do not invent links between adjacent product IDs. + if len(prior) == 0 { + continue + } + rules[key] = prior + } + s.predecessors = rules + return nil +} + +func (s *Service) available(d gamedata.CashProductDesign) bool { + if s.eventShopGroups[d.Key.GroupID] { + now := uint64(s.now().UnixMilli()) + for _, w := range s.eventShopWindows[d.Key] { + if w[0] <= now && now < w[1] { + return true + } + } + return false + } + w, ok := s.shopWindows[d.Key] + if !ok { + return d.TimeLimitType == 0 + } + now := uint64(s.now().UnixMilli()) + return w[0] <= now && (w[1] == 0 || now < w[1]) +} +func (s *Service) IsAvailable(ctx command.Context, key gamedata.CashProductKey) bool { + d, ok := s.catalog.Design(key) + return ok && s.available(d) +} + +func (s *Service) SetClock(now func() time.Time, resetSeconds int64) { + s.now = now + s.resetSeconds = resetSeconds +} +func skuKey(k gamedata.CashProductKey) string { + return fmt.Sprintf("%d:%d:%d", k.GroupID, k.ProductID, k.SaleGroup) +} +func (s *Service) load(ctx command.Context) (purchaseState, error) { + v := purchaseState{Receipts: map[string]purchaseReceipt{}, Billing: map[string]string{}, Counts: map[string]purchaseCount{}} + b, e := s.store.Load(ctx.State, "commerce") + if e != nil || b == nil { + return v, e + } + if e = stateio.RequireExactJSONObject(b, "receipts", "billing", "counts"); e != nil { + return v, e + } + if e = json.Unmarshal(b, &v); e != nil { + return v, e + } + if v.Receipts == nil || v.Billing == nil || v.Counts == nil { + return v, fmt.Errorf("commerce: invalid state") + } + for identity, receipt := range v.Receipts { + digest, err := hex.DecodeString(receipt.Digest) + if identity == "" || err != nil || len(digest) != sha256.Size || len(receipt.Response) == 0 { + return v, fmt.Errorf("commerce: invalid saved receipt") + } + } + for _, identity := range v.Billing { + if _, ok := v.Receipts[identity]; !ok { + return v, fmt.Errorf("commerce: billing identity has no purchase receipt") + } + } + for key, count := range v.Counts { + if key == "" || count.Period == "" || count.Count == 0 || count.Count > math.MaxInt32 || count.Lifetime < count.Count { + return v, fmt.Errorf("commerce: invalid saved purchase count") + } + } + return v, nil +} +func (s *Service) period(typ uint64) string { + t := s.now().UTC().Add(-time.Duration(s.resetSeconds) * time.Second) + switch typ { + case 1: + return t.Format("2006-01-02") + case 2: + y, w := t.ISOWeek() + return fmt.Sprintf("%d-W%02d", y, w) + case 3: + return t.Format("2006-01") + default: + return "account" + } +} +func (s *Service) count(v purchaseState, d gamedata.CashProductDesign) uint64 { + c := v.Counts[skuKey(d.Key)] + if c.Period != s.period(d.PurchaseLimitType) { + return 0 + } + return c.Count +} + +type buyLine struct { + Key gamedata.CashProductKey + Count uint64 +} + +// LifetimePurchaseTotal derives cumulative bonuses from committed purchases, +// including previous reset periods, without keeping a second purchase counter. +func (s *Service) LifetimePurchaseTotal(ctx command.Context, keys []gamedata.CashProductKey) (uint64, error) { + + v, err := s.load(ctx) + if err != nil { + return 0, err + } + var total uint64 + for _, k := range keys { + n := v.Counts[skuKey(k)].Lifetime + if n > math.MaxInt32-total { + return 0, fmt.Errorf("commerce: bonus purchase count overflow") + } + total += n + } + return total, nil +} +func (s *Service) HasPurchased(ctx command.Context, k gamedata.CashProductKey) bool { + + v, e := s.load(ctx) + if e != nil { + return false + } + return v.Counts[skuKey(k)].Count > 0 +} + +// ConsumeEntitlement is an authorization callback within the caller's account +// transaction. Each successful cash activation consumes one lifetime purchase. +// It must not be called recursively from the purchase hook or delegate. +func (s *Service) ConsumeEntitlement(ctx command.Context, k gamedata.CashProductKey) bool { + + v, err := s.load(ctx) + if err != nil { + return false + } + n := v.Counts[skuKey(k)].Lifetime + used := map[string]uint64{} + raw, err := s.store.Load(ctx.State, "commerce_pass_receipts") + if err != nil { + return false + } + if raw != nil { + if err = json.Unmarshal(raw, &used); err != nil || used == nil { + return false + } + } + key := skuKey(k) + if used[key] >= n { + return false + } + used[key]++ + raw, err = json.Marshal(used) + if err != nil { + return false + } + return s.store.Save(ctx.State, "commerce_pass_receipts", raw) == nil +} diff --git a/go/internal/server/domain/events/actions/battle.go b/go/internal/server/domain/events/actions/battle.go new file mode 100644 index 0000000..4c26fcf --- /dev/null +++ b/go/internal/server/domain/events/actions/battle.go @@ -0,0 +1,61 @@ +package eventactions + +import ( + "bd2server/internal/server/domain/events" + "crypto/sha256" + "encoding/hex" +) + +func (s *Service) tacticsLines(uid uint64) uint64 { + v, _ := s.registry.Resolve(uid) + g, ok := s.design.Row("TacticsBingoGroupTable", 3, v.ID) + if !ok || g.V(4) > 2 { + return 0 + } + n := uint64(5) - g.V(4) + clear := s.state.Tactics[uid] + var lines uint64 + for i := range n { + row, col := true, true + for j := range n { + row = row && contains(clear, i*n+j+1) + col = col && contains(clear, j*n+i+1) + } + if row { + lines++ + } + if col { + lines++ + } + } + a, b := true, true + for i := range n { + a = a && contains(clear, i*n+i+1) + b = b && contains(clear, i*n+(n-i)) + } + if a { + lines++ + } + if b { + lines++ + } + return lines +} + +func (s *Service) AssociatedMissionGroup(v events.Schedule) uint64 { + if v.Type == 20 { + r, ok := s.design.Row("TacticsBingoGroupTable", 3, v.ID) + if ok { + return r.V(2) + } + } + if v.Type == 23 { + r, ok := s.design.Row("VotingEventTable", 6, v.ID) + if ok { + return r.V(12) + } + } + return 0 +} +func receiptRecord(b []byte, digest string) receipt { return receipt{Digest: digest, Reply: b} } +func battleDigest(b []byte) string { sum := sha256.Sum256(b); return hex.EncodeToString(sum[:]) } diff --git a/go/internal/server/domain/events/actions/mini_content.go b/go/internal/server/domain/events/actions/mini_content.go new file mode 100644 index 0000000..9d800ca --- /dev/null +++ b/go/internal/server/domain/events/actions/mini_content.go @@ -0,0 +1,64 @@ +package eventactions + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + "errors" + "fmt" +) + +type MiniContentResolver interface { + ResolveMiniContentUID(ctx command.Context, _ uint64) (uint64, uint64, int64, int64, error) + ListMiniContentRoutes(ctx command.Context) ([]gamedata.MiniContentRoute, error) +} + +func (s *Service) AttachMiniContent(ctx command.Context, resolver MiniContentResolver, design *gamedata.MiniContentDesign) error { + if resolver == nil || design == nil { + return errors.New("eventactions: mini content dependencies missing") + } + routes, err := resolver.ListMiniContentRoutes(ctx) + if err != nil { + return err + } + for _, route := range routes { + if route.ContentType == 14 { + ids := design.Groups[route.ContentID] + if len(ids) == 0 { + return fmt.Errorf("eventactions: published mini story group %d missing", route.ContentID) + } + for _, id := range ids { + if _, exists := design.Stories[id]; !exists { + return fmt.Errorf("eventactions: published daily story %d missing", id) + } + } + } + if route.ContentType == 13 { + found := false + for _, row := range s.design.Tables["NpcQuizTable"] { + if row.V(1) == route.ContentID { + found = true + break + } + } + if !found { + return fmt.Errorf("eventactions: published NPC quiz group %d missing", route.ContentID) + } + } + } + s.miniContent, s.miniDesign = resolver, design + return nil +} + +func (s *Service) resolveQuiz(ctx command.Context, uid uint64) (events.Schedule, error) { + if s.miniContent != nil { + typ, id, start, end, err := s.miniContent.ResolveMiniContentUID(ctx, uid) + if err == nil { + if typ != 13 { + return events.Schedule{}, errors.New("eventactions: content UID is not a quiz") + } + return events.Schedule{UID: uid, ID: id, Start: start, End: end}, nil + } + } + return s.registry.Resolve(uid) +} diff --git a/go/internal/server/domain/events/actions/protocol_adapter.go b/go/internal/server/domain/events/actions/protocol_adapter.go new file mode 100644 index 0000000..de9c865 --- /dev/null +++ b/go/internal/server/domain/events/actions/protocol_adapter.go @@ -0,0 +1,937 @@ +package eventactions + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + "bd2server/internal/server/protocol/wire" + "crypto/sha256" + "encoding/hex" + "encoding/json" + "errors" + "fmt" + "math" + "slices" + "sort" + "strconv" + "strings" + "time" +) + +func voteWire(v *vote) []byte { + b := wire.AppendVarint(nil, 1, v.Round) + b = wire.AppendVarint(b, 2, v.Candidate) + b = wire.AppendVarint(b, 3, v.Normal+v.Additional) + b = wire.AppendVarint(b, 4, v.Normal) + return wire.AppendVarint(b, 5, v.Additional) +} + +func (s *Service) voteHandle(ctx command.Context, path string, b []byte, identity string) ([]byte, error) { + v, r, ok := s.voteEvent() + if !ok { + if path == "/CharVoteInfo" || path == "/CharVoteSeasonRanking" { + return nil, nil + } + if path == "/CharVoteTotalRanking" { + return s.totalRanking(val(b, 2)), nil + } + return nil, errors.New("eventactions: voting event inactive") + } + round, rounds := s.rounds(v, r) + ids := s.candidates(r) + ids = s.advancedCandidates(v, r, round, ids) + switch path { + case "/CharVoteInfo": + out := wire.AppendVarint(nil, 1, round) + for _, id := range ids { + out = wire.AppendVarint(out, 2, id) + } + for k, x := range s.state.Votes { + var uid, ro, c uint64 + if _, e := fmt.Sscanf(k, "%d/%d/%d", &uid, &ro, &c); e == nil && uid == v.UID { + out = wire.AppendBytes(out, 3, voteWire(x)) + } + } + for k := range s.state.VoteRewards { + var uid, id uint64 + if _, e := fmt.Sscanf(k, "%d/%d", &uid, &id); e == nil && uid == v.UID { + x := wire.AppendVarint(nil, 1, v.ID) + x = wire.AppendVarint(x, 2, id) + out = wire.AppendBytes(out, 4, x) + } + } + for _, id := range ids { + if s.state.Favorites[id] { + out = wire.AppendVarint(out, 5, id) + } + if s.state.NormalVoted[key(v.UID, id)] { + out = wire.AppendVarint(out, 6, id) + } + } + out = wire.AppendVarint(out, 7, uint64(s.now().UTC().Truncate(24*time.Hour).Add(24*time.Hour).UnixMilli())) + for i, x := range rounds { + start := v.Start + int64(x.V(4))*86400000 + end := v.End + if i+1 < len(rounds) { + end = v.Start + int64(rounds[i+1].V(4))*86400000 + } + a := wire.AppendVarint(nil, 1, x.V(3)) + a = wire.AppendVarint(a, 2, uint64(start)) + a = wire.AppendVarint(a, 3, uint64(end)) + out = wire.AppendBytes(out, 9, a) + } + return out, nil + case "/CharVoteFavoriteAdd", "/CharVoteFavoriteDelete": + id := val(b, 2) + if !contains(ids, id) { + return nil, errors.New("eventactions: unknown candidate") + } + if path == "/CharVoteFavoriteAdd" { + s.state.Favorites[id] = true + } else { + delete(s.state.Favorites, id) + } + return nil, nil + case "/CharVoteSave": + id, count, typ := val(b, 2), val(b, 4), val(b, 5) + if !contains(ids, id) || count == 0 || count > 2147483647 || typ > 1 || round == 0 { + return nil, errors.New("eventactions: invalid vote") + } + normalKey := key(v.UID, id) + if typ == 0 && (s.state.NormalVoted[normalKey] || count != 1) { + return nil, errors.New("eventactions: daily normal vote exhausted") + } + cost := gamedata.Reward{Type: r.V(9), ID: r.V(8), Count: r.V(7) * count} + if typ == 1 { + cost = gamedata.Reward{Type: r.V(3), ID: r.V(2), Count: r.V(1) * count} + } + // A first vote may use the additional ticket when the normal ticket + // is unavailable; the client still sends vote_type=0. + if typ == 0 { + var submittedID uint64 + _ = wire.Walk(b, func(f wire.Field) error { + if f.Number == 3 && f.Type == 2 { + submittedID = val(f.Value, 2) + } + return nil + }) + if submittedID == r.V(2) { + cost = gamedata.Reward{Type: r.V(3), ID: r.V(2), Count: r.V(1) * count} + } + } + if cost.Type == 0 || cost.Count == 0 { + return nil, errors.New("eventactions: missing vote cost") + } + vk := key(v.UID, round, id) + current := s.state.Votes[vk] + if current == nil { + current = &vote{Round: round, Candidate: id} + } + var total uint64 + for k, x := range s.state.Votes { + var uid, ro, c uint64 + if _, e := fmt.Sscanf(k, "%d/%d/%d", &uid, &ro, &c); e == nil && uid == v.UID { + total += x.Normal + x.Additional + } + } + var rewards []gamedata.Reward + var newIDs []uint64 + for _, rr := range s.design.Tables["VotingCountRewardTable"] { + if rr.V(1) == r.V(11) && rr.V(6) <= total+count && !s.state.VoteRewards[key(v.UID, rr.V(2))] { + rewards = append(rewards, rr.Rewards...) + newIDs = append(newIDs, rr.V(2)) + } + } + bundle, e := s.economy.Apply(ctx, identity, []gamedata.Reward{cost}, rewards) + if e != nil { + return nil, e + } + s.state.Votes[vk] = current + if typ == 0 { + current.Normal += count + s.state.NormalVoted[normalKey] = true + } else { + current.Additional += count + } + out := wire.AppendBytes(nil, 1, bundle) + for _, id := range newIDs { + s.state.VoteRewards[key(v.UID, id)] = true + x := wire.AppendVarint(nil, 1, v.ID) + x = wire.AppendVarint(x, 2, id) + out = wire.AppendBytes(out, 2, x) + } + out = wire.AppendBytes(out, 4, voteWire(current)) + return out, nil + case "/CharVoteRanking": + wanted := val(b, 2) + if wanted == 0 { + wanted = round + } + out := wire.AppendVarint(nil, 1, v.ID) + out = wire.AppendVarint(out, 2, wanted) + for _, x := range s.rankRows(v.UID, wanted, ids) { + out = wire.AppendBytes(out, 3, x) + } + return out, nil + case "/CharVoteTotalRanking": + return s.totalRanking(val(b, 2)), nil + case "/CharVoteSeasonRanking": + var out []byte + for _, schedule := range s.registry.List() { + if schedule.Type != 23 { + continue + } + for _, id := range ids { + var total uint64 + for k, x := range s.state.Votes { + var uid, ro, c uint64 + if _, e := fmt.Sscanf(k, "%d/%d/%d", &uid, &ro, &c); e == nil && uid == schedule.UID && c == id { + total += x.Normal + x.Additional + } + } + if total == 0 { + continue + } + x := wire.AppendVarint(nil, 1, schedule.ID) + x = wire.AppendVarint(x, 2, id) + x = wire.AppendVarint(x, 3, total) + x = wire.AppendVarint(x, 4, total) + out = wire.AppendBytes(out, 1, x) + } + } + return out, nil + } + return nil, errors.New("eventactions: unknown voting operation") +} + +func (s *Service) rankRows(uid, round uint64, ids []uint64) [][]byte { + type result struct{ id, count, last uint64 } + var rows []result + var totals map[uint64]uint64 + if s.voteTotals != nil { + totals, _ = s.voteTotals(uid, round) + } + for _, id := range ids { + r := result{id: id} + for k, x := range s.state.Votes { + var u, ro, c uint64 + if _, e := fmt.Sscanf(k, "%d/%d/%d", &u, &ro, &c); e == nil && u == uid && c == id && (round == 0 || round == ro) { + r.count += x.Normal + x.Additional + if ro > r.last { + r.last = ro + } + } + } + if totals != nil { + r.count = totals[id] + } + rows = append(rows, r) + } + sort.Slice(rows, func(i, j int) bool { + if rows[i].count == rows[j].count { + return rows[i].id < rows[j].id + } + return rows[i].count > rows[j].count + }) + var out [][]byte + var lastCount, rank uint64 + for i, r := range rows { + if i == 0 || r.count != lastCount { + rank = uint64(i + 1) + } + lastCount = r.count + x := wire.AppendVarint(nil, 1, r.id) + x = wire.AppendVarint(x, 2, r.count) + x = wire.AppendVarint(x, 3, rank) + x = wire.AppendVarint(x, 4, r.last) + out = append(out, x) + } + return out +} + +func (s *Service) advancedCandidates(v events.Schedule, event gamedata.EventActionRow, round uint64, ids []uint64) []uint64 { + for previous := uint64(1); previous < round; previous++ { + var count uint64 + for _, r := range s.design.Tables["VotingRoundTable"] { + if r.V(2) == event.V(14) && r.V(3) == previous { + count = r.V(1) + } + } + if count == 0 { + continue + } + ranked := s.rankRows(v.UID, previous, ids) + var next []uint64 + for _, b := range ranked { + if val(b, 3) <= count { + next = append(next, val(b, 1)) + } + } + ids = next + } + return ids +} + +func (s *Service) totalRanking(event uint64) []byte { + out := wire.AppendVarint(nil, 1, event) + for _, v := range s.registry.List() { + if v.Type != 23 || v.ID != event { + continue + } + r, ok := s.design.Row("VotingEventTable", 6, event) + if !ok { + continue + } + for _, row := range s.rankRows(v.UID, 0, s.candidates(r)) { + out = wire.AppendBytes(out, 2, row) + } + } + return out +} + +func decodeTacticsDeck(request []byte) ([]tacticsDeckEntry, error) { + var entries []tacticsDeckEntry + indices, characters, positions, sequences := map[uint64]bool{}, map[uint64]bool{}, map[uint64]bool{}, map[uint64]bool{} + err := wire.Walk(request, func(f wire.Field) error { + if f.Number != 2 { + return nil + } + if f.Type != 2 { + return errors.New("eventactions: malformed tactics deck entry") + } + seen := map[int]bool{} + if err := wire.Walk(f.Value, func(v wire.Field) error { + if v.Number >= 1 && v.Number <= 5 { + if v.Type != 0 || seen[v.Number] { + return errors.New("eventactions: malformed tactics deck field") + } + seen[v.Number] = true + } + return nil + }); err != nil { + return err + } + e := tacticsDeckEntry{val(f.Value, 1), val(f.Value, 2), val(f.Value, 3), val(f.Value, 4), val(f.Value, 5)} + // These indices are client-generated virtual identities, not owned + // inventory. Still enforce their signed protocol range and uniqueness. + if e.index == 0 || e.index > math.MaxInt64 || e.character == 0 || e.character > math.MaxInt32 || e.costume == 0 || e.costume > math.MaxInt32 || e.position >= tacticsGridSize || e.sequence == 0 || e.sequence > tacticsPartySize || indices[e.index] || characters[e.character] || positions[e.position] || sequences[e.sequence] { + return errors.New("eventactions: invalid tactics deck") + } + indices[e.index], characters[e.character], positions[e.position], sequences[e.sequence] = true, true, true, true + entries = append(entries, e) + return nil + }) + if err != nil { + return nil, err + } + if len(entries) == 0 || len(entries) > tacticsPartySize { + return nil, errors.New("eventactions: invalid tactics deck size") + } + for sequence := uint64(1); sequence <= uint64(len(entries)); sequence++ { + if !sequences[sequence] { + return nil, errors.New("eventactions: invalid tactics deck sequence") + } + } + return entries, nil +} + +func (s *Service) validateTacticsDeck(request []byte) error { + entries, err := decodeTacticsDeck(request) + if err != nil { + return err + } + // The request only carries seq and deck_info. The client normally enters + // the stage before saving; bind the saved party to that server context. + if s.state.BattleUID != 0 { + event, err := s.resolve(s.state.BattleUID, 20) + if err != nil { + return err + } + group, ok := s.design.Row("TacticsBingoGroupTable", 3, event.ID) + stage, found := s.row("TacticsBingoTable", 4, 5, group.V(1), s.state.BattleStage) + if !ok || !found || stage.V(2) != s.state.BattleDeck { + return errors.New("eventactions: tactics battle context missing") + } + party, err := s.tacticsBlueParty(stage.V(1)) + if err != nil { + return err + } + if matchesTacticsParty(entries, party) { + return nil + } + return errors.New("eventactions: tactics deck does not match entered stage") + } + // Without a bound stage, accept only a complete authored party belonging + // to a currently active tactics schedule, never an arbitrary mixed roster. + for _, event := range s.registry.List() { + if event.Type != 20 || !s.active(event) { + continue + } + group, ok := s.design.Row("TacticsBingoGroupTable", 3, event.ID) + if !ok { + continue + } + for _, stage := range s.design.Tables["TacticsBingoTable"] { + if stage.V(4) != group.V(1) { + continue + } + party, err := s.tacticsBlueParty(stage.V(1)) + if err != nil { + return err + } + if matchesTacticsParty(entries, party) { + return nil + } + } + } + return errors.New("eventactions: tactics deck has no active authored party") +} + +func val(p []byte, n int) uint64 { v, _, _ := wire.Varint(p, n); return v } + +func (s *Service) Handle(ctx command.Context, path string, b []byte) (int, []byte, bool, error) { + codes := map[string]int{"/FieldEventSpawnInfo": 551, "/FieldEventSpawnStart": 552, "/FieldEventSpawnReward": 553, "/FireWorksInfo": 554, "/FireWorksReward": 555, "/CharVoteInfo": 581, "/CharVoteSave": 582, "/CharVoteRanking": 583, "/CharVoteSeasonRanking": 584, "/CharVoteFavoriteAdd": 585, "/CharVoteFavoriteDelete": 586, "/CharVoteTotalRanking": 587, "/FriendshipSpecialEpisodeInfo": 623, "/FriendshipSpecialEpisodeClear": 624, "/ChargeCostInfo": 123, "/NpcQuizInfo": 535, "/NpcQuizClear": 536, "/TacticsBingoInfo": 546, "/TacticsBingoDeckSave": 550} + code, ok := codes[path] + if path == "/CafeteriaEventNpcInteractionReward" { + code, ok = 414, true + } + if path == "/DailyStoryInfo" { + code, ok = 538, true + } + if path == "/DailyStoryClear" { + code, ok = 539, true + } + if !ok { + return 0, nil, false, nil + } + + seq, found, e := wire.Varint(b, 1) + if e != nil || !found || seq == 0 { + return code, nil, true, errors.New("eventactions: invalid request") + } + if e = wire.Walk(b, func(wire.Field) error { return nil }); e != nil { + return code, nil, true, e + } + sum := sha256.Sum256(append([]byte(path), b...)) + digest := hex.EncodeToString(sum[:]) + rk := ctx.SessionID + ":" + path + ":" + key(seq) + if r, ok := s.state.Receipts[rk]; ok { + if r.Digest != digest { + return code, nil, true, errors.New("eventactions: replay changed") + } + return code, r.Reply, true, nil + } + before, _ := json.Marshal(s.state) + s.roll() + out, e := s.handle(ctx, path, b, rk) + if e != nil { + _ = json.Unmarshal(before, &s.state) + return code, nil, true, e + } + s.state.Receipts[rk] = receipt{digest, out} + if e = s.save(ctx); e != nil { + _ = json.Unmarshal(before, &s.state) + return code, nil, true, e + } + return code, out, true, nil +} + +func (s *Service) spawnWire(p *spawn) []byte { + if p == nil { + return nil + } + b := wire.AppendVarint(nil, 1, uint64(p.Start)) + b = wire.AppendVarint(b, 2, p.UID) + b = wire.AppendVarint(b, 3, p.ID) + b = wire.AppendVarint(b, 4, p.Group) + var ids []string + for k := range p.Caught { + ids = append(ids, k) + } + sort.Strings(ids) + for _, k := range ids { + var g, id uint64 + _, _ = fmt.Sscanf(k, "%d/%d", &g, &id) + x := wire.AppendVarint(nil, 1, p.ID) + x = wire.AppendVarint(x, 2, g) + x = wire.AppendVarint(x, 3, id) + b = wire.AppendBytes(b, 5, x) + } + return b +} + +func (s *Service) handle(ctx command.Context, path string, b []byte, identity string) ([]byte, error) { + switch path { + case "/DailyStoryInfo", "/DailyStoryClear": + return s.dailyStory(ctx, path, b, identity) + case "/CafeteriaEventNpcInteractionReward": + return s.cafeteriaReward(ctx, b, identity) + case "/ChargeCostInfo": + if s.chargeInfo == nil { + return nil, errors.New("eventactions: charge state source unavailable") + } + return s.chargeInfo(ctx) + case "/FieldEventSpawnInfo": + uid := val(b, 2) + if _, e := s.resolve(uid, 21); e != nil { + return nil, e + } + var out []byte + if p := s.state.Spawns[uid]; p != nil { + out = wire.AppendBytes(out, 1, s.spawnWire(p)) + } + out = wire.AppendVarint(out, 2, s.state.DailyNormal) + out = wire.AppendVarint(out, 3, s.state.DailySpecial) + return out, nil + case "/FieldEventSpawnStart": + v, e := s.resolve(val(b, 2), 21) + if e != nil { + return nil, e + } + g, id := val(b, 3), val(b, 4) + r, ok := s.row("FieldSpawnEventTable", 4, 5, g, id) + if !ok || g != v.ID { + return nil, errors.New("eventactions: invalid spawn event") + } + if err := s.spawnWindow(r); err != nil { + return nil, err + } + if p := s.state.Spawns[v.UID]; p != nil { + defaults, _ := s.design.Row("FieldEventDefaultTable", 9, 0) + expired := defaults.V(12) > 0 && s.now().UnixMilli()-p.Start >= int64(defaults.V(12))*1000 + if expired && (p.Group != g || p.ID != id) { + delete(s.state.Spawns, v.UID) + } else { + if p.Group != g || p.ID != id { + return nil, errors.New("eventactions: spawn already started") + } + return wire.AppendBytes(nil, 1, s.spawnWire(p)), nil + } + } + _ = r + s.state.Spawns[v.UID] = &spawn{v.UID, g, id, s.now().UnixMilli(), map[string]bool{}} + return wire.AppendBytes(nil, 1, s.spawnWire(s.state.Spawns[v.UID])), nil + case "/FieldEventSpawnReward": + v, e := s.resolve(val(b, 6), 21) + if e != nil { + return nil, e + } + p := s.state.Spawns[v.UID] + if p == nil || p.ID != val(b, 2) || p.Group != val(b, 4) { + return nil, errors.New("eventactions: spawn not started") + } + spawnRow, ok := s.row("FieldSpawnEventTable", 4, 5, p.Group, p.ID) + if !ok { + return nil, errors.New("eventactions: spawn definition absent") + } + g, id := val(b, 5), val(b, 3) + r, ok := s.row("FieldEventMonsterTable", 3, 4, g, id) + if !ok || g != spawnRow.V(2) { + return nil, errors.New("eventactions: invalid spawn monster") + } + ck := key(g, id) + if p.Caught[ck] { + // A new transport sequence must not turn a confirmed catch into + // either another grant or an error/recovery loop. + return wire.AppendBytes(nil, 1, nil), nil + } + defaults, _ := s.design.Row("FieldEventDefaultTable", 9, 0) + if defaults.V(12) > 0 && s.now().UnixMilli()-p.Start >= int64(defaults.V(12))*1000 { + return nil, errors.New("eventactions: spawn time expired") + } + special := r.V(2) != 0 + limited := special && s.state.DailySpecial >= defaults.V(5) || !special && s.state.DailyNormal >= defaults.V(4) + if limited { + // Participation remains available after the daily reward quota. + // Completing the capture without a grant also prevents client retries. + p.Caught[ck] = true + return wire.AppendBytes(nil, 1, nil), nil + } + reward, ok := s.design.SpawnRewards[[2]uint64{r.V(5), r.V(1)}] + if !ok || reward.Count == 0 { + return nil, errors.New("eventactions: spawn reward missing") + } + bundle, e := s.economy.Apply(ctx, identity, nil, []gamedata.Reward{reward}) + if e != nil { + return nil, e + } + p.Caught[ck] = true + if special { + s.state.DailySpecial++ + } else { + s.state.DailyNormal++ + } + if s.progress != nil { + if e = s.progress(ctx, 349, r.V(2), 1); e != nil { + return nil, e + } + } + return wire.AppendBytes(nil, 1, bundle), nil + case "/FireWorksInfo": + var out []byte + for k := range s.state.Claims { + var uid, g uint64 + if _, e := fmt.Sscanf(k, "fire:%d/%d", &uid, &g); e == nil { + out = wire.AppendVarint(out, 1, g) + } + } + return out, nil + case "/FireWorksReward": + v, e := s.resolve(val(b, 2), 22) + if e != nil { + return nil, e + } + g := val(b, 3) + if g != v.ID { + return nil, errors.New("eventactions: wrong fireworks group") + } + ck := "fire:" + key(v.UID, g) + if s.state.Claims[ck] { + return nil, errors.New("eventactions: fireworks already received") + } + var rs []gamedata.Reward + for _, r := range s.design.Tables["FireworksTable"] { + if r.V(4) == g { + rs = append(rs, r.Rewards...) + break + } + } + if len(rs) == 0 { + return nil, errors.New("eventactions: fireworks design missing") + } + bundle, e := s.economy.Apply(ctx, identity, nil, rs) + if e != nil { + return nil, e + } + s.state.Claims[ck] = true + return wire.AppendBytes(nil, 1, bundle), nil + case "/FriendshipSpecialEpisodeInfo": + var out []byte + for k := range s.state.Claims { + var g, id uint64 + if _, e := fmt.Sscanf(k, "friend:%d/%d", &g, &id); e == nil { + x := wire.AppendVarint(nil, 1, g) + x = wire.AppendVarint(x, 2, id) + out = wire.AppendBytes(out, 1, x) + } + } + return out, nil + case "/FriendshipSpecialEpisodeClear": + g, id := val(b, 2), val(b, 3) + r, ok := s.row("FriendshipSpecialEpisodeTable", 5, 6, g, id) + if !ok { + return nil, errors.New("eventactions: unknown episode") + } + if s.friendship == nil || s.friendship(g) < r.V(12) { + return nil, errors.New("eventactions: friendship level insufficient") + } + if id > 1 && !s.state.Claims["friend:"+key(g, id-1)] { + return nil, errors.New("eventactions: previous episode incomplete") + } + ck := "friend:" + key(g, id) + if s.state.Claims[ck] { + return nil, errors.New("eventactions: episode already cleared") + } + rs := append([]gamedata.Reward(nil), r.Rewards...) + for _, v := range s.registry.List() { + if v.Type == 24 && v.ID == g && s.active(v) { + rs = append(rs, r.EventRewards...) + break + } + } + bundle, e := s.economy.Apply(ctx, identity, nil, rs) + if e != nil { + return nil, e + } + s.state.Claims[ck] = true + x := wire.AppendVarint(nil, 1, g) + x = wire.AppendVarint(x, 2, id) + out := wire.AppendBytes(nil, 1, bundle) + out = wire.AppendBytes(out, 2, x) + return out, nil + case "/NpcQuizInfo": + uid := val(b, 2) + v, e := s.resolveQuiz(ctx, uid) + if e != nil { + return nil, e + } + var out []byte + for _, r := range s.design.Tables["NpcQuizTable"] { + if r.V(1) != v.ID { + continue + } + if s.state.Claims["quiz:"+key(uid, r.V(1), r.V(2))] { + x := wire.AppendVarint(nil, 1, uid) + x = wire.AppendVarint(x, 2, r.V(1)) + x = wire.AppendVarint(x, 3, r.V(2)) + out = wire.AppendBytes(out, 1, x) + } + } + return out, nil + case "/NpcQuizClear": + uid, g, id := val(b, 2), val(b, 3), val(b, 4) + v, e := s.resolveQuiz(ctx, uid) + if e != nil { + return nil, e + } + if !s.active(v) { + return nil, errors.New("eventactions: quiz event inactive") + } + r, ok := s.row("NpcQuizTable", 1, 2, g, id) + if !ok || g != v.ID || s.now().UnixMilli() < v.Start+int64(r.V(8))*86400000 { + return nil, errors.New("eventactions: quiz locked") + } + ck := "quiz:" + key(uid, g, id) + if s.state.Claims[ck] { + return nil, errors.New("eventactions: quiz already cleared") + } + bundle, e := s.economy.Apply(ctx, identity, nil, r.Rewards) + if e != nil { + return nil, e + } + s.state.Claims[ck] = true + x := wire.AppendVarint(nil, 1, uid) + x = wire.AppendVarint(x, 2, g) + x = wire.AppendVarint(x, 3, id) + out := wire.AppendBytes(nil, 1, bundle) + return wire.AppendBytes(out, 2, x), nil + case "/TacticsBingoInfo": + uid := val(b, 2) + v, e := s.registry.Resolve(uid) + if e != nil { + return nil, e + } + if v.Type != 20 { + return nil, errors.New("eventactions: wrong tactics event") + } + out := wire.AppendVarint(nil, 1, uid) + out = wire.AppendVarint(out, 2, v.ID) + for _, id := range s.state.Tactics[uid] { + out = wire.AppendVarint(out, 3, id) + } + if !s.active(v) { + out = wire.AppendVarint(out, 4, 1) + } + return out, nil + case "/TacticsBingoDeckSave": + if err := s.validateTacticsDeck(b); err != nil { + return nil, err + } + s.state.Deck = append([]byte(nil), b...) + return nil, nil + default: + return s.voteHandle(ctx, path, b, identity) + } +} + +func (s *Service) dailyStory(ctx command.Context, path string, b []byte, identity string) ([]byte, error) { + if path == "/DailyStoryInfo" { + var ids []uint64 + for k, claimed := range s.state.Claims { + if claimed && strings.HasPrefix(k, "daily-story:") { + id, err := strconv.ParseUint(strings.TrimPrefix(k, "daily-story:"), 10, 64) + if err == nil { + ids = append(ids, id) + } + } + } + slices.Sort(ids) + var out []byte + for _, id := range ids { + out = wire.AppendVarint(out, 1, id) + } + return out, nil + } + id := val(b, 2) + if id == 0 || s.miniContent == nil || s.miniDesign == nil { + return nil, errors.New("eventactions: daily story unavailable") + } + reward, ok := s.miniDesign.Stories[id] + if !ok { + return nil, errors.New("eventactions: unknown daily story") + } + ck := fmt.Sprintf("daily-story:%d", id) + if s.state.Claims[ck] { + return wire.AppendVarint(wire.AppendBytes(nil, 1, nil), 2, id), nil + } + routes, err := s.miniContent.ListMiniContentRoutes(ctx) + if err != nil { + return nil, err + } + available := false + now := s.now().UnixMilli() + for _, route := range routes { + if route.ContentType != 14 || now < route.Start || now > route.End { + continue + } + for _, story := range s.miniDesign.Groups[route.ContentID] { + if story == id { + available = true + } + } + } + if !available { + return nil, errors.New("eventactions: daily story event inactive") + } + var rewards []gamedata.Reward + if reward.Count > 0 { + rewards = append(rewards, reward) + } + bundle, err := s.economy.Apply(ctx, identity, nil, rewards) + if err != nil { + return nil, err + } + s.state.Claims[ck] = true + return wire.AppendVarint(wire.AppendBytes(nil, 1, bundle), 2, id), nil +} + +func (s *Service) cafeteriaReward(ctx command.Context, req []byte, identity string) ([]byte, error) { + group, id := val(req, 2), val(req, 3) + row, ok := s.row("CafeteriaEventTable", 5, 6, group, id) + if !ok || row.V(13) == 0 || row.V(11) == 0 { + return nil, errors.New("eventactions: cafeteria interaction absent") + } + var uid uint64 + for _, v := range s.registry.List() { + if v.Type == 25 && s.active(v) && (v.SubID == group || v.SubID == 0 && v.ID == group) { + uid = v.UID + break + } + } + if uid == 0 { + return nil, errors.New("eventactions: cafeteria event inactive") + } + if len(s.design.Tables["CafeteriaDefaultTable"]) != 1 { + return nil, errors.New("eventactions: cafeteria default missing") + } + defaults := s.design.Tables["CafeteriaDefaultTable"][0] + cap := defaults.V(8) + if cap == 0 || s.state.CafeteriaCurrency >= cap { + return nil, errors.New("eventactions: cafeteria daily currency limit") + } + term := defaults.V(3) + if row.V(4) != 1 { + term = defaults.V(28) + } + receiptKey := key(uid, group, id) + now := s.now().UnixMilli() + if last := s.state.CafeteriaLast[receiptKey]; last > 0 && now-last < int64(term)*1000 { + return nil, errors.New("eventactions: cafeteria interaction cooldown") + } + count := min(row.V(11), cap-s.state.CafeteriaCurrency) + rewards := []gamedata.Reward{{Type: row.V(13), ID: row.V(12), Count: count}} + bundle, err := s.economy.Apply(ctx, identity, nil, rewards) + if err != nil { + return nil, err + } + s.state.CafeteriaCurrency += count + s.state.CafeteriaLast[receiptKey] = now + out := wire.AppendBytes(nil, 1, bundle) + out = wire.AppendVarint(out, 2, s.state.CafeteriaCurrency) + return out, nil +} + +// Tactics uses the client's custom battle stage fields 8/9 and has fixed +// tutorial decks. Character health and ordinary pack rewards do not apply. +func (s *Service) HandlesBattle(mode uint64) bool { return mode == 29 } +func (s *Service) EnterBattle(ctx command.Context, req []byte, receipt string) ([]byte, error) { + + digest := battleDigest(req) + if r, ok := s.state.Receipts[ctx.SessionID+":enter:"+receipt]; ok { + if r.Digest != digest { + return nil, errors.New("eventactions: changed battle enter replay") + } + return r.Reply, nil + } + before, _ := json.Marshal(s.state) + if val(req, 5) != 29 { + return nil, errors.New("eventactions: wrong tactics battle mode") + } + group, stage, deck := val(req, 8), val(req, 9), val(req, 4) + row, ok := s.row("TacticsBingoTable", 4, 5, group, stage) + if !ok || row.V(2) != deck { + return nil, errors.New("eventactions: invalid tactics stage/deck") + } + var uid uint64 + for _, v := range s.registry.List() { + if v.Type != 20 || !s.active(v) { + continue + } + g, ok := s.design.Row("TacticsBingoGroupTable", 3, v.ID) + if ok && g.V(1) == group { + uid = v.UID + break + } + } + if uid == 0 { + return nil, errors.New("eventactions: tactics event inactive") + } + if _, err := s.tacticsBlueParty(row.V(1)); err != nil { + return nil, err + } + // The client sends BattleEnter before the new scene's DeckSave. Its + // battle reset clears the prior stage's virtual party; do the same here. + s.state.Deck = nil + s.state.BattleUID, s.state.BattleStage, s.state.BattleDeck = uid, stage, deck + s.state.Receipts[ctx.SessionID+":enter:"+receipt] = receiptRecord(nil, digest) + if e := s.save(ctx); e != nil { + _ = json.Unmarshal(before, &s.state) + return nil, e + } + return nil, nil +} + +func (s *Service) CompleteBattle(ctx command.Context, req []byte, receipt string) ([]byte, error) { + + if r, ok := s.state.Receipts[ctx.SessionID+":battle:"+receipt]; ok { + if r.Digest != battleDigest(req) { + return nil, errors.New("eventactions: changed battle completion replay") + } + return r.Reply, nil + } + before, _ := json.Marshal(s.state) + uid, stage := s.state.BattleUID, s.state.BattleStage + if uid == 0 || stage == 0 { + return nil, errors.New("eventactions: tactics battle not entered") + } + if _, e := s.resolve(uid, 20); e != nil { + return nil, e + } + if val(req, 2) == 1 && !contains(s.state.Tactics[uid], stage) { + priorLines := s.tacticsLines(uid) + s.state.Tactics[uid] = append(s.state.Tactics[uid], stage) + if s.progress != nil { + if e := s.progress(ctx, 346, stage, 1); e != nil { + _ = json.Unmarshal(before, &s.state) + return nil, e + } + lines := s.tacticsLines(uid) + if lines > priorLines { + if e := s.progress(ctx, 347, 0, lines-priorLines); e != nil { + _ = json.Unmarshal(before, &s.state) + return nil, e + } + } + v, _ := s.registry.Resolve(uid) + g, _ := s.design.Row("TacticsBingoGroupTable", 3, v.ID) + n := uint64(5) - g.V(4) + if uint64(len(s.state.Tactics[uid])) == n*n { + if e := s.progress(ctx, 348, 0, 1); e != nil { + _ = json.Unmarshal(before, &s.state) + return nil, e + } + } + } + } + var out []byte + for _, id := range s.state.Tactics[uid] { + out = wire.AppendVarint(out, 29, id) + } + s.state.BattleUID, s.state.BattleStage, s.state.BattleDeck = 0, 0, 0 + s.state.Receipts[ctx.SessionID+":battle:"+receipt] = receiptRecord(out, battleDigest(req)) + if e := s.save(ctx); e != nil { + _ = json.Unmarshal(before, &s.state) + return nil, e + } + return out, nil +} diff --git a/go/internal/server/domain/events/actions/service.go b/go/internal/server/domain/events/actions/service.go new file mode 100644 index 0000000..8b6b1c8 --- /dev/null +++ b/go/internal/server/domain/events/actions/service.go @@ -0,0 +1,219 @@ +// Package eventactions owns repeatable event interaction state. It uses the +// injected calendar and shared transactional reward service. +package eventactions + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "errors" + "fmt" + "slices" + "strings" + + "time" +) + +type Economy interface { + Apply(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) +} +type receipt struct { + Digest string + Reply []byte +} +type spawn struct { + UID, Group, ID uint64 + Start int64 + Caught map[string]bool +} +type vote struct{ Round, Candidate, Normal, Additional uint64 } +type snapshot struct { + Version string + BattleUID, BattleStage, BattleDeck uint64 + Claims map[string]bool + Favorites map[uint64]bool + Votes map[string]*vote + NormalVoted map[string]bool + VoteRewards map[string]bool + Spawns map[uint64]*spawn + DailyNormal, DailySpecial uint64 + CafeteriaCurrency uint64 + CafeteriaLast map[string]int64 + Day string + Tactics map[uint64][]uint64 + Deck []byte + Receipts map[string]receipt +} +type Service struct { + store stateio.Store + design *gamedata.EventActionsDesign + registry events.Resolver + economy Economy + state snapshot + now func() time.Time + + friendship func(uint64) uint64 + chargeInfo func(ctx command.Context) ([]byte, error) + progress func(ctx command.Context, _ uint64, _ uint64, _ uint64) error + voteTotals func(uint64, uint64) (map[uint64]uint64, error) + miniContent MiniContentResolver + miniDesign *gamedata.MiniContentDesign +} + +func Open(ctx command.Context, store stateio.Store, d *gamedata.EventActionsDesign, r events.Resolver, e Economy) (*Service, error) { + if store == nil || d == nil || r == nil || e == nil { + return nil, errors.New("eventactions: missing dependency") + } + s := &Service{store: store, design: d, registry: r, economy: e, now: time.Now} + b, err := store.Load(ctx.State, "eventactions") + if err != nil { + return nil, err + } + if b != nil { + if err = stateio.RequireExactJSONObject(b, "Version", "BattleUID", "BattleStage", "BattleDeck", "Claims", "Favorites", "Votes", "NormalVoted", "VoteRewards", "Spawns", "DailyNormal", "DailySpecial", "CafeteriaCurrency", "CafeteriaLast", "Day", "Tactics", "Deck", "Receipts"); err != nil { + return nil, err + } + if err = json.Unmarshal(b, &s.state); err != nil { + return nil, err + } + if s.state.Version != versionconfig.State() || s.state.Claims == nil || s.state.Favorites == nil || s.state.Votes == nil || s.state.NormalVoted == nil || s.state.VoteRewards == nil || s.state.Spawns == nil || s.state.Tactics == nil || s.state.Receipts == nil || s.state.CafeteriaLast == nil { + return nil, errors.New("eventactions: invalid state layout") + } + for _, v := range s.state.Votes { + if v == nil || v.Candidate == 0 || v.Round == 0 { + return nil, errors.New("eventactions: invalid saved vote") + } + } + for _, p := range s.state.Spawns { + if p == nil || p.Caught == nil || p.ID == 0 { + return nil, errors.New("eventactions: invalid saved spawn") + } + } + } else { + s.state.Version = versionconfig.State() + } + s.init() + return s, nil +} +func (s *Service) init() { + if s.state.CafeteriaLast == nil { + s.state.CafeteriaLast = map[string]int64{} + } + if s.state.Claims == nil { + s.state.Claims = map[string]bool{} + } + if s.state.Favorites == nil { + s.state.Favorites = map[uint64]bool{} + } + if s.state.Votes == nil { + s.state.Votes = map[string]*vote{} + } + if s.state.NormalVoted == nil { + s.state.NormalVoted = map[string]bool{} + } + if s.state.VoteRewards == nil { + s.state.VoteRewards = map[string]bool{} + } + if s.state.Spawns == nil { + s.state.Spawns = map[uint64]*spawn{} + } + if s.state.Tactics == nil { + s.state.Tactics = map[uint64][]uint64{} + } + if s.state.Receipts == nil { + s.state.Receipts = map[string]receipt{} + } +} + +func (s *Service) AttachFriendshipLevel(f func(uint64) uint64) { s.friendship = f } +func (s *Service) AttachChargeInfo(f func(ctx command.Context) ([]byte, error)) { s.chargeInfo = f } +func (s *Service) AttachProgress(f func(ctx command.Context, _ uint64, _ uint64, _ uint64) error) { + s.progress = f +} + +// The current server owns one account save. Its real votes are the default +// leaderboard; a future account aggregator can supply broader server totals. +func (s *Service) AttachVoteTotals(f func(uint64, uint64) (map[uint64]uint64, error)) { + s.voteTotals = f +} +func (s *Service) save(ctx command.Context) error { + b, e := json.Marshal(s.state) + if e != nil { + return e + } + return s.store.Save(ctx.State, "eventactions", b) +} + +func key(v ...uint64) string { + r := []string{} + for _, n := range v { + r = append(r, fmt.Sprint(n)) + } + return strings.Join(r, "/") +} +func (s *Service) active(v events.Schedule) bool { + return s.now().UnixMilli() >= v.Start && s.now().UnixMilli() < v.End +} +func (s *Service) resolve(uid, t uint64) (events.Schedule, error) { + v, e := s.registry.Resolve(uid) + if e != nil { + return v, e + } + if v.Type != t || !s.active(v) { + return v, errors.New("eventactions: event inactive or wrong type") + } + return v, nil +} +func (s *Service) find(t uint64) (events.Schedule, bool) { + for _, v := range s.registry.List() { + if v.Type == t && s.active(v) { + return v, true + } + } + return events.Schedule{}, false +} +func (s *Service) roll() { + day := s.now().UTC().Format("2006-01-02") + if s.state.Day != day { + s.state.Day = day + s.state.DailyNormal = 0 + s.state.DailySpecial = 0 + s.state.Spawns = map[uint64]*spawn{} + s.state.CafeteriaCurrency = 0 + s.state.CafeteriaLast = map[string]int64{} + s.state.NormalVoted = map[string]bool{} + } +} + +func (s *Service) row(table string, groupField, idField int, g, id uint64) (gamedata.EventActionRow, bool) { + for _, r := range s.design.Tables[table] { + if r.V(groupField) == g && r.V(idField) == id { + return r, true + } + } + return gamedata.EventActionRow{}, false +} + +func (s *Service) RecordTacticsClear(ctx command.Context, uid, stage uint64) error { + + v, e := s.resolve(uid, 20) + if e != nil { + return e + } + group, ok := s.design.Row("TacticsBingoGroupTable", 3, v.ID) + if !ok { + return errors.New("eventactions: tactics group missing") + } + if _, ok = s.row("TacticsBingoTable", 4, 5, group.V(1), stage); !ok { + return errors.New("eventactions: invalid tactics stage") + } + if slices.Contains(s.state.Tactics[uid], stage) { + return nil + } + s.state.Tactics[uid] = append(s.state.Tactics[uid], stage) + slices.Sort(s.state.Tactics[uid]) + return s.save(ctx) +} diff --git a/go/internal/server/eventactions/spawn_window.go b/go/internal/server/domain/events/actions/spawn_window.go similarity index 95% rename from go/internal/server/eventactions/spawn_window.go rename to go/internal/server/domain/events/actions/spawn_window.go index a2152fb..cb131a1 100644 --- a/go/internal/server/eventactions/spawn_window.go +++ b/go/internal/server/domain/events/actions/spawn_window.go @@ -1,7 +1,7 @@ package eventactions import ( - "bd2server/internal/server/gamedata" + "bd2server/internal/server/design/gamedata" "fmt" "time" ) diff --git a/go/internal/server/domain/events/actions/tactics_deck.go b/go/internal/server/domain/events/actions/tactics_deck.go new file mode 100644 index 0000000..4e69cbf --- /dev/null +++ b/go/internal/server/domain/events/actions/tactics_deck.go @@ -0,0 +1,57 @@ +package eventactions + +import ( + "bd2server/internal/server/design/gamedata" + + "errors" + "sort" +) + +// BattlePlayLibrary gives TacticsBingo the standard 3 by 4 grid and five +// characters. Positions are zero-based; DeckSave sequences start at one. +const tacticsGridSize = 3 * 4 +const tacticsPartySize = 5 + +type tacticsDeckEntry struct { + index, character, costume, position, sequence uint64 +} + +// SetPlayDataTarosTactics builds the first five authored CharGroup rows with +// their CharTable default costumes; player ownership never enters this path. +func (s *Service) tacticsBlueParty(group uint64) (map[uint64]uint64, error) { + var rows []gamedata.EventActionRow + for _, row := range s.design.Tables["CharGroupTable"] { + if group != 0 && row.V(2) == group { + rows = append(rows, row) + } + } + sort.Slice(rows, func(i, j int) bool { return rows[i].V(3) < rows[j].V(3) }) + if len(rows) > tacticsPartySize { + rows = rows[:tacticsPartySize] + } + if len(rows) == 0 { + return nil, errors.New("eventactions: tactics blue party missing") + } + party := map[uint64]uint64{} + for _, row := range rows { + id := row.V(1) + character, ok := s.design.Row("CharTable", 12, id) + if id == 0 || party[id] != 0 || row.V(5) == 0 || !ok || character.V(5) == 0 { + return nil, errors.New("eventactions: invalid tactics blue party design") + } + party[id] = character.V(5) + } + return party, nil +} + +func matchesTacticsParty(entries []tacticsDeckEntry, party map[uint64]uint64) bool { + if len(entries) != len(party) { + return false + } + for _, e := range entries { + if party[e.character] != e.costume { + return false + } + } + return true +} diff --git a/go/internal/server/domain/events/actions/voting.go b/go/internal/server/domain/events/actions/voting.go new file mode 100644 index 0000000..1464da7 --- /dev/null +++ b/go/internal/server/domain/events/actions/voting.go @@ -0,0 +1,46 @@ +package eventactions + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/events" + "slices" + "sort" +) + +func (s *Service) voteEvent() (events.Schedule, gamedata.EventActionRow, bool) { + v, ok := s.find(23) + if !ok { + return v, gamedata.EventActionRow{}, false + } + r, ok := s.design.Row("VotingEventTable", 6, v.ID) + return v, r, ok +} +func (s *Service) rounds(v events.Schedule, r gamedata.EventActionRow) (uint64, []gamedata.EventActionRow) { + var rows []gamedata.EventActionRow + current := uint64(0) + for _, x := range s.design.Tables["VotingRoundTable"] { + if x.V(2) == r.V(14) { + rows = append(rows, x) + } + } + sort.Slice(rows, func(i, j int) bool { return rows[i].V(3) < rows[j].V(3) }) + for _, x := range rows { + if s.now().UnixMilli() >= v.Start+int64(x.V(4))*86400000 { + current = x.V(3) + } + } + return current, rows +} +func (s *Service) candidates(r gamedata.EventActionRow) []uint64 { + var ids []uint64 + for _, x := range s.design.Tables["VotingCandidateTable"] { + if x.V(1) == r.V(10) { + ids = append(ids, x.V(2)) + } + } + slices.Sort(ids) + return ids +} +func contains(ids []uint64, id uint64) bool { + return slices.Contains(ids, id) +} diff --git a/go/internal/server/events/ap_refresh.go b/go/internal/server/domain/events/ap_refresh.go similarity index 58% rename from go/internal/server/events/ap_refresh.go rename to go/internal/server/domain/events/ap_refresh.go index ca479de..af014b9 100644 --- a/go/internal/server/events/ap_refresh.go +++ b/go/internal/server/domain/events/ap_refresh.go @@ -1,10 +1,10 @@ package events import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/storage/stateio" "encoding/json" "fmt" "maps" @@ -17,8 +17,7 @@ type economyAPClock struct { } func (e *Economy) AttachAPRefresh(caps map[uint64]uint64, design gamedata.HuntingAPDesign) error { - e.mu.Lock() - defer e.mu.Unlock() + for typ, max := range caps { if (typ != 30 && typ != 32) || max == 0 || max > math.MaxInt32 { return fmt.Errorf("events: invalid free AP refresh") @@ -29,14 +28,14 @@ func (e *Economy) AttachAPRefresh(caps map[uint64]uint64, design gamedata.Huntin e.now = time.Now return nil } -func (e *Economy) refreshAP(s *economySnapshot) error { +func (e *Economy) refreshAP(ctx command.Context, s *economySnapshot) error { if len(e.apCaps) == 0 { return nil } now := e.now().UnixMilli() offset := (e.resetSeconds - 9*3600) * 1000 next := ((now-offset)/86400000+1)*86400000 + offset - b, err := e.store.Load("eventapclock") + b, err := e.store.Load(ctx.State, "eventapclock") if err != nil { return err } @@ -60,27 +59,26 @@ func (e *Economy) refreshAP(s *economySnapshot) error { if err != nil { return err } - if err = e.store.Save("eventapclock", raw); err != nil { + if err = e.store.Save(ctx.State, "eventapclock", raw); err != nil { return err } raw, err = json.Marshal(s) if err != nil { return err } - return e.store.Save("event_economy", raw) + return e.store.Save(ctx.State, "event_economy", raw) } -func (e *Economy) CanApply(costs []gamedata.Reward) error { - e.mu.Lock() - defer e.mu.Unlock() - s, err := e.load() +func (e *Economy) CanApply(ctx command.Context, costs []gamedata.Reward) error { + + s, err := e.load(ctx) if err != nil { return err } - if err = e.refreshAP(&s); err != nil { + if err = e.refreshAP(ctx, &s); err != nil { return err } var wallet, ap []gamedata.Reward - var items []player.Item + var items []assets.Item totals := map[[2]uint64]uint64{} for _, r := range costs { k := [2]uint64{r.Type, r.ID} @@ -101,7 +99,7 @@ func (e *Economy) CanApply(costs []gamedata.Reward) error { return fmt.Errorf("events: insufficient currency") } case inventoryType(r.Type): - selected, err := e.items.SelectMutable(r.Type, r.ID, n) + selected, err := e.items.SelectMutable(ctx, r.Type, r.ID, n) if err != nil { return err } @@ -110,54 +108,19 @@ func (e *Economy) CanApply(costs []gamedata.Reward) error { return fmt.Errorf("events: unsupported cost type %d", r.Type) } } - if err = e.wallet.CanExchange(wallet, nil); err != nil { + if err = e.wallet.CanExchange(ctx, wallet, nil); err != nil { return err } if len(ap) > 0 { if e.hunting == nil { return fmt.Errorf("events: AP unavailable") } - if err = e.hunting.CanExchangeAP(ap, nil); err != nil { + if err = e.hunting.CanExchangeAP(ctx, ap, nil); err != nil { return err } } if len(items) > 0 { - return e.items.CanConsume(items) + return e.items.CanConsume(ctx, items) } return nil } - -func (e *Economy) ChargeInfo() ([]byte, error) { - e.mu.Lock() - defer e.mu.Unlock() - s, err := e.load() - if err != nil { - return nil, err - } - if err = e.refreshAP(&s); err != nil { - return nil, err - } - raw, err := e.store.Load("eventapclock") - if err != nil { - return nil, err - } - if raw == nil { - return nil, nil - } - var clock economyAPClock - if err = json.Unmarshal(raw, &clock); err != nil { - return nil, err - } - var out []byte - for _, typ := range []uint64{30, 32} { - if _, ok := e.apCaps[typ]; !ok { - continue - } - info := wire.AppendVarint(nil, 1, uint64(clock.Next-86400000)) - item := wire.AppendVarint(nil, 3, typ) - item = wire.AppendVarint(item, 4, s.Balances[typ]) - info = wire.AppendBytes(info, 2, item) - out = wire.AppendBytes(out, 1, info) - } - return out, nil -} diff --git a/go/internal/server/domain/events/boxes.go b/go/internal/server/domain/events/boxes.go new file mode 100644 index 0000000..63524f1 --- /dev/null +++ b/go/internal/server/domain/events/boxes.go @@ -0,0 +1,21 @@ +package events + +import ( + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/storage/stateio" + "fmt" +) + +type BoxService struct { + items *assets.Inventory + economy *Economy + store stateio.Store +} +type boxReceipt struct{ Request, Response []byte } + +func OpenBoxes(store stateio.Store, items *assets.Inventory, economy *Economy) (*BoxService, error) { + if store == nil || items == nil || economy == nil { + return nil, fmt.Errorf("events: invalid box runtime") + } + return &BoxService{store: store, items: items, economy: economy}, nil +} diff --git a/go/internal/server/events/buff_rewards.go b/go/internal/server/domain/events/buff_rewards.go similarity index 79% rename from go/internal/server/events/buff_rewards.go rename to go/internal/server/domain/events/buff_rewards.go index 9bbb371..214a39c 100644 --- a/go/internal/server/events/buff_rewards.go +++ b/go/internal/server/domain/events/buff_rewards.go @@ -7,10 +7,10 @@ import ( "fmt" "math" "sort" - "sync" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/storage/stateio" ) type buffRewardState struct { @@ -18,22 +18,21 @@ type buffRewardState struct { Receipts map[string]string } type BuffRewards struct { - mu sync.Mutex store stateio.Store design map[uint64]gamedata.PictorialBuffStat } -func OpenBuffRewards(store stateio.Store, design map[uint64]gamedata.PictorialBuffStat) (*BuffRewards, error) { +func OpenBuffRewards(ctx command.Context, store stateio.Store, design map[uint64]gamedata.PictorialBuffStat) (*BuffRewards, error) { if store == nil || len(design) == 0 { return nil, fmt.Errorf("events: buff reward dependencies unavailable") } s := &BuffRewards{store: store, design: design} - _, err := s.load() + _, err := s.load(ctx) return s, err } -func (s *BuffRewards) load() (buffRewardState, error) { +func (s *BuffRewards) load(ctx command.Context) (buffRewardState, error) { v := buffRewardState{Counts: map[uint64]uint64{}, Receipts: map[string]string{}} - b, err := s.store.Load("buff_item_rewards") + b, err := s.store.Load(ctx.State, "buff_item_rewards") if err != nil || b == nil { return v, err } @@ -61,16 +60,15 @@ func (s *BuffRewards) Validate(rewards []gamedata.Reward) error { } return nil } -func (s *BuffRewards) GrantOnce(identity string, rewards []gamedata.Reward) error { +func (s *BuffRewards) GrantOnce(ctx command.Context, identity string, rewards []gamedata.Reward) error { if identity == "" { return fmt.Errorf("events: empty buff reward identity") } if err := s.Validate(rewards); err != nil { return err } - s.mu.Lock() - defer s.mu.Unlock() - v, err := s.load() + + v, err := s.load(ctx) if err != nil { return err } @@ -94,12 +92,11 @@ func (s *BuffRewards) GrantOnce(identity string, rewards []gamedata.Reward) erro if err != nil { return err } - return s.store.Save("buff_item_rewards", raw) + return s.store.Save(ctx.State, "buff_item_rewards", raw) } -func (s *BuffRewards) SnapshotBuffs() ([]gamedata.PictorialBuffStat, error) { - s.mu.Lock() - defer s.mu.Unlock() - v, err := s.load() +func (s *BuffRewards) SnapshotBuffs(ctx command.Context) ([]gamedata.PictorialBuffStat, error) { + + v, err := s.load(ctx) if err != nil { return nil, err } diff --git a/go/internal/server/domain/events/calendar/binary.go b/go/internal/server/domain/events/calendar/binary.go new file mode 100644 index 0000000..bf3f33e --- /dev/null +++ b/go/internal/server/domain/events/calendar/binary.go @@ -0,0 +1,75 @@ +package calendar + +import ( + "fmt" +) + +const MaxFileSize = 8 * 1024 * 1024 +const maxRows = 100000 +const maxString = 64 * 1024 +const headerSize = 46 + +var magic = []byte{'B', 'D', '2', 'S', 'C', 'H', 0, 0} + +type encoder struct { + data []byte + err error + rows uint64 +} + +func (e *encoder) b(v bool) { + if !e.room(1) { + return + } + if v { + e.data = append(e.data, 1) + } else { + e.data = append(e.data, 0) + } +} + +func (e *encoder) room(n int) bool { + if e.err != nil { + return false + } + if n > MaxFileSize-headerSize-len(e.data) { + e.err = fmt.Errorf("calendar: file size limit exceeded") + return false + } + return true +} + +type decoder struct { + data []byte + pos int + err error + rows uint64 +} + +func (d *decoder) take(n int) []byte { + if d.err != nil { + return nil + } + if n < 0 || n > len(d.data)-d.pos { + d.err = fmt.Errorf("calendar: truncated payload") + return nil + } + v := d.data[d.pos : d.pos+n] + d.pos += n + return v +} + +func (d *decoder) b() bool { + v := d.take(1) + if len(v) != 1 { + return false + } + if v[0] > 1 { + d.err = fmt.Errorf("calendar: invalid boolean/presence") + } + return v[0] == 1 +} + +// MarshalBinary emits format 1's fixed-order record payload and checksum. + +// UnmarshalBinary rejects damaged, unsupported, oversized and trailing data. diff --git a/go/internal/server/calendar/load.go b/go/internal/server/domain/events/calendar/load.go similarity index 61% rename from go/internal/server/calendar/load.go rename to go/internal/server/domain/events/calendar/load.go index 310537f..8a940d5 100644 --- a/go/internal/server/calendar/load.go +++ b/go/internal/server/domain/events/calendar/load.go @@ -1,19 +1,15 @@ package calendar import ( - "bd2server/internal/server/events" - "bd2server/internal/server/gacha" - "bd2server/internal/server/schedule" + "bd2server/internal/server/design/schedule" + "bd2server/internal/server/domain/commerce/gacha" + "bd2server/internal/server/domain/events" "fmt" "math" - "os" - "path/filepath" - "sort" - "strings" "time" ) -func timestamp(raw string) (uint64, error) { +func ParseTimestamp(raw string) (uint64, error) { t, e := time.Parse(time.RFC3339Nano, raw) if e != nil { return 0, fmt.Errorf("invalid RFC3339 timestamp %q", raw) @@ -25,11 +21,11 @@ func timestamp(raw string) (uint64, error) { return uint64(ms), nil } func window(start, end string, equal bool) (uint64, uint64, error) { - a, e := timestamp(start) + a, e := ParseTimestamp(start) if e != nil { return 0, 0, e } - b, e := timestamp(end) + b, e := ParseTimestamp(end) if e != nil { return 0, 0, e } @@ -60,111 +56,6 @@ func convertSeason(s Season) (schedule.Season, error) { return schedule.Season{ID: s.ID, StartMilliseconds: a, EndMilliseconds: b, RankRewardGroupID: s.RankRewardGroupID, Error: s.Error, Return: s.Return}, nil } -// LoadDirectory validates the whole directory before returning any calendar. -// Every calendar file must use the .bd2schedule suffix. -// Directories and symlinks are -// rejected so an accidentally omitted calendar never produces a partial set. -func LoadDirectory(dir, gameVersion, gameDataVersion string) (*Set, error) { - entries, e := os.ReadDir(dir) - if e != nil { - return nil, fmt.Errorf("calendar: read directory: %w", e) - } - set := &Set{GachaSeed: &gacha.ScheduleSeed{ClientVersion: gameVersion}} - seen := map[string]bool{} - claim := func(key string) error { - if seen[key] { - return fmt.Errorf("duplicate calendar identity %s", key) - } - seen[key] = true - return nil - } - files := 0 - for _, entry := range entries { - if !entry.Type().IsRegular() || !strings.HasSuffix(entry.Name(), ".bd2schedule") { - return nil, fmt.Errorf("calendar: unsupported entry %s", entry.Name()) - } - info, e := entry.Info() - if e != nil { - return nil, e - } - if info.Size() > MaxFileSize { - return nil, fmt.Errorf("calendar: %s: file size limit exceeded", entry.Name()) - } - raw, e := os.ReadFile(filepath.Join(dir, entry.Name())) - if e != nil { - return nil, e - } - m, e := UnmarshalBinary(raw) - if e != nil { - return nil, fmt.Errorf("calendar: %s: %w", entry.Name(), e) - } - if m.SchemaVersion != 1 || strings.TrimSpace(m.Revision) == "" || m.GameVersion != gameVersion || m.GameDataVersion != gameDataVersion || gameVersion == "" || gameDataVersion == "" { - return nil, fmt.Errorf("calendar: %s: incompatible schema/version or missing revision", entry.Name()) - } - if e = merge(set, m, claim); e != nil { - return nil, fmt.Errorf("calendar: %s: %w", entry.Name(), e) - } - set.Revisions = append(set.Revisions, m.Revision) - files++ - } - if files == 0 { - return nil, fmt.Errorf("calendar: empty directory") - } - sort.Slice(set.EventHubs, func(i, j int) bool { return set.EventHubs[i].UID < set.EventHubs[j].UID }) - sort.Slice(set.MiniGameHubs, func(i, j int) bool { return set.MiniGameHubs[i].Slot < set.MiniGameHubs[j].Slot }) - sort.Slice(set.Events, func(i, j int) bool { - a, b := set.Events[i], set.Events[j] - if a.UID != b.UID { - return a.UID < b.UID - } - if a.Type != b.Type { - return a.Type < b.Type - } - if a.ID != b.ID { - return a.ID < b.ID - } - return a.SubID < b.SubID - }) - sortWindows := func(w []gacha.ScheduleWindow) { - sort.Slice(w, func(i, j int) bool { - if w[i].GroupID != w[j].GroupID { - return w[i].GroupID < w[j].GroupID - } - return w[i].StartTime < w[j].StartTime - }) - } - sortWindows(set.GachaSeed.Schedules) - sortWindows(set.GachaSeed.StepUps) - if e = set.GachaSeed.Validate(gameVersion); e != nil { - return nil, e - } - if set.RegularService != nil { - sort.Slice(set.RegularService.Contents, func(i, j int) bool { return set.RegularService.Contents[i].ID < set.RegularService.Contents[j].ID }) - sort.Slice(set.RegularService.Regular, func(i, j int) bool { - return set.RegularService.Regular[i].ContentID < set.RegularService.Regular[j].ContentID - }) - if e = set.RegularService.Validate(); e != nil { - return nil, e - } - } - sort.Slice(set.CashProducts, func(i, j int) bool { - a, b := set.CashProducts[i], set.CashProducts[j] - if a.GroupID != b.GroupID { - return a.GroupID < b.GroupID - } - if a.ProductID != b.ProductID { - return a.ProductID < b.ProductID - } - return a.SaleGroup < b.SaleGroup - }) - if set.MonsterHunt != nil { - sort.Slice(set.MonsterHunt.Seasons, func(i, j int) bool { - return set.MonsterHunt.Seasons[i].Season.ID < set.MonsterHunt.Seasons[j].Season.ID - }) - sort.Slice(set.MonsterHunt.History, func(i, j int) bool { return set.MonsterHunt.History[i].Season < set.MonsterHunt.History[j].Season }) - } - return set, nil -} func merge(s *Set, m Manifest, claim func(string) error) error { for _, v := range m.EventHubs { if e := claim(fmt.Sprintf("hub:%d", v.UID)); e != nil { @@ -180,7 +71,7 @@ func merge(s *Set, m Manifest, claim func(string) error) error { if e != nil { return e } - play, e := timestamp(v.PlayEnd) + play, e := ParseTimestamp(v.PlayEnd) if e != nil { return e } @@ -291,11 +182,11 @@ func merge(s *Set, m Manifest, claim func(string) error) error { if _, e := convertSeason(v.Season); e != nil { return e } - end, e := timestamp(v.CalculateEndAt) + end, e := ParseTimestamp(v.CalculateEndAt) if e != nil { return e } - seasonEnd, _ := timestamp(v.Season.End) + seasonEnd, _ := ParseTimestamp(v.Season.End) if end < seasonEnd { return fmt.Errorf("hunt calculation ends before season") } @@ -338,13 +229,13 @@ func merge(s *Set, m Manifest, claim func(string) error) error { var a, b uint64 var e error if v.Start != "" { - a, e = timestamp(v.Start) + a, e = ParseTimestamp(v.Start) if e != nil { return e } } if v.End != "" { - b, e = timestamp(v.End) + b, e = ParseTimestamp(v.End) if e != nil { return e } diff --git a/go/internal/server/domain/events/calendar/protocol_adapter.go b/go/internal/server/domain/events/calendar/protocol_adapter.go new file mode 100644 index 0000000..5094da6 --- /dev/null +++ b/go/internal/server/domain/events/calendar/protocol_adapter.go @@ -0,0 +1,818 @@ +package calendar + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/design/schedule" + + "bd2server/internal/server/domain/commerce/gacha" + "bd2server/internal/server/domain/events" + "bd2server/internal/server/protocol/wire" + "bytes" + "crypto/sha256" + "encoding/binary" + "fmt" + "os" + "path/filepath" + "sort" + "strings" + "unicode/utf8" +) + +func (e *encoder) manifest(v Manifest) { + e.str(v.Revision) + e.str(v.GameVersion) + e.str(v.GameDataVersion) + e.count(len(v.Events)) + for _, item := range v.Events { + e.event(item) + } + e.count(len(v.Gacha)) + for _, item := range v.Gacha { + e.gacha(item) + } + e.count(len(v.StepUps)) + for _, item := range v.StepUps { + e.gacha(item) + } + e.b(v.Regular != nil) + if v.Regular != nil { + e.regular(*v.Regular) + } + e.b(v.MonsterHunt != nil) + if v.MonsterHunt != nil { + e.monsterhunt(*v.MonsterHunt) + } + e.count(len(v.CashProducts)) + for _, item := range v.CashProducts { + e.cashproduct(item) + } + e.count(len(v.EventHubs)) + for _, item := range v.EventHubs { + e.eventhub(item) + } + e.count(len(v.MiniGameHubs)) + for _, item := range v.MiniGameHubs { + e.minigamehub(item) + } +} + +func (d *decoder) manifest() Manifest { + var v Manifest + v.Revision = d.str() + v.GameVersion = d.str() + v.GameDataVersion = d.str() + for n := d.count(); n > 0 && d.err == nil; n-- { + v.Events = append(v.Events, d.event()) + } + for n := d.count(); n > 0 && d.err == nil; n-- { + v.Gacha = append(v.Gacha, d.gacha()) + } + for n := d.count(); n > 0 && d.err == nil; n-- { + v.StepUps = append(v.StepUps, d.gacha()) + } + if d.b() { + item := d.regular() + v.Regular = &item + } + if d.b() { + item := d.monsterhunt() + v.MonsterHunt = &item + } + for n := d.count(); n > 0 && d.err == nil; n-- { + v.CashProducts = append(v.CashProducts, d.cashproduct()) + } + for n := d.count(); n > 0 && d.err == nil; n-- { + v.EventHubs = append(v.EventHubs, d.eventhub()) + } + for n := d.count(); n > 0 && d.err == nil; n-- { + v.MiniGameHubs = append(v.MiniGameHubs, d.minigamehub()) + } + return v +} + +func (e *encoder) event(v Event) { + e.u64(v.UID) + e.u64(v.Type) + e.u64(v.ID) + e.u64(v.SubID) + e.str(v.Start) + e.str(v.End) +} + +func (d *decoder) event() Event { + var v Event + v.UID = d.u64() + v.Type = d.u64() + v.ID = d.u64() + v.SubID = d.u64() + v.Start = d.str() + v.End = d.str() + return v +} + +func (e *encoder) gacha(v Gacha) { + e.u64(v.GroupID) + e.str(v.Start) + e.str(v.End) + e.b(v.FreeCountBonus) + e.b(v.CashCountBonus) +} + +func (d *decoder) gacha() Gacha { + var v Gacha + v.GroupID = d.u64() + v.Start = d.str() + v.End = d.str() + v.FreeCountBonus = d.b() + v.CashCountBonus = d.b() + return v +} + +func (e *encoder) season(v Season) { + e.u64(v.ID) + e.str(v.Start) + e.str(v.End) + e.u64(v.RankRewardGroupID) + e.b(v.Error) + e.b(v.Return) +} + +func (d *decoder) season() Season { + var v Season + v.ID = d.u64() + v.Start = d.str() + v.End = d.str() + v.RankRewardGroupID = d.u64() + v.Error = d.b() + v.Return = d.b() + return v +} + +func (e *encoder) content(v Content) { + e.u64(v.ID) + e.season(v.Current) + e.season(v.Next) +} + +func (d *decoder) content() Content { + var v Content + v.ID = d.u64() + v.Current = d.season() + v.Next = d.season() + return v +} + +func (e *encoder) regular(v Regular) { + e.u64(v.CalculateMilliseconds) + e.count(len(v.Contents)) + for _, item := range v.Contents { + e.content(item) + } + e.count(len(v.Regular)) + for _, item := range v.Regular { + e.regularseason(item) + } +} + +func (d *decoder) regular() Regular { + var v Regular + v.CalculateMilliseconds = d.u64() + for n := d.count(); n > 0 && d.err == nil; n-- { + v.Contents = append(v.Contents, d.content()) + } + for n := d.count(); n > 0 && d.err == nil; n-- { + v.Regular = append(v.Regular, d.regularseason()) + } + return v +} + +func (e *encoder) hunt(v Hunt) { + e.season(v.Season) + e.u64(v.HuntID) + e.u64(v.InfoOpenDay) + e.str(v.CalculateEndAt) + e.b(v.ErrorFlag) + e.b(v.IndependentFlag) + e.u64(v.RankRewardGroupID) + e.count(len(v.CostumeBanIDs)) + for _, item := range v.CostumeBanIDs { + e.u64(item) + } + e.count(len(v.BurstBanIDs)) + for _, item := range v.BurstBanIDs { + e.u64(item) + } +} + +func (d *decoder) hunt() Hunt { + var v Hunt + v.Season = d.season() + v.HuntID = d.u64() + v.InfoOpenDay = d.u64() + v.CalculateEndAt = d.str() + v.ErrorFlag = d.b() + v.IndependentFlag = d.b() + v.RankRewardGroupID = d.u64() + for n := d.count(); n > 0 && d.err == nil; n-- { + v.CostumeBanIDs = append(v.CostumeBanIDs, d.u64()) + } + for n := d.count(); n > 0 && d.err == nil; n-- { + v.BurstBanIDs = append(v.BurstBanIDs, d.u64()) + } + return v +} + +func (e *encoder) hunthistory(v HuntHistory) { + e.u64(v.Season) + e.u64(v.HuntID) + e.b(v.ErrorFlag) + e.b(v.Hidden) +} + +func (d *decoder) hunthistory() HuntHistory { + var v HuntHistory + v.Season = d.u64() + v.HuntID = d.u64() + v.ErrorFlag = d.b() + v.Hidden = d.b() + return v +} + +func (e *encoder) monsterhunt(v MonsterHunt) { + e.count(len(v.Seasons)) + for _, item := range v.Seasons { + e.hunt(item) + } + e.u64(v.StartRegularSeason) + e.count(len(v.History)) + for _, item := range v.History { + e.hunthistory(item) + } +} + +func (d *decoder) monsterhunt() MonsterHunt { + var v MonsterHunt + for n := d.count(); n > 0 && d.err == nil; n-- { + v.Seasons = append(v.Seasons, d.hunt()) + } + v.StartRegularSeason = d.u64() + for n := d.count(); n > 0 && d.err == nil; n-- { + v.History = append(v.History, d.hunthistory()) + } + return v +} + +func (e *encoder) cashproduct(v CashProduct) { + e.u64(v.GroupID) + e.u64(v.ProductID) + e.u64(v.SaleGroup) + e.str(v.Start) + e.str(v.End) + e.u64(v.EndDelayMinutes) + e.u64(v.EventIndex) +} + +func (d *decoder) cashproduct() CashProduct { + var v CashProduct + v.GroupID = d.u64() + v.ProductID = d.u64() + v.SaleGroup = d.u64() + v.Start = d.str() + v.End = d.str() + v.EndDelayMinutes = d.u64() + v.EventIndex = d.u64() + return v +} + +func (e *encoder) hubsetting(v HubSetting) { + e.u64(v.Slot) + e.u64(v.ProgressType) + e.count(len(v.EventUIDs)) + for _, item := range v.EventUIDs { + e.u64(item) + } +} + +func (d *decoder) hubsetting() HubSetting { + var v HubSetting + v.Slot = d.u64() + v.ProgressType = d.u64() + for n := d.count(); n > 0 && d.err == nil; n-- { + v.EventUIDs = append(v.EventUIDs, d.u64()) + } + return v +} + +func (e *encoder) eventhub(v EventHub) { + e.u64(v.UID) + e.u64(v.HubID) + e.str(v.Start) + e.str(v.PlayEnd) + e.str(v.End) + e.count(len(v.Settings)) + for _, item := range v.Settings { + e.hubsetting(item) + } +} + +func (d *decoder) eventhub() EventHub { + var v EventHub + v.UID = d.u64() + v.HubID = d.u64() + v.Start = d.str() + v.PlayEnd = d.str() + v.End = d.str() + for n := d.count(); n > 0 && d.err == nil; n-- { + v.Settings = append(v.Settings, d.hubsetting()) + } + return v +} + +func (e *encoder) minigamehub(v MiniGameHub) { + e.u64(v.Slot) + e.u64(v.EventUID) + e.u64(v.ProgressType) +} + +func (d *decoder) minigamehub() MiniGameHub { + var v MiniGameHub + v.Slot = d.u64() + v.EventUID = d.u64() + v.ProgressType = d.u64() + return v +} + +func (e *encoder) regularseason(v schedule.RegularSeason) { + e.u64(v.ContentID) + e.u64(v.Season) +} + +func (d *decoder) regularseason() schedule.RegularSeason { + var v schedule.RegularSeason + v.ContentID = d.u64() + v.Season = d.u64() + return v +} + +// validateMiniHubBindings rejects invalid published routes at startup instead +// of letting a client open an incompatible prefab or use another activity UID. +func (s *Set) validateMiniHubBindings(design *gamedata.EventPlayCatalog) error { + uid := make(map[uint64]events.Schedule) + for _, v := range s.Events { + if v.UID != 0 { + uid[v.UID] = v + } + } + value := func(row []byte, field int) uint64 { v, _, _ := wire.Varint(row, field); return v } + for _, hub := range s.EventHubs { + table, err := design.Row("PackEventHubTable", 14, hub.HubID) + if err != nil { + return err + } + if value(table, 13) != 1 { + continue + } + if hub.UID == 0 { + return fmt.Errorf("calendar: mini hub %d requires a nonzero UID", hub.HubID) + } + start, err := ParseTimestamp(hub.Start) + if err != nil { + return err + } + playEnd, err := ParseTimestamp(hub.PlayEnd) + if err != nil { + return err + } + end, err := ParseTimestamp(hub.End) + if err != nil { + return err + } + for _, binding := range hub.Settings { + var slot []byte + for _, row := range design.Rows("PackEventListTable", 6, hub.HubID) { + if value(row, 11) != binding.Slot { + continue + } + if slot != nil { + return fmt.Errorf("calendar: mini hub %d slot %d is ambiguous in GameData", hub.HubID, binding.Slot) + } + slot = row + } + if slot == nil { + return fmt.Errorf("calendar: mini hub %d slot %d missing GameData", hub.HubID, binding.Slot) + } + contentType, contentID := value(slot, 9), value(slot, 7) + if binding.ProgressType != contentType { + return fmt.Errorf("calendar: mini hub %d slot %d content type %d, want %d", hub.HubID, binding.Slot, binding.ProgressType, contentType) + } + eventType, supported := gamedata.MiniHubEventType(contentType) + // Mini stories and NPC quizzes have their own slot UID namespace, + // distinct from Define_EventType (13 there means bingo). Their group + // and availability derive from this static slot and the hub window. + if contentType == 13 || contentType == 14 { + if len(binding.EventUIDs) != 1 || binding.EventUIDs[0] == 0 { + return fmt.Errorf("calendar: mini hub %d slot %d requires one local content UID", hub.HubID, binding.Slot) + } + if _, collision := uid[binding.EventUIDs[0]]; collision { + return fmt.Errorf("calendar: mini hub %d slot %d content UID collides with global event", hub.HubID, binding.Slot) + } + if value(slot, 4) != 0 { + return fmt.Errorf("calendar: mini hub %d story/quiz slot has invalid end type", hub.HubID) + } + continue + } + if !supported { + return fmt.Errorf("calendar: mini hub %d slot %d content type %d has no supported scheduled route", hub.HubID, binding.Slot, contentType) + } + endType := value(slot, 4) + if endType > 1 { + return fmt.Errorf("calendar: mini hub %d slot %d unsupported end type %d", hub.HubID, binding.Slot, endType) + } + matches := 0 + for _, id := range binding.EventUIDs { + child, ok := uid[id] + if !ok || child.Type != eventType || child.ID != contentID || child.SubID != 0 { + return fmt.Errorf("calendar: mini hub %d slot %d references incompatible event UID %d", hub.HubID, binding.Slot, id) + } + childStart, childEnd := child.Start, child.End + if childStart < int64(start) { + childStart = int64(start) + } + if childEnd > int64(end) { + childEnd = int64(end) + } + // Project policy: EndDateType=0 closes the slot at PlayEnd; + // EndDateType=1 allows it through the final hub End window. + if endType == 0 && childEnd > int64(playEnd) { + childEnd = int64(playEnd) + } + if childStart < childEnd { + matches++ + } + } + if matches != 1 { + return fmt.Errorf("calendar: mini hub %d slot %d has %d schedules in its window, want one", hub.HubID, binding.Slot, matches) + } + } + } + return nil +} + +// LoadDirectory validates the whole directory before returning any calendar. +// Every calendar file must use the .bd2schedule suffix. +// Directories and symlinks are +// rejected so an accidentally omitted calendar never produces a partial set. +func LoadDirectory(dir, gameVersion, gameDataVersion string) (*Set, error) { + entries, e := os.ReadDir(dir) + if e != nil { + return nil, fmt.Errorf("calendar: read directory: %w", e) + } + set := &Set{GachaSeed: &gacha.ScheduleSeed{ClientVersion: gameVersion}} + seen := map[string]bool{} + claim := func(key string) error { + if seen[key] { + return fmt.Errorf("duplicate calendar identity %s", key) + } + seen[key] = true + return nil + } + files := 0 + for _, entry := range entries { + if !entry.Type().IsRegular() || !strings.HasSuffix(entry.Name(), ".bd2schedule") { + return nil, fmt.Errorf("calendar: unsupported entry %s", entry.Name()) + } + info, e := entry.Info() + if e != nil { + return nil, e + } + if info.Size() > MaxFileSize { + return nil, fmt.Errorf("calendar: %s: file size limit exceeded", entry.Name()) + } + raw, e := os.ReadFile(filepath.Join(dir, entry.Name())) + if e != nil { + return nil, e + } + m, e := UnmarshalBinary(raw) + if e != nil { + return nil, fmt.Errorf("calendar: %s: %w", entry.Name(), e) + } + if m.SchemaVersion != 1 || strings.TrimSpace(m.Revision) == "" || m.GameVersion != gameVersion || m.GameDataVersion != gameDataVersion || gameVersion == "" || gameDataVersion == "" { + return nil, fmt.Errorf("calendar: %s: incompatible schema/version or missing revision", entry.Name()) + } + if e = merge(set, m, claim); e != nil { + return nil, fmt.Errorf("calendar: %s: %w", entry.Name(), e) + } + set.Revisions = append(set.Revisions, m.Revision) + files++ + } + if files == 0 { + return nil, fmt.Errorf("calendar: empty directory") + } + sort.Slice(set.EventHubs, func(i, j int) bool { return set.EventHubs[i].UID < set.EventHubs[j].UID }) + sort.Slice(set.MiniGameHubs, func(i, j int) bool { return set.MiniGameHubs[i].Slot < set.MiniGameHubs[j].Slot }) + sort.Slice(set.Events, func(i, j int) bool { + a, b := set.Events[i], set.Events[j] + if a.UID != b.UID { + return a.UID < b.UID + } + if a.Type != b.Type { + return a.Type < b.Type + } + if a.ID != b.ID { + return a.ID < b.ID + } + return a.SubID < b.SubID + }) + sortWindows := func(w []gacha.ScheduleWindow) { + sort.Slice(w, func(i, j int) bool { + if w[i].GroupID != w[j].GroupID { + return w[i].GroupID < w[j].GroupID + } + return w[i].StartTime < w[j].StartTime + }) + } + sortWindows(set.GachaSeed.Schedules) + sortWindows(set.GachaSeed.StepUps) + if e = set.GachaSeed.Validate(gameVersion); e != nil { + return nil, e + } + if set.RegularService != nil { + sort.Slice(set.RegularService.Contents, func(i, j int) bool { return set.RegularService.Contents[i].ID < set.RegularService.Contents[j].ID }) + sort.Slice(set.RegularService.Regular, func(i, j int) bool { + return set.RegularService.Regular[i].ContentID < set.RegularService.Regular[j].ContentID + }) + if e = set.RegularService.Validate(); e != nil { + return nil, e + } + } + sort.Slice(set.CashProducts, func(i, j int) bool { + a, b := set.CashProducts[i], set.CashProducts[j] + if a.GroupID != b.GroupID { + return a.GroupID < b.GroupID + } + if a.ProductID != b.ProductID { + return a.ProductID < b.ProductID + } + return a.SaleGroup < b.SaleGroup + }) + if set.MonsterHunt != nil { + sort.Slice(set.MonsterHunt.Seasons, func(i, j int) bool { + return set.MonsterHunt.Seasons[i].Season.ID < set.MonsterHunt.Seasons[j].Season.ID + }) + sort.Slice(set.MonsterHunt.History, func(i, j int) bool { return set.MonsterHunt.History[i].Season < set.MonsterHunt.History[j].Season }) + } + return set, nil +} + +// ValidateDesign checks playable identities against their domain's installed +// GameData. Announcement-only types 3/6/14/15/16/18 have no local gameplay +// design resolver and receive only the manifest's protocol/format validation. +func (s *Set) ValidateDesign(root, version string) error { + cash, e := gamedata.LoadCashCatalog(root, version) + if e != nil { + return e + } + keys := map[gamedata.CashProductKey]bool{} + for _, v := range cash.Products { + keys[v.Key] = true + } + for _, v := range s.CashProducts { + if !keys[gamedata.CashProductKey{GroupID: v.GroupID, ProductID: v.ProductID, SaleGroup: v.SaleGroup}] { + return fmt.Errorf("calendar: cash product %d/%d/%d missing GameData", v.GroupID, v.ProductID, v.SaleGroup) + } + } + if s.MonsterHunt != nil { + ids, e := gamedata.LoadMonsterHuntIDs(root, version) + if e != nil { + return e + } + for _, v := range s.MonsterHunt.Seasons { + if !ids[v.HuntID] { + return fmt.Errorf("calendar: active/playable hunt %d missing GameData", v.HuntID) + } + } + } + play, e := gamedata.LoadEventPlayCatalog(root, version) + if e != nil { + return e + } + tasks, e := gamedata.LoadEventTasksDesign(root, version) + if e != nil { + return e + } + exchange, e := gamedata.LoadEventExchangeCatalog(root, version) + if e != nil { + return e + } + actions, e := gamedata.LoadEventActionsDesign(root, version) + if e != nil { + return e + } + packs, e := gamedata.LoadCalendarPackIDs(root, version) + if e != nil { + return e + } + for _, h := range s.EventHubs { + if _, e = play.Row("PackEventHubTable", 14, h.HubID); e != nil { + return fmt.Errorf("calendar: hub %d: %w", h.HubID, e) + } + } + if err := s.validateMiniHubBindings(play); err != nil { + return err + } + uid := map[uint64]events.Schedule{} + for _, v := range s.Events { + if v.UID != 0 { + uid[v.UID] = v + } + } + for _, h := range s.MiniGameHubs { + v, ok := uid[h.EventUID] + if !ok || v.Type != 11 { + return fmt.Errorf("calendar: mini game slot %d references non-minigame event %d", h.Slot, h.EventUID) + } + if _, e := play.Row("PackEventMiniGameTable", 8, v.ID); e != nil { + return e + } + + } + for _, v := range s.Events { + valid := true + var err error + switch v.Type { + case 0: + _, valid = tasks.Attendance[v.ID] + case 1: + valid = false + for k := range tasks.LimitRewards { + if k[0] == v.ID { + valid = true + break + } + } + case 4: + _, valid = tasks.MissionGroups[v.ID] + case 5: + _, valid = tasks.Passes[v.ID] + case 7: + _, valid = exchange.Groups[v.ID] + case 8: + valid = packs[v.ID] + case 9: + _, err = play.Row("PackEventBattleGroupTable", 3, v.ID) + case 10: + _, err = play.Row("PackEventStoryGroupTable", 1, v.ID) + case 11: + _, err = play.Row("PackEventMiniGameTable", 8, v.ID) + case 12, 13, 17, 19: + _, err = gamedata.LoadEventGame(root, version, v.Type, v.ID) + case 20: + _, valid = actions.Row("TacticsBingoGroupTable", 3, v.ID) + case 21: + _, valid = actions.Row("FieldSpawnEventTable", 5, v.ID) + case 22: + _, valid = actions.Row("FireworksTable", 4, v.ID) + case 23: + _, valid = actions.Row("VotingEventTable", 6, v.ID) + case 24: + _, valid = actions.Row("FriendshipSpecialEpisodeTable", 5, v.ID) + case 25: + id := v.SubID + if id == 0 { + id = v.ID + } + _, valid = actions.Row("CafeteriaEventTable", 5, id) + } + if !valid || err != nil { + return fmt.Errorf("calendar: event uid %d type %d id %d missing GameData: %v", v.UID, v.Type, v.ID, err) + } + } + return nil +} + +func (e *encoder) u64(v uint64) { + if e.room(8) { + e.data = binary.LittleEndian.AppendUint64(e.data, v) + } +} + +func (e *encoder) u32(v uint32) { + if e.room(4) { + e.data = binary.LittleEndian.AppendUint32(e.data, v) + } +} + +func (e *encoder) str(v string) { + if e.err != nil { + return + } + if len(v) > maxString || !utf8.ValidString(v) { + e.err = fmt.Errorf("calendar: invalid/oversized UTF8 string") + return + } + if !e.room(4 + len(v)) { + return + } + e.u32(uint32(len(v))) + e.data = append(e.data, v...) +} + +func (e *encoder) count(n int) { + if e.err != nil { + return + } + e.rows += uint64(n) + if n > maxRows || e.rows > maxRows { + e.err = fmt.Errorf("calendar: row limit exceeded") + return + } + e.u32(uint32(n)) +} + +func (d *decoder) u64() uint64 { + v := d.take(8) + if len(v) != 8 { + return 0 + } + return binary.LittleEndian.Uint64(v) +} + +func (d *decoder) u32() uint32 { + v := d.take(4) + if len(v) != 4 { + return 0 + } + return binary.LittleEndian.Uint32(v) +} + +func (d *decoder) str() string { + n := d.u32() + if n > maxString { + d.err = fmt.Errorf("calendar: string limit exceeded") + return "" + } + v := d.take(int(n)) + if !utf8.Valid(v) { + d.err = fmt.Errorf("calendar: invalid UTF8") + } + return string(v) +} + +func (d *decoder) count() int { + n := d.u32() + d.rows += uint64(n) + if n > maxRows || d.rows > maxRows || uint64(n) > uint64(len(d.data)-d.pos) { + d.err = fmt.Errorf("calendar: row count limit/truncation") + return 0 + } + return int(n) +} + +func MarshalBinary(m Manifest) ([]byte, error) { + if m.SchemaVersion != 1 { + return nil, fmt.Errorf("calendar: unsupported schema") + } + e := &encoder{} + e.manifest(m) + if e.err != nil { + return nil, e.err + } + if len(e.data) > MaxFileSize-headerSize { + return nil, fmt.Errorf("calendar: file size limit exceeded") + } + out := append([]byte(nil), magic...) + out = binary.LittleEndian.AppendUint16(out, 1) + out = binary.LittleEndian.AppendUint32(out, uint32(len(e.data))) + sum := sha256.Sum256(e.data) + out = append(out, sum[:]...) + return append(out, e.data...), nil +} + +func UnmarshalBinary(raw []byte) (Manifest, error) { + var m Manifest + if len(raw) < headerSize || len(raw) > MaxFileSize { + return m, fmt.Errorf("calendar: invalid file size") + } + if !bytes.Equal(raw[:8], magic) || binary.LittleEndian.Uint16(raw[8:10]) != 1 { + return m, fmt.Errorf("calendar: unsupported magic/format") + } + n := binary.LittleEndian.Uint32(raw[10:14]) + if uint64(n) != uint64(len(raw)-headerSize) { + return m, fmt.Errorf("calendar: payload length mismatch") + } + payload := raw[headerSize:] + sum := sha256.Sum256(payload) + if !bytes.Equal(raw[14:46], sum[:]) { + return m, fmt.Errorf("calendar: checksum mismatch") + } + d := &decoder{data: payload} + m = d.manifest() + if d.err != nil { + return Manifest{}, d.err + } + if d.pos != len(payload) { + return Manifest{}, fmt.Errorf("calendar: trailing payload") + } + m.SchemaVersion = 1 + return m, nil +} diff --git a/go/internal/server/calendar/schema.go b/go/internal/server/domain/events/calendar/schema.go similarity index 94% rename from go/internal/server/calendar/schema.go rename to go/internal/server/domain/events/calendar/schema.go index c17451d..7169ca3 100644 --- a/go/internal/server/calendar/schema.go +++ b/go/internal/server/domain/events/calendar/schema.go @@ -2,9 +2,9 @@ package calendar import ( - "bd2server/internal/server/events" - "bd2server/internal/server/gacha" - "bd2server/internal/server/schedule" + "bd2server/internal/server/design/schedule" + "bd2server/internal/server/domain/commerce/gacha" + "bd2server/internal/server/domain/events" ) type Manifest struct { diff --git a/go/internal/server/domain/events/economy.go b/go/internal/server/domain/events/economy.go new file mode 100644 index 0000000..47e8d22 --- /dev/null +++ b/go/internal/server/domain/events/economy.go @@ -0,0 +1,150 @@ +package events + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "bd2server/internal/server/storage/stateio" + "crypto/sha256" + "encoding/json" + "fmt" + "maps" + "math" + + "time" +) + +type HuntingAP interface { + CanExchangeAP(ctx command.Context, costs, rewards []gamedata.Reward) error + ExchangeAPOnce(ctx command.Context, identity string, costs, rewards []gamedata.Reward) error +} + +type rewardResolver interface { + Resolve([]gamedata.BattleReward) ([]gamedata.BattleReward, error) +} +type equipmentRoller interface { + RollOptions(uint64) ([]gamedata.EquipmentOptionChoice, []gamedata.EquipmentOptionChoice, *gamedata.EquipmentOptionChoice, error) +} + +type economyReceipt struct{ Digest, Definition, Bundle []byte } +type economySnapshot struct { + Balances map[uint64]uint64 `json:"balances"` + Receipts map[string]economyReceipt `json:"receipts"` +} + +// Economy dispatches verified static rewards to their owning domains. It is +// called inside the same account transaction as the gameplay operation. +type Economy struct { + ownedDesign map[uint64]map[uint64]bool + avatarRewards *gamedata.AvatarRewardDesign + buffRewards *BuffRewards + + store stateio.Store + items *assets.Inventory + wallet *assets.Wallet + collection *roster.CollectionStore + equipment *assets.EquipmentInventory + costumes roster.CostumeDesignSource + options equipmentRoller + graph rewardResolver + hunting HuntingAP + prestige map[uint64]uint64 + prestigePortrait func() uint64 + apCaps map[uint64]uint64 + resetSeconds int64 + now func() time.Time + initial map[uint64]uint64 +} + +func NewEconomy(ctx command.Context, store stateio.Store, items *assets.Inventory, wallet *assets.Wallet, collection *roster.CollectionStore, equipment *assets.EquipmentInventory, costumes roster.CostumeDesignSource, options equipmentRoller, graph rewardResolver, initial map[uint64]uint64) (*Economy, error) { + if store == nil || items == nil || wallet == nil || collection == nil || equipment == nil || costumes == nil || options == nil || graph == nil { + return nil, fmt.Errorf("events: invalid economy configuration") + } + e := &Economy{store: store, items: items, wallet: wallet, collection: collection, equipment: equipment, costumes: costumes, options: options, graph: graph, initial: map[uint64]uint64{}} + for t, n := range initial { + if _, ok := AdditionalCurrencyFields[t]; !ok || n > math.MaxInt32 { + return nil, fmt.Errorf("events: invalid initial currency %d", t) + } + e.initial[t] = n + } + _, err := e.load(ctx) + return e, err +} + +// AdditionalCurrencyFields maps EElementType to its UserDBInfo field. These +// balances use a gameplay entry, leaving the frozen wallet schema unchanged. +var AdditionalCurrencyFields = map[uint64]int{15: 16, 16: 17, 18: 18, 24: 22, 30: 34, 31: 35, 32: 36, 33: 37, 38: 42, 39: 44, 40: 45, 43: 48, 44: 54, 54: 61, 60: 60, 70: 71} + +func (e *Economy) AttachAvatarRewards(d *gamedata.AvatarRewardDesign) { e.avatarRewards = d } + +func (e *Economy) AttachHuntingAP(h HuntingAP) { e.hunting = h } +func (e *Economy) AdditionalCurrencies(ctx command.Context) (map[int]uint64, error) { + + s, err := e.load(ctx) + if err != nil { + return nil, err + } + if err = e.refreshAP(ctx, &s); err != nil { + return nil, err + } + out := map[int]uint64{} + for t, f := range AdditionalCurrencyFields { + out[f] = s.Balances[t] + } + return out, nil +} +func (e *Economy) load(ctx command.Context) (economySnapshot, error) { + s := economySnapshot{Balances: map[uint64]uint64{}, Receipts: map[string]economyReceipt{}} + maps.Copy(s.Balances, e.initial) + b, err := e.store.Load(ctx.State, "event_economy") + if err != nil || b == nil { + return s, err + } + if err = stateio.RequireExactJSONObject(b, "balances", "receipts"); err != nil { + return s, err + } + if err = json.Unmarshal(b, &s); err != nil { + return s, err + } + if s.Balances == nil || s.Receipts == nil { + return s, fmt.Errorf("events: invalid saved economy") + } + for t, n := range s.Balances { + if _, ok := AdditionalCurrencyFields[t]; !ok || n > math.MaxInt32 { + return s, fmt.Errorf("events: invalid saved currency %d", t) + } + } + for id, r := range s.Receipts { + if id == "" || len(r.Digest) != sha256.Size { + return s, fmt.Errorf("events: invalid economy receipt") + } + } + return s, nil +} + +func walletCurrency(t uint64) bool { + switch t { + case 2, 3, 4, 12, 20, 22, 68: + return true + } + return false +} +func extraCurrency(t uint64) bool { _, ok := AdditionalCurrencyFields[t]; return ok } +func apCurrency(t uint64) bool { return t == 21 || t == 23 } +func inventoryType(t uint64) bool { + switch t { + case 5, 7, 8, 9, 13, 14, 17, 19, 25, 26, 27, 29, 34, 45, 46, 47, 49, 50, 61, 69: + return true + } + return false +} + +func (e *Economy) resolveGranted(rewards []gamedata.BattleReward) ([]gamedata.BattleReward, error) { + if g, ok := e.graph.(interface { + ResolveGranted([]gamedata.BattleReward) ([]gamedata.BattleReward, error) + }); ok { + return g.ResolveGranted(rewards) + } + return e.graph.Resolve(rewards) +} diff --git a/go/internal/server/eventexchange/service.go b/go/internal/server/domain/events/exchange/protocol_adapter.go similarity index 53% rename from go/internal/server/eventexchange/service.go rename to go/internal/server/domain/events/exchange/protocol_adapter.go index 7c8b1ee..c022abd 100644 --- a/go/internal/server/eventexchange/service.go +++ b/go/internal/server/domain/events/exchange/protocol_adapter.go @@ -1,162 +1,33 @@ package eventexchange import ( - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/protocol/wire" "bytes" - "crypto/rand" - "encoding/json" "fmt" "math" - "math/big" "sort" "strconv" - "sync" - "time" ) -type Runtime interface { - Apply(string, []gamedata.Reward, []gamedata.Reward) ([]byte, error) - ConsumeAndGrant(string, []player.Item, []gamedata.BattleReward) ([]byte, error) -} -type progress struct { - Group, Page, Key, Free uint64 - Reset int64 - Counts map[string]uint64 -} -type receipt struct { - Fingerprint, Response []byte - Code int -} -type snapshot struct { - Version string `json:"version"` - Progress map[string]progress `json:"progress"` - Receipts map[string]receipt `json:"receipts"` -} -type Service struct { - mu sync.Mutex - store stateio.Store - design *gamedata.EventExchangeCatalog - registry events.Resolver - runtime Runtime - now func() time.Time - draw func(uint64) (uint64, error) -} - -func Open(store stateio.Store, design *gamedata.EventExchangeCatalog, registry events.Resolver, runtime Runtime) (*Service, error) { - if store == nil || design == nil || registry == nil || runtime == nil { - return nil, fmt.Errorf("eventexchange: invalid configuration") - } - s := &Service{store: store, design: design, registry: registry, runtime: runtime, now: time.Now} - s.draw = func(n uint64) (uint64, error) { - if n == 0 { - return 0, fmt.Errorf("eventexchange: empty reward pool") - } - v, e := rand.Int(rand.Reader, new(big.Int).SetUint64(n)) - if e != nil { - return 0, e - } - return v.Uint64(), nil - } - _, err := s.load() - return s, err -} -func (s *Service) load() (snapshot, error) { - st := snapshot{versionconfig.State(), map[string]progress{}, map[string]receipt{}} - b, err := s.store.Load("eventexchange") - if err != nil { - return st, err - } - if b == nil { - return st, nil - } - if err = stateio.RequireExactJSONObject(b, "version", "progress", "receipts"); err != nil { - return st, err - } - dec := json.NewDecoder(bytes.NewReader(b)) - dec.DisallowUnknownFields() - if err = dec.Decode(&st); err != nil { - return st, err - } - if st.Version != versionconfig.State() || st.Progress == nil || st.Receipts == nil { - return st, fmt.Errorf("eventexchange: incompatible state") - } - for uid, p := range st.Progress { - n, e := strconv.ParseUint(uid, 10, 64) - g, ok := s.design.Groups[p.Group] - if e != nil || n == 0 || !ok || p.Page < g.StartPage || (!g.Repeat && p.Page > g.EndPage) || p.Counts == nil || p.Free > g.FreeCount { - return st, fmt.Errorf("eventexchange: invalid saved progress") - } - known := map[string]uint64{} - for _, entry := range g.Page(p.Page) { - known[strconv.FormatUint(entry.ID, 10)] = entry.SetCount - } - for id, count := range p.Counts { - limit, ok := known[id] - if !ok || count > limit { - return st, fmt.Errorf("eventexchange: invalid reward count") - } - } - } - return st, nil -} -func (s *Service) save(st snapshot) error { - b, err := json.Marshal(st) - if err != nil { - return err - } - return s.store.Save("eventexchange", b) -} -func (s *Service) active(uid uint64) (events.Schedule, gamedata.EventExchangeGroup, error) { - a, e := s.registry.Resolve(uid) - if e != nil { - return a, gamedata.EventExchangeGroup{}, e - } - g, ok := s.design.Groups[a.ID] - now := s.now().UnixMilli() - if !ok || a.Type != 7 || now < a.Start || now >= a.End { - return a, g, fmt.Errorf("eventexchange: event unavailable") - } - return a, g, nil -} -func (s *Service) init(st *snapshot, a events.Schedule, g gamedata.EventExchangeGroup) progress { - k := strconv.FormatUint(a.UID, 10) - p, ok := st.Progress[k] - if !ok { - p = progress{Group: g.ID, Page: g.StartPage, Free: g.FreeCount, Counts: map[string]uint64{}} - } - if g.FreeType == 2 { - now := s.now() - if p.Reset <= now.UnixMilli() { - p.Free = g.FreeCount - p.Reset = now.UTC().Truncate(24 * time.Hour).Add(24 * time.Hour).UnixMilli() - } - } - return p -} -func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { - return s.HandleSession(path, req, "local") -} -func (s *Service) HandleSession(path string, req []byte, session string) (int, []byte, bool, error) { +func (s *Service) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID code, ok := map[string]int{"/EventExchangeInfo": 183, "/EventExchangeReward": 184, "/EventExchangeNextPageOpen": 185}[path] if !ok { return 0, nil, false, nil } - s.mu.Lock() - defer s.mu.Unlock() + seq, found, err := wire.Varint(req, 1) if err != nil || !found || seq == 0 || seq > math.MaxInt32 { return code, nil, true, fmt.Errorf("eventexchange: invalid sequence") } - st, err := s.load() + st, err := s.load(ctx) if err != nil { return code, nil, true, err } - key := session + ":" + strconv.FormatUint(seq, 10) + key := commandSession + ":" + strconv.FormatUint(seq, 10) fingerprint := append([]byte(path), req...) if r, ok := st.Receipts[key]; ok { if !bytes.Equal(fingerprint, r.Fingerprint) { @@ -180,7 +51,7 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ out = wire.AppendBytes(out, 2, countWire(a.UID, g.ID, entry.ID, p.Counts[strconv.FormatUint(entry.ID, 10)])) } } - if err = s.save(st); err != nil { + if err = s.save(ctx, st); err != nil { return code, nil, true, err } return code, out, true, nil @@ -289,19 +160,19 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ } var bundle []byte if free { - bundle, err = s.runtime.Apply("eventexchange:"+key, nil, rewards) + bundle, err = s.runtime.Apply(ctx, "eventexchange:"+key, nil, rewards) } else { currency := g.Cost.Type == 2 || g.Cost.Type == 3 || g.Cost.Type == 4 || g.Cost.Type == 12 || g.Cost.Type == 20 if currency { cost := g.Cost cost.Count *= count - bundle, err = s.runtime.Apply("eventexchange:"+key, []gamedata.Reward{cost}, rewards) + bundle, err = s.runtime.Apply(ctx, "eventexchange:"+key, []gamedata.Reward{cost}, rewards) } else { rs := make([]gamedata.BattleReward, len(rewards)) for i, r := range rewards { rs[i] = gamedata.BattleReward{Type: r.Type, ID: r.ID, Count: r.Count} //nolint:staticcheck // S1016 } - bundle, err = s.runtime.ConsumeAndGrant("eventexchange:"+key, uses, rs) + bundle, err = s.runtime.ConsumeAndGrant(ctx, "eventexchange:"+key, uses, rs) } } if err != nil { @@ -313,43 +184,12 @@ func (s *Service) HandleSession(path string, req []byte, session string) (int, [ } st.Progress[strconv.FormatUint(uid, 10)] = p st.Receipts[key] = receipt{fingerprint, out, code} - if err = s.save(st); err != nil { + if err = s.save(ctx, st); err != nil { return code, nil, true, err } return code, out, true, nil } -func exhausted(g gamedata.EventExchangeGroup, p progress) bool { - for _, e := range g.Page(p.Page) { - if p.Counts[strconv.FormatUint(e.ID, 10)] < e.SetCount { - return false - } - } - return true -} -func canAdvance(g gamedata.EventExchangeGroup, p progress) bool { - if p.Page >= g.EndPage && !g.Repeat { - return false - } - var keys uint64 - for _, e := range g.Page(p.Page) { - if e.KeyType == 1 { - keys++ - } - } - return exhausted(g, p) || (keys > 0 && p.Key >= keys) -} -func advance(g gamedata.EventExchangeGroup, p *progress) error { - if p.Page >= math.MaxInt32 { - return fmt.Errorf("eventexchange: page overflow") - } - if p.Page >= g.EndPage && !g.Repeat { - return fmt.Errorf("eventexchange: final page") - } - p.Page++ - p.Key = 0 - p.Counts = map[string]uint64{} - return nil -} + func progressWire(uid uint64, p progress) []byte { var out []byte for _, f := range []struct { @@ -362,6 +202,7 @@ func progressWire(uid uint64, p progress) []byte { } return out } + func countWire(uid, group, id, count uint64) []byte { b := wire.AppendVarint(nil, 1, uid) b = wire.AppendVarint(b, 2, group) @@ -371,8 +212,9 @@ func countWire(uid, group, id, count uint64) []byte { } return b } -func parseUses(req []byte) ([]player.Item, error) { - var out []player.Item + +func parseUses(req []byte) ([]assets.Item, error) { + var out []assets.Item seen := map[uint64]bool{} err := wire.Walk(req, func(f wire.Field) error { if f.Number != 4 { @@ -381,7 +223,7 @@ func parseUses(req []byte) ([]player.Item, error) { if f.Type != 2 { return fmt.Errorf("eventexchange: malformed use item") } - var item player.Item + var item assets.Item for n, p := range map[int]*uint64{1: &item.InvenIndex, 2: &item.ID, 3: &item.Type, 4: &item.Count} { v, _, e := wire.Varint(f.Value, n) if e != nil { diff --git a/go/internal/server/domain/events/exchange/service.go b/go/internal/server/domain/events/exchange/service.go new file mode 100644 index 0000000..25876e5 --- /dev/null +++ b/go/internal/server/domain/events/exchange/service.go @@ -0,0 +1,171 @@ +package eventexchange + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/storage/stateio" + "bytes" + "crypto/rand" + "encoding/json" + "fmt" + "math" + "math/big" + "strconv" + + "time" +) + +type Runtime interface { + Apply(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) + ConsumeAndGrant(ctx command.Context, _ string, _ []assets.Item, _ []gamedata.BattleReward) ([]byte, error) +} +type progress struct { + Group, Page, Key, Free uint64 + Reset int64 + Counts map[string]uint64 +} +type receipt struct { + Fingerprint, Response []byte + Code int +} +type snapshot struct { + Version string `json:"version"` + Progress map[string]progress `json:"progress"` + Receipts map[string]receipt `json:"receipts"` +} +type Service struct { + store stateio.Store + design *gamedata.EventExchangeCatalog + registry events.Resolver + runtime Runtime + now func() time.Time + draw func(uint64) (uint64, error) +} + +func Open(ctx command.Context, store stateio.Store, design *gamedata.EventExchangeCatalog, registry events.Resolver, runtime Runtime) (*Service, error) { + if store == nil || design == nil || registry == nil || runtime == nil { + return nil, fmt.Errorf("eventexchange: invalid configuration") + } + s := &Service{store: store, design: design, registry: registry, runtime: runtime, now: time.Now} + s.draw = func(n uint64) (uint64, error) { + if n == 0 { + return 0, fmt.Errorf("eventexchange: empty reward pool") + } + v, e := rand.Int(rand.Reader, new(big.Int).SetUint64(n)) + if e != nil { + return 0, e + } + return v.Uint64(), nil + } + _, err := s.load(ctx) + return s, err +} +func (s *Service) load(ctx command.Context) (snapshot, error) { + st := snapshot{versionconfig.State(), map[string]progress{}, map[string]receipt{}} + b, err := s.store.Load(ctx.State, "eventexchange") + if err != nil { + return st, err + } + if b == nil { + return st, nil + } + if err = stateio.RequireExactJSONObject(b, "version", "progress", "receipts"); err != nil { + return st, err + } + dec := json.NewDecoder(bytes.NewReader(b)) + dec.DisallowUnknownFields() + if err = dec.Decode(&st); err != nil { + return st, err + } + if st.Version != versionconfig.State() || st.Progress == nil || st.Receipts == nil { + return st, fmt.Errorf("eventexchange: incompatible state") + } + for uid, p := range st.Progress { + n, e := strconv.ParseUint(uid, 10, 64) + g, ok := s.design.Groups[p.Group] + if e != nil || n == 0 || !ok || p.Page < g.StartPage || (!g.Repeat && p.Page > g.EndPage) || p.Counts == nil || p.Free > g.FreeCount { + return st, fmt.Errorf("eventexchange: invalid saved progress") + } + known := map[string]uint64{} + for _, entry := range g.Page(p.Page) { + known[strconv.FormatUint(entry.ID, 10)] = entry.SetCount + } + for id, count := range p.Counts { + limit, ok := known[id] + if !ok || count > limit { + return st, fmt.Errorf("eventexchange: invalid reward count") + } + } + } + return st, nil +} +func (s *Service) save(ctx command.Context, st snapshot) error { + b, err := json.Marshal(st) + if err != nil { + return err + } + return s.store.Save(ctx.State, "eventexchange", b) +} +func (s *Service) active(uid uint64) (events.Schedule, gamedata.EventExchangeGroup, error) { + a, e := s.registry.Resolve(uid) + if e != nil { + return a, gamedata.EventExchangeGroup{}, e + } + g, ok := s.design.Groups[a.ID] + now := s.now().UnixMilli() + if !ok || a.Type != 7 || now < a.Start || now >= a.End { + return a, g, fmt.Errorf("eventexchange: event unavailable") + } + return a, g, nil +} +func (s *Service) init(st *snapshot, a events.Schedule, g gamedata.EventExchangeGroup) progress { + k := strconv.FormatUint(a.UID, 10) + p, ok := st.Progress[k] + if !ok { + p = progress{Group: g.ID, Page: g.StartPage, Free: g.FreeCount, Counts: map[string]uint64{}} + } + if g.FreeType == 2 { + now := s.now() + if p.Reset <= now.UnixMilli() { + p.Free = g.FreeCount + p.Reset = now.UTC().Truncate(24 * time.Hour).Add(24 * time.Hour).UnixMilli() + } + } + return p +} + +func exhausted(g gamedata.EventExchangeGroup, p progress) bool { + for _, e := range g.Page(p.Page) { + if p.Counts[strconv.FormatUint(e.ID, 10)] < e.SetCount { + return false + } + } + return true +} +func canAdvance(g gamedata.EventExchangeGroup, p progress) bool { + if p.Page >= g.EndPage && !g.Repeat { + return false + } + var keys uint64 + for _, e := range g.Page(p.Page) { + if e.KeyType == 1 { + keys++ + } + } + return exhausted(g, p) || (keys > 0 && p.Key >= keys) +} +func advance(g gamedata.EventExchangeGroup, p *progress) error { + if p.Page >= math.MaxInt32 { + return fmt.Errorf("eventexchange: page overflow") + } + if p.Page >= g.EndPage && !g.Repeat { + return fmt.Errorf("eventexchange: final page") + } + p.Page++ + p.Key = 0 + p.Counts = map[string]uint64{} + return nil +} diff --git a/go/internal/server/eventgames/SPEC.md b/go/internal/server/domain/events/games/SPEC.md similarity index 100% rename from go/internal/server/eventgames/SPEC.md rename to go/internal/server/domain/events/games/SPEC.md diff --git a/go/internal/server/domain/events/games/games.go b/go/internal/server/domain/events/games/games.go new file mode 100644 index 0000000..c01f340 --- /dev/null +++ b/go/internal/server/domain/events/games/games.go @@ -0,0 +1,148 @@ +// Package eventgames implements this server's persistent seasonal minigames. +package eventgames + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/storage/stateio" + "crypto/rand" + "encoding/json" + "fmt" + "math/big" + "slices" + + "time" +) + +type Resolver interface { + Resolve(uint64) (events.Schedule, error) +} +type Rewards interface { + ConsumeAndGrant(ctx command.Context, _ string, _ []assets.Item, _ []gamedata.BattleReward) ([]byte, error) +} +type GameState struct { + UID, Type, ID, Position, Clear, Free, Reset, Tries, SinceSpecial uint64 + Board, Opened, Lines []uint64 + Special bool +} +type reply struct{ Request, Body []byte } +type snapshot struct { + Version int `json:"version"` + Games map[string]GameState `json:"games"` + Replies map[string]reply `json:"replies"` +} + +// RuleLoader returns shared immutable game rules. +type RuleLoader func(uint64, uint64) (*gamedata.EventGame, error) + +type Service struct { + store stateio.Store + resolver Resolver + rewards Rewards + state snapshot + load func(uint64, uint64) (*gamedata.EventGame, error) + sample func(uint64) (uint64, error) + now func() time.Time +} + +func Open(ctx command.Context, store stateio.Store, load RuleLoader, resolver Resolver, rewards Rewards) (*Service, error) { + if store == nil || load == nil || resolver == nil || rewards == nil { + return nil, fmt.Errorf("eventgames: incomplete runtime") + } + s := &Service{store: store, resolver: resolver, rewards: rewards, now: time.Now, state: snapshot{1, map[string]GameState{}, map[string]reply{}}} + s.load = load + s.sample = func(max uint64) (uint64, error) { + if max == 0 { + return 0, fmt.Errorf("eventgames: empty sample") + } + n, e := rand.Int(rand.Reader, new(big.Int).SetUint64(max)) + if e != nil { + return 0, e + } + return n.Uint64(), nil + } + raw, e := store.Load(ctx.State, "eventgames") + if e != nil { + return nil, e + } + if raw != nil { + if e = stateio.RequireExactJSONObject(raw, "version", "games", "replies"); e != nil { + return nil, e + } + if e = json.Unmarshal(raw, &s.state); e != nil { + return nil, e + } + if s.state.Version != 1 || s.state.Games == nil || s.state.Replies == nil { + return nil, fmt.Errorf("eventgames: incompatible state") + } + } + for key, g := range s.state.Games { + d, e := s.load(g.Type, g.ID) + if e != nil { + return nil, e + } + if e = validateSaved(key, g, d); e != nil { + return nil, e + } + } + return s, nil +} + +func contains(a []uint64, v uint64) bool { + return slices.Contains(a, v) +} + +var routes = map[string]struct { + Code int + Kind uint64 +}{"/MiniGameBoardInfo": {223, 12}, "/MiniGameBoardPlay": {224, 12}, "/MiniGameBingoInfo": {256, 13}, "/MiniGameBingoPlay": {257, 13}, "/MiniPuzzleInfo": {343, 17}, "/MiniPuzzleOpen": {344, 17}, "/MiniPuzzleAllOpen": {345, 17}, "/MiniPuzzleRenew": {346, 17}, "/MiniGameRouletteInfo": {415, 19}, "/MiniGameRouletteDraw": {416, 19}} + +func (s *Service) initialize(g *GameState, d *gamedata.EventGame) error { + if d.Type != 13 && d.Type != 17 { + return nil + } + count := g.Clear + if d.Type == 17 && count == 0 { + count = 1 + g.Clear = 1 + } + var cells []gamedata.EventGameReward + max := uint64(0) + for _, r := range d.Cells { + if r.Count > max { + max = r.Count + } + } + if count > max { + count = max + } + for _, r := range d.Cells { + if r.Count == count { + cells = append(cells, r) + } + } + if len(cells) == 0 { + return fmt.Errorf("eventgames: board rewards missing") + } + g.Board = nil + g.Opened = nil + g.Lines = nil + for _, r := range cells { + g.Board = append(g.Board, r.ID) + } + if d.Type == 13 { + for i := len(g.Board) - 1; i > 0; i-- { + j, e := s.sample(uint64(i + 1)) + if e != nil { + return e + } + if j >= uint64(i+1) { + return fmt.Errorf("eventgames: sampler out of range") + } + g.Board[i], g.Board[j] = g.Board[j], g.Board[i] + } + } + return nil +} diff --git a/go/internal/server/domain/events/games/play.go b/go/internal/server/domain/events/games/play.go new file mode 100644 index 0000000..b33debc --- /dev/null +++ b/go/internal/server/domain/events/games/play.go @@ -0,0 +1,44 @@ +package eventgames + +import ( + "bd2server/internal/server/design/gamedata" +) + +func rewardCount(d *gamedata.EventGame, count uint64) uint64 { + max := uint64(0) + for _, r := range d.Cells { + if r.Count > max { + max = r.Count + } + } + if count > max { + return max + } + return count +} +func completeLine(g *GameState, n, typ, index uint64) bool { + if n == 0 { + return false + } + for i := range n { + var pos uint64 + switch typ { + case 1: + pos = index*n + i + case 2: + pos = i*n + index + case 0: + if index == 0 { + pos = i*n + i + } else { + pos = i*n + (n - 1 - i) + } + default: + return false + } + if pos >= uint64(len(g.Board)) || !contains(g.Opened, pos) { + return false + } + } + return true +} diff --git a/go/internal/server/eventgames/play.go b/go/internal/server/domain/events/games/protocol_adapter.go similarity index 50% rename from go/internal/server/eventgames/play.go rename to go/internal/server/domain/events/games/protocol_adapter.go index b1037b8..ced384d 100644 --- a/go/internal/server/eventgames/play.go +++ b/go/internal/server/domain/events/games/protocol_adapter.go @@ -1,10 +1,17 @@ package eventgames import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/protocol/wire" + "bytes" + "encoding/binary" + "encoding/json" "fmt" "math" + "strconv" + "strings" ) func (s *Service) play(path string, req []byte, g *GameState, d *gamedata.EventGame) ([]byte, uint64, map[int][]gamedata.BattleReward, error) { @@ -254,41 +261,243 @@ func (s *Service) play(path string, req []byte, g *GameState, d *gamedata.EventG } return body, cost, rewards, nil } -func rewardCount(d *gamedata.EventGame, count uint64) uint64 { - max := uint64(0) - for _, r := range d.Cells { - if r.Count > max { - max = r.Count + +func nums(b []byte, n int) ([]uint64, error) { + var out []uint64 + e := wire.Walk(b, func(f wire.Field) error { + if f.Number != n { + return nil } - } - if count > max { - return max - } - return count -} -func completeLine(g *GameState, n, typ, index uint64) bool { - if n == 0 { - return false - } - for i := range n { - var pos uint64 - switch typ { - case 1: - pos = index*n + i - case 2: - pos = i*n + index - case 0: - if index == 0 { - pos = i*n + i - } else { - pos = i*n + (n - 1 - i) + if f.Type == 0 { + v, _ := binary.Uvarint(f.Value) + out = append(out, v) + return nil + } + if f.Type != 2 { + return wire.ErrMalformed + } + for p := f.Value; len(p) > 0; { + v, k := binary.Uvarint(p) + if k <= 0 { + return wire.ErrMalformed } - default: - return false + out = append(out, v) + p = p[k:] } - if pos >= uint64(len(g.Board)) || !contains(g.Opened, pos) { - return false + return nil + }) + return out, e +} + +func scalar(b []byte, n int) uint64 { v, _, _ := wire.Varint(b, n); return v } + +func (s *Service) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + r, ok := routes[path] + if !ok { + return 0, nil, false, nil + } + + fail := func(e error) (int, []byte, bool, error) { return r.Code, nil, true, e } + if e := wire.Walk(req, func(f wire.Field) error { + if f.Type == 0 && f.Number != 3 { + v, _ := binary.Uvarint(f.Value) + if v > math.MaxInt32 { + return wire.ErrMalformed + } + } + return nil + }); e != nil { + return fail(e) + } + seq := scalar(req, 1) + if seq == 0 || seq > math.MaxInt32 { + return fail(fmt.Errorf("eventgames: invalid sequence")) + } + key := fmt.Sprintf("%s:%s:%d", commandSession, path, seq) + if v, ok := s.state.Replies[key]; ok { + if !bytes.Equal(v.Request, req) { + return fail(fmt.Errorf("eventgames: changed request retry")) + } + return r.Code, v.Body, true, nil + } + raw, _ := json.Marshal(s.state) + var next snapshot + _ = json.Unmarshal(raw, &next) + uids, e := nums(req, 2) + if e != nil || len(uids) == 0 && !strings.HasSuffix(path, "Info") { + return fail(fmt.Errorf("eventgames: missing schedule")) + } + if len(uids) == 0 { + return r.Code, nil, true, nil + } + if !strings.HasSuffix(path, "Info") && len(uids) != 1 { + return fail(fmt.Errorf("eventgames: operation requires one schedule")) + } + var out []byte + seen := map[uint64]bool{} + for _, uid := range uids { + if uid == 0 || seen[uid] { + return fail(fmt.Errorf("eventgames: invalid duplicate schedule")) + } + seen[uid] = true + c, e := s.resolver.Resolve(uid) + if e != nil { + return fail(e) + } + if c.UID != uid || c.Type != r.Kind { + return fail(fmt.Errorf("eventgames: schedule/game mismatch")) + } + d, e := s.load(c.Type, c.ID) + if e != nil { + return fail(e) + } + k := strconv.FormatUint(uid, 10) + g, exists := next.Games[k] + if !exists { + g = GameState{UID: uid, Type: c.Type, ID: c.ID, Free: d.Free} + if e = s.initialize(&g, d); e != nil { + return fail(e) + } + } + if g.Type != c.Type || g.ID != c.ID { + return fail(fmt.Errorf("eventgames: saved game identity changed")) + } + now := uint64(s.now().UnixMilli()) + if c.Type == 19 && g.Reset <= now { + g.Free = d.Free + g.Reset = (now/86400000 + 1) * 86400000 + } + if strings.HasSuffix(path, "Info") { + out = wire.AppendBytes(out, 1, stateWire(g, d)) + next.Games[k] = g + continue + } + if int64(now) < c.Start || int64(now) >= c.End { + return fail(fmt.Errorf("eventgames: event is closed")) + } + before := stateWire(g, d) + body, cost, rewards, e := s.play(path, req, &g, d) + if e != nil { + return fail(e) + } + consumeField := 3 + if path == "/MiniPuzzleOpen" { + consumeField = 4 + } + if path == "/MiniGameRouletteDraw" { + consumeField = 5 + } + items, e := consume(req, consumeField, d, cost) + if e != nil { + return fail(e) + } + bundles := map[int][]byte{} + fields := []int{3, 4, 5, 6, 7} + if c.Type == 19 { + fields = []int{2, 3} + } + for _, field := range fields { + rr := rewards[field] + if len(rr) == 0 { + continue + } + consumed := []assets.Item(nil) + if len(items) > 0 { + consumed = items + items = nil + } + b, e := s.rewards.ConsumeAndGrant(ctx, fmt.Sprintf("eventgames:%s:%d", key, field), consumed, rr) + if e != nil { + return fail(e) + } + bundles[field] = b + } + if len(items) > 0 { + if _, e = s.rewards.ConsumeAndGrant(ctx, "eventgames:"+key+":cost", items, nil); e != nil { + return fail(e) + } + } + switch c.Type { + case 12: + body = wire.AppendBytes(body, 1, stateWire(g, d)) + case 13, 17: + body = wire.AppendBytes(body, 1, before) + body = wire.AppendBytes(body, 2, stateWire(g, d)) + case 19: + body = wire.AppendBytes(body, 1, stateWire(g, d)) + } + for _, field := range fields { + if b := bundles[field]; len(b) > 0 { + body = wire.AppendBytes(body, field, b) + } + } + out = body + next.Games[k] = g + } + next.Replies[key] = reply{append([]byte(nil), req...), out} + payload, e := json.Marshal(next) + if e != nil { + return fail(e) + } + if e = s.store.Save(ctx.State, "eventgames", payload); e != nil { + return fail(e) + } + s.state = next + return r.Code, out, true, nil +} + +func consume(req []byte, n int, d *gamedata.EventGame, count uint64) ([]assets.Item, error) { + var out []assets.Item + total := uint64(0) + seen := map[uint64]bool{} + e := wire.Walk(req, func(f wire.Field) error { + if f.Number != n { + return nil + } + if f.Type != 2 { + return wire.ErrMalformed + } + id, typ, qty, index := scalar(f.Value, 2), scalar(f.Value, 3), scalar(f.Value, 4), scalar(f.Value, 1) + if count == 0 || id != d.CostID || typ != d.CostType || qty == 0 || index == 0 || seen[index] || qty > math.MaxInt32 || total > math.MaxUint64-qty { + return fmt.Errorf("eventgames: invalid submitted cost") + } + seen[index] = true + total += qty + out = append(out, assets.Item{InvenIndex: index, ID: id, Type: typ, Count: qty}) + return nil + }) + if e != nil { + return nil, e + } + if d.Cost > 0 && count > math.MaxUint64/d.Cost || total != count*d.Cost { + return nil, fmt.Errorf("eventgames: cost does not match design") + } + return out, nil +} + +func stateWire(g GameState, d *gamedata.EventGame) []byte { + b := wire.AppendVarint(nil, 1, g.UID) + switch g.Type { + case 12: + b = wire.AppendVarint(b, 2, d.ScaffoldGroup) + b = wire.AppendVarint(b, 3, d.Cells[g.Position].ID) + b = wire.AppendVarint(b, 4, g.Clear) + case 13, 17: + b = wire.AppendVarint(b, 2, g.Clear) + for _, v := range g.Board { + b = wire.AppendVarint(b, 3, v) + } + for _, v := range g.Opened { + b = wire.AppendVarint(b, 4, v) + } + case 19: + b = wire.AppendVarint(b, 2, g.Free) + b = wire.AppendVarint(b, 3, g.Reset) + if g.Special { + b = wire.AppendVarint(b, 4, 1) } + b = wire.AppendVarint(b, 5, g.Tries) } - return true + return b } diff --git a/go/internal/server/eventgames/validation.go b/go/internal/server/domain/events/games/validation.go similarity index 98% rename from go/internal/server/eventgames/validation.go rename to go/internal/server/domain/events/games/validation.go index fa94b84..64e77e0 100644 --- a/go/internal/server/eventgames/validation.go +++ b/go/internal/server/domain/events/games/validation.go @@ -1,7 +1,7 @@ package eventgames import ( - "bd2server/internal/server/gamedata" + "bd2server/internal/server/design/gamedata" "fmt" "strconv" ) diff --git a/go/internal/server/events/native_costs.go b/go/internal/server/domain/events/native_costs.go similarity index 78% rename from go/internal/server/events/native_costs.go rename to go/internal/server/domain/events/native_costs.go index 0e6135c..c020162 100644 --- a/go/internal/server/events/native_costs.go +++ b/go/internal/server/domain/events/native_costs.go @@ -4,12 +4,13 @@ import ( "fmt" "math" - "bd2server/internal/server/gamedata" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" ) // Native shops' FreeJewelry prices use total diamonds: free first, then paid. // Cash conversion policy and explicitly paid-only prices keep their own rules. -func (e *Economy) NativePurchaseCosts(cost gamedata.Reward) ([]gamedata.Reward, error) { +func (e *Economy) NativePurchaseCosts(ctx command.Context, cost gamedata.Reward) ([]gamedata.Reward, error) { if cost.Count == 0 || cost.Count > math.MaxInt32 { return nil, fmt.Errorf("events: invalid native purchase cost") } @@ -19,7 +20,7 @@ func (e *Economy) NativePurchaseCosts(cost gamedata.Reward) ([]gamedata.Reward, if cost.ID != 0 { return nil, fmt.Errorf("events: diamond cost has an item ID") } - balance := e.wallet.Snapshot() + balance := e.wallet.Snapshot(ctx) free := min(cost.Count, balance.FreeJewelry) paid := cost.Count - free if paid > balance.Jewelry { diff --git a/go/internal/server/domain/events/owned_items.go b/go/internal/server/domain/events/owned_items.go new file mode 100644 index 0000000..30d8bda --- /dev/null +++ b/go/internal/server/domain/events/owned_items.go @@ -0,0 +1,3 @@ +package events + +func (e *Economy) AttachOwnedItemDesign(d map[uint64]map[uint64]bool) { e.ownedDesign = d } diff --git a/go/internal/server/eventplay/SPEC.md b/go/internal/server/domain/events/play/SPEC.md similarity index 100% rename from go/internal/server/eventplay/SPEC.md rename to go/internal/server/domain/events/play/SPEC.md diff --git a/go/internal/server/domain/events/play/battle.go b/go/internal/server/domain/events/play/battle.go new file mode 100644 index 0000000..aa50bb0 --- /dev/null +++ b/go/internal/server/domain/events/play/battle.go @@ -0,0 +1,12 @@ +package eventplay + +import ( + "bd2server/internal/server/design/gamedata" +) + +func (s *Service) HandlesBattle(mode uint64) bool { return mode == 17 } + +func (s *Service) AttachBattleChallenges(d gamedata.EventBattleChallenges) { + + s.battleChallenges = d +} diff --git a/go/internal/server/eventplay/field_binding.go b/go/internal/server/domain/events/play/field_binding.go similarity index 94% rename from go/internal/server/eventplay/field_binding.go rename to go/internal/server/domain/events/play/field_binding.go index 52a1f23..9d07d34 100644 --- a/go/internal/server/eventplay/field_binding.go +++ b/go/internal/server/domain/events/play/field_binding.go @@ -1,7 +1,7 @@ package eventplay import ( - "bd2server/internal/server/versionconfig" + "bd2server/internal/server/platform/versionconfig" "bytes" "encoding/json" "fmt" @@ -37,8 +37,7 @@ func (s *Service) AttachFieldBindingsFile(path string) error { if seed.Version != versionconfig.Current().GameVersion { return fmt.Errorf("eventplay: field binding version mismatch") } - s.mu.Lock() - defer s.mu.Unlock() + seen := map[int]bool{} for _, b := range seed.Bindings { pack, ok := s.design.FieldPacks[b.PackID] diff --git a/go/internal/server/domain/events/play/mini_event_hubs.go b/go/internal/server/domain/events/play/mini_event_hubs.go new file mode 100644 index 0000000..d5f4eab --- /dev/null +++ b/go/internal/server/domain/events/play/mini_event_hubs.go @@ -0,0 +1,33 @@ +package eventplay + +import ( + "bd2server/internal/server/design/gamedata" +) + +// miniHubSchedule is a project calendar identity; HubID is distinct from the +// pack ID used by EventPack schedules. +type miniHubBinding struct { + Slot, ContentType uint64 + UIDs []uint64 +} + +type miniHubSchedule struct { + UID, HubID uint64 + Start, PlayEnd, End int64 + Bindings []miniHubBinding +} + +// Hub content and EventType are separate client enums. MiniGame schedules +// are design identities; calendar SubID is zero for the supported slot routes. +func (s *Service) miniSlotSchedule(contentType, id uint64) (uint64, uint64, bool) { + eventType, known := gamedata.MiniHubEventType(contentType) + if !known { + return 0, 0, false + } + if contentType == 6 { + if _, err := s.design.Row("PackEventMiniGameTable", 8, id); err != nil { + return 0, 0, false + } + } + return eventType, 0, true +} diff --git a/go/internal/server/domain/events/play/play.go b/go/internal/server/domain/events/play/play.go new file mode 100644 index 0000000..5308b22 --- /dev/null +++ b/go/internal/server/domain/events/play/play.go @@ -0,0 +1,123 @@ +// Package eventplay owns event stories and client-simulated minigame runs. +package eventplay + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + "bd2server/internal/server/protocol/staticdata" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "fmt" + "math" + "strings" + + "time" +) + +type Rewards interface { + Apply(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) +} +type Run struct { + UID, Game, Stage, Mode, Started, Score, HP uint64 + Family, Session string + Objects, Skills []uint64 + Coin, Exp, Char, MapGroup uint64 + SkillLevels map[uint64]uint64 + MaxHP, Rerolls, Level, SkillCredits uint64 + Offers []uint64 + LevelExp uint64 + Killed, Picked, Available map[uint64]uint64 +} +type record struct { + UID, Game, Stage, Mode, Best, Paid, Date uint64 + Family string + Clears map[string]bool + PendingBundle []byte +} +type reply struct{ Request, Body []byte } +type snapshot struct { + Version int `json:"version"` + Records map[string]record `json:"records"` + Runs map[string]Run `json:"runs"` + Replies map[string]reply `json:"replies"` + Stories map[string]bool `json:"stories"` + Upgrades map[string]uint64 `json:"upgrades"` +} + +// FieldRuleLoader returns shared immutable field game rules. +type FieldRuleLoader func(uint64) (*gamedata.EventField, error) + +type Service struct { + fieldRules FieldRuleLoader + battleChallenges gamedata.EventBattleChallenges + + store stateio.Store + registry events.Resolver + economy Rewards + design *gamedata.EventPlayCatalog + state snapshot + now func() time.Time + rooms RoomRuntime + onProgress func(command.Context, uint64, uint64, uint64) error + hubCalendars *readonly.Seed + fieldBindings []FieldBinding +} + +func Open(ctx command.Context, store stateio.Store, design *gamedata.EventPlayCatalog, fieldRules FieldRuleLoader, registry events.Resolver, economy Rewards) (*Service, error) { + if store == nil || design == nil || fieldRules == nil || registry == nil || economy == nil { + return nil, fmt.Errorf("eventplay: incomplete configuration") + } + s := &Service{store: store, registry: registry, economy: economy, design: design, fieldRules: fieldRules, now: time.Now, state: snapshot{1, map[string]record{}, map[string]Run{}, map[string]reply{}, map[string]bool{}, map[string]uint64{}}} + raw, e := store.Load(ctx.State, "eventplay") + if e != nil { + return nil, e + } + if raw != nil { + if e = stateio.RequireExactJSONObject(raw, "version", "records", "runs", "replies", "stories", "upgrades"); e != nil { + return nil, e + } + if e = json.Unmarshal(raw, &s.state); e != nil { + return nil, e + } + if s.state.Version != 1 || s.state.Records == nil || s.state.Runs == nil || s.state.Replies == nil || s.state.Stories == nil || s.state.Upgrades == nil { + return nil, fmt.Errorf("eventplay: incompatible save") + } + } + for _, r := range s.state.Records { + if r.UID == 0 || r.Game == 0 || r.Best > math.MaxInt32 || r.Clears == nil { + return nil, fmt.Errorf("eventplay: invalid saved record") + } + if _, e = s.design.Row("PackEventMiniGameTable", 8, r.Game); e != nil { + return nil, e + } + } + return s, nil +} + +// AttachHubCalendars installs project-maintained hub layouts, including the +// separate play/end windows and slots referencing multiple domain identities. +func (s *Service) AttachHubCalendars(seed *readonly.Seed) { + s.hubCalendars = seed +} + +func (s *Service) grant(ctx command.Context, identity string, rewards []gamedata.BattleReward) ([]byte, error) { + rs := make([]gamedata.Reward, 0, len(rewards)) + for _, r := range rewards { + rs = append(rs, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 + } + return s.economy.Apply(ctx, identity, nil, rs) +} + +func family(path string) string { + p := strings.TrimPrefix(path, "/MiniGame") + for _, f := range []string{"Hopscotch", "Survival", "Defense", "Rhythm", "Sichuan", "Action", "Field", "Run"} { + if strings.HasPrefix(p, f) { + return f + } + } + return "" +} +func recKey(uid, stage, mode uint64, f string) string { + return fmt.Sprintf("%d:%s:%d:%d", uid, f, stage, mode) +} diff --git a/go/internal/server/domain/events/play/progress.go b/go/internal/server/domain/events/play/progress.go new file mode 100644 index 0000000..3dae2b8 --- /dev/null +++ b/go/internal/server/domain/events/play/progress.go @@ -0,0 +1,7 @@ +package eventplay + +import "bd2server/internal/server/domain/command" + +func (s *Service) AttachProgress(fn func(ctx command.Context, condition, sub, count uint64) error) { + s.onProgress = fn +} diff --git a/go/internal/server/domain/events/play/protocol_adapter.go b/go/internal/server/domain/events/play/protocol_adapter.go new file mode 100644 index 0000000..e89df66 --- /dev/null +++ b/go/internal/server/domain/events/play/protocol_adapter.go @@ -0,0 +1,1948 @@ +package eventplay + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + "bd2server/internal/server/protocol/wire" + "bytes" + "encoding/binary" + "encoding/json" + "fmt" + "math" + "slices" + "sort" + "strconv" + "strings" +) + +func (s *Service) upgrade(ctx command.Context, path string, req []byte, uid uint64, next *snapshot, key string) ([]byte, error) { + prefix := fmt.Sprintf("%d:", uid) + var costs, rewards []gamedata.Reward + if path == "/MiniGameSurvivalCharUpgrade" { + id, target := num(req, 2), num(req, 3) + k := prefix + strconv.FormatUint(id, 10) + if target == 0 || target != next.Upgrades[k]+1 { + return nil, fmt.Errorf("eventplay: upgrade level is not next") + } + var row []byte + for _, r := range s.design.Rows("FieldMiniGameUpgradeTable", 1, id) { + if num(r, 2) == target { + row = r + break + } + } + if row == nil { + return nil, fmt.Errorf("eventplay: upgrade design missing") + } + costs = []gamedata.Reward{{Type: 43, Count: num(row, 3)}} + if _, e := s.economy.Apply(ctx, "eventplay:"+key, costs, nil); e != nil { + return nil, e + } + next.Upgrades[k] = target + return nil, nil + } + for _, r := range s.design.Tables["FieldMiniGameUpgradeTable"] { + id, level := num(r, 1), num(r, 2) + if next.Upgrades[prefix+strconv.FormatUint(id, 10)] >= level { + rewards = append(rewards, gamedata.Reward{Type: 43, Count: num(r, 3)}) + } + } + bundle, e := s.economy.Apply(ctx, "eventplay:"+key, nil, rewards) + if e != nil { + return nil, e + } + for k := range next.Upgrades { + if len(k) >= len(prefix) && k[:len(prefix)] == prefix { + delete(next.Upgrades, k) + } + } + return bundle, nil +} + +func (s *Service) survivalProgress(req []byte, a *Run) (uint64, uint64, error) { + char, e := s.design.Row("FieldMiniGameCharTable", 13, a.Char) + if e != nil { + return 0, 0, e + } + var mapRow []byte + for _, r := range s.design.Rows("FieldMiniGameMapTable", 2, a.MapGroup) { + if num(r, 3) == a.Stage { + mapRow = r + } + } + if mapRow == nil { + return 0, 0, fmt.Errorf("eventplay: survival map missing") + } + monsters := s.design.Rows("FieldMiniGameMonsterTable", 2, num(mapRow, 4)) + if a.Killed == nil { + a.Killed = map[uint64]uint64{} + } + if a.Picked == nil { + a.Picked = map[uint64]uint64{} + } + if a.Available == nil { + a.Available = map[uint64]uint64{} + } + elapsed := uint64(0) + if uint64(s.now().UnixMilli()) > a.Started { + elapsed = (uint64(s.now().UnixMilli()) - a.Started) / 1000 + } + e = wire.Walk(req, func(f wire.Field) error { + if f.Number != 6 { + return nil + } + if f.Type != 2 { + return wire.ErrMalformed + } + id, count := num(f.Value, 1), num(f.Value, 2) + if count == 0 { + return fmt.Errorf("eventplay: empty kill count") + } + var row []byte + for _, r := range monsters { + if num(r, 5) == id { + row = r + break + } + } + if row == nil { + return fmt.Errorf("eventplay: monster outside survival map") + } + max := num(row, 12) + repeats := num(row, 11) + start := num(row, 16) + interval := num(row, 13) + if elapsed < start { + return fmt.Errorf("eventplay: monster not spawned yet") + } + waves := uint64(1) + if interval > 0 { + waves += (elapsed - start) / interval + } + if repeats > 0 && waves > repeats { + waves = repeats + } + if max == 0 || a.Killed[id] > max*waves || count > max*waves-a.Killed[id] { + return fmt.Errorf("eventplay: kill count exceeds designed spawns") + } + a.Killed[id] += count + box := num(row, 6) + if box != 0 { + items, err := s.boxItems(box) + if err != nil { + return err + } + for _, item := range items { + a.Available[item] += count + } + } + return nil + }) + if e != nil { + return 0, 0, e + } + exp, coin := uint64(0), uint64(0) + healing := uint64(0) + e = wire.Walk(req, func(f wire.Field) error { + if f.Number != 5 { + return nil + } + if f.Type != 2 { + return wire.ErrMalformed + } + id, count := num(f.Value, 1), num(f.Value, 2) + if count == 0 || count > a.Available[id] { + return fmt.Errorf("eventplay: item count exceeds committed drops") + } + row, e := s.design.Row("FieldMiniGameSurvivalItemTable", 1, id) + if e != nil { + return e + } + v := num(row, 4) + if count > math.MaxInt32 || v > math.MaxInt32/count { + return fmt.Errorf("eventplay: item value overflow") + } + a.Available[id] -= count + a.Picked[id] += count + switch num(row, 3) { + case 0: + healing += v * count + case 3: + exp += uint64(math.Floor(float64(v*count) * doubleField(char, 8, 1))) + case 4: + coin += uint64(math.Floor(float64(v*count) * doubleField(char, 10, 1))) + case 5: + a.SkillCredits += count + } + return nil + }) + if e != nil { + return 0, 0, e + } + boxes, e := list(req, 7) + if e != nil { + return 0, 0, e + } + for _, id := range boxes { + if _, e := s.design.Row("FieldMiniGameSurvivalBoxTable", 2, id); e != nil { + return 0, 0, e + } + } + hp := num(req, 2) + if hp > a.MaxHP || hp > a.HP+healing { + return 0, 0, fmt.Errorf("eventplay: invalid survival HP") + } + a.HP = hp + a.Exp += exp + a.Coin += coin + a.LevelExp += exp + if a.Exp > math.MaxInt32 || a.Coin > math.MaxInt32 { + return 0, 0, fmt.Errorf("eventplay: survival progress exceeds protocol range") + } + growth := num(char, 2) + for { + var current, next []byte + for _, r := range s.design.Rows("FieldMiniGameCharLevelTable", 1, growth) { + if num(r, 2) == a.Level { + current = r + } + if num(r, 2) == a.Level+1 { + next = r + } + } + if current == nil || next == nil || num(current, 3) == 0 || a.LevelExp < num(current, 3) { + break + } + a.LevelExp -= num(current, 3) + a.Level++ + a.SkillCredits++ + } + return exp, coin, nil +} + +func doubleField(b []byte, n int, def float64) float64 { + v := def + _ = wire.Walk(b, func(f wire.Field) error { + if f.Number == n && f.Type == 1 { + v = math.Float64frombits(binary.LittleEndian.Uint64(f.Value)) + } + return nil + }) + return v +} + +func (s *Service) boxItems(id uint64) ([]uint64, error) { + r, e := s.design.Row("FieldMiniGameSurvivalBoxTable", 2, id) + if e != nil { + return nil, e + } + values, e := list(r, 3) + if e != nil { + return nil, e + } + return values, nil +} + +func (s *Service) survivalOffers(a *Run) ([]byte, error) { + if a.SkillCredits == 0 { + return nil, fmt.Errorf("eventplay: no earned skill selection") + } + if len(a.Offers) > 0 { + if a.Rerolls == 0 { + return nil, fmt.Errorf("eventplay: no remaining rerolls") + } + a.Rerolls-- + } + a.Offers = nil + rows := append([][]byte(nil), s.design.Tables["FieldMiniGameSkillGroupTable"]...) + sort.Slice(rows, func(i, j int) bool { return num(rows[i], 1) < num(rows[j], 1) }) + for _, r := range rows { + id := num(r, 1) + if a.SkillLevels[id] < num(r, 2) { + a.Offers = append(a.Offers, id) + if len(a.Offers) == 3 { + break + } + } + } + out := wire.AppendVarint(nil, 1, a.Rerolls) + for _, id := range a.Offers { + out = wire.AppendVarint(out, 2, id) + } + return out, nil +} + +func summaryWire(st *snapshot, record bool) []byte { + keys := []string{} + for k, v := range st.Records { + if v.Best > 0 { + keys = append(keys, k) + } + } + sort.Strings(keys) + var out []byte + for _, k := range keys { + v := st.Records[k] + field := map[string]int{"Survival": 1, "Sichuan": 2, "Action": 3, "Rhythm": 5, "Hopscotch": 6}[v.Family] + if field == 0 { + continue + } + var b []byte + switch v.Family { + case "Survival": + b = wire.AppendVarint(nil, 2, v.Best) + case "Sichuan": + b = wire.AppendDouble(nil, 2, float64(v.Best)) + case "Action": + b = wire.AppendVarint(nil, 2, v.Best) + b = wire.AppendVarint(b, 3, v.Stage) + case "Rhythm": + b = wire.AppendVarint(nil, 2, v.Best) + b = wire.AppendVarint(b, 3, v.Stage) + case "Hopscotch": + b = wire.AppendVarint(nil, 1, v.Stage) + b = wire.AppendVarint(b, 3, v.Best) + } + if record { + switch v.Family { + case "Survival": + b = wire.AppendVarint(nil, 1, v.Best) + b = wire.AppendDouble(b, 2, 100) + case "Sichuan": + b = wire.AppendDouble(nil, 1, float64(v.Best)) + b = wire.AppendDouble(b, 2, 100) + case "Action": + b = wire.AppendVarint(nil, 1, v.Stage) + b = wire.AppendVarint(b, 2, v.Best) + b = wire.AppendDouble(b, 3, 100) + case "Rhythm": + b = wire.AppendVarint(nil, 1, v.Best) + b = wire.AppendVarint(b, 2, v.Stage) + b = wire.AppendDouble(b, 3, 100) + case "Hopscotch": + b = wire.AppendVarint(nil, 1, v.Stage) + b = wire.AppendVarint(b, 2, v.Best) + b = wire.AppendDouble(b, 4, 100) + } + } + out = wire.AppendBytes(out, field, b) + } + return out +} + +func (s *Service) endScore(f string, req []byte, a Run) (uint64, error) { + var score uint64 + switch f { + case "Run": + score = num(req, 3) + case "Field": + score = a.Score + case "Action": + score = num(req, 7) + case "Rhythm": + score = num(req, 4) + r, e := s.design.Row("RhythmGameMusicTable", 5, a.Stage) + if e != nil { + return 0, e + } + maxField := 8 + if a.Mode == 1 { //nolint:staticcheck // QF1003 + maxField = 2 + } else if a.Mode == 2 { + maxField = 1 + } + if score > num(r, maxField) { + return 0, fmt.Errorf("eventplay: rhythm score exceeds static maximum") + } + judgments := uint64(0) + err := wire.Walk(req, func(f wire.Field) error { + if f.Number == 3 { + if f.Type != 2 { + return wire.ErrMalformed + } + typ, count := num(f.Value, 1), num(f.Value, 2) + if typ > 10 || count > math.MaxInt32 { + return fmt.Errorf("eventplay: invalid note judgment") + } + judgments += count + } + return nil + }) + if err != nil { + return 0, err + } + noteField := 10 + if a.Mode > 0 { + noteField = 4 + } + if judgments > num(r, noteField) { + return 0, fmt.Errorf("eventplay: judgment count exceeds designed notes") + } + case "Sichuan": + score = num(req, 4) + case "Hopscotch": + score = num(req, 2) + if score > 10000 { + return 0, fmt.Errorf("eventplay: captured area exceeds 100 percent") + } + case "Survival": + score = num(req, 3) + d := s.design.Tables["FieldMiniGameSurvivalTable"][0] + if score > num(d, 24) { + return 0, fmt.Errorf("eventplay: survival time exceeds design") + } + case "Defense": + score = num(req, 3) + } + if score > math.MaxInt32 { + return 0, fmt.Errorf("eventplay: submitted score exceeds protocol range") + } + return score, nil +} + +func (s *Service) scoreRewards(game, score uint64) (uint64, []gamedata.BattleReward, error) { + d, e := s.design.Row("PackEventMiniGameTable", 8, game) + if e != nil { + return 0, nil, e + } + return s.design.ScoreRewards(num(d, 10), score) +} + +func (s *Service) progress(ctx command.Context, path string, req []byte, rk string, next *snapshot, key string) ([]byte, error) { + a, ok := next.Runs[rk] + if !ok { + return nil, fmt.Errorf("eventplay: no active run") + } + var out []byte + switch path { + case "/MiniGameFieldScore": + d, e := s.fieldRules(a.Game) + if e != nil { + return nil, e + } + ids, e := list(req, 3) + if e != nil || len(ids) == 0 { + return nil, fmt.Errorf("eventplay: missing field objects") + } + for _, id := range ids { + v, ok := d.Objects[id] + if !ok || has(a.Objects, id) { + return nil, fmt.Errorf("eventplay: object invalid or already scored") + } + a.Objects = append(a.Objects, id) + if v.Type == 1 { //nolint:staticcheck // QF1003 + a.Score += v.Point + } else if v.Type == 2 { + a.HP += v.Point + } + } + out = wire.AppendVarint(nil, 1, a.Score) + out = wire.AppendVarint(out, 2, a.HP) + case "/MiniGameSurvivalPlay": + exp, coin, e := s.survivalProgress(req, &a) + if e != nil { + return nil, e + } + if s.onProgress != nil { + total := uint64(0) + for id, count := range a.Killed { + previous := next.Runs[rk].Killed[id] + if count > previous { + total += count - previous + } + } + if total > 0 { + if e = s.onProgress(ctx, 319, a.Game, total); e != nil { + return nil, e + } + } + } + out = wire.AppendVarint(nil, 1, coin) + out = wire.AppendVarint(out, 2, exp) + case "/MiniGameSurvivalSkillSelectList": + var e error + out, e = s.survivalOffers(&a) + if e != nil { + return nil, e + } + case "/MiniGameSurvivalSkillUp": + id := num(req, 2) + if a.SkillCredits == 0 || !has(a.Offers, id) { + return nil, fmt.Errorf("eventplay: skill not earned or offered") + } + group, err := s.design.Row("FieldMiniGameSkillGroupTable", 1, id) + if err != nil { + return nil, fmt.Errorf("eventplay: invalid skill") + } + if item := num(req, 3); item != 0 { + if _, e := s.design.Row("FieldMiniGameSurvivalItemTable", 1, item); e != nil { + return nil, e + } + } + if a.SkillLevels == nil { + a.SkillLevels = map[uint64]uint64{} + } + level := a.SkillLevels[id] + 1 + if level > num(group, 2) { + return nil, fmt.Errorf("eventplay: survival skill at maximum") + } + capacity := uint64(0) + for _, r := range s.design.Tables["FieldMiniGameSurvivalTable"] { + if num(r, 21) > capacity { + capacity = num(r, 21) + } + } + if !has(a.Skills, id) && uint64(len(a.SkillLevels)) >= capacity { + return nil, fmt.Errorf("eventplay: survival skill capacity reached") + } + a.SkillLevels[id] = level + a.SkillCredits-- + a.Offers = nil + if level == num(group, 2) && s.onProgress != nil { + if e := s.onProgress(ctx, 318, a.Game, 1); e != nil { + return nil, e + } + if e := s.onProgress(ctx, 321, id, 1); e != nil { + return nil, e + } + } + if !has(a.Skills, id) { + a.Skills = append(a.Skills, id) + } + v := wire.AppendVarint(nil, 1, id) + v = wire.AppendVarint(v, 2, level) + out = wire.AppendBytes(nil, 1, v) + default: + if strings.Contains(path, "QuickReward") || strings.Contains(path, "Reward") { + return nil, fmt.Errorf("eventplay: reward requires a completed unpaid score record") + } + return nil, fmt.Errorf("eventplay: unsupported run transition") + } + next.Runs[rk] = a + _ = key + return out, nil +} + +// LockRoomRun installs the stage selected by native room matching before the +// client's HTTP Start. It does not award rewards or trust HTTP stage changes. +func (s *Service) LockRoomRun(ctx command.Context, session, family, guid string, uid, group, stage, monster uint64) error { + + if family != "Defense" && family != "Action" { + return fmt.Errorf("eventplay: invalid room family") + } + c, e := s.calendar(uid, family) + if e != nil { + return e + } + raw, _ := json.Marshal(s.state) + var next snapshot + _ = json.Unmarshal(raw, &next) + next.Runs[session+":"+family+":native"] = Run{UID: uid, Game: c.ID, Stage: stage, Mode: group, Char: monster, Family: family, Session: guid} + raw, _ = json.Marshal(next) + if e = s.store.Save(ctx.State, "eventplay", raw); e != nil { + return e + } + s.state = next + return nil +} + +func specializedInfo(st *snapshot, uid uint64, f string) []byte { + var out []byte + if f == "Action" || f == "Survival" || f == "Defense" { + out = wire.AppendVarint(out, 1, uid) + } + for _, r := range st.Records { + if r.UID != uid || r.Family != f { + continue + } + switch f { + case "Sichuan": + v := wire.AppendVarint(nil, 1, uid) + v = wire.AppendVarint(v, 6, r.Best) + out = wire.AppendBytes(out, 1, v) + case "Rhythm": + v := wire.AppendVarint(nil, 1, r.Stage) + v = wire.AppendVarint(v, 2, r.Mode) + v = wire.AppendVarint(v, 3, r.Best) + out = wire.AppendBytes(out, 1, v) + case "Action": + v := wire.AppendVarint(nil, 1, r.Stage) + v = wire.AppendVarint(v, 2, r.Best) + out = wire.AppendBytes(out, 2, v) + case "Hopscotch": + v := wire.AppendVarint(nil, 1, r.Stage) + if r.Best > 0 { + v = wire.AppendVarint(v, 2, 1) + } + out = wire.AppendBytes(out, 2, v) + case "Survival": + out = wire.AppendVarint(out, 4, r.Best) + for stage, clear := range r.Clears { + if !clear { + continue + } + var group, id uint64 + if _, e := fmt.Sscanf(stage, "map:%d:%d", &group, &id); e == nil { + v := wire.AppendVarint(nil, 1, group) + v = wire.AppendVarint(v, 2, id) + out = wire.AppendBytes(out, 8, v) + } + } + } + } + if f == "Survival" { + prefix := fmt.Sprintf("%d:", uid) + for key, level := range st.Upgrades { + if !strings.HasPrefix(key, prefix) { + continue + } + var id uint64 + _, _ = fmt.Sscanf(key[len(prefix):], "%d", &id) + v := wire.AppendVarint(nil, 1, id) + v = wire.AppendVarint(v, 2, level) + out = wire.AppendBytes(out, 10, v) + } + } + return out +} + +func rankingWire(st *snapshot, uid uint64, f string, record bool) []byte { + keys := []string{} + for k, r := range st.Records { + if r.UID == uid && r.Family == f && r.Best > 0 { + keys = append(keys, k) + } + } + sort.Strings(keys) + var out []byte + for _, k := range keys { + r := st.Records[k] + var b []byte + switch f { + case "Survival": + b = wire.AppendVarint(nil, 1, 1) + b = wire.AppendVarint(b, 4, r.Best) + if record { + b = wire.AppendVarint(nil, 1, r.Best) + b = wire.AppendDouble(b, 2, 100) + } + case "Hopscotch": + b = wire.AppendVarint(nil, 1, 1) + b = wire.AppendVarint(b, 4, r.Best) + if record { + b = wire.AppendVarint(nil, 1, r.Stage) + b = wire.AppendVarint(b, 2, r.Best) + b = wire.AppendDouble(b, 4, 100) + } + case "Action": + b = wire.AppendVarint(nil, 3, 1) + b = wire.AppendVarint(b, 4, r.Best) + if record { + b = wire.AppendVarint(nil, 1, r.Stage) + b = wire.AppendVarint(b, 2, r.Best) + b = wire.AppendDouble(b, 3, 100) + } + case "Sichuan": + b = wire.AppendVarint(nil, 7, 1) + b = wire.AppendVarint(b, 8, r.Best) + if record { + b = wire.AppendVarint(nil, 1, r.Best) + b = wire.AppendDouble(b, 2, 100) + } + case "Rhythm": + b = wire.AppendVarint(nil, 1, 1) + b = wire.AppendVarint(b, 4, r.Best) + if record { + b = wire.AppendVarint(nil, 1, r.Best) + b = wire.AppendVarint(b, 2, r.Stage) + b = wire.AppendDouble(b, 3, 100) + } + default: + b = wire.AppendVarint(nil, 1, 1) + b = wire.AppendVarint(b, 4, r.Best) + } + out = wire.AppendBytes(out, 1, b) + } + return out +} + +func (s *Service) claimScore(ctx command.Context, path string, uid uint64, f string, next *snapshot, key string) ([]byte, error) { + for k, r := range next.Records { + if r.UID != uid || r.Family != f || r.Best == 0 { + continue + } + point, rewards, e := s.scoreRewards(r.Game, r.Best) + if e != nil { + return nil, e + } + day := uint64(s.now().UnixMilli()) / 86400000 + if r.Date != day { + r.Paid = 0 + } + if point <= r.Paid { + return nil, fmt.Errorf("eventplay: daily best reward already paid") + } + _, old, e := s.scoreRewards(r.Game, r.Paid) + if e != nil { + return nil, e + } + bundle, e := s.grant(ctx, "eventplay:"+key, rewardDifference(old, rewards)) + if e != nil { + return nil, e + } + r.Paid = point + r.Date = day + next.Records[k] = r + if strings.HasSuffix(path, "QuickReward") { + return wire.AppendBytes(nil, 1, bundle), nil + } + out := wire.AppendVarint(nil, 1, point) + return wire.AppendBytes(out, 2, bundle), nil + } + return nil, fmt.Errorf("eventplay: no completed score to reward") +} + +func (s *Service) AssociatedMissionGroup(c events.Schedule) (uint64, error) { + if c.Type != 11 { + return 0, nil + } + game, e := s.design.Row("PackEventMiniGameTable", 8, c.ID) + if e != nil { + return 0, e + } + switch num(game, 3) { + case 5: + row, e := s.design.Row("FieldMiniGameSurvivalTable", 9, num(game, 4)) + if e != nil { + return 0, e + } + return num(row, 8), nil + case 7: + if len(s.design.Tables["MGDDefaultTable"]) > 0 { + return num(s.design.Tables["MGDDefaultTable"][0], 5), nil + } + case 8: + if len(s.design.Tables["ActionGameDefaultTable"]) > 0 { + return num(s.design.Tables["ActionGameDefaultTable"][0], 8), nil + } + case 12: + if len(s.design.Tables["HopscotchDefaultTable"]) > 0 { + return num(s.design.Tables["HopscotchDefaultTable"][0], 20), nil + } + } + return 0, nil +} + +func num(b []byte, n int) uint64 { v, _, _ := wire.Varint(b, n); return v } + +func list(b []byte, n int) ([]uint64, error) { + var out []uint64 + e := wire.Walk(b, func(f wire.Field) error { + if f.Number != n { + return nil + } + if f.Type == 0 { + v, _ := binary.Uvarint(f.Value) + out = append(out, v) + return nil + } + if f.Type != 2 { + return wire.ErrMalformed + } + for p := f.Value; len(p) > 0; { + v, k := binary.Uvarint(p) + if k <= 0 { + return wire.ErrMalformed + } + out = append(out, v) + p = p[k:] + } + return nil + }) + return out, e +} + +func (s *Service) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + code, ok := codes[path] + if !ok { + return 0, nil, false, nil + } + + fail := func(e error) (int, []byte, bool, error) { return code, nil, true, e } + seq := num(req, 1) + if seq == 0 || seq > math.MaxInt32 { + return fail(fmt.Errorf("eventplay: invalid sequence")) + } + if e := wire.Walk(req, func(f wire.Field) error { + if f.Type == 0 { + v, _ := binary.Uvarint(f.Value) + if v > math.MaxInt64 { + return wire.ErrMalformed + } + } + return nil + }); e != nil { + return fail(e) + } + // Public hub calendars are derived afresh, never stored as account replies. + if path == "/EventHubInfo" || path == "/MiniEventHubInfo" || path == "/MiniGameHubInfo" { + if s.hubCalendars != nil && path == "/MiniGameHubInfo" { + return s.hubCalendars.Handle(ctx, path, req) + } + if s.hubCalendars != nil && path == "/EventHubInfo" { + out, err := s.publicEventHubs(ctx, req) + return code, out, true, err + } + if path == "/MiniEventHubInfo" { + out, err := s.miniEventHubs(ctx, req) + return code, out, true, err + } + return code, nil, true, nil + } + key := fmt.Sprintf("%s:%s:%d", commandSession, path, seq) + if r, ok := s.state.Replies[key]; ok { + if !bytes.Equal(r.Request, req) { + return fail(fmt.Errorf("eventplay: changed request retry")) + } + return code, r.Body, true, nil + } + raw, _ := json.Marshal(s.state) + var next snapshot + _ = json.Unmarshal(raw, &next) + var out []byte + var e error + switch path { + case "/MiniGameUserRecordInfo", "/MiniGameRanking": + out = summaryWire(&next, path == "/MiniGameUserRecordInfo") + case "/MiniGameRelayServerInfo": + return fail(fmt.Errorf("eventplay: native relay channels are not configured")) + + case "/PackEventStoryInfo", "/PackEventBattleInfo": + uids, err := list(req, 2) + if err != nil { + return fail(err) + } + for _, uid := range uids { + c, err := s.registry.Resolve(uid) + if err != nil { + return fail(err) + } + if path == "/PackEventStoryInfo" { + for _, r := range s.design.Rows("PackEventStoryTable", 1, c.ID) { + id := num(r, 2) + if next.Stories[fmt.Sprintf("%d:%d:%d", uid, c.ID, id)] { + v := wire.AppendVarint(nil, 1, uid) + v = wire.AppendVarint(v, 2, c.ID) + v = wire.AppendVarint(v, 3, id) + out = wire.AppendBytes(out, 1, v) + } + } + } else { + for _, r := range s.design.Rows("PackEventBattleTable", 4, c.ID) { + id := num(r, 5) + if !next.Stories[fmt.Sprintf("battle:%d:%d:%d", uid, c.ID, id)] { + continue + } + v := wire.AppendVarint(nil, 1, uid) + v = wire.AppendVarint(v, 2, c.ID) + v = wire.AppendVarint(v, 3, id) + for _, idx := range s.battleChallengeIndexesLocked(uid, c.ID, id) { + v = wire.AppendVarint(v, 4, idx) + } + out = wire.AppendBytes(out, 1, v) + } + } + } + case "/PackEventStoryClear", "/PackEventStoryReplayClear": + uid, group, id := num(req, 2), num(req, 3), num(req, 4) + if path == "/PackEventStoryReplayClear" { + group, id = num(req, 2), num(req, 3) + uid = 0 + } + if uid != 0 { + c, err := s.registry.Resolve(uid) + if err != nil { + return fail(err) + } + if c.ID != group || s.now().UnixMilli() < c.Start || s.now().UnixMilli() >= c.End { + return fail(fmt.Errorf("eventplay: story schedule unavailable")) + } + } + r, err := s.design.Story(group, id) + if err != nil { + return fail(err) + } + storyKey := fmt.Sprintf("%d:%d:%d", uid, group, id) + if !next.Stories[storyKey] { + rs, err := gamedata.EventPlayRewards(r, 11, 10, 9) + if err != nil { + return fail(err) + } + out, e = s.grant(ctx, "eventplay:story:"+storyKey, rs) + if e != nil { + return fail(e) + } + out = wire.AppendBytes(nil, 1, out) + next.Stories[storyKey] = true + } + if path == "/PackEventStoryClear" { + v := wire.AppendVarint(nil, 1, uid) + v = wire.AppendVarint(v, 2, group) + v = wire.AppendVarint(v, 3, id) + out = wire.AppendBytes(out, 2, v) + } + default: + out, e = s.game(ctx, path, req, commandSession, &next, key) + } + if e != nil { + return fail(e) + } + next.Replies[key] = reply{append([]byte(nil), req...), out} + b, e := json.Marshal(next) + if e != nil { + return fail(e) + } + if e = s.store.Save(ctx.State, "eventplay", b); e != nil { + return fail(e) + } + s.state = next + return code, out, true, nil +} + +func recordWire(r record) []byte { + b := wire.AppendVarint(nil, 1, r.UID) + b = wire.AppendVarint(b, 2, r.Paid) + b = wire.AppendVarint(b, 3, r.Best) + if r.Best > 0 && r.Paid < r.Best { + b = wire.AppendVarint(b, 4, 1) + } + return b +} + +func (s *Service) calendar(uid uint64, f string) (events.Schedule, error) { + if uid == 0 { + for _, c := range s.registry.List() { + if c.Type == 11 { + d, e := s.design.Row("PackEventMiniGameTable", 8, c.ID) + if e != nil { + continue + } + kind := num(d, 3) + expected := map[string]uint64{"Field": 1, "Run": 3, "Survival": 5, "Sichuan": 6, "Defense": 7, "Action": 8, "Rhythm": 9, "Hopscotch": 12}[f] + if f == "Run" && (kind == 2 || kind == 4) { + expected = kind + } + if kind != expected { + continue + } + uid = c.UID + break + } + } + } + c, e := s.registry.Resolve(uid) + if e != nil { + return c, e + } + if c.Type != 11 { + return c, fmt.Errorf("eventplay: minigame schedule mismatch") + } + if _, e = s.design.Row("PackEventMiniGameTable", 8, c.ID); e != nil { + return c, e + } + d, _ := s.design.Row("PackEventMiniGameTable", 8, c.ID) + kind := num(d, 3) + expected := map[string]uint64{"Field": 1, "Run": 3, "Survival": 5, "Sichuan": 6, "Defense": 7, "Action": 8, "Rhythm": 9, "Hopscotch": 12}[f] + if f == "Run" && (kind == 2 || kind == 4) { + expected = kind + } + if kind != expected { + return c, fmt.Errorf("eventplay: configured minigame family mismatch") + } + return c, nil +} + +func (s *Service) game(ctx command.Context, path string, req []byte, session string, next *snapshot, key string) ([]byte, error) { + f := family(path) + if f == "" { + return nil, fmt.Errorf("eventplay: unknown family") + } + start := strings.HasSuffix(path, "Start") + end := strings.HasSuffix(path, "End") + runKey := session + ":" + f + uid := num(req, 2) + if end || path == "/MiniGameFieldScore" || path == "/MiniGameSurvivalPlay" || strings.Contains(path, "Skill") { + if a, ok := next.Runs[runKey]; ok { + uid = a.UID + } else { + return nil, fmt.Errorf("eventplay: no active run") + } + } + c, e := s.calendar(uid, f) + if e != nil { + return nil, e + } + uid = c.UID + stage, mode := num(req, 3), num(req, 4) + if f == "Rhythm" { + mode = num(req, 5) + } + if f == "Survival" { + stage = num(req, 5) + } + if f == "Sichuan" { + stage = num(req, 4) + } + rk := recKey(uid, stage, mode, f) + r := next.Records[rk] + if r.Clears == nil { + r = record{UID: uid, Game: c.ID, Stage: stage, Mode: mode, Family: f, Clears: map[string]bool{}} + } + if path == "/MiniGameSurvivalCharUpgrade" || path == "/MiniGameSurvivalCharUpgradeReset" { + return s.upgrade(ctx, path, req, uid, next, key) + } + if path == "/MiniGameDefenseMatching" { + matching := num(req, 3) + if matching > 2 { + return nil, fmt.Errorf("eventplay: unknown matching mode") + } + next.Runs[runKey+":matching"] = Run{UID: uid, Game: c.ID, Family: f, Session: session} + return nil, nil + } + if start { + if s.now().UnixMilli() < c.Start || s.now().UnixMilli() >= c.End { + return nil, fmt.Errorf("eventplay: season closed") + } + if _, active := next.Runs[runKey]; active { + return nil, fmt.Errorf("eventplay: run already active") + } + if e = s.validateStage(f, stage, mode); e != nil { + return nil, e + } + a := Run{UID: uid, Game: c.ID, Stage: stage, Mode: mode, Started: uint64(s.now().UnixMilli()), Family: f, Session: session, HP: 100} + if f == "Action" { + game, _ := s.design.Row("PackEventMiniGameTable", 8, c.ID) + group := num(game, 4) + rows := s.design.Rows("ActionGameStageTable", 3, group) + if len(rows) == 0 { + return nil, fmt.Errorf("eventplay: action stage group missing") + } + a.Stage = num(rows[0], 4) + a.Mode = group + a.Char = num(rows[0], 1) + } + if f == "Action" || f == "Defense" { + if locked, ok := next.Runs[runKey+":native"]; ok { + a.Stage, a.Mode, a.Char = locked.Stage, locked.Mode, locked.Char + a.Session = locked.Session + if s.rooms != nil { + if e = s.rooms.ValidateRoom(session, a.Session, uid); e != nil { + return nil, e + } + } + } + } + var startReply []byte + if f == "Field" { + d, e := s.fieldRules(c.ID) + if e != nil { + return nil, e + } + a.HP = d.HP + ids := make([]uint64, 0, len(d.Objects)) + for id := range d.Objects { + ids = append(ids, id) + } + slices.Sort(ids) + for _, id := range ids { + startReply = wire.AppendVarint(startReply, 1, id) + } + } + if f == "Survival" { + a.Char = num(req, 3) + a.MapGroup = num(req, 4) + char, e := s.design.Row("FieldMiniGameCharTable", 13, a.Char) + if e != nil { + return nil, e + } + a.HP = num(char, 20) + a.MaxHP = a.HP + a.Rerolls = num(char, 19) + a.Level = 1 + a.SkillCredits = 0 + a.Killed = map[uint64]uint64{} + a.Picked = map[uint64]uint64{} + a.Available = map[uint64]uint64{} + skillGroup := num(char, 3) + skills := s.design.Rows("FieldMiniGameSkillTable", 7, skillGroup) + if len(skills) == 0 { + return nil, fmt.Errorf("eventplay: character starting skill missing") + } + a.Skills = []uint64{num(skills[0], 8)} + a.SkillLevels = map[uint64]uint64{skillGroup: 1} + found := false + for _, m := range s.design.Rows("FieldMiniGameMapTable", 2, a.MapGroup) { + if num(m, 3) == a.Stage { + found = true + } + } + if !found { + return nil, fmt.Errorf("eventplay: survival map missing") + } + } + next.Runs[runKey] = a + if s.onProgress != nil { + if e = s.onProgress(ctx, 343, c.ID, 1); e != nil { + return nil, e + } + if e = s.onProgress(ctx, 315, c.ID, 1); e != nil { + return nil, e + } + } + next.Records[rk] = r + if f == "Survival" { + v := wire.AppendVarint(nil, 1, a.Skills[0]) + v = wire.AppendVarint(v, 2, 1) + out := wire.AppendBytes(nil, 1, v) + return wire.AppendVarint(out, 2, seqSeed(a.Started)), nil + } + return startReply, nil + } + if end { + a := next.Runs[runKey] + if a.UID != uid { + return nil, fmt.Errorf("eventplay: run schedule changed") + } + if f == "Action" && (num(req, 3) != a.Mode || num(req, 4) != a.Stage || num(req, 6) != a.Char) { + return nil, fmt.Errorf("eventplay: action submitted stage/monster differs from configured start") + } + if f == "Action" || f == "Defense" { + guid, _, err := wire.Bytes(req, 2) + if err != nil { + return nil, err + } + if a.Session != session && !bytes.Equal(guid, []byte(a.Session)) { + return nil, fmt.Errorf("eventplay: native room guid mismatch") + } + if s.rooms != nil { + if e = s.rooms.ValidateRoom(session, string(guid), uid); e != nil { + return nil, e + } + } + } + score, e := s.endScore(f, req, a) + if e != nil { + return nil, e + } + if s.onProgress != nil { + switch f { + case "Survival": + for condition, count := range map[uint64]uint64{316: a.Level, 317: score} { + if e = s.onProgress(ctx, condition, c.ID, count); e != nil { + return nil, e + } + } + if num(req, 2) == 0 { + if e = s.onProgress(ctx, 320, a.Stage, 1); e != nil { + return nil, e + } + } + case "Defense": + if e = s.onProgress(ctx, 322, c.ID, score); e != nil { + return nil, e + } + case "Action": + if e = s.onProgress(ctx, 323, a.Char, 1); e != nil { + return nil, e + } + for condition, field := range map[uint64]int{326: 9, 327: 10, 329: 12, 330: 13, 333: 14, 334: 15, 335: 16, 336: 17} { + if count := num(req, field); count > 0 { + if e = s.onProgress(ctx, condition, a.Char, count); e != nil { + return nil, e + } + } + } + } + } + rk = recKey(uid, a.Stage, a.Mode, f) + r = next.Records[rk] + if r.Clears == nil { + r.Clears = map[string]bool{} + } + if score > r.Best { + r.Best = score + } + var out []byte + point, rewards, e := s.scoreRewards(c.ID, score) + if e != nil { + return nil, e + } + if f == "Defense" { + for _, row := range s.design.Tables["MGDRewardTable"] { + id, wave := num(row, 1), num(row, 6) + ck := fmt.Sprintf("wave:%d", id) + if wave <= score && !r.Clears[ck] { + rr, err := gamedata.EventPlayRewards(row, 5, 3, 2) + if err != nil { + return nil, err + } + b, err := s.grant(ctx, "eventplay:"+key+":"+ck, rr) + if err != nil { + return nil, err + } + out = wire.AppendBytes(out, 1, b) + out = wire.AppendVarint(out, 2, id) + r.Clears[ck] = true + } + } + } + if f == "Survival" { + if num(req, 5) > a.Coin || num(req, 6) > a.Exp { + return nil, fmt.Errorf("eventplay: end coin/exp exceeds committed play progress") + } + out = wire.AppendVarint(out, 2, a.Coin) + out = wire.AppendVarint(out, 3, a.Exp) + if num(req, 2) == 0 { + r.Clears[fmt.Sprintf("map:%d:%d", a.MapGroup, a.Stage)] = true + for _, m := range s.design.Rows("FieldMiniGameMapTable", 2, a.MapGroup) { + if num(m, 3) != a.Stage { + continue + } + rr, err := gamedata.EventPlayRewards(m, 8, 7, 6) + if err != nil { + return nil, err + } + if len(rr) > 0 { + b, err := s.grant(ctx, fmt.Sprintf("eventplay:map:%d:%d:%d", uid, a.MapGroup, a.Stage), rr) + if err != nil { + return nil, err + } + out = wire.AppendBytes(out, 1, b) + } + } + } + if a.Coin > 0 { + b, err := s.grant(ctx, "eventplay:"+key+":coin", []gamedata.BattleReward{{Type: 43, Count: a.Coin}}) + if err != nil { + return nil, err + } + out = wire.AppendBytes(out, 1, b) + } + } + day := uint64(s.now().UnixMilli()) / 86400000 + if r.Date != day { + r.Paid = 0 + } + if point > r.Paid { + _, previous, e := s.scoreRewards(c.ID, r.Paid) + if e != nil { + return nil, e + } + bundle, e := s.grant(ctx, "eventplay:"+key, rewardDifference(previous, rewards)) + if e != nil { + return nil, e + } + field := 1 + if f == "Run" { //nolint:staticcheck // QF1003 + field = 3 + } else if f == "Field" { + field = 5 + } else if f == "Action" { + field = 3 + } else if f == "Hopscotch" { + field = 4 + } + if f == "Rhythm" { + r.PendingBundle = append(r.PendingBundle, bundle...) + } else { + out = wire.AppendBytes(out, field, bundle) + } + r.Paid = point + r.Date = day + } + switch f { + case "Run": + out = wire.AppendVarint(out, 1, r.Best) + out = wire.AppendVarint(out, 2, r.Paid) + case "Field": + out = wire.AppendVarint(out, 1, uid) + out = wire.AppendVarint(out, 2, score) + out = wire.AppendVarint(out, 3, r.Best) + out = wire.AppendVarint(out, 4, r.Paid) + case "Sichuan": + out = wire.AppendDouble(out, 4, float64(r.Best)) + case "Rhythm": + v := wire.AppendVarint(nil, 1, a.Stage) + v = wire.AppendVarint(v, 2, a.Mode) + v = wire.AppendVarint(v, 3, r.Best) + v = wire.AppendVarint(v, 4, num(req, 2)) + v = wire.AppendVarint(v, 5, num(req, 6)) + out = wire.AppendBytes(out, 1, v) + case "Action": + out = wire.AppendVarint(out, 1, r.Best) + case "Hopscotch": + v := wire.AppendVarint(nil, 1, a.Stage) + if score > 0 { + v = wire.AppendVarint(v, 2, 1) + } + out = wire.AppendBytes(out, 2, v) + } + next.Records[rk] = r + delete(next.Runs, runKey) + if s.rooms != nil && (f == "Action" || f == "Defense") { + if e = s.rooms.CompleteRoom(session, a.Session, score); e != nil { + return nil, e + } + } + return out, nil + } + if strings.Contains(path, "Ranking") || strings.Contains(path, "RecordInfo") { + return rankingWire(next, uid, f, strings.Contains(path, "RecordInfo")), nil + } + if strings.HasSuffix(path, "Info") { + var out []byte + if f == "Rhythm" { + for k, v := range next.Records { + if v.UID == uid && v.Family == f && len(v.PendingBundle) > 0 { + out = wire.AppendBytes(out, 2, v.PendingBundle) + v.PendingBundle = nil + next.Records[k] = v + } + } + } + if f != "Run" && f != "Field" { + if _, exists := next.Records[rk]; !exists { + next.Records[rk] = r + } + return append(out, specializedInfo(next, uid, f)...), nil + } + for _, v := range next.Records { + if v.UID == uid && v.Family == f { + out = wire.AppendBytes(out, 1, recordWire(v)) + } + } + if len(out) == 0 { + next.Records[rk] = r + out = wire.AppendBytes(out, 1, recordWire(r)) + } + return out, nil + } + if strings.Contains(path, "QuickReward") || path == "/MiniGameFieldReward" { + return s.claimScore(ctx, path, uid, f, next, key) + } + return s.progress(ctx, path, req, runKey, next, key) +} + +// ListMiniContentRoutes exposes the same validated hub-scoped calendar used +// for presentation, including routes whose play period has already ended. +func (s *Service) ListMiniContentRoutes(ctx command.Context) ([]gamedata.MiniContentRoute, error) { + body, err := s.miniEventHubs(ctx, wire.AppendVarint(nil, 1, 1)) + if err != nil { + return nil, err + } + var routes []gamedata.MiniContentRoute + seen := map[uint64]bool{} + err = wire.Walk(body, func(hub wire.Field) error { + if hub.Number != 1 || hub.Type != 2 { + return nil + } + return wire.Walk(hub.Value, func(slot wire.Field) error { + if slot.Number != 6 || slot.Type != 2 { + return nil + } + typ := num(slot.Value, 2) + if typ != 13 && typ != 14 { + return nil + } + uid := num(slot.Value, 4) + if seen[uid] { + return fmt.Errorf("eventplay: ambiguous mini content UID %d", uid) + } + seen[uid] = true + routes = append(routes, gamedata.MiniContentRoute{UID: uid, ContentType: typ, ContentID: num(slot.Value, 3), Start: int64(num(slot.Value, 5)), End: int64(num(slot.Value, 6))}) + return nil + }) + }) + if err != nil { + return nil, err + } + sort.Slice(routes, func(i, j int) bool { return routes[i].UID < routes[j].UID }) + return routes, nil +} + +// ResolveMiniContentUID resolves hub-scoped story/quiz identities without +// manufacturing EventScheduleInfo rows. Callers enforce the returned window. +func (s *Service) ResolveMiniContentUID(ctx command.Context, uid uint64) (contentType, contentID uint64, start, end int64, err error) { + if uid == 0 { + return 0, 0, 0, 0, fmt.Errorf("eventplay: missing mini content UID") + } + body, err := s.miniEventHubs(ctx, wire.AppendVarint(nil, 1, 1)) + if err != nil { + return 0, 0, 0, 0, err + } + found := false + err = wire.Walk(body, func(hub wire.Field) error { + if hub.Number != 1 || hub.Type != 2 { + return nil + } + return wire.Walk(hub.Value, func(slot wire.Field) error { + if slot.Number != 6 || slot.Type != 2 || num(slot.Value, 4) != uid { + return nil + } + typ := num(slot.Value, 2) + if typ != 13 && typ != 14 { + return nil + } + if found { + return fmt.Errorf("eventplay: ambiguous mini content UID %d", uid) + } + found = true + contentType, contentID = typ, num(slot.Value, 3) + start, end = int64(num(slot.Value, 5)), int64(num(slot.Value, 6)) + return nil + }) + }) + if err != nil { + return 0, 0, 0, 0, err + } + if !found { + return 0, 0, 0, 0, fmt.Errorf("eventplay: mini content UID %d unavailable", uid) + } + return contentType, contentID, start, end, nil +} + +func (s *Service) miniEventHubs(ctx command.Context, req []byte) ([]byte, error) { + var hubs []miniHubSchedule + if s.hubCalendars == nil { + return nil, nil + } + _, body, handled, err := s.hubCalendars.Handle(ctx, "/EventHubInfo", req) + if err != nil { + return nil, err + } + if handled { + err = wire.Walk(body, func(f wire.Field) error { + if f.Number != 1 || f.Type != 2 { + return nil + } + hub := miniHubSchedule{UID: num(f.Value, 1), HubID: num(f.Value, 2), Start: int64(num(f.Value, 3)), PlayEnd: int64(num(f.Value, 4)), End: int64(num(f.Value, 5))} + if err := wire.Walk(f.Value, func(setting wire.Field) error { + if setting.Number != 6 || setting.Type != 2 { + return nil + } + uids, err := list(setting.Value, 3) + if err != nil { + return err + } + hub.Bindings = append(hub.Bindings, miniHubBinding{num(setting.Value, 1), num(setting.Value, 2), uids}) + return nil + }); err != nil { + return err + } + hubs = append(hubs, hub) + return nil + }) + if err != nil { + return nil, err + } + } + sort.Slice(hubs, func(i, j int) bool { return hubs[i].UID < hubs[j].UID }) + rows := s.registry.List() + var out []byte + for _, hub := range hubs { + table, err := s.design.Row("PackEventHubTable", 14, hub.HubID) + if err != nil || num(table, 13) != 1 { + continue + } + v := wire.AppendVarint(nil, 1, hub.UID) + v = wire.AppendVarint(v, 2, hub.HubID) + v = wire.AppendVarint(v, 3, uint64(hub.Start)) + v = wire.AppendVarint(v, 4, uint64(hub.PlayEnd)) + v = wire.AppendVarint(v, 5, uint64(hub.End)) + slots := s.design.Rows("PackEventListTable", 6, hub.HubID) + sort.Slice(slots, func(i, j int) bool { return num(slots[i], 10) < num(slots[j], 10) }) + for _, binding := range hub.Bindings { + found := false + for _, slot := range slots { + if binding.Slot == num(slot, 11) { + found = true + break + } + } + if !found { + return nil, fmt.Errorf("eventplay: mini hub %d static slot index %d missing", hub.UID, binding.Slot) + } + } + for _, slot := range slots { + contentType, contentID := num(slot, 9), num(slot, 7) + bound := map[uint64]bool{} + for _, binding := range hub.Bindings { + if binding.Slot == num(slot, 11) { + if binding.ContentType != contentType { + return nil, fmt.Errorf("eventplay: mini hub %d slot %d content type mismatch", hub.UID, num(slot, 10)) + } + for _, uid := range binding.UIDs { + bound[uid] = true + } + } + } + if len(bound) == 0 { + continue + } + // Mini hub stories and NPC quizzes have hub-scoped UIDs, rather + // than entries in EventScheduleInfo. Their explicit binding and + // static EndDateType supply the calendar; never invent a global + // event type for these separate client protocols. + if contentType == 13 || contentType == 14 { + if len(bound) != 1 || bound[0] { + return nil, fmt.Errorf("eventplay: mini hub %d slot %d requires one nonzero hub-scoped UID", hub.UID, num(slot, 10)) + } + var uid uint64 + for id := range bound { + uid = id + } + for _, row := range rows { + if row.UID == uid { + return nil, fmt.Errorf("eventplay: mini hub %d slot %d hub-scoped UID collides with global schedule", hub.UID, num(slot, 10)) + } + } + end := hub.End + if num(slot, 4) == 0 { + end = hub.PlayEnd + } + if end <= hub.Start { + continue + } + b := wire.AppendVarint(nil, 1, num(slot, 10)) + b = wire.AppendVarint(b, 2, contentType) + b = wire.AppendVarint(b, 3, contentID) + b = wire.AppendVarint(b, 4, uid) + b = wire.AppendVarint(b, 5, uint64(hub.Start)) + b = wire.AppendVarint(b, 6, uint64(end)) + v = wire.AppendBytes(v, 6, b) + continue + } + eventType, subID, known := s.miniSlotSchedule(contentType, contentID) + if !known { + return nil, fmt.Errorf("eventplay: mini hub %d slot %d content type %d unsupported", hub.UID, num(slot, 10), contentType) + } + for uid := range bound { + found := false + for _, row := range rows { + if row.UID == uid && row.Type == eventType && row.ID == contentID && row.SubID == subID { + found = true + break + } + } + if !found { + return nil, fmt.Errorf("eventplay: mini hub %d slot %d schedule %d identity mismatch", hub.UID, num(slot, 10), uid) + } + } + var matches []events.Schedule + for _, row := range rows { + if bound[row.UID] && row.Type == eventType && row.ID == contentID && row.SubID == subID && row.UID != 0 && row.Start < hub.End && hub.Start < row.End { + matches = append(matches, row) + } + } + if len(matches) == 0 { + continue + } + if len(matches) > 1 { + return nil, fmt.Errorf("eventplay: mini hub %d slot %d has ambiguous schedules", hub.UID, num(slot, 10)) + } + child := matches[0] + start, end := child.Start, child.End + if start < hub.Start { + start = hub.Start + } + if end > hub.End { + end = hub.End + } + // Project policy follows static EndDateType: play content closes at + // PlayEnd; exchange content can remain through the declared End. + if num(slot, 4) == 0 && end > hub.PlayEnd { + end = hub.PlayEnd + } + if end <= start { + continue + } + b := wire.AppendVarint(nil, 1, num(slot, 10)) + b = wire.AppendVarint(b, 2, contentType) + b = wire.AppendVarint(b, 3, contentID) + b = wire.AppendVarint(b, 4, child.UID) + b = wire.AppendVarint(b, 5, uint64(start)) + b = wire.AppendVarint(b, 6, uint64(end)) + v = wire.AppendBytes(v, 6, b) + } + out = wire.AppendBytes(out, 1, v) + } + return out, nil +} + +// Public event packs retain their declared wire settings and both end windows. +// Mini hub prefabs are served exclusively by MiniEventHubInfo. +func (s *Service) publicEventHubs(ctx command.Context, req []byte) ([]byte, error) { + _, body, _, err := s.hubCalendars.Handle(ctx, "/EventHubInfo", req) + if err != nil { + return nil, err + } + var out []byte + err = wire.Walk(body, func(f wire.Field) error { + if f.Number != 1 || f.Type != 2 { + return nil + } + row, err := s.design.Row("PackEventHubTable", 14, num(f.Value, 2)) + if err != nil { + return nil + } + if num(row, 13) == 0 { + out = wire.AppendBytes(out, 1, f.Value) + } + return nil + }) + return out, err +} + +// ListEventFieldPacks authorizes hidden fields through installed pack rules +// and the live project calendar. A hub field and a minigame field are separate +// routes: some public hubs have no PackEventMiniGameTable row at all. +func (s *Service) ListEventFieldPacks(ctx command.Context) ([]gamedata.EventFieldPack, error) { + + return s.eventFieldPacks(ctx) +} + +func (s *Service) ResolveEventFieldPack(ctx command.Context, id int) (gamedata.EventFieldPack, bool, error) { + + packs, err := s.eventFieldPacks(ctx) + if err != nil { + return gamedata.EventFieldPack{}, false, err + } + for _, p := range packs { + if p.ID == id { + return p, true, nil + } + } + return gamedata.EventFieldPack{}, false, nil +} + +func (s *Service) eventFieldPacks(ctx command.Context) ([]gamedata.EventFieldPack, error) { + now := s.now().UnixMilli() + result := map[int]gamedata.EventFieldPack{} + // Bound minigames use the hub play window even when their global schedule + // has a longer archive window. Zero means a known, currently closed binding. + boundEnds := map[uint64]int64{} + add := func(id int, uid, game, hub, mapID, point uint64, end int64) error { + base, ok := s.design.FieldPacks[id] + if !ok { + return nil + } + if len(base.MapIDs) == 0 { + return fmt.Errorf("eventplay: hidden pack %d has no installed map", id) + } + if mapID != 0 { + found := false + for _, m := range base.MapIDs { + if uint64(m) == mapID { + found = true + } + } + if !found { + return fmt.Errorf("eventplay: pack %d calendar map %d mismatch", id, mapID) + } + } + if old, exists := result[id]; exists && old.End >= end { + return nil + } + base.MapIDs = append([]int(nil), base.MapIDs...) + base.BuyRewards = append([]gamedata.Reward(nil), base.BuyRewards...) + base.ScheduleUID, base.GameID, base.HubID, base.End = uid, game, hub, end + base.InitialMapID, base.PointPositionID = mapID, point + if base.InitialMapID == 0 { + base.InitialMapID = uint64(base.MapIDs[0]) + } + result[id] = base + return nil + } + if s.hubCalendars != nil { + _, body, handled, err := s.hubCalendars.Handle(ctx, "/EventHubInfo", wire.AppendVarint(nil, 1, 1)) + if err != nil { + return nil, err + } + if handled { + err = wire.Walk(body, func(f wire.Field) error { + if f.Number != 1 || f.Type != 2 { + return nil + } + start, end := int64(num(f.Value, 3)), int64(num(f.Value, 4)) + if err := wire.Walk(f.Value, func(setting wire.Field) error { + if setting.Number != 6 || setting.Type != 2 || num(setting.Value, 2) != 6 { + return nil + } + uids, err := list(setting.Value, 3) + if err != nil { + return err + } + for _, uid := range uids { + if _, exists := boundEnds[uid]; !exists { + boundEnds[uid] = 0 + } + if now >= start && now < end && boundEnds[uid] < end { + boundEnds[uid] = end + } + } + return nil + }); err != nil { + return err + } + if now < start || now >= end { + return nil + } + hubID := num(f.Value, 2) + hub, e := s.design.Row("PackEventHubTable", 14, hubID) + if e != nil { + return e + } + return add(int(num(hub, 20)), num(f.Value, 1), 0, hubID, 0, 0, end) + }) + if err != nil { + return nil, err + } + } + } + for _, c := range s.registry.List() { + for _, binding := range s.fieldBindings { + if c.Type == binding.EventType && now >= c.Start && now < c.End { + if err := add(binding.PackID, c.UID, c.ID, 0, 0, 0, c.End); err != nil { + return nil, err + } + p := result[binding.PackID] + p.ContentOpenType = binding.ContentOpenType + result[binding.PackID] = p + } + } + if c.Type != 11 || now < c.Start || now >= c.End { + continue + } + end := c.End + if hubEnd, bound := boundEnds[c.UID]; bound { + if hubEnd == 0 { + continue + } + if hubEnd < end { + end = hubEnd + } + } + game, err := s.design.Row("PackEventMiniGameTable", 8, c.ID) + if err != nil { + return nil, err + } + if err = add(int(num(game, 12)), c.UID, c.ID, 0, num(game, 9), num(game, 13), end); err != nil { + return nil, err + } + } + var packs []gamedata.EventFieldPack + for _, p := range result { + packs = append(packs, p) + } + sort.Slice(packs, func(i, j int) bool { return packs[i].ID < packs[j].ID }) + return packs, nil +} + +// FieldObjectEventPeriod follows EventLostCoinInfo: the exact public calendar +// must be active, and the installed event must explicitly contain this pack. +func (s *Service) FieldObjectEventPeriod(pack int) (string, int64, error) { + + now := s.now().UnixMilli() + period := "" + var end int64 + for _, schedule := range s.registry.List() { + if schedule.Type != 14 || now < schedule.Start || now >= schedule.End { + continue + } + row, err := s.design.Row("EventLostCoinTable", 4, schedule.ID) + if err != nil { + return "", 0, err + } + packs, err := list(row, 1) + if err != nil { + return "", 0, err + } + matches := false + for _, id := range packs { + matches = matches || id == uint64(pack) + } + if !matches { + continue + } + if period != "" { + return "", 0, fmt.Errorf("eventplay: ambiguous active lost coin calendar for pack %d", pack) + } + period = fmt.Sprintf("event:%d:%d:%d:%d", schedule.UID, schedule.ID, schedule.Start, schedule.End) + end = schedule.End + } + if period == "" { + return "", 0, fmt.Errorf("eventplay: no active lost coin calendar for pack %d", pack) + } + return period, end, nil +} + +func (s *Service) EnterBattle(ctx command.Context, req []byte, receipt string) ([]byte, error) { + + if num(req, 5) != 17 || receipt == "" { + return nil, fmt.Errorf("eventplay: invalid event battle") + } + group, id, deck := num(req, 8), num(req, 9), num(req, 4) + var row []byte + for _, r := range s.design.Rows("PackEventBattleTable", 4, group) { + if num(r, 5) == id && num(r, 1) == deck { + row = r + break + } + } + if row == nil { + return nil, fmt.Errorf("eventplay: event battle deck/stage mismatch") + } + var uid uint64 + for _, c := range s.registry.List() { + if c.ID == group && (c.Type == 9 || c.Type == 8) && s.now().UnixMilli() >= c.Start && s.now().UnixMilli() < c.End { + uid = c.UID + break + } + } + if uid == 0 { + return nil, fmt.Errorf("eventplay: event battle schedule unavailable") + } + previous := uint64(0) + for _, r := range s.design.Rows("PackEventBattleTable", 4, group) { + stage := num(r, 5) + if stage < id && stage > previous { + previous = stage + } + } + if previous != 0 && !s.state.Stories[fmt.Sprintf("battle:%d:%d:%d", uid, group, previous)] { + return nil, fmt.Errorf("eventplay: previous event battle not cleared") + } + if validator, ok := s.economy.(interface { + CanApply(ctx command.Context, _ []gamedata.Reward) error + }); ok { + cost := num(row, 3) + if cost > 0 { + if e := validator.CanApply(ctx, []gamedata.Reward{{Type: 30, Count: cost}}); e != nil { + return nil, e + } + } + } + payload, _ := json.Marshal(s.state) + var next snapshot + _ = json.Unmarshal(payload, &next) + next.Runs["battle:"+receipt] = Run{UID: uid, Game: group, Stage: id, Mode: 17, Started: uint64(s.now().UnixMilli()), Family: "Battle", Session: strings.Split(receipt, ":")[0]} + payload, _ = json.Marshal(next) + if e := s.store.Save(ctx.State, "eventplay", payload); e != nil { + return nil, e + } + s.state = next + return nil, nil +} + +func (s *Service) CompleteBattle(ctx command.Context, req []byte, receipt string) ([]byte, error) { + + key := "eventbattle:" + receipt + if r, ok := s.state.Replies[key]; ok { + if !bytes.Equal(r.Request, req) { + return nil, fmt.Errorf("eventplay: changed battle retry") + } + return r.Body, nil + } + a, ok := s.state.Runs["battle:"+receipt] + if !ok { + return nil, fmt.Errorf("eventplay: no entered event battle") + } + var row []byte + for _, r := range s.design.Rows("PackEventBattleTable", 4, a.Game) { + if num(r, 5) == a.Stage { + row = r + break + } + } + if row == nil { + return nil, fmt.Errorf("eventplay: saved battle design missing") + } + raw, _ := json.Marshal(s.state) + var next snapshot + _ = json.Unmarshal(raw, &next) + var out []byte + if num(req, 2) == 1 { + rs, e := gamedata.EventPlayRewards(row, 9, 10, 8) + if e != nil { + return nil, e + } + definitions := s.battleChallenges[num(row, 1)] + claims, e := gamedata.VerifySubmittedChallenges(req, definitions) + if e != nil { + return nil, e + } + for _, index := range claims { + marker := fmt.Sprintf("challenge:%d:%d:%d:%d", a.UID, a.Game, a.Stage, index) + if !next.Stories[marker] { + rs = append(rs, definitions[index].Reward) + next.Stories[marker] = true + } + } + rewards := []gamedata.Reward{} + for _, r := range rs { + rewards = append(rewards, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 + } + cost := num(row, 3) + var costs []gamedata.Reward + if cost > 0 { + costs = []gamedata.Reward{{Type: 30, Count: cost}} + } + bundle, e := s.economy.Apply(ctx, key, costs, rewards) + if e != nil { + return nil, e + } + out = wire.AppendBytes(out, 5, bundle) + info := wire.AppendVarint(nil, 1, a.UID) + info = wire.AppendVarint(info, 2, a.Game) + info = wire.AppendVarint(info, 3, a.Stage) + for i := range definitions { + if next.Stories[fmt.Sprintf("challenge:%d:%d:%d:%d", a.UID, a.Game, a.Stage, i)] { + info = wire.AppendVarint(info, 4, uint64(i)) + } + } + out = wire.AppendBytes(out, 16, info) + next.Stories[fmt.Sprintf("battle:%d:%d:%d", a.UID, a.Game, a.Stage)] = true + } + delete(next.Runs, "battle:"+receipt) + next.Replies[key] = reply{append([]byte(nil), req...), out} + raw, _ = json.Marshal(next) + if e := s.store.Save(ctx.State, "eventplay", raw); e != nil { + return nil, e + } + s.state = next + return out, nil +} + +func (s *Service) BattleChallengeIndexes(uid, group, stage uint64) []uint64 { + + return s.battleChallengeIndexesLocked(uid, group, stage) +} + +func (s *Service) battleChallengeIndexesLocked(uid, group, stage uint64) []uint64 { + var deck uint64 + for _, row := range s.design.Rows("PackEventBattleTable", 4, group) { + if num(row, 5) == stage { + deck = num(row, 1) + break + } + } + var out []uint64 + for i := range s.battleChallenges[deck] { + if s.state.Stories[fmt.Sprintf("challenge:%d:%d:%d:%d", uid, group, stage, i)] { + out = append(out, uint64(i)) + } + } + return out +} diff --git a/go/internal/server/domain/events/play/room.go b/go/internal/server/domain/events/play/room.go new file mode 100644 index 0000000..1c45f43 --- /dev/null +++ b/go/internal/server/domain/events/play/room.go @@ -0,0 +1,11 @@ +package eventplay + +// RoomRuntime is implemented by the native matching/relay service. HTTP +// gameplay settles only runs whose native room is authenticated. + +type RoomRuntime interface { + ValidateRoom(session, guid string, uid uint64) error + CompleteRoom(session, guid string, score uint64) error +} + +func (s *Service) AttachRooms(r RoomRuntime) { s.rooms = r } diff --git a/go/internal/server/eventplay/routes.go b/go/internal/server/domain/events/play/routes.go similarity index 100% rename from go/internal/server/eventplay/routes.go rename to go/internal/server/domain/events/play/routes.go diff --git a/go/internal/server/domain/events/play/runtime.go b/go/internal/server/domain/events/play/runtime.go new file mode 100644 index 0000000..6e319a3 --- /dev/null +++ b/go/internal/server/domain/events/play/runtime.go @@ -0,0 +1,40 @@ +package eventplay + +import ( + "fmt" + "slices" +) + +func (s *Service) validateStage(f string, id, mode uint64) error { + switch f { + case "Rhythm": + r, e := s.design.Row("RhythmGameMusicTable", 5, id) + if e != nil { + return e + } + if mode > 2 { + return fmt.Errorf("eventplay: unknown rhythm mode") + } + _ = r + case "Hopscotch": + _, e := s.design.Row("HopscotchStageTable", 6, id) + return e + case "Sichuan": + _, e := s.design.Row("SichuanStageTable", 2, id) + return e + case "Survival": + if len(s.design.Tables["FieldMiniGameSurvivalTable"]) == 0 { + return fmt.Errorf("eventplay: survival design missing") + } + case "Action": + if len(s.design.Tables["ActionGameDefaultTable"]) == 0 { + return fmt.Errorf("eventplay: action design missing") + } + } + return nil +} + +func has(a []uint64, id uint64) bool { + return slices.Contains(a, id) +} +func seqSeed(n uint64) uint64 { return n & 2147483647 } diff --git a/go/internal/server/domain/events/play/upgrade.go b/go/internal/server/domain/events/play/upgrade.go new file mode 100644 index 0000000..919e960 --- /dev/null +++ b/go/internal/server/domain/events/play/upgrade.go @@ -0,0 +1,21 @@ +package eventplay + +import ( + "bd2server/internal/server/design/gamedata" +) + +func rewardDifference(old, next []gamedata.BattleReward) []gamedata.BattleReward { + paid := map[[2]uint64]uint64{} + for _, r := range old { + paid[[2]uint64{r.Type, r.ID}] += r.Count + } + var out []gamedata.BattleReward + for _, r := range next { + n := paid[[2]uint64{r.Type, r.ID}] + if r.Count > n { + r.Count -= n + out = append(out, r) + } + } + return out +} diff --git a/go/internal/server/domain/events/protocol_adapter.go b/go/internal/server/domain/events/protocol_adapter.go new file mode 100644 index 0000000..c363e9f --- /dev/null +++ b/go/internal/server/domain/events/protocol_adapter.go @@ -0,0 +1,581 @@ +package events + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "bd2server/internal/server/protocol/wire" + "bytes" + "crypto/sha256" + "encoding/json" + "fmt" + "math" + "sort" + "strconv" +) + +func (h SkinHandler) Handle(ctx command.Context, path string, request []byte) (int, []byte, bool, error) { + + if code, response, handled, err := h.Economy.OwnedItemInfo(ctx, path, request); handled || err != nil { + return code, response, handled, err + } + return h.Economy.Handle(ctx, path, request) +} + +func (e *Economy) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + if path != "/PrestigeSkinSet" { + return e.PrestigeSkinInfo(ctx, path, req) + } + + if err := wire.Walk(req, func(f wire.Field) error { + if (f.Number == 1 && f.Type != 0) || (f.Number == 2 && f.Type != 2) { + return fmt.Errorf("events: invalid prestige request field") + } + return nil + }); err != nil { + return 426, nil, true, err + } + seq, _, err := wire.Varint(req, 1) + info, found, err2 := wire.Bytes(req, 2) + if err != nil || err2 != nil || !found || seq == 0 || seq > math.MaxInt32 || commandSession == "" { + return 426, nil, true, fmt.Errorf("events: invalid prestige set request") + } + if err := wire.Walk(info, func(f wire.Field) error { + if f.Number >= 1 && f.Number <= 4 && f.Type != 0 { + return fmt.Errorf("events: invalid prestige info field") + } + return nil + }); err != nil { + return 426, nil, true, err + } + costume, _, err := wire.Varint(info, 1) + design, _, err2 := wire.Varint(info, 2) + setting, _, err3 := wire.Varint(info, 3) + timeValue, _, err4 := wire.Varint(info, 4) + if err != nil || err2 != nil || err3 != nil || err4 != nil || costume == 0 || costume > math.MaxInt32 || design == 0 || design > math.MaxInt32 || setting > 1 || timeValue > math.MaxInt64 || e.prestige[design] != costume { + return 426, nil, true, fmt.Errorf("events: invalid prestige skin") + } + s, err := e.loadPrestigeSkins(ctx) + if err != nil { + return 426, nil, true, err + } + key := commandSession + ":" + strconv.FormatUint(seq, 10) + if receipt, ok := s.Receipts[key]; ok { + if !bytes.Equal(receipt.Request, req) { + return 426, nil, true, fmt.Errorf("events: prestige sequence conflict") + } + return 426, append([]byte(nil), receipt.Response...), true, nil + } + owned := false + for _, item := range e.items.All(ctx) { + if item.Type == 45 && item.ID == design && item.Count > 0 { + owned = true + break + } + } + if !owned { + return 426, nil, true, fmt.Errorf("events: prestige skin not owned") + } + if setting == 1 { + s.Selections[costume] = design + } else if s.Selections[costume] == design { + delete(s.Selections, costume) + } + portrait := costume + if e.prestigePortrait != nil { + portrait = e.prestigePortrait() + } + out := wire.AppendVarint(nil, 1, portrait) + out = wire.AppendVarint(out, 2, s.Selections[portrait]) + s.Receipts[key] = prestigeSkinReceipt{Request: append([]byte(nil), req...), Response: append([]byte(nil), out...)} + raw, err := json.Marshal(s) + if err == nil { + err = e.store.Save(ctx.State, "prestige_skin_sets", raw) + } + if err != nil { + return 426, nil, true, err + } + return 426, out, true, nil +} + +func (e *Economy) PrestigeSkinInfo(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + if path != "/PrestigeSkinInfo" { + return 0, nil, false, nil + } + seq, _, err := wire.Varint(req, 1) + if err != nil || seq == 0 || seq > math.MaxInt32 { + return 425, nil, true, fmt.Errorf("events: missing sequence") + } + + s, err := e.loadPrestigeSkins(ctx) + if err != nil { + return 425, nil, true, err + } + var out []byte + seen := map[uint64]bool{} + for _, item := range e.items.All(ctx) { + if item.Type != 45 || item.Count == 0 || seen[item.ID] { + continue + } + costume, ok := e.prestige[item.ID] + if !ok { + return 425, nil, true, fmt.Errorf("events: owned skin missing design") + } + seen[item.ID] = true + b := wire.AppendVarint(nil, 1, costume) + b = wire.AppendVarint(b, 2, item.ID) + if s.Selections[costume] == item.ID { + b = wire.AppendVarint(b, 3, 1) + } + b = wire.AppendVarint(b, 4, item.TimeValue) + out = wire.AppendBytes(out, 1, b) + } + return 425, out, true, nil +} + +func (r *Registry) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + if path != "/EventScheduleInfo" { + return 0, nil, false, nil + } + seq, found, err := wire.Varint(req, 1) + if err != nil || !found || seq == 0 { + return 163, nil, true, fmt.Errorf("events: invalid sequence") + } + var out []byte + for _, s := range r.List() { + b := wire.AppendVarint(nil, 1, s.UID) + b = wire.AppendVarint(b, 2, s.Type) + b = wire.AppendVarint(b, 3, s.ID) + if s.SubID != 0 { + b = wire.AppendVarint(b, 4, s.SubID) + } + b = wire.AppendVarint(b, 5, uint64(s.Start)) + b = wire.AppendVarint(b, 6, uint64(s.End)) + if now := r.now().UnixMilli(); now >= s.Start && now < s.End { + b = wire.AppendVarint(b, 7, 1) + } + out = wire.AppendBytes(out, 1, b) + } + return 163, out, true, nil +} + +// ID card items are loaded through ItemInfo; there is no IdCardInfo ownership +// endpoint. AvatarInfo separately includes its owned ItemDBInfo list. +func (e *Economy) OwnedItemInfo(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + if path != "/AvatarInfo" { + return 0, nil, false, nil + } + seq, _, err := wire.Varint(req, 1) + if err != nil || seq == 0 { + return 467, nil, true, fmt.Errorf("events: missing sequence") + } + var out []byte + for _, item := range e.items.All(ctx) { + if item.Type == 49 || item.Type == 50 || item.Type == 61 { + out = wire.AppendBytes(out, 2, assets.ItemWire(item)) + } + } + return 467, out, true, nil +} + +func (e *Economy) Apply(ctx command.Context, identity string, costs, rewards []gamedata.Reward) ([]byte, error) { + definition, _ := json.Marshal(struct{ Costs, Rewards []gamedata.Reward }{costs, rewards}) + d := sha256.Sum256(definition) + + s, err := e.load(ctx) + if err != nil { + + return nil, err + } + if r, ok := s.Receipts[identity]; ok { + + if !bytes.Equal(r.Definition, d[:]) { + return nil, fmt.Errorf("events: reward identity reused with different definition") + } + return append([]byte(nil), r.Bundle...), nil + } + + var selected []assets.Item + // Merge repeated definitions before selecting stacks so the same quantity + // cannot be selected twice from a single inventory entry. + totals := map[[2]uint64]uint64{} + for _, r := range costs { + k := [2]uint64{r.Type, r.ID} + if r.Count == 0 || r.Count > math.MaxInt64 || totals[k] > math.MaxInt64-r.Count { + return nil, fmt.Errorf("events: invalid cost") + } + totals[k] += r.Count + } + keys := make([][2]uint64, 0, len(totals)) + for k := range totals { + keys = append(keys, k) + } + sort.Slice(keys, func(i, j int) bool { + if keys[i][0] != keys[j][0] { + return keys[i][0] < keys[j][0] + } + return keys[i][1] < keys[j][1] + }) + for _, k := range keys { + n := totals[k] + if walletCurrency(k[0]) || extraCurrency(k[0]) || apCurrency(k[0]) { + if k[1] != 0 { + return nil, fmt.Errorf("events: currency has item id") + } + selected = append(selected, assets.Item{Type: k[0], Count: n}) + continue + } + items, err := e.items.SelectMutable(ctx, k[0], k[1], n) + if err != nil { + return nil, err + } + selected = append(selected, items...) + } + var rs []gamedata.BattleReward + for _, r := range rewards { + rs = append(rs, gamedata.BattleReward(r)) + } + return e.consumeAndGrant(ctx, identity, selected, rs, d[:], false) +} + +func (e *Economy) ConsumeAndGrant(ctx command.Context, identity string, consumed []assets.Item, rewards []gamedata.BattleReward) ([]byte, error) { + return e.consumeAndGrant(ctx, identity, consumed, rewards, nil, false) +} + +func (e *Economy) consumeAndGrant(ctx command.Context, identity string, consumed []assets.Item, rewards []gamedata.BattleReward, definition []byte, openBox bool) ([]byte, error) { + + if identity == "" { + return nil, fmt.Errorf("events: empty reward identity") + } + s, err := e.load(ctx) + if err != nil { + return nil, err + } + raw, err := json.Marshal(struct { + Costs []assets.Item + Rewards []gamedata.BattleReward + }{consumed, rewards}) + if err != nil { + return nil, err + } + digest := sha256.Sum256(raw) + if receipt, ok := s.Receipts[identity]; ok { + if !bytes.Equal(receipt.Digest, digest[:]) { + return nil, fmt.Errorf("events: reward identity reused with different contents") + } + return append([]byte(nil), receipt.Bundle...), nil + } + if err = e.refreshAP(ctx, &s); err != nil { + return nil, err + } + var itemCosts []assets.Item + var walletCosts, apCosts []gamedata.Reward + for _, c := range consumed { + if c.Count == 0 || c.Count > math.MaxInt32 { + return nil, fmt.Errorf("events: invalid consumption count") + } + r := gamedata.Reward{Type: c.Type, ID: c.ID, Count: c.Count} + switch { + case walletCurrency(c.Type): + if c.ID != 0 || c.InvenIndex != 0 { + return nil, fmt.Errorf("events: invalid currency cost") + } + walletCosts = append(walletCosts, r) + case apCurrency(c.Type): + if c.ID != 0 || c.InvenIndex != 0 { + return nil, fmt.Errorf("events: invalid hunting AP cost") + } + apCosts = append(apCosts, r) + case extraCurrency(c.Type): + if c.ID != 0 || c.InvenIndex != 0 || s.Balances[c.Type] < c.Count { + return nil, fmt.Errorf("events: insufficient currency %d", c.Type) + } + s.Balances[c.Type] -= c.Count + default: + if !inventoryType(c.Type) { + return nil, fmt.Errorf("events: unsupported consumed item type %d", c.Type) + } + itemCosts = append(itemCosts, c) + } + } + if len(itemCosts) > 0 { + if err = e.items.CanConsume(ctx, itemCosts); err != nil { + return nil, err + } + } + var expanded []gamedata.BattleReward + if openBox { + expanded, err = e.graph.Resolve(rewards) + } else { + expanded, err = e.resolveGranted(rewards) + } + if err != nil { + return nil, err + } + expanded, err = e.avatarRewards.Expand(expanded) + if err != nil { + return nil, err + } + var walletRewards, apRewards, buffItems []gamedata.Reward + var itemRewards []gamedata.BattleReward + var costumes []uint64 + var equips []assets.Equipment + var bundle []byte + for _, r := range expanded { + if r.Count == 0 || r.Count > math.MaxInt32 { + return nil, fmt.Errorf("events: invalid reward count") + } + switch { + case walletCurrency(r.Type): + if r.ID != 0 { + return nil, fmt.Errorf("events: currency reward has item id") + } + walletRewards = append(walletRewards, gamedata.Reward(r)) + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(assets.Item{Type: r.Type, Count: r.Count})) + case apCurrency(r.Type): + if r.ID != 0 { + return nil, fmt.Errorf("events: invalid AP reward") + } + apRewards = append(apRewards, gamedata.Reward(r)) + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(assets.Item{Type: r.Type, Count: r.Count})) + case extraCurrency(r.Type): + if r.ID != 0 || s.Balances[r.Type] > math.MaxInt32-r.Count { + return nil, fmt.Errorf("events: currency overflow") + } + s.Balances[r.Type] += r.Count + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(assets.Item{Type: r.Type, Count: r.Count})) + case r.Type == 63: + if e.buffRewards == nil { + return nil, fmt.Errorf("events: buff reward runtime unavailable") + } + buffItems = append(buffItems, gamedata.Reward(r)) + bundle = wire.AppendBytes(bundle, 6, assets.ItemWire(assets.Item{Type: r.Type, ID: r.ID, Count: r.Count})) + case r.Type == 11: + if _, ok := e.costumes.Character(r.ID); !ok || r.Count > 1000 { + return nil, fmt.Errorf("events: unknown or excessive costume reward %d", r.ID) + } + for n := uint64(0); n < r.Count; n++ { + costumes = append(costumes, r.ID) + } + case r.Type == 10: + if r.Count > 1000 { + return nil, fmt.Errorf("events: excessive equipment reward") + } + for n := uint64(0); n < r.Count; n++ { + main, sub, private, rollErr := e.options.RollOptions(r.ID) + if rollErr != nil { + return nil, rollErr + } + entry := assets.Equipment{ID: r.ID, Rank: []uint64{0, 0, 0}} + for _, o := range main { + entry.MainOption = append(entry.MainOption, assets.EquipmentOption{GroupID: o.GroupID, ID: o.ID}) + } + for _, o := range sub { + entry.SubOption = append(entry.SubOption, assets.EquipmentOption{GroupID: o.GroupID, ID: o.ID}) + } + if private != nil { + entry.PrivateOption = &assets.EquipmentOption{GroupID: private.GroupID, ID: private.ID} + } + equips = append(equips, entry) + } + case inventoryType(r.Type): + if r.Type == 47 || r.Type == 69 { + if !e.ownedDesign[r.Type][r.ID] { + return nil, fmt.Errorf("events: unknown owned item design %d:%d", r.Type, r.ID) + } + } + if r.Type == 49 || r.Type == 50 || r.Type == 61 { + valid := e.avatarRewards != nil && e.avatarRewards.Items[r.Type][r.ID] + if !valid && !e.ownedDesign[r.Type][r.ID] { + return nil, fmt.Errorf("events: unknown avatar member %d:%d", r.Type, r.ID) + } + } + if r.Type == 45 { + if _, ok := e.prestige[r.ID]; !ok { + return nil, fmt.Errorf("events: unknown prestige skin %d", r.ID) + } + } + if r.ID == 0 { + return nil, fmt.Errorf("events: zero inventory reward id") + } + itemRewards = append(itemRewards, r) + default: + return nil, fmt.Errorf("events: unsupported reward type %d", r.Type) + } + } + if len(buffItems) > 0 { + if err = e.buffRewards.Validate(buffItems); err != nil { + return nil, err + } + } + if err = e.wallet.CanExchange(ctx, walletCosts, walletRewards); err != nil { + return nil, err + } + if len(apCosts)+len(apRewards) > 0 { + if e.hunting == nil { + return nil, fmt.Errorf("events: hunting AP runtime unavailable") + } + if err = e.hunting.CanExchangeAP(ctx, apCosts, apRewards); err != nil { + return nil, err + } + } + // Validation and random generation finish before the first account write. + if len(itemCosts) > 0 { + if err = e.items.Consume(ctx, itemCosts); err != nil { + return nil, err + } + } + if err = e.wallet.ExchangeOnce(ctx, identity+":wallet", walletCosts, walletRewards); err != nil { + return nil, err + } + if len(apCosts)+len(apRewards) > 0 { + if err = e.hunting.ExchangeAPOnce(ctx, identity+":ap", apCosts, apRewards); err != nil { + return nil, err + } + } + if len(buffItems) > 0 { + if err = e.buffRewards.GrantOnce(ctx, identity+":buff-items", buffItems); err != nil { + return nil, err + } + } + items, err := e.items.GrantOnce(ctx, identity+":items", itemRewards) + if err != nil { + return nil, err + } + for _, item := range items { + bundle = wire.AppendBytes(bundle, 1, assets.ItemWire(item)) + bundle = wire.AppendBytes(bundle, 6, assets.ItemWire(assets.Item{Type: item.Type, ID: item.ID, Count: item.Count})) + } + if len(costumes) > 0 { + grant, err := e.collection.GrantCostumes(ctx, identity+":costumes", costumes, e.costumes) + if err != nil { + return nil, err + } + var overflow uint64 + for _, x := range grant.Exchanges { + if x.ExchangeItemType != 20 || x.ExchangeCount > math.MaxInt64-overflow { + return nil, fmt.Errorf("events: invalid costume exchange") + } + overflow += x.ExchangeCount + } + if overflow > 0 { + if _, err = e.wallet.GrantMileageOnce(ctx, identity+":overflow", overflow); err != nil { + return nil, err + } + } + bundle = append(bundle, roster.CollectionRewardBundle(e.collection, grant)...) + } + for i, entry := range equips { + saved, err := e.equipment.GrantGeneratedOnce(ctx, identity+":equipment:"+strconv.Itoa(i), entry) + if err != nil { + return nil, err + } + bundle = wire.AppendBytes(bundle, 4, assets.EquipmentWire(saved)) + bundle = wire.AppendBytes(bundle, 6, assets.ItemWire(assets.Item{Type: 10, ID: saved.ID, Count: 1})) + } + s.Receipts[identity] = economyReceipt{Digest: digest[:], Definition: definition, Bundle: bundle} + data, err := json.Marshal(s) + if err != nil { + return nil, err + } + if err = e.store.Save(ctx.State, "event_economy", data); err != nil { + return nil, err + } + return bundle, nil +} + +func (e *Economy) OpenBox(ctx command.Context, identity string, box assets.Item) ([]byte, error) { + return e.consumeAndGrant(ctx, identity, []assets.Item{box}, []gamedata.BattleReward{{Type: 9, ID: box.ID, Count: box.Count}}, nil, true) +} + +func (s *BoxService) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { + commandSession := ctx.SessionID + if path != "/UseRandomBox" { + return 0, nil, false, nil + } + + seq, _, e := wire.Varint(req, 1) + index, _, e2 := wire.Varint(req, 2) + count, _, e3 := wire.Varint(req, 3) + if e != nil || e2 != nil || e3 != nil || seq == 0 || index == 0 || count == 0 || count > 1000 || seq > math.MaxInt32 || index > math.MaxInt64 { + return 143, nil, true, fmt.Errorf("events: invalid box request") + } + key := commandSession + ":" + strconv.FormatUint(seq, 10) + receipts := map[string]boxReceipt{} + raw, e := s.store.Load(ctx.State, "eventboxes") + if e != nil { + return 143, nil, true, e + } + if raw != nil { + if e = json.Unmarshal(raw, &receipts); e != nil { + return 143, nil, true, e + } + } + if r, ok := receipts[key]; ok { + if !bytes.Equal(req, r.Request) { + return 143, nil, true, fmt.Errorf("events: box sequence conflict") + } + return 143, r.Response, true, nil + } + var box assets.Item + for _, item := range s.items.All(ctx) { + if item.InvenIndex == index { + box = item + break + } + } + if box.Type != 9 || box.Count < count { + return 143, nil, true, fmt.Errorf("events: box unavailable") + } + box.Count = count + bundle, e := s.economy.OpenBox(ctx, "usebox:"+key, box) + if e != nil { + return 143, nil, true, e + } + out := wire.AppendBytes(nil, 1, bundle) + receipts[key] = boxReceipt{append([]byte(nil), req...), out} + raw, e = json.Marshal(receipts) + if e != nil { + return 143, nil, true, e + } + if e = s.store.Save(ctx.State, "eventboxes", raw); e != nil { + return 143, nil, true, e + } + return 143, out, true, nil +} + +func (e *Economy) ChargeInfo(ctx command.Context) ([]byte, error) { + + s, err := e.load(ctx) + if err != nil { + return nil, err + } + if err = e.refreshAP(ctx, &s); err != nil { + return nil, err + } + raw, err := e.store.Load(ctx.State, "eventapclock") + if err != nil { + return nil, err + } + if raw == nil { + return nil, nil + } + var clock economyAPClock + if err = json.Unmarshal(raw, &clock); err != nil { + return nil, err + } + var out []byte + for _, typ := range []uint64{30, 32} { + if _, ok := e.apCaps[typ]; !ok { + continue + } + info := wire.AppendVarint(nil, 1, uint64(clock.Next-86400000)) + item := wire.AppendVarint(nil, 3, typ) + item = wire.AppendVarint(item, 4, s.Balances[typ]) + info = wire.AppendBytes(info, 2, item) + out = wire.AppendBytes(out, 1, info) + } + return out, nil +} diff --git a/go/internal/server/events/registry.go b/go/internal/server/domain/events/registry.go similarity index 63% rename from go/internal/server/events/registry.go rename to go/internal/server/domain/events/registry.go index bfc2aef..92a3b7c 100644 --- a/go/internal/server/events/registry.go +++ b/go/internal/server/domain/events/registry.go @@ -3,10 +3,8 @@ package events import ( "fmt" - "sync" - "time" - "bd2server/internal/server/wire" + "time" ) type Schedule struct { @@ -21,7 +19,6 @@ type Resolver interface { } type Registry struct { - mu sync.RWMutex rows []Schedule now func() time.Time } @@ -42,21 +39,18 @@ func (r *Registry) Replace(rows []Schedule) error { } seen[identity] = true } - r.mu.Lock() - defer r.mu.Unlock() + r.rows = append([]Schedule(nil), rows...) return nil } func (r *Registry) List() []Schedule { - r.mu.RLock() - defer r.mu.RUnlock() + return append([]Schedule(nil), r.rows...) } func (r *Registry) Resolve(uid uint64) (Schedule, error) { - r.mu.RLock() - defer r.mu.RUnlock() + var result Schedule found := false for _, s := range r.rows { @@ -72,29 +66,3 @@ func (r *Registry) Resolve(uid uint64) (Schedule, error) { } return Schedule{}, fmt.Errorf("events: unknown schedule %d", uid) } - -func (r *Registry) Handle(path string, req []byte) (int, []byte, bool, error) { - if path != "/EventScheduleInfo" { - return 0, nil, false, nil - } - seq, found, err := wire.Varint(req, 1) - if err != nil || !found || seq == 0 { - return 163, nil, true, fmt.Errorf("events: invalid sequence") - } - var out []byte - for _, s := range r.List() { - b := wire.AppendVarint(nil, 1, s.UID) - b = wire.AppendVarint(b, 2, s.Type) - b = wire.AppendVarint(b, 3, s.ID) - if s.SubID != 0 { - b = wire.AppendVarint(b, 4, s.SubID) - } - b = wire.AppendVarint(b, 5, uint64(s.Start)) - b = wire.AppendVarint(b, 6, uint64(s.End)) - if now := r.now().UnixMilli(); now >= s.Start && now < s.End { - b = wire.AppendVarint(b, 7, 1) - } - out = wire.AppendBytes(out, 1, b) - } - return 163, out, true, nil -} diff --git a/go/internal/server/domain/events/skin_handler.go b/go/internal/server/domain/events/skin_handler.go new file mode 100644 index 0000000..6a62402 --- /dev/null +++ b/go/internal/server/domain/events/skin_handler.go @@ -0,0 +1,4 @@ +package events + +// SkinHandler restores skin ownership through the client's dedicated query. +type SkinHandler struct{ Economy *Economy } diff --git a/go/internal/server/domain/events/skins.go b/go/internal/server/domain/events/skins.go new file mode 100644 index 0000000..dce7f91 --- /dev/null +++ b/go/internal/server/domain/events/skins.go @@ -0,0 +1,47 @@ +package events + +import ( + "bd2server/internal/server/domain/command" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "fmt" +) + +func (e *Economy) AttachPrestigeSkins(design map[uint64]uint64) { e.prestige = design } +func (e *Economy) AttachPrestigePortrait(portrait func() uint64) { e.prestigePortrait = portrait } + +type prestigeSkinReceipt struct{ Request, Response []byte } +type prestigeSkinState struct { + Selections map[uint64]uint64 `json:"selections"` + Receipts map[string]prestigeSkinReceipt `json:"receipts"` +} + +func (e *Economy) loadPrestigeSkins(ctx command.Context) (prestigeSkinState, error) { + s := prestigeSkinState{Selections: map[uint64]uint64{}, Receipts: map[string]prestigeSkinReceipt{}} + raw, err := e.store.Load(ctx.State, "prestige_skin_sets") + if err != nil || raw == nil { + return s, err + } + if err = stateio.RequireExactJSONObject(raw, "selections", "receipts"); err != nil { + return s, err + } + if err = json.Unmarshal(raw, &s); err != nil { + return s, err + } + if s.Selections == nil || s.Receipts == nil { + return s, fmt.Errorf("events: invalid saved prestige skins") + } + for costume, design := range s.Selections { + if costume == 0 || design == 0 || e.prestige[design] != costume { + return s, fmt.Errorf("events: invalid saved prestige selection") + } + } + return s, nil +} + +// PrestigeSkinSelections returns a detached projection for costume responses. +func (e *Economy) PrestigeSkinSelections(ctx command.Context) (map[uint64]uint64, error) { + + s, err := e.loadPrestigeSkins(ctx) + return s.Selections, err +} diff --git a/go/internal/server/domain/events/tasks/inventory_projection.go b/go/internal/server/domain/events/tasks/inventory_projection.go new file mode 100644 index 0000000..e6c61b2 --- /dev/null +++ b/go/internal/server/domain/events/tasks/inventory_projection.go @@ -0,0 +1,63 @@ +package eventtasks + +import ( + "bd2server/internal/server/domain/command" + assets "bd2server/internal/server/domain/inventory" + "bd2server/internal/server/domain/roster" + "fmt" +) + +type InventoryEquipment struct{ Level uint64 } +type InventoryCostume struct{ ID, Level uint64 } +type InventorySnapshot struct { + Items map[[2]uint64]uint64 + Equipment map[uint64]InventoryEquipment + Costumes map[uint64]InventoryCostume +} +type InventoryProvider interface { + InventorySnapshot(ctx command.Context) (InventorySnapshot, error) +} +type ItemSource interface { + All(ctx command.Context) []assets.Item +} +type EquipmentSource interface { + All(ctx command.Context) []assets.Equipment +} +type CostumeSource interface{ Costumes() []roster.Costume } +type InventoryProjection struct { + Items ItemSource + Equipment EquipmentSource + Costumes CostumeSource + StateVersion func() uint64 +} + +func (p *InventoryProjection) ObservationVersion() string { + if p.StateVersion == nil { + return "" + } + var costumes uint64 + if source, ok := p.Costumes.(interface{ ObservationVersion() uint64 }); ok { + costumes = source.ObservationVersion() + } + return fmt.Sprintf("%d/%d", p.StateVersion(), costumes) +} + +func (p *InventoryProjection) InventorySnapshot(ctx command.Context) (InventorySnapshot, error) { + s := InventorySnapshot{Items: map[[2]uint64]uint64{}, Equipment: map[uint64]InventoryEquipment{}, Costumes: map[uint64]InventoryCostume{}} + if p.Items != nil { + for _, v := range p.Items.All(ctx) { + s.Items[[2]uint64{v.Type, v.ID}] += v.Count + } + } + if p.Equipment != nil { + for _, v := range p.Equipment.All(ctx) { + s.Equipment[v.InvenIndex] = InventoryEquipment{Level: v.Level} + } + } + if p.Costumes != nil { + for _, v := range p.Costumes.Costumes() { + s.Costumes[v.InvenIndex] = InventoryCostume{ID: v.ID, Level: v.Level} + } + } + return s, nil +} diff --git a/go/internal/server/domain/events/tasks/notify_changes.go b/go/internal/server/domain/events/tasks/notify_changes.go new file mode 100644 index 0000000..95caa3f --- /dev/null +++ b/go/internal/server/domain/events/tasks/notify_changes.go @@ -0,0 +1,65 @@ +package eventtasks + +import "bd2server/internal/server/domain/command" + +import ( + "fmt" + "strings" +) + +// The baseline belongs to the current account transaction, not a client session. +// Reads and retries therefore do not replay historical progress as notifications. +type missionNoticeKey struct { + Schedule string + Event, Group, Task uint64 +} + +// visibleMissionValues is a read-only view. In particular, it must not call +// mission(), which initializes and resets persisted progress as a side effect. +func (s *Service) visibleMissionValues(ctx command.Context) map[missionNoticeKey]uint64 { + schedules := s.taskSchedules() + version := s.inventoryObservationVersion() + if version != "" { + var signature strings.Builder + now := s.now().UTC() + year, week := now.ISOWeek() + fmt.Fprintf(&signature, "%s/%s/%d:%d", version, now.Format("2006-01-02"), year, week) + for _, v := range schedules { + fmt.Fprintf(&signature, "/%s:%d:%d:%d:%t:%d", scheduleKey(v), v.ID, v.Start, v.End, s.active(v), (now.UnixMilli()-v.Start)/86400000) + } + version = signature.String() + if s.visibleMissions != nil && s.visibleVersion == version { + return s.visibleMissions + } + } + out := map[missionNoticeKey]uint64{} + for _, v := range schedules { + if v.Type != 4 || !s.active(v) { + continue + } + group := s.design.MissionGroups[v.ID] + period := "" + if group.Type == 1 { + period = s.day() + } + if group.Type == 2 { + y, w := s.now().UTC().ISOWeek() + period = fmt.Sprintf("%d-%d", y, w) + } + for _, t := range s.design.Missions { + if !s.availableTask(ctx, v, t) { + continue + } + k := missionNoticeKey{scheduleKey(v), v.ID, t.Group, t.ID} + value := uint64(0) + if m := s.state.Missions[k.Schedule+"/"+key(t.ID)]; m != nil && m.Period == period { + value = m.Value + } + out[k] = value + } + } + if version != "" { + s.visibleMissions, s.visibleVersion = out, version + } + return out +} diff --git a/go/internal/server/domain/events/tasks/observer.go b/go/internal/server/domain/events/tasks/observer.go new file mode 100644 index 0000000..ba3b90e --- /dev/null +++ b/go/internal/server/domain/events/tasks/observer.go @@ -0,0 +1,66 @@ +package eventtasks + +import ( + "bd2server/internal/server/domain/command" + "encoding/json" +) + +func (s *Service) BeginLogin(ctx command.Context) { + + s.inventoryReady = false +} + +func (s *Service) inventoryObservationVersion() string { + if provider, ok := s.provider.(interface{ ObservationVersion() string }); ok { + return provider.ObservationVersion() + } + return "" +} +func (s *Service) inventorySnapshot(ctx command.Context) (InventorySnapshot, error) { + return s.provider.InventorySnapshot(ctx) +} +func (s *Service) BeforeDispatch(ctx command.Context, _ string, _ []byte) error { + + s.beforeMissions = s.visibleMissionValues(ctx) + + if s.provider == nil { + return nil + } + version := s.inventoryObservationVersion() + if s.inventoryReady && version != "" && version == s.observationVersion { + return nil + } + s.inventoryReady = false + var e error + s.before, e = s.inventorySnapshot(ctx) + if e == nil { + s.observationVersion = s.inventoryObservationVersion() + s.inventoryReady = true + } + return e +} +func (s *Service) AttachInventoryProvider(p InventoryProvider) { s.provider = p } + +func (s *Service) CompleteSingleTargetEvent(ctx command.Context, condition uint64, unlocked func(command.Context, uint64, uint64) bool) error { + + before, _ := json.Marshal(s.state) + for _, v := range s.taskSchedules() { + if !s.active(v) { + continue + } + for _, t := range s.design.Missions { + if t.Type != condition || t.SubType != 0 || t.Target != 1 || len(t.Params) != 0 || !s.availableTask(ctx, v, t) { + continue + } + if (t.UnlockPack > 0 || t.UnlockQuest > 0) && (unlocked == nil || !unlocked(ctx, t.UnlockPack, t.UnlockQuest)) { + continue + } + s.mission(v, t.ID).Value = 1 + } + } + if e := s.save(ctx); e != nil { + _ = json.Unmarshal(before, &s.state) + return e + } + return nil +} diff --git a/go/internal/server/eventtasks/service.go b/go/internal/server/domain/events/tasks/protocol_adapter.go similarity index 53% rename from go/internal/server/eventtasks/service.go rename to go/internal/server/domain/events/tasks/protocol_adapter.go index efb3a94..b158156 100644 --- a/go/internal/server/eventtasks/service.go +++ b/go/internal/server/domain/events/tasks/protocol_adapter.go @@ -1,15 +1,9 @@ -// Package eventtasks owns attendance, scheduled missions and pass progress. -// Calendar policy is supplied by events.Resolver; requests never create dates -// or manufacture progress. The parent request transaction includes all saves. package eventtasks import ( - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" - "bd2server/internal/server/world" + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/protocol/wire" "crypto/sha256" "encoding/hex" "encoding/json" @@ -18,342 +12,10 @@ import ( "slices" "sort" "strconv" - "sync" "time" ) -type Economy interface { - Apply(string, []gamedata.Reward, []gamedata.Reward) ([]byte, error) -} - -// AttendanceMailIssuer delivers attendance attachments inside the parent -// account transaction. Its durable identity prevents duplicate mail on replay. -type AttendanceMailIssuer interface { - IssueAttachmentsOnce(identity, title, body string, rewards []gamedata.Reward, sentAt time.Time) error -} - -type attendance struct { - Group, Count uint64 - LastDay string - Obtained map[string]bool - History map[string]bool -} -type mission struct { - Value uint64 - Claimed bool - Period string -} -type pass struct { - Exp uint64 - Premium bool - Claimed map[string]bool -} -type receipt struct { - Digest string - Code int - Reply []byte -} -type snapshot struct { - Version string - NewbieStep uint64 - Attendance map[string]*attendance - Missions map[string]*mission - Passes map[string]*pass - LoginDays map[string]int64 - Receipts map[string]receipt -} -type Service struct { - provider world.GameplayAchievementProvider - before world.GameplayAchievementSnapshot - beforeMissions map[missionNoticeKey]uint64 - mu sync.Mutex - store stateio.Store - design *gamedata.EventTasksDesign - registry events.Resolver - economy Economy - state snapshot - now func() time.Time - session string - unlocked func(uint64, uint64) bool - authorizeCash func(uint64, uint64) bool - attendancePremium func(uint64) bool - attendanceMail AttendanceMailIssuer - associated func(events.Schedule) uint64 -} - -func Open(store stateio.Store, d *gamedata.EventTasksDesign, r events.Resolver, e Economy) (*Service, error) { - if store == nil || d == nil || r == nil || e == nil { - return nil, errors.New("eventtasks: missing dependency") - } - s := &Service{store: store, design: d, registry: r, economy: e, now: time.Now} - b, err := store.Load("eventtasks") - if err != nil { - return nil, err - } - if b != nil { - if err = stateio.RequireExactJSONObject(b, "Version", "NewbieStep", "Attendance", "Missions", "Passes", "LoginDays", "Receipts"); err != nil { - return nil, err - } - if err = json.Unmarshal(b, &s.state); err != nil { - return nil, err - } - if s.state.Version != versionconfig.State() || s.state.Attendance == nil || s.state.Missions == nil || s.state.Passes == nil || s.state.LoginDays == nil || s.state.Receipts == nil { - return nil, errors.New("eventtasks: invalid state layout") - } - for _, a := range s.state.Attendance { - if a == nil || a.Obtained == nil || a.History == nil { - return nil, errors.New("eventtasks: invalid attendance") - } - } - for _, m := range s.state.Missions { - if m == nil { - return nil, errors.New("eventtasks: invalid mission") - } - } - for _, p := range s.state.Passes { - if p == nil || p.Claimed == nil { - return nil, errors.New("eventtasks: invalid pass") - } - } - } else { - s.state.Version = versionconfig.State() - } - s.init() - return s, nil -} -func (s *Service) init() { - if s.state.Attendance == nil { - s.state.Attendance = map[string]*attendance{} - } - if s.state.Missions == nil { - s.state.Missions = map[string]*mission{} - } - if s.state.Passes == nil { - s.state.Passes = map[string]*pass{} - } - if s.state.LoginDays == nil { - s.state.LoginDays = map[string]int64{} - } - if s.state.Receipts == nil { - s.state.Receipts = map[string]receipt{} - } -} -func (s *Service) SetSession(v string) { s.mu.Lock(); defer s.mu.Unlock(); s.session = v } -func (s *Service) AttachUnlockResolver(f func(uint64, uint64) bool) { s.unlocked = f } -func (s *Service) AttachCashAuthorization(f func(uint64, uint64) bool) { s.authorizeCash = f } -func (s *Service) AttachAttendancePremium(f func(uint64) bool) { s.attendancePremium = f } -func (s *Service) AttachAssociatedMissionGroup(f func(events.Schedule) uint64) { s.associated = f } - -func (s *Service) AttachAttendanceMail(issuer AttendanceMailIssuer) { s.attendanceMail = issuer } - -func (s *Service) mailAttendance(identity string, rewards []gamedata.Reward) error { - if s.attendanceMail == nil { - return errors.New("eventtasks: attendance mailbox unavailable") - } - return s.attendanceMail.IssueAttachmentsOnce(identity, "Attendance Rewards", "Your attendance rewards are ready. Please claim the attachments in your mailbox.", rewards, s.now()) -} - -func (s *Service) SetNewbieStep(step uint64) error { - s.mu.Lock() - defer s.mu.Unlock() - if step > s.state.NewbieStep { - s.state.NewbieStep = step - } - return s.save() -} -func (s *Service) NewbieStep() uint64 { s.mu.Lock(); defer s.mu.Unlock(); return s.state.NewbieStep } -func (s *Service) save() error { - b, e := json.Marshal(s.state) - if e != nil { - return e - } - return s.store.Save("eventtasks", b) -} -func scalar(b []byte, n int) uint64 { v, _, _ := wire.Varint(b, n); return v } -func key(v ...uint64) string { - var k string - for i, n := range v { - if i > 0 { - k += "/" //nolint:modernize // stringsbuilder - } - k += strconv.FormatUint(n, 10) - } - return k -} - -// scheduleKey preserves the public UID where it is present. UID zero is a -// protocol default shared by permanent groups, so its local identity also -// includes the event type and design group. -func scheduleKey(v events.Schedule) string { - if v.UID != 0 { - return key(v.UID) - } - return key(0, v.Type, v.ID, v.SubID) -} -func (s *Service) active(v events.Schedule) bool { - n := s.now().UnixMilli() - return n >= v.Start && n < v.End -} -func (s *Service) find(t, id uint64) (events.Schedule, error) { - rows := s.registry.List() - if t == 4 { - rows = s.taskSchedules() - } - for _, v := range rows { - if v.Type == t && v.ID == id && s.active(v) { - return v, nil - } - } - return events.Schedule{}, fmt.Errorf("eventtasks: event %d/%d inactive", t, id) -} - -func (s *Service) taskSchedules() []events.Schedule { - rows := s.registry.List() - var out []events.Schedule - seen := map[uint64]bool{} - for _, v := range rows { - if v.Type == 4 { - out = append(out, v) - if s.active(v) { - seen[v.ID] = true - } - } - } - for _, v := range rows { - if v.Type == 5 && s.active(v) { - g := s.design.Passes[v.ID].MissionGroup - if g > 0 && !seen[g] { - v.Type = 4 - v.ID = g - out = append(out, v) - seen[g] = true - } - } - } - if s.associated != nil { - for _, v := range rows { - if !s.active(v) { - continue - } - g := s.associated(v) - if g > 0 && !seen[g] { - v.Type = 4 - v.ID = g - out = append(out, v) - seen[g] = true - } - } - } - return out -} -func (s *Service) resolve(uid uint64) (events.Schedule, error) { - v, e := s.registry.Resolve(uid) - if e != nil { - return v, e - } - if !s.active(v) { - return v, errors.New("eventtasks: event expired") - } - return v, nil -} -func (s *Service) day() string { return s.now().UTC().Format("2006-01-02") } -func (s *Service) attendance(v events.Schedule) *attendance { - k := scheduleKey(v) - a := s.state.Attendance[k] - if a == nil { - g := s.design.Attendance[v.ID].Group - var first uint64 - for _, x := range s.design.AttendanceGroups { - if x.Group == g && (first == 0 || x.ID < first) { - first = x.ID - } - } - if first > 0 { - g = first - } - a = &attendance{Group: g, Obtained: map[string]bool{}, History: map[string]bool{}} - s.state.Attendance[k] = a - } - return a -} - -// attendanceRewards is shared by automatic and explicit claims. Obtained is -// the current cycle ledger; History is the client-visible historical ledger. -func (s *Service) attendanceRewards(v events.Schedule, a *attendance, group, id uint64) []gamedata.Reward { - if a.Obtained[key(group, id)] { - return nil - } - if v.Type == 0 { - if group != a.Group { - return nil - } - for _, r := range s.design.AttendanceRewards[group] { - if r.ID != id || r.Day > a.Count { - continue - } - var rewards []gamedata.Reward - if r.Basic.Count > 0 { - rewards = append(rewards, r.Basic) - } - if ticket := s.design.Attendance[v.ID].Ticket; ticket != 0 && s.attendancePremium != nil && s.attendancePremium(ticket) && r.Premium.Count > 0 { - rewards = append(rewards, r.Premium) - } - return rewards - } - } else if v.Type == 1 && group == v.ID && id > 0 { - box, ok := s.design.LimitRewards[[2]uint64{group, id}] - date := time.UnixMilli(v.Start).UTC().AddDate(0, 0, int(id)-1).Format("2006-01-02") - _, logged := s.state.LoginDays[date] - if ok && box > 0 && logged && date == s.day() { - return []gamedata.Reward{{Type: 9, ID: box, Count: 1}} - } - } - return nil -} -func (s *Service) pass(v events.Schedule) *pass { - k := scheduleKey(v) - p := s.state.Passes[k] - if p == nil { - p = &pass{Claimed: map[string]bool{}} - s.state.Passes[k] = p - } - return p -} -func (s *Service) mission(v events.Schedule, id uint64) *mission { - k := scheduleKey(v) + "/" + key(id) - m := s.state.Missions[k] - if m == nil { - m = &mission{} - s.state.Missions[k] = m - } - g := s.design.MissionGroups[v.ID] - period := "" - if g.Type == 1 { - period = s.day() - } - if g.Type == 2 { - y, w := s.now().UTC().ISOWeek() - period = fmt.Sprintf("%d-%d", y, w) - } - if m.Period != period { - *m = mission{Period: period} - } - return m -} -func (s *Service) availableTask(v events.Schedule, t gamedata.EventTask) bool { - g := s.design.MissionGroups[v.ID] - for i, n := range g.Groups { - if n == t.Group { - if g.Type == 3 && int64(i) > (s.now().UnixMilli()-v.Start)/86400000 { - return false - } - return t.UnlockPack == 0 && t.UnlockQuest == 0 || s.unlocked != nil && s.unlocked(t.UnlockPack, t.UnlockQuest) - } - } - return false -} - -func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { +func (s *Service) Handle(ctx command.Context, path string, req []byte) (int, []byte, bool, error) { codes := map[string]int{"/Attendance": 0, "/AttendanceInfo": 0, "/EventRewardHistory": 0, "/EventReward": 151, "/EventMissionInfo": 127, "/MissionClear": 120, "/MissionUpdate": 119, "/PassInfo": 124, "/PassReward": 126, "/PassBuy": 129, "/LoginEvent": 0} code, found := codes[path] if !found { @@ -374,8 +36,7 @@ func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { return 0, nil, false, nil } } - s.mu.Lock() - defer s.mu.Unlock() + seq, present, err := wire.Varint(req, 1) if err != nil || !present || seq == 0 { return code, nil, true, errors.New("eventtasks: invalid sequence") @@ -385,7 +46,7 @@ func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { } digest := sha256.Sum256(append([]byte(path), req...)) ds := hex.EncodeToString(digest[:]) - rk := s.session + ":" + path + ":" + key(seq) + rk := ctx.SessionID + ":" + path + ":" + key(seq) if r, ok := s.state.Receipts[rk]; ok { if r.Digest != ds { return code, nil, true, errors.New("eventtasks: changed replay request") @@ -393,14 +54,14 @@ func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { return r.Code, append([]byte(nil), r.Reply...), true, nil } before, _ := json.Marshal(s.state) - out, err := s.handle(path, req, rk) + out, err := s.handle(ctx, path, req, rk) if err != nil { s.state = snapshot{} _ = json.Unmarshal(before, &s.state) return code, nil, true, err } s.state.Receipts[rk] = receipt{ds, code, out} - if err = s.save(); err != nil { + if err = s.save(ctx); err != nil { s.state = snapshot{} _ = json.Unmarshal(before, &s.state) return code, nil, true, err @@ -408,7 +69,9 @@ func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { return code, out, true, nil } -func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) { +func scalar(b []byte, n int) uint64 { v, _, _ := wire.Varint(b, n); return v } + +func (s *Service) handle(ctx command.Context, path string, b []byte, identity string) ([]byte, error) { switch path { case "/Attendance", "/LoginEvent": today := s.day() @@ -501,7 +164,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) } slices.Sort(ids) for _, id := range ids { - rewards := s.attendanceRewards(v, a, group, id) + rewards := s.attendanceRewards(ctx, v, a, group, id) if len(rewards) == 0 { continue } @@ -515,7 +178,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) } } if len(granted) > 0 { - if err := s.mailAttendance(identity, granted); err != nil { + if err := s.mailAttendance(ctx, identity, granted); err != nil { return nil, err } } @@ -574,11 +237,11 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) if a.Obtained[ck] { return nil, errors.New("eventtasks: attendance already received") } - rewards := s.attendanceRewards(v, a, g, id) + rewards := s.attendanceRewards(ctx, v, a, g, id) if len(rewards) == 0 { return nil, errors.New("eventtasks: attendance day unavailable") } - if err := s.mailAttendance(identity, rewards); err != nil { + if err := s.mailAttendance(ctx, identity, rewards); err != nil { return nil, err } a.Obtained[ck] = true @@ -591,7 +254,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) continue } for _, t := range s.design.Missions { - if !s.availableTask(v, t) { + if !s.availableTask(ctx, v, t) { continue } m := s.mission(v, t.ID) @@ -617,7 +280,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) return e } t, ok := s.design.Missions[id] - if !ok || t.Group != scalar(f.Value, 1) || !s.availableTask(v, t) { + if !ok || t.Group != scalar(f.Value, 1) || !s.availableTask(ctx, v, t) { return errors.New("eventtasks: invalid mission update") } if value > s.mission(v, id).Value { @@ -637,7 +300,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) var done []uint64 var exp uint64 for _, t := range s.design.Missions { - if t.Group != g || !s.availableTask(v, t) || !all && t.ID != id { + if t.Group != g || !s.availableTask(ctx, v, t) || !all && t.ID != id { continue } m := s.mission(v, t.ID) @@ -650,7 +313,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) if len(done) == 0 { return nil, errors.New("eventtasks: no completed unclaimed mission") } - bundle, e := s.economy.Apply(identity, nil, rewards) + bundle, e := s.economy.Apply(ctx, identity, nil, rewards) if e != nil { return nil, e } @@ -660,12 +323,12 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) // Completion-of-mission conditions are derived from durable claims, // rather than a client-supplied total. for _, t := range s.design.Missions { - if !s.availableTask(v, t) || (t.Type != 1001 && t.Type != 1002) { + if !s.availableTask(ctx, v, t) || (t.Type != 1001 && t.Type != 1002) { continue } var completed uint64 for _, other := range s.design.Missions { - if !s.availableTask(v, other) || !s.mission(v, other.ID).Claimed { + if !s.availableTask(ctx, v, other) || !s.mission(v, other.ID).Claimed { continue } match := other.Group == t.Group @@ -776,7 +439,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) } var bundle []byte if len(claims) > 0 { - bundle, e = s.economy.Apply(identity, nil, rewards) + bundle, e = s.economy.Apply(ctx, identity, nil, rewards) if e != nil { return nil, fmt.Errorf("eventtasks: pass reward grant failed pass=%d level=%d reward_type=%d all=%t: %w", v.ID, id, rt, all, e) } @@ -799,7 +462,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) complete = false } else { for _, t := range s.design.Missions { - if s.availableTask(mv, t) && !s.mission(mv, t.ID).Claimed { + if s.availableTask(ctx, mv, t) && !s.mission(mv, t.ID).Claimed { complete = false } } @@ -853,7 +516,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) } // CashShopBuy charges paid diamonds and grants the pass ticket first. // Activation consumes that purchase entitlement and the design ticket. - if chosen.CashID != 0 && (s.authorizeCash == nil || !s.authorizeCash(v.ID, typ)) { + if chosen.CashID != 0 && (s.authorizeCash == nil || !s.authorizeCash(ctx, v.ID, typ)) { return nil, fmt.Errorf("eventtasks: cash entitlement required pass=%d type=%d sku=%d/%d/%d", v.ID, typ, chosen.CashGroup, chosen.CashID, chosen.CashSales) } var costs []gamedata.Reward @@ -865,7 +528,7 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) } costs = append(costs, cost) } - if _, e = s.economy.Apply(identity, costs, chosen.Rewards); e != nil { + if _, e = s.economy.Apply(ctx, identity, costs, chosen.Rewards); e != nil { return nil, fmt.Errorf("eventtasks: pass payment failed pass=%d type=%d: %w", v.ID, typ, e) } if chosen.LevelsGranted > 0 { @@ -881,73 +544,8 @@ func (s *Service) handle(path string, b []byte, identity string) ([]byte, error) return nil, errors.New("eventtasks: unsupported operation") } -// RecordEvent is called by authoritative gameplay. It returns no rewards and -// caps every matching task at its design threshold. -func (s *Service) RecordEvent(condition, sub, count uint64, unlocked func(uint64, uint64) bool) error { - s.mu.Lock() - defer s.mu.Unlock() - if !s.recordEventLocked(condition, sub, count, unlocked) { - return nil - } - return s.save() -} +func (s *Service) Notify(ctx command.Context) ([]byte, error) { -// recordEventLocked reports every persistent mutation, including initializing -// a matching task or rolling its period. Irrelevant and already capped events -// leave the account snapshot untouched. -func (s *Service) recordEventLocked(condition, sub, count uint64, unlocked func(uint64, uint64) bool) bool { - changed := false - if count == 0 { - return false - } - for _, v := range s.taskSchedules() { - if v.Type != 4 || !s.active(v) { - continue - } - for _, t := range s.design.Missions { - if t.Type != condition || t.SubType != 0 && t.SubType != sub || condition == 349 && t.SubType != sub || !s.availableTask(v, t) { - continue - } - if (t.UnlockPack > 0 || t.UnlockQuest > 0) && (unlocked == nil || !unlocked(t.UnlockPack, t.UnlockQuest)) { - continue - } - match := len(t.Params) == 0 - for _, n := range t.Params { - if n == sub { - match = true - } - } - if !match { - continue - } - previous := s.state.Missions[scheduleKey(v)+"/"+key(t.ID)] - var prior mission - if previous != nil { - prior = *previous - } - m := s.mission(v, t.ID) - if previous == nil || prior != *m { - changed = true - } - if m.Claimed { - continue - } - if m.Value >= t.Target { - continue - } - if count >= t.Target-m.Value { - m.Value = t.Target - } else { - m.Value += count - } - changed = true - } - } - return changed -} -func (s *Service) Notify() ([]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() var out []byte for _, v := range s.taskSchedules() { if v.Type != 4 || !s.active(v) { @@ -955,7 +553,7 @@ func (s *Service) Notify() ([]byte, error) { } groups := map[uint64][]byte{} for _, t := range s.design.Missions { - if !s.availableTask(v, t) { + if !s.availableTask(ctx, v, t) { continue } m := s.mission(v, t.ID) @@ -979,17 +577,147 @@ func (s *Service) Notify() ([]byte, error) { return out, nil } -// RecordLogin shares the same daily marker as Attendance, preventing repeated -// reconnects from inflating event login tasks. Main should invoke this once -// through the regular mission observer before serving attendance queries. -func (s *Service) RecordLogin(unlocked func(uint64, uint64) bool) error { - s.mu.Lock() - marker := "mission-login:" + s.day() - if _, ok := s.state.LoginDays[marker]; ok { - s.mu.Unlock() - return nil +func (s *Service) AfterDispatch(ctx command.Context, path string, request, response []byte) ([]byte, error) { + if s.provider != nil { + version := s.inventoryObservationVersion() + if s.inventoryReady && version != "" && version == s.observationVersion { + return s.notifyMissionChanges(ctx) + } + s.inventoryReady = false + after, e := s.inventorySnapshot(ctx) + if e != nil { + return nil, e + } + rk := fmt.Sprintf("observer:%s:%s:%d", ctx.SessionID, path, scalar(request, 1)) + + _, seen := s.state.Receipts[rk] + + if !seen { + type delta struct{ condition, sub, count uint64 } + var deltas []delta + for kind, old := range s.before.Items { + current := after.Items[kind] + if current < old { + condition := uint64(12) + if kind[1] == 0 { + condition = 11 + } + deltas = append(deltas, delta{condition, kind[0], old - current}) + } + } + for kind, current := range after.Items { + old := s.before.Items[kind] + if current > old { + deltas = append(deltas, delta{32, kind[0], current - old}) + } + } + for idx, current := range after.Equipment { + old, ok := s.before.Equipment[idx] + if ok && current.Level > old.Level { + deltas = append(deltas, delta{14, 0, current.Level - old.Level}) + } + } + for idx, current := range after.Costumes { + old, ok := s.before.Costumes[idx] + if ok && current.Level > old.Level { + deltas = append(deltas, delta{104, current.ID, current.Level - old.Level}) + } + } + if len(deltas) > 0 { + + before, e := json.Marshal(s.state) + if e != nil { + + return nil, e + } + changed := false + for _, d := range deltas { + changed = s.recordEventLocked(ctx, d.condition, d.sub, d.count, s.unlocked) || changed + } + if changed { + s.state.Receipts[rk] = receipt{Digest: "observer"} + e = s.save(ctx) + } + if e != nil { + s.state = snapshot{} + _ = json.Unmarshal(before, &s.state) + } + + if e != nil { + return nil, e + } + } + } + s.before = after + s.observationVersion = s.inventoryObservationVersion() + s.inventoryReady = true } - s.state.LoginDays[marker] = s.now().UnixMilli() - s.mu.Unlock() - return s.RecordEvent(0, 0, 1, unlocked) + return s.notifyMissionChanges(ctx) +} + +func (s *Service) notifyMissionChanges(ctx command.Context) ([]byte, error) { + + if s.beforeMissions == nil { + return nil, nil + } + current := s.visibleMissionValues(ctx) + var keys []missionNoticeKey + for k, value := range current { + if value != s.beforeMissions[k] { + keys = append(keys, k) + } + } + sort.Slice(keys, func(i, j int) bool { + a, b := keys[i], keys[j] + if a.Event != b.Event { + return a.Event < b.Event + } + if a.Group != b.Group { + return a.Group < b.Group + } + if a.Task != b.Task { + return a.Task < b.Task + } + return a.Schedule < b.Schedule + }) + var out, groupMap, rows []byte + var eventID, groupID uint64 + flushGroup := func() { + if len(rows) == 0 { + return + } + entry := wire.AppendVarint(nil, 1, groupID) + entry = wire.AppendBytes(entry, 2, rows) + groupMap = wire.AppendBytes(groupMap, 1, entry) + rows = nil + } + flushEvent := func() { + flushGroup() + if len(groupMap) == 0 { + return + } + entry := wire.AppendVarint(nil, 1, eventID) + entry = wire.AppendBytes(entry, 2, groupMap) + out = wire.AppendBytes(out, 4, entry) + groupMap = nil + } + for _, k := range keys { + if k.Event != eventID { + flushEvent() + eventID = k.Event + groupID = k.Group + } + if k.Group != groupID { + flushGroup() + groupID = k.Group + } + entry := wire.AppendVarint(nil, 1, k.Task) + entry = wire.AppendVarint(entry, 2, current[k]) + rows = wire.AppendBytes(rows, 1, entry) + } + flushEvent() + // Consume only this request's baseline; the next BeforeDispatch takes a fresh + // snapshot even when a transaction rolls back or the session changes. + s.beforeMissions = nil + return out, nil } diff --git a/go/internal/server/domain/events/tasks/service.go b/go/internal/server/domain/events/tasks/service.go new file mode 100644 index 0000000..ee9055d --- /dev/null +++ b/go/internal/server/domain/events/tasks/service.go @@ -0,0 +1,435 @@ +// Package eventtasks owns attendance, scheduled missions and pass progress. +// Calendar policy is supplied by events.Resolver; requests never create dates +// or manufacture progress. The parent request transaction includes all saves. +package eventtasks + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/domain/events" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "errors" + "fmt" + "strconv" + + "time" +) + +type Economy interface { + Apply(ctx command.Context, _ string, _ []gamedata.Reward, _ []gamedata.Reward) ([]byte, error) +} + +// AttendanceMailIssuer delivers attendance attachments inside the parent +// account transaction. Its durable identity prevents duplicate mail on replay. +type AttendanceMailIssuer interface { + IssueAttachmentsOnce(ctx command.Context, identity, title, body string, rewards []gamedata.Reward, sentAt time.Time) error +} + +type attendance struct { + Group, Count uint64 + LastDay string + Obtained map[string]bool + History map[string]bool +} +type mission struct { + Value uint64 + Claimed bool + Period string +} +type pass struct { + Exp uint64 + Premium bool + Claimed map[string]bool +} +type receipt struct { + Digest string + Code int + Reply []byte +} +type snapshot struct { + Version string + NewbieStep uint64 + Attendance map[string]*attendance + Missions map[string]*mission + Passes map[string]*pass + LoginDays map[string]int64 + Receipts map[string]receipt +} +type Service struct { + provider InventoryProvider + before InventorySnapshot + observationVersion string + inventoryReady bool + beforeMissions map[missionNoticeKey]uint64 + visibleMissions map[missionNoticeKey]uint64 + visibleVersion string + + store stateio.Store + design *gamedata.EventTasksDesign + registry events.Resolver + economy Economy + state snapshot + now func() time.Time + + unlocked func(command.Context, uint64, uint64) bool + authorizeCash func(command.Context, uint64, uint64) bool + attendancePremium func(command.Context, uint64) bool + attendanceMail AttendanceMailIssuer + associated func(events.Schedule) uint64 +} + +func Open(ctx command.Context, store stateio.Store, d *gamedata.EventTasksDesign, r events.Resolver, e Economy) (*Service, error) { + if store == nil || d == nil || r == nil || e == nil { + return nil, errors.New("eventtasks: missing dependency") + } + s := &Service{store: store, design: d, registry: r, economy: e, now: time.Now} + b, err := store.Load(ctx.State, "eventtasks") + if err != nil { + return nil, err + } + if b != nil { + if err = stateio.RequireExactJSONObject(b, "Version", "NewbieStep", "Attendance", "Missions", "Passes", "LoginDays", "Receipts"); err != nil { + return nil, err + } + if err = json.Unmarshal(b, &s.state); err != nil { + return nil, err + } + if s.state.Version != versionconfig.State() || s.state.Attendance == nil || s.state.Missions == nil || s.state.Passes == nil || s.state.LoginDays == nil || s.state.Receipts == nil { + return nil, errors.New("eventtasks: invalid state layout") + } + for _, a := range s.state.Attendance { + if a == nil || a.Obtained == nil || a.History == nil { + return nil, errors.New("eventtasks: invalid attendance") + } + } + for _, m := range s.state.Missions { + if m == nil { + return nil, errors.New("eventtasks: invalid mission") + } + } + for _, p := range s.state.Passes { + if p == nil || p.Claimed == nil { + return nil, errors.New("eventtasks: invalid pass") + } + } + } else { + s.state.Version = versionconfig.State() + } + s.init() + return s, nil +} +func (s *Service) init() { + if s.state.Attendance == nil { + s.state.Attendance = map[string]*attendance{} + } + if s.state.Missions == nil { + s.state.Missions = map[string]*mission{} + } + if s.state.Passes == nil { + s.state.Passes = map[string]*pass{} + } + if s.state.LoginDays == nil { + s.state.LoginDays = map[string]int64{} + } + if s.state.Receipts == nil { + s.state.Receipts = map[string]receipt{} + } +} + +func (s *Service) AttachUnlockResolver(f func(command.Context, uint64, uint64) bool) { s.unlocked = f } +func (s *Service) AttachCashAuthorization(f func(command.Context, uint64, uint64) bool) { + s.authorizeCash = f +} +func (s *Service) AttachAttendancePremium(f func(command.Context, uint64) bool) { + s.attendancePremium = f +} +func (s *Service) AttachAssociatedMissionGroup(f func(events.Schedule) uint64) { s.associated = f } + +func (s *Service) AttachAttendanceMail(issuer AttendanceMailIssuer) { s.attendanceMail = issuer } + +func (s *Service) mailAttendance(ctx command.Context, identity string, rewards []gamedata.Reward) error { + if s.attendanceMail == nil { + return errors.New("eventtasks: attendance mailbox unavailable") + } + return s.attendanceMail.IssueAttachmentsOnce(ctx, identity, "Attendance Rewards", "Your attendance rewards are ready. Please claim the attachments in your mailbox.", rewards, s.now()) +} + +func (s *Service) SetNewbieStep(ctx command.Context, step uint64) error { + + if step > s.state.NewbieStep { + s.state.NewbieStep = step + } + return s.save(ctx) +} +func (s *Service) NewbieStep() uint64 { return s.state.NewbieStep } +func (s *Service) save(ctx command.Context) error { + b, e := json.Marshal(s.state) + if e != nil { + return e + } + return s.store.Save(ctx.State, "eventtasks", b) +} + +func key(v ...uint64) string { + var k string + for i, n := range v { + if i > 0 { + k += "/" //nolint:modernize // stringsbuilder + } + k += strconv.FormatUint(n, 10) + } + return k +} + +// scheduleKey preserves the public UID where it is present. UID zero is a +// protocol default shared by permanent groups, so its local identity also +// includes the event type and design group. +func scheduleKey(v events.Schedule) string { + if v.UID != 0 { + return key(v.UID) + } + return key(0, v.Type, v.ID, v.SubID) +} +func (s *Service) active(v events.Schedule) bool { + n := s.now().UnixMilli() + return n >= v.Start && n < v.End +} +func (s *Service) find(t, id uint64) (events.Schedule, error) { + rows := s.registry.List() + if t == 4 { + rows = s.taskSchedules() + } + for _, v := range rows { + if v.Type == t && v.ID == id && s.active(v) { + return v, nil + } + } + return events.Schedule{}, fmt.Errorf("eventtasks: event %d/%d inactive", t, id) +} + +func (s *Service) taskSchedules() []events.Schedule { + rows := s.registry.List() + var out []events.Schedule + seen := map[uint64]bool{} + for _, v := range rows { + if v.Type == 4 { + out = append(out, v) + if s.active(v) { + seen[v.ID] = true + } + } + } + for _, v := range rows { + if v.Type == 5 && s.active(v) { + g := s.design.Passes[v.ID].MissionGroup + if g > 0 && !seen[g] { + v.Type = 4 + v.ID = g + out = append(out, v) + seen[g] = true + } + } + } + if s.associated != nil { + for _, v := range rows { + if !s.active(v) { + continue + } + g := s.associated(v) + if g > 0 && !seen[g] { + v.Type = 4 + v.ID = g + out = append(out, v) + seen[g] = true + } + } + } + return out +} +func (s *Service) resolve(uid uint64) (events.Schedule, error) { + v, e := s.registry.Resolve(uid) + if e != nil { + return v, e + } + if !s.active(v) { + return v, errors.New("eventtasks: event expired") + } + return v, nil +} +func (s *Service) day() string { return s.now().UTC().Format("2006-01-02") } +func (s *Service) attendance(v events.Schedule) *attendance { + k := scheduleKey(v) + a := s.state.Attendance[k] + if a == nil { + g := s.design.Attendance[v.ID].Group + var first uint64 + for _, x := range s.design.AttendanceGroups { + if x.Group == g && (first == 0 || x.ID < first) { + first = x.ID + } + } + if first > 0 { + g = first + } + a = &attendance{Group: g, Obtained: map[string]bool{}, History: map[string]bool{}} + s.state.Attendance[k] = a + } + return a +} + +// attendanceRewards is shared by automatic and explicit claims. Obtained is +// the current cycle ledger; History is the client-visible historical ledger. +func (s *Service) attendanceRewards(ctx command.Context, v events.Schedule, a *attendance, group, id uint64) []gamedata.Reward { + if a.Obtained[key(group, id)] { + return nil + } + if v.Type == 0 { + if group != a.Group { + return nil + } + for _, r := range s.design.AttendanceRewards[group] { + if r.ID != id || r.Day > a.Count { + continue + } + var rewards []gamedata.Reward + if r.Basic.Count > 0 { + rewards = append(rewards, r.Basic) + } + if ticket := s.design.Attendance[v.ID].Ticket; ticket != 0 && s.attendancePremium != nil && s.attendancePremium(ctx, ticket) && r.Premium.Count > 0 { + rewards = append(rewards, r.Premium) + } + return rewards + } + } else if v.Type == 1 && group == v.ID && id > 0 { + box, ok := s.design.LimitRewards[[2]uint64{group, id}] + date := time.UnixMilli(v.Start).UTC().AddDate(0, 0, int(id)-1).Format("2006-01-02") + _, logged := s.state.LoginDays[date] + if ok && box > 0 && logged && date == s.day() { + return []gamedata.Reward{{Type: 9, ID: box, Count: 1}} + } + } + return nil +} +func (s *Service) pass(v events.Schedule) *pass { + k := scheduleKey(v) + p := s.state.Passes[k] + if p == nil { + p = &pass{Claimed: map[string]bool{}} + s.state.Passes[k] = p + } + return p +} +func (s *Service) mission(v events.Schedule, id uint64) *mission { + k := scheduleKey(v) + "/" + key(id) + m := s.state.Missions[k] + if m == nil { + m = &mission{} + s.state.Missions[k] = m + } + g := s.design.MissionGroups[v.ID] + period := "" + if g.Type == 1 { + period = s.day() + } + if g.Type == 2 { + y, w := s.now().UTC().ISOWeek() + period = fmt.Sprintf("%d-%d", y, w) + } + if m.Period != period { + *m = mission{Period: period} + } + return m +} +func (s *Service) availableTask(ctx command.Context, v events.Schedule, t gamedata.EventTask) bool { + g := s.design.MissionGroups[v.ID] + for i, n := range g.Groups { + if n == t.Group { + if g.Type == 3 && int64(i) > (s.now().UnixMilli()-v.Start)/86400000 { + return false + } + return t.UnlockPack == 0 && t.UnlockQuest == 0 || s.unlocked != nil && s.unlocked(ctx, t.UnlockPack, t.UnlockQuest) + } + } + return false +} + +// RecordEvent is called by authoritative gameplay. It returns no rewards and +// caps every matching task at its design threshold. +func (s *Service) RecordEvent(ctx command.Context, condition, sub, count uint64, unlocked func(command.Context, uint64, uint64) bool) error { + + if !s.recordEventLocked(ctx, condition, sub, count, unlocked) { + return nil + } + return s.save(ctx) +} + +// recordEventLocked reports every persistent mutation, including initializing +// a matching task or rolling its period. Irrelevant and already capped events +// leave the account snapshot untouched. +func (s *Service) recordEventLocked(ctx command.Context, condition, sub, count uint64, unlocked func(command.Context, uint64, uint64) bool) bool { + changed := false + if count == 0 { + return false + } + for _, v := range s.taskSchedules() { + if v.Type != 4 || !s.active(v) { + continue + } + for _, t := range s.design.Missions { + if t.Type != condition || t.SubType != 0 && t.SubType != sub || condition == 349 && t.SubType != sub || !s.availableTask(ctx, v, t) { + continue + } + if (t.UnlockPack > 0 || t.UnlockQuest > 0) && (unlocked == nil || !unlocked(ctx, t.UnlockPack, t.UnlockQuest)) { + continue + } + match := len(t.Params) == 0 + for _, n := range t.Params { + if n == sub { + match = true + } + } + if !match { + continue + } + previous := s.state.Missions[scheduleKey(v)+"/"+key(t.ID)] + var prior mission + if previous != nil { + prior = *previous + } + m := s.mission(v, t.ID) + if previous == nil || prior != *m { + changed = true + } + if m.Claimed { + continue + } + if m.Value >= t.Target { + continue + } + if count >= t.Target-m.Value { + m.Value = t.Target + } else { + m.Value += count + } + changed = true + } + } + return changed +} + +// RecordLogin shares the same daily marker as Attendance, preventing repeated +// reconnects from inflating event login tasks. Main should invoke this once +// through the regular mission observer before serving attendance queries. +func (s *Service) RecordLogin(ctx command.Context, unlocked func(command.Context, uint64, uint64) bool) error { + + marker := "mission-login:" + s.day() + if _, ok := s.state.LoginDays[marker]; ok { + + return nil + } + s.state.LoginDays[marker] = s.now().UnixMilli() + + return s.RecordEvent(ctx, 0, 0, 1, unlocked) +} diff --git a/go/internal/server/domain/identity/registration_test.go b/go/internal/server/domain/identity/registration_test.go new file mode 100644 index 0000000..588b658 --- /dev/null +++ b/go/internal/server/domain/identity/registration_test.go @@ -0,0 +1,129 @@ +package identity_test + +import ( + "bd2server/internal/server/domain/identity" + identitystore "bd2server/internal/server/storage/identity" + "context" + "errors" + "path/filepath" + "testing" + "time" +) + +// Registration authority must distinguish players even when clients use the +// same device flow. Admin authority belongs to the retained first identity. +func TestPlayerIdentityAndAdministrativeAuthorityRemainSeparate(t *testing.T) { + path := filepath.Join(t.TempDir(), "identity.db") + open := func() *identitystore.Store { + t.Helper() + key := make([]byte, 32) + for i := range key { + key[i] = byte(i + 1) + } + s, e := identitystore.Open(path, key) + if e != nil { + t.Fatal(e) + } + return s + } + store := open() + service, e := identity.New(identity.Config{Providers: map[string]string{"discord": "client"}, DeviceTTL: time.Minute, AccessTTL: time.Hour, RefreshTTL: 24 * time.Hour}, store) + if e != nil { + t.Fatal(e) + } + login := func(subject string) identity.TokenSet { + t.Helper() + device, e := service.CreateDevice("discord", "127.0.0.1") + if e != nil { + t.Fatal(e) + } + auth, e := service.Start("discord", device.ID, device.StartTicket) + if e != nil { + t.Fatal(e) + } + if e = service.CompleteDevice(auth.ID, "discord", identity.ProviderIdentity{Issuer: "https://discord.com", Subject: subject}); e != nil { + t.Fatal(e) + } + result, e := service.Poll(device.ID, device.Secret) + if e != nil { + t.Fatal(e) + } + if _, e = service.Poll(device.ID, device.Secret); e == nil { + t.Fatal("device credentials delivered twice") + } + return result.Tokens + } + a, b := login("111"), login("222") + aID, e := service.ValidateAccess(a.AccessToken) + if e != nil { + t.Fatal(e) + } + bID, e := service.ValidateAccess(b.AccessToken) + if e != nil { + t.Fatal(e) + } + if aID == bID || aID == "" || bID == "" { + t.Fatal("distinct provider subjects share one asset owner") + } + profileA, e := store.GameIdentity(context.Background(), aID) + if e != nil { + t.Fatal(e) + } + profileB, e := store.GameIdentity(context.Background(), bID) + if e != nil { + t.Fatal(e) + } + if profileA.OwnerIndex <= 0 || profileB.OwnerIndex <= 0 || profileA.OwnerIndex == profileB.OwnerIndex || profileA.UserID == profileB.UserID { + t.Fatal("different asset owners share client-visible identity") + } + owner, e := store.OwnerAccountID(context.Background()) + if e != nil || owner != aID { + t.Fatalf("original owner not retained: %s %v", owner, e) + } + if _, e = service.AuthorizeAdministrator(a.AccessToken); e != nil { + t.Fatal("original owner lost authority", e) + } + if _, e = service.AuthorizeAdministrator(b.AccessToken); !errors.Is(e, identity.ErrUnauthorized) { + t.Fatal("new player gained administrative authority", e) + } + refreshed, e := service.Refresh(a.RefreshToken, "attempt-owner-00001") + if e != nil { + t.Fatal(e) + } + retried, e := service.Refresh(a.RefreshToken, "attempt-owner-00001") + if e != nil || refreshed.AccessToken != retried.AccessToken || refreshed.RefreshToken != retried.RefreshToken { + t.Fatal("network retry issued different credentials") + } + if _, e = service.Refresh(a.RefreshToken, "attempt-owner-00002"); e == nil { + t.Fatal("used refresh token accepted as new operation") + } + if _, e = service.ValidateAccess(refreshed.AccessToken); e == nil { + t.Fatal("replayed owner's family remained valid") + } + if actual, e := service.ValidateAccess(b.AccessToken); e != nil || actual != bID { + t.Fatal("owner replay revoked unrelated player", e) + } + if e = store.Close(); e != nil { + t.Fatal(e) + } + store = open() + defer func() { _ = store.Close() }() + service, e = identity.New(identity.Config{Providers: map[string]string{"discord": "client"}, DeviceTTL: time.Minute, AccessTTL: time.Hour, RefreshTTL: 24 * time.Hour}, store) + if e != nil { + t.Fatal(e) + } + if actual, e := service.ValidateAccess(b.AccessToken); e != nil || actual != bID { + t.Fatal("restart mixed player credentials", e) + } + retainedA, e := store.GameIdentity(context.Background(), aID) + if e != nil || retainedA != profileA { + t.Fatal("restart changed original player public identity", e) + } + retainedB, e := store.GameIdentity(context.Background(), bID) + if e != nil || retainedB != profileB { + t.Fatal("restart changed other player public identity", e) + } + if owner, e = store.OwnerAccountID(context.Background()); e != nil || owner != aID { + t.Fatal("restart reassigned administrative owner") + } +} diff --git a/go/internal/server/domain/identity/repository.go b/go/internal/server/domain/identity/repository.go new file mode 100644 index 0000000..e0f887d --- /dev/null +++ b/go/internal/server/domain/identity/repository.go @@ -0,0 +1,104 @@ +package identity + +import "time" + +type ProviderIdentity struct{ Issuer, Subject string } +type Config struct { + Providers map[string]string + AccessTTL, RefreshTTL, DeviceTTL time.Duration +} +type TokenSet struct { + Provider, AccessToken, RefreshToken string + AccessExpiresIn, RefreshExpiresIn int64 +} +type StoredRefreshResult struct { + Provider, AccessToken, RefreshToken string + AccessExpiresAt, RefreshExpiresAt int64 +} + +func (r StoredRefreshResult) Tokens(now int64) TokenSet { + return TokenSet{Provider: r.Provider, AccessToken: r.AccessToken, RefreshToken: r.RefreshToken, AccessExpiresIn: max(r.AccessExpiresAt-now, 0), RefreshExpiresIn: max(r.RefreshExpiresAt-now, 0)} +} + +type CreatedDevice struct{ ID, Secret, StartTicket string } +type Authorization struct{ ID, State, Verifier, Nonce string } +type StartRecord struct { + Matches bool + Provider, Status string + Expires int64 +} +type AuthorizationRecord struct { + ID, Provider, Status string + Expires int64 + VerifierCipher, NonceCipher []byte +} +type AccountRecord struct{ ID, Status string } +type DeviceRecord struct { + Matches bool + Status, ErrorCode string + Expires int64 + ResultCipher []byte +} +type RefreshRecord struct { + FamilyID, AccountID, Provider, AccountStatus string + TokenExpires, FamilyExpires int64 + Used, Revoked bool +} +type RefreshAttempt struct { + Matches bool + Result StoredRefreshResult +} +type AccessRecord struct { + FamilyID, AccountID, AccountStatus string + Expires int64 + Revoked bool +} +type DeviceInsert struct { + ID, ClientIP, Secret, StartTicket, Provider string + Created, Expires int64 +} +type FamilyInsert struct { + ID, AccountID, Provider string + Created, Expires int64 +} + +type Repository interface { + Now() time.Time + Begin() (Transaction, error) + StartRecord(id, ticket string) (StartRecord, error) + StartAuthorization(id, state, verifier, nonce string) (bool, error) + CancelAuthorization(provider, state string, now int64) (bool, error) + AuthorizationRecord(state string) (AuthorizationRecord, error) + AuthorizationSecrets(AuthorizationRecord) (verifier, nonce []byte, err error) + FailAuthorization(id string) error + DecodeDeviceResult(id string, cipher []byte) (TokenSet, error) + AccessToken(token string) (AccessRecord, error) + AdministratorAccountID() (string, error) +} + +type Transaction interface { + CleanupExpired(now int64) error + PendingDevices(clientIP string, now int64) (int, error) + InsertDevice(DeviceInsert) error + Identity(ProviderIdentity) (AccountRecord, bool, error) + AccountCount() (int, error) + SetAdministratorAccount(accountID string) error + RejectDevice(id, provider, code string) (bool, error) + InsertAccount(accountID, provider string, identity ProviderIdentity, now int64) error + TouchIdentity(identity ProviderIdentity, now int64) error + InsertFamily(FamilyInsert) error + CompleteDevice(id, provider string, result TokenSet) (bool, error) + PollDevice(id, secret string) (DeviceRecord, error) + ConsumeDevice(id string) (bool, error) + RefreshToken(token string) (RefreshRecord, error) + RefreshAttempt(familyID, attemptID, requestToken string) (RefreshAttempt, bool, error) + ConsumeRefreshToken(token string, now int64) (bool, error) + RevokeFamily(id string, now int64) (bool, error) + DeleteFamilyAccess(id string) error + InsertAccess(token, familyID, accountID string, created, expires int64) error + InsertRefresh(token, familyID string, created, expires int64) error + InsertRefreshAttempt(familyID, attemptID, requestToken string, result StoredRefreshResult, created, expires int64) error + AccessToken(token string) (AccessRecord, error) + Commit() error + Rollback() error +} diff --git a/go/internal/server/domain/identity/service.go b/go/internal/server/domain/identity/service.go new file mode 100644 index 0000000..2b77e95 --- /dev/null +++ b/go/internal/server/domain/identity/service.go @@ -0,0 +1,429 @@ +package identity + +import ( + "crypto/rand" + "encoding/base64" + "errors" + "time" +) + +type FailureKind uint8 + +const ( + Unavailable FailureKind = iota + Invalid + Forbidden + Expired + Conflict + Unauthorized + TooManyPending +) + +type Failure struct { + Kind FailureKind + Message string + RefreshInvalid bool +} + +func (e *Failure) Error() string { return e.Message } +func failure(kind FailureKind, message string) error { return &Failure{Kind: kind, Message: message} } + +var ( + ErrUnauthorized = errors.New("auth: unauthorized") + ErrConsumed = errors.New("auth: consumed") + ErrNotAllowed = errors.New("auth: identity is not allowed") +) + +type Service struct { + config Config + store Repository +} + +func New(config Config, store Repository) (*Service, error) { + if store == nil { + return nil, errors.New("auth: missing identity repository") + } + return &Service{config: config, store: store}, nil +} +func (s *Service) Now() time.Time { return s.store.Now() } +func randomToken(size int) (string, error) { + b := make([]byte, size) + if _, e := rand.Read(b); e != nil { + return "", e + } + return base64.RawURLEncoding.EncodeToString(b), nil +} +func (s *Service) CreateDevice(provider, clientIP string) (CreatedDevice, error) { + if _, ok := s.config.Providers[provider]; !ok { + return CreatedDevice{}, failure(Invalid, "provider is not enabled") + } + id, e := randomToken(18) + if e != nil { + return CreatedDevice{}, e + } + secret, e := randomToken(32) + if e != nil { + return CreatedDevice{}, e + } + ticket, e := randomToken(32) + if e != nil { + return CreatedDevice{}, e + } + now := s.Now() + tx, e := s.store.Begin() + if e != nil { + return CreatedDevice{}, e + } + defer func() { _ = tx.Rollback() }() + if e = tx.CleanupExpired(now.Unix()); e != nil { + return CreatedDevice{}, e + } + pending, e := tx.PendingDevices(clientIP, now.Unix()) + if e != nil { + return CreatedDevice{}, e + } + if pending >= 5 { + return CreatedDevice{}, failure(TooManyPending, "too many pending login transactions") + } + if e = tx.InsertDevice(DeviceInsert{ID: id, ClientIP: clientIP, Secret: secret, StartTicket: ticket, Provider: provider, Created: now.Unix(), Expires: now.Add(s.config.DeviceTTL).Unix()}); e != nil { + return CreatedDevice{}, e + } + if e = tx.Commit(); e != nil { + return CreatedDevice{}, e + } + return CreatedDevice{ID: id, Secret: secret, StartTicket: ticket}, nil +} +func (s *Service) Start(provider, id, ticket string) (Authorization, error) { + row, e := s.store.StartRecord(id, ticket) + if e != nil || !row.Matches || row.Provider != provider || row.Status != "created" { + return Authorization{}, failure(Forbidden, "invalid login transaction") + } + if s.Now().Unix() >= row.Expires { + return Authorization{}, failure(Expired, "login transaction expired") + } + state, e := randomToken(32) + if e != nil { + return Authorization{}, e + } + verifier, e := randomToken(32) + if e != nil { + return Authorization{}, e + } + nonce, e := randomToken(24) + if e != nil { + return Authorization{}, e + } + updated, e := s.store.StartAuthorization(id, state, verifier, nonce) + if e != nil { + return Authorization{}, failure(Conflict, "could not start authorization") + } + if !updated { + return Authorization{}, failure(Conflict, "could not start authorization") + } + return Authorization{ID: id, State: state, Verifier: verifier, Nonce: nonce}, nil +} +func (s *Service) CancelAuthorization(provider, state string) error { + updated, e := s.store.CancelAuthorization(provider, state, s.Now().Unix()) + if e != nil { + return failure(Unavailable, "authorization state unavailable") + } + if !updated { + return failure(Forbidden, "invalid or expired authorization state") + } + return nil +} +func (s *Service) Authorization(provider, state string) (Authorization, error) { + row, e := s.store.AuthorizationRecord(state) + if e != nil || row.Provider != provider || row.Status != "authorizing" || s.Now().Unix() >= row.Expires { + return Authorization{}, failure(Forbidden, "invalid or expired authorization state") + } + verifier, nonce, e := s.store.AuthorizationSecrets(row) + if e != nil { + return Authorization{}, failure(Unavailable, "authorization state unavailable") + } + defer clear(verifier) + defer clear(nonce) + return Authorization{ID: row.ID, State: state, Verifier: string(verifier), Nonce: string(nonce)}, nil +} +func (s *Service) RejectProvider(id string) { _ = s.store.FailAuthorization(id) } +func ValidateProviderIdentity(provider string, id ProviderIdentity) error { + switch provider { + case "discord": + if id.Issuer != "https://discord.com" || len(id.Subject) == 0 || len(id.Subject) > 32 { + return errors.New("auth: invalid Discord identity") + } + for _, c := range id.Subject { + if c < '0' || c > '9' { + return errors.New("auth: invalid Discord identity") + } + } + case "google": + if id.Issuer != "https://accounts.google.com" || len(id.Subject) == 0 || len(id.Subject) > 255 { + return errors.New("auth: invalid Google identity") + } + default: + return errors.New("auth: unsupported identity provider") + } + return nil +} +func (s *Service) ValidateGoogleClaims(issuer, audience, subject, nonce, expectedNonce string, expires int64) bool { + validIssuer := issuer == "https://accounts.google.com" || issuer == "accounts.google.com" + return validIssuer && audience == s.config.Providers["google"] && subject != "" && nonce == expectedNonce && s.Now().Unix() < expires +} +func (s *Service) CompleteDevice(deviceID, provider string, id ProviderIdentity) error { + if e := ValidateProviderIdentity(provider, id); e != nil { + return e + } + now := s.Now() + tx, e := s.store.Begin() + if e != nil { + return e + } + defer func() { _ = tx.Rollback() }() + account, found, e := tx.Identity(id) + if e != nil { + return e + } + reject := func() error { + updated, e := tx.RejectDevice(deviceID, provider, "not_allowed") + if e != nil { + return e + } + if !updated { + return ErrConsumed + } + if e = tx.Commit(); e != nil { + return e + } + return ErrNotAllowed + } + if !found { + count, e := tx.AccountCount() + if e != nil { + return e + } + account.ID, e = randomToken(18) + if e != nil { + return e + } + if e = tx.InsertAccount(account.ID, provider, id, now.Unix()); e != nil { + return e + } + if count == 0 { + if e = tx.SetAdministratorAccount(account.ID); e != nil { + return e + } + } + account.Status = "active" + } else if e = tx.TouchIdentity(id, now.Unix()); e != nil { + return e + } + if account.Status != "active" { + return reject() + } + family, e := randomToken(18) + if e != nil { + return e + } + access, e := randomToken(32) + if e != nil { + return e + } + refresh, e := randomToken(32) + if e != nil { + return e + } + if e = tx.InsertFamily(FamilyInsert{ID: family, AccountID: account.ID, Provider: provider, Created: now.Unix(), Expires: now.Add(s.config.RefreshTTL).Unix()}); e != nil { + return e + } + if e = tx.InsertAccess(access, family, account.ID, now.Unix(), now.Add(s.config.AccessTTL).Unix()); e != nil { + return e + } + if e = tx.InsertRefresh(refresh, family, now.Unix(), now.Add(s.config.RefreshTTL).Unix()); e != nil { + return e + } + tokens := TokenSet{Provider: provider, AccessToken: access, RefreshToken: refresh, AccessExpiresIn: int64(s.config.AccessTTL.Seconds()), RefreshExpiresIn: int64(s.config.RefreshTTL.Seconds())} + updated, e := tx.CompleteDevice(deviceID, provider, tokens) + if e != nil { + return e + } + if !updated { + return ErrConsumed + } + return tx.Commit() +} + +type PollResult struct { + Status, ErrorCode string + Tokens TokenSet +} + +func (s *Service) Poll(id, secret string) (PollResult, error) { + tx, e := s.store.Begin() + if e != nil { + return PollResult{}, failure(Unavailable, "login result unavailable") + } + defer func() { _ = tx.Rollback() }() + row, e := tx.PollDevice(id, secret) + if e != nil || !row.Matches { + return PollResult{}, failure(Forbidden, "invalid device transaction") + } + if s.Now().Unix() >= row.Expires { + return PollResult{}, failure(Expired, "device transaction expired") + } + switch row.Status { + case "created", "authorizing": + return PollResult{Status: "pending"}, nil + case "failed": + return PollResult{Status: "failed", ErrorCode: row.ErrorCode}, nil + case "complete": + tokens, e := s.store.DecodeDeviceResult(id, row.ResultCipher) + if e != nil { + return PollResult{}, failure(Unavailable, "login result unavailable") + } + updated, e := tx.ConsumeDevice(id) + if e != nil || !updated { + return PollResult{}, failure(Expired, "login result already consumed") + } + if e = tx.Commit(); e != nil { + return PollResult{}, failure(Unavailable, "login result unavailable") + } + return PollResult{Status: "complete", Tokens: tokens}, nil + default: + return PollResult{}, failure(Expired, "device transaction consumed") + } +} +func ValidRefreshAttemptID(v string) bool { + if len(v) < 16 || len(v) > 128 { + return false + } + for _, c := range v { + valid := c >= 'a' && c <= 'z' || c >= 'A' && c <= 'Z' || c >= '0' && c <= '9' || c == '-' || c == '_' + if !valid { + return false + } + } + return true +} +func invalidRefresh(kind FailureKind, message string) error { + return &Failure{Kind: kind, Message: message, RefreshInvalid: true} +} +func (s *Service) Refresh(token, attemptID string) (TokenSet, error) { + if token == "" || !ValidRefreshAttemptID(attemptID) { + return TokenSet{}, failure(Invalid, "refresh_token and valid attempt_id required") + } + now := s.Now() + tx, e := s.store.Begin() + if e != nil { + return TokenSet{}, e + } + defer func() { _ = tx.Rollback() }() + if e = tx.CleanupExpired(now.Unix()); e != nil { + return TokenSet{}, e + } + row, e := tx.RefreshToken(token) + if e != nil || row.Revoked || row.AccountStatus != "active" || now.Unix() >= row.FamilyExpires { + return TokenSet{}, invalidRefresh(Unauthorized, "refresh token invalid") + } + saved, found, e := tx.RefreshAttempt(row.FamilyID, attemptID, token) + if e != nil { + return TokenSet{}, e + } + if found { + if !saved.Matches { + return TokenSet{}, invalidRefresh(Conflict, "refresh attempt_id already belongs to another request") + } + if saved.Result.Provider == "" || saved.Result.AccessToken == "" || saved.Result.RefreshToken == "" { + return TokenSet{}, failure(Unavailable, "refresh unavailable") + } + return saved.Result.Tokens(now.Unix()), nil + } + if now.Unix() >= row.TokenExpires { + return TokenSet{}, invalidRefresh(Unauthorized, "refresh token expired") + } + if row.Used { + if _, e = tx.RevokeFamily(row.FamilyID, now.Unix()); e != nil { + return TokenSet{}, e + } + if e = tx.Commit(); e != nil { + return TokenSet{}, e + } + return TokenSet{}, invalidRefresh(Unauthorized, "refresh token replayed") + } + access, e := randomToken(32) + if e != nil { + return TokenSet{}, e + } + refresh, e := randomToken(32) + if e != nil { + return TokenSet{}, e + } + updated, e := tx.ConsumeRefreshToken(token, now.Unix()) + if e != nil { + return TokenSet{}, e + } + if !updated { + return TokenSet{}, invalidRefresh(Unauthorized, "refresh token invalid") + } + if e = tx.DeleteFamilyAccess(row.FamilyID); e != nil { + return TokenSet{}, e + } + accessExpiry := now.Add(s.config.AccessTTL).Unix() + refreshExpiry := min(row.FamilyExpires, now.Add(s.config.RefreshTTL).Unix()) + if e = tx.InsertAccess(access, row.FamilyID, row.AccountID, now.Unix(), accessExpiry); e != nil { + return TokenSet{}, e + } + if e = tx.InsertRefresh(refresh, row.FamilyID, now.Unix(), refreshExpiry); e != nil { + return TokenSet{}, e + } + result := StoredRefreshResult{Provider: row.Provider, AccessToken: access, RefreshToken: refresh, AccessExpiresAt: accessExpiry, RefreshExpiresAt: refreshExpiry} + if e = tx.InsertRefreshAttempt(row.FamilyID, attemptID, token, result, now.Unix(), refreshExpiry); e != nil { + return TokenSet{}, e + } + if e = tx.Commit(); e != nil { + return TokenSet{}, e + } + return result.Tokens(now.Unix()), nil +} +func (s *Service) Revoke(token string) error { + now := s.Now().Unix() + tx, e := s.store.Begin() + if e != nil { + return e + } + defer func() { _ = tx.Rollback() }() + row, e := tx.AccessToken(token) + if e != nil || row.AccountStatus != "active" || row.Revoked || now >= row.Expires { + return failure(Unauthorized, "access token invalid") + } + updated, e := tx.RevokeFamily(row.FamilyID, now) + if e != nil { + return e + } + if !updated { + return failure(Unauthorized, "access token invalid") + } + return tx.Commit() +} +func (s *Service) ValidateAccess(token string) (string, error) { + if token == "" { + return "", ErrUnauthorized + } + row, e := s.store.AccessToken(token) + if e != nil || row.AccountStatus != "active" || row.Revoked || s.Now().Unix() >= row.Expires { + return "", ErrUnauthorized + } + return row.AccountID, nil +} + +func (s *Service) AuthorizeAdministrator(token string) (string, error) { + accountID, err := s.ValidateAccess(token) + if err != nil { + return "", err + } + owner, err := s.store.AdministratorAccountID() + if err != nil || owner == "" || owner != accountID { + return "", ErrUnauthorized + } + return accountID, nil +} diff --git a/go/internal/server/domain/inventory/character_source.go b/go/internal/server/domain/inventory/character_source.go new file mode 100644 index 0000000..0abf5b2 --- /dev/null +++ b/go/internal/server/domain/inventory/character_source.go @@ -0,0 +1,16 @@ +package inventory + +import "bd2server/internal/server/domain/command" + +type EquipmentCharacter struct { + InvenIndex uint64 + ID uint64 + TalentLevel uint64 + TalentExp uint64 + Response []byte +} +type EquipmentCharacterSource interface { + EquipmentCharacters() []EquipmentCharacter + EquipmentCharacter(ctx command.Context, _ uint64) (EquipmentCharacter, bool) + AddEquipmentTalentExperience(ctx command.Context, _ uint64, _ uint64, _ uint64) (EquipmentCharacter, error) +} diff --git a/go/internal/server/player/commerce_items.go b/go/internal/server/domain/inventory/commerce_items.go similarity index 85% rename from go/internal/server/player/commerce_items.go rename to go/internal/server/domain/inventory/commerce_items.go index 4b52131..b99cd3d 100644 --- a/go/internal/server/player/commerce_items.go +++ b/go/internal/server/domain/inventory/commerce_items.go @@ -1,18 +1,18 @@ -package player +package inventory import ( + "bd2server/internal/server/domain/command" "fmt" "math" ) // GrantCommerceOnce preserves the exact item expiry supplied by the commerce // entitlement policy. It shares the existing atomic inventory layout. -func (s *Inventory) GrantCommerceOnce(identity string, rewards []Item) ([]Item, error) { +func (s *Inventory) GrantCommerceOnce(ctx command.Context, identity string, rewards []Item) ([]Item, error) { if identity == "" { return nil, fmt.Errorf("player: missing commerce reward identity") } - s.mu.Lock() - defer s.mu.Unlock() + if s.owned.Granted[identity] { var out []Item for _, idx := range s.owned.GrantItems[identity] { @@ -55,17 +55,16 @@ func (s *Inventory) GrantCommerceOnce(identity string, rewards []Item) ([]Item, out = append(out, r) } next.Granted[identity] = true - if err := s.commitOwned(next); err != nil { + if err := s.commitOwned(ctx, next); err != nil { return nil, err } s.owned = next return out, nil } func (s *Inventory) ContentTicketExpiry(id uint64) uint64 { - s.mu.Lock() - defer s.mu.Unlock() + var expiry uint64 - for _, items := range [][]Item{s.starter.Items, s.owned.Items} { + for _, items := range [][]Item{s.starter, s.owned.Items} { for _, i := range items { if i.Type == 19 && i.ID == id && i.Count > 0 && i.ExpiryTime > expiry { expiry = i.ExpiryTime diff --git a/go/internal/server/domain/inventory/content_open.go b/go/internal/server/domain/inventory/content_open.go new file mode 100644 index 0000000..cfa164c --- /dev/null +++ b/go/internal/server/domain/inventory/content_open.go @@ -0,0 +1,29 @@ +package inventory + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/storage/stateio" +) + +const contentOpenPacketCode = 622 + +type contentOpenReceipt struct { + Digest string `json:"digest"` + Body []byte `json:"body"` +} + +type contentOpenState struct { + Receipts map[string]contentOpenReceipt `json:"receipts"` +} + +// ContentOpenService grants the completion ticket for the client's sole +// implemented ContentOpen type (1). The prerequisite ticket is never spent. +// Handle must run in the account request transaction: inventory and receipt +// writes commit together, and account recovery reconstructs Inventory after +// a dirty rollback. Receipts are reloaded for every request. +type ContentOpenService struct { + design *gamedata.ContentOpeningDesign + inventory *Inventory + store stateio.Store + squadLevel func() (uint64, error) +} diff --git a/go/internal/server/player/entry_ledger.go b/go/internal/server/domain/inventory/entry_ledger.go similarity index 63% rename from go/internal/server/player/entry_ledger.go rename to go/internal/server/domain/inventory/entry_ledger.go index cb23aba..4e2e34c 100644 --- a/go/internal/server/player/entry_ledger.go +++ b/go/internal/server/domain/inventory/entry_ledger.go @@ -1,14 +1,15 @@ -package player +package inventory import ( "encoding/json" "fmt" - "bd2server/internal/server/stateio" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/storage/stateio" ) -func loadBoolEntries(store stateio.AtomicEntryStore, domain, bucket string) (map[string]bool, error) { - entries, err := store.ListEntries(domain, bucket) +func loadBoolEntries(ctx command.Context, store stateio.ScopedEntryStore, domain, bucket string) (map[string]bool, error) { + entries, err := store.ListEntries(ctx.State, domain, bucket) if err != nil { return nil, err } @@ -22,8 +23,8 @@ func loadBoolEntries(store stateio.AtomicEntryStore, domain, bucket string) (map return result, nil } -func loadUintEntries(store stateio.AtomicEntryStore, domain, bucket string) (map[string]uint64, error) { - entries, err := store.ListEntries(domain, bucket) +func loadUintEntries(ctx command.Context, store stateio.ScopedEntryStore, domain, bucket string) (map[string]uint64, error) { + entries, err := store.ListEntries(ctx.State, domain, bucket) if err != nil { return nil, err } diff --git a/go/internal/server/domain/inventory/equipment.go b/go/internal/server/domain/inventory/equipment.go new file mode 100644 index 0000000..7bd53c9 --- /dev/null +++ b/go/internal/server/domain/inventory/equipment.go @@ -0,0 +1,927 @@ +package inventory + +import ( + "bd2server/internal/server/design/gamedata" + "bd2server/internal/server/domain/command" + "bd2server/internal/server/platform/versionconfig" + "bd2server/internal/server/storage/stateio" + "encoding/json" + "errors" + "fmt" + "maps" + "math" + "reflect" + "sort" + "strconv" + "strings" + + "unicode/utf8" +) + +// Equipment is one server-owned equipment instance. The immutable definition +// (name, icon, slot and base stats) remains in the client's local GameData. +type Equipment struct { + InvenIndex uint64 `json:"inven_index"` + ID uint64 `json:"id"` + Level uint64 `json:"level"` + UseChar uint64 `json:"use_char,omitempty"` + KeepFlag uint64 `json:"keep_flag,omitempty"` + LockFlag uint64 `json:"lock_flag,omitempty"` + SortID uint64 `json:"sort_id,omitempty"` + Mark string `json:"mark,omitempty"` + MainOption []EquipmentOption `json:"main_option,omitempty"` + SubOption []EquipmentOption `json:"sub_option,omitempty"` + PrivateOption *EquipmentOption `json:"private_option,omitempty"` + Rank []uint64 `json:"rank,omitempty"` + UpgradeAttempts uint64 `json:"upgrade_attempts"` +} + +// EquipmentOption is the exact EquipmentOptionTable composite key used by an +// equipment instance. The client resolves the displayed stat curve locally. +type EquipmentOption struct { + GroupID uint64 `json:"group_id"` + ID uint64 `json:"id"` +} + +type equipmentSnapshot struct { + Version string `json:"version"` + NextIndex uint64 `json:"next_index"` + Equipment []Equipment `json:"-"` + Granted map[string]uint64 `json:"-"` +} + +// EquipmentInventory persists non-stackable equipment independently from +// ItemDBInfo inventory because the wire protocols are different types. +type EquipmentInventory struct { + store stateio.ScopedEntryStore + owned equipmentSnapshot + persisted equipmentSnapshot + corePresent bool + characters EquipmentCharacterSource + slots map[uint64]uint64 + upgrade *gamedata.EquipmentUpgradeDesign + craft *gamedata.EquipmentCraftDesign + smelting *gamedata.EquipmentSmeltingDesign + optionReroll *gamedata.EquipmentOptionRerollDesign + statDesign *gamedata.EquipmentStatDesign + wallet *Wallet + inventory *Inventory + + smeltCache map[string]smeltingReply + pendingReroll *equipmentOptionRerollPending + presets map[equipmentPresetKey]equipmentPreset +} + +type smeltingReply struct { + code int + body []byte +} + +type equipmentOptionRerollPending struct { + Equipment Equipment `json:"equipment"` +} + +func (s *EquipmentInventory) AttachUpgrade(ctx command.Context, design *gamedata.EquipmentUpgradeDesign, wallet *Wallet, inventory *Inventory) error { + if design == nil || wallet == nil || inventory == nil { + return errors.New("player: incomplete equipment upgrade configuration") + } + + s.upgrade, s.wallet, s.inventory = design, wallet, inventory + return nil +} + +func (s *EquipmentInventory) AttachCraft(ctx command.Context, design *gamedata.EquipmentCraftDesign) error { + if design == nil { + return errors.New("player: incomplete equipment crafting configuration") + } + + s.craft = design + return nil +} + +func (s *EquipmentInventory) AttachSmelting(ctx command.Context, design *gamedata.EquipmentSmeltingDesign, wallet *Wallet, inventory *Inventory) error { + if design == nil || wallet == nil || inventory == nil { + return errors.New("player: incomplete equipment smelting configuration") + } + + s.smelting, s.wallet, s.inventory = design, wallet, inventory + return nil +} + +func (s *EquipmentInventory) AttachOptionReroll(ctx command.Context, design *gamedata.EquipmentOptionRerollDesign, wallet *Wallet, inventory *Inventory) error { + if design == nil || wallet == nil || inventory == nil { + return errors.New("player: incomplete equipment option reroll configuration") + } + + s.optionReroll, s.wallet, s.inventory = design, wallet, inventory + if s.pendingReroll != nil { + if err := s.validatePendingRerollLocked(s.pendingReroll); err != nil { + return err + } + } + return nil +} + +// BeginLogin resets the in-memory request replay cache. A repeated protobuf +// sequence in one login must return the first refinement result without a +// second roll or charge. +func (s *EquipmentInventory) BeginLogin(ctx command.Context) { + id := ctx.SessionID + + if id == "" { + return + } + s.smeltCache = make(map[string]smeltingReply) +} + +func (s *EquipmentInventory) AttachSlots(ctx command.Context, slots map[uint64]uint64) error { + if len(slots) == 0 { + return errors.New("player: empty equipment slot design") + } + + s.slots = make(map[uint64]uint64, len(slots)) + for id, slot := range slots { + if id == 0 || slot > 4 { + return errors.New("player: invalid equipment slot design") + } + s.slots[id] = slot + } + return nil +} + +func (s *EquipmentInventory) AttachCharacters(ctx command.Context, characters EquipmentCharacterSource) error { + if characters == nil { + return errors.New("player: nil character store") + } + known := make(map[uint64]bool) + for _, character := range characters.EquipmentCharacters() { + known[character.InvenIndex] = true + } + + for key := range s.presets { + if !known[key.CharacterIndex] { + return fmt.Errorf("player: equipment preset references unknown character %d", key.CharacterIndex) + } + } + s.characters = characters + return nil +} + +func OpenEquipmentInventory(ctx command.Context, store stateio.Store) (*EquipmentInventory, error) { + entries, ok := store.(stateio.ScopedEntryStore) + if !ok { + return nil, errors.New("player: nil equipment store") + } + s := &EquipmentInventory{store: entries, smeltCache: make(map[string]smeltingReply), presets: make(map[equipmentPresetKey]equipmentPreset), owned: equipmentSnapshot{ + Version: versionconfig.State(), NextIndex: 910000001, Granted: map[string]uint64{}, + }} + data, err := store.Load(ctx.State, "equipment") + if err != nil { + return nil, err + } + if data != nil { + s.corePresent = true + if err := stateio.RequireExactJSONObject(data, "version", "next_index"); err != nil { + return nil, fmt.Errorf("player: incompatible equipment layout: %w", err) + } + var shape map[string]json.RawMessage + if err := json.Unmarshal(data, &shape); err != nil { + return nil, fmt.Errorf("player: decode equipment shape: %w", err) + } + for _, name := range []string{"equipment", "granted"} { + if _, exists := shape[name]; exists { + return nil, fmt.Errorf("player: equipment %s must use entries", name) + } + } + if err := json.Unmarshal(data, &s.owned); err != nil { + return nil, fmt.Errorf("player: decode equipment: %w", err) + } + } else if err := stateio.RequireNoEntries(entries, ctx.State, "equipment", "equipment", "granted", "reroll_pending", "presets"); err != nil { + return nil, fmt.Errorf("player: invalid equipment storage: %w", err) + } + if s.owned.Version != versionconfig.State() || s.owned.NextIndex < 910000001 { + return nil, errors.New("player: invalid saved equipment") + } + s.owned.Granted, err = loadUintEntries(ctx, entries, "equipment", "granted") + if err != nil { + return nil, err + } + rawEquipment, err := entries.ListEntries(ctx.State, "equipment", "equipment") + if err != nil { + return nil, err + } + for key, value := range rawEquipment { + var entry Equipment + index, parseErr := strconv.ParseUint(key, 10, 64) + var shape map[string]json.RawMessage + if parseErr != nil || json.Unmarshal(value, &shape) != nil || json.Unmarshal(value, &entry) != nil || entry.InvenIndex != index { + return nil, fmt.Errorf("player: invalid equipment entry %q", key) + } + if _, present := shape["upgrade_attempts"]; !present { + return nil, errors.New("player: equipment save requires upgrade_attempts; migrate the development save") + } + s.owned.Equipment = append(s.owned.Equipment, entry) + } + sort.Slice(s.owned.Equipment, func(i, j int) bool { return s.owned.Equipment[i].InvenIndex < s.owned.Equipment[j].InvenIndex }) + for _, entry := range s.owned.Equipment { + if len(entry.Rank) != 3 { + return nil, fmt.Errorf("player: equipment %d requires exactly three rank slots, found %d; repair the development save before starting", entry.InvenIndex, len(entry.Rank)) + } + } + pendingEntries, err := entries.ListEntries(ctx.State, "equipment", "reroll_pending") + if err != nil { + return nil, err + } + if len(pendingEntries) > 1 { + return nil, errors.New("player: multiple equipment option reroll candidates") + } + if raw, ok := pendingEntries["current"]; ok { + if err := stateio.RequireExactJSONObject(raw, "equipment"); err != nil { + return nil, fmt.Errorf("player: incompatible equipment option reroll candidate: %w", err) + } + var pending equipmentOptionRerollPending + if err := json.Unmarshal(raw, &pending); err != nil { + return nil, fmt.Errorf("player: decode equipment option reroll candidate: %w", err) + } + s.pendingReroll = &pending + if err := s.validatePendingRerollLocked(s.pendingReroll); err != nil { + return nil, err + } + } else if len(pendingEntries) != 0 { + return nil, errors.New("player: invalid equipment option reroll candidate key") + } + if err := s.loadEquipmentPresets(ctx, entries); err != nil { + return nil, err + } + s.persisted = cloneEquipmentSnapshot(s.owned) + return s, nil +} + +func (s *EquipmentInventory) EnsurePersisted(ctx command.Context) error { + + data, err := s.store.Load(ctx.State, "equipment") + if err != nil { + return err + } + if data != nil { + return nil + } + return s.commitLocked(ctx, cloneEquipmentSnapshot(s.owned), "initial account generation") +} + +// GrantOnce returns the same instance on a retry, allowing QuestClear response +// retries without duplicating ownership. +func (s *EquipmentInventory) GrantOnce(ctx command.Context, identity string, equipmentID uint64) (Equipment, error) { + if identity == "" || equipmentID == 0 { + return Equipment{}, errors.New("player: invalid equipment grant") + } + + if index := s.owned.Granted[identity]; index != 0 { + for _, current := range s.owned.Equipment { + if current.InvenIndex == index { + return current, nil + } + } + return Equipment{}, errors.New("player: equipment grant index is missing") + } + return s.grantLocked(ctx, identity, Equipment{ID: equipmentID}) +} + +// GrantGeneratedOnce saves an independently generated gacha instance. Retry +// calls return the original rolls rather than creating a second copy. +func (s *EquipmentInventory) GrantGeneratedOnce(ctx command.Context, identity string, entry Equipment) (Equipment, error) { + if identity == "" || entry.ID == 0 || len(entry.Rank) != 3 { + return Equipment{}, errors.New("player: invalid generated equipment") + } + + return s.grantLocked(ctx, identity, entry) +} + +func (s *EquipmentInventory) grantLocked(ctx command.Context, identity string, entry Equipment) (Equipment, error) { + if index := s.owned.Granted[identity]; index != 0 { + for _, current := range s.owned.Equipment { + if current.InvenIndex == index { + return current, nil + } + } + return Equipment{}, errors.New("player: equipment grant index is missing") + } + if len(entry.Rank) == 0 { + // Rank is a fixed three-slot client field (unlock thresholds +3/+6/+9). + // Even an unenhanced item must have three explicit zero values. + entry.Rank = []uint64{0, 0, 0} + } + if len(entry.Rank) != 3 { + return Equipment{}, errors.New("player: equipment requires exactly three rank slots") + } + entry.InvenIndex = s.owned.NextIndex + next := equipmentSnapshot{Version: s.owned.Version, NextIndex: s.owned.NextIndex + 1, + Equipment: append([]Equipment(nil), s.owned.Equipment...), Granted: make(map[string]uint64, len(s.owned.Granted)+1)} + next.Equipment = append(next.Equipment, entry) + maps.Copy(next.Granted, s.owned.Granted) + next.Granted[identity] = entry.InvenIndex + if err := s.commitLocked(ctx, next, "grant"); err != nil { + return Equipment{}, err + } + return entry, nil +} + +const ( + equipUpgradeSuccess = iota + equipUpgradeFail + equipUpgradeStopMaxLevel + equipUpgradeStopSuccess + equipUpgradeStopNotEnough + equipUpgradeStopGoldLimit + equipUpgradeStopTargetLevel + equipUpgradeStopMaxTryCount +) + +func (s *EquipmentInventory) smeltingCacheKey(ctx command.Context, kind string, seq uint64) string { + return kind + ":" + ctx.SessionID + ":seq:" + strconv.FormatUint(seq, 10) +} + +func (s *EquipmentInventory) smeltingEquipmentLocked(index uint64) (int, Equipment, error) { + if s.smelting == nil || s.wallet == nil || s.inventory == nil { + return -1, Equipment{}, errors.New("player: equipment smelting unavailable") + } + position := s.equipmentPositionLocked(index) + if position < 0 { + return -1, Equipment{}, fmt.Errorf("player: unknown equipment %d", index) + } + entry := s.owned.Equipment[position] + design, ok := s.smelting.Equipment[entry.ID] + if !ok || entry.Level != design.MaxLevel || len(entry.Rank) != 3 { + return -1, Equipment{}, fmt.Errorf("player: equipment %d is not ready for smelting", index) + } + if _, err := s.smelting.Score(entry.ID, entry.Rank); err != nil { + return -1, Equipment{}, fmt.Errorf("player: equipment %d has invalid smelting rank: %w", index, err) + } + return position, entry, nil +} + +func validateSmeltingMaterials(costs []gamedata.PromotionCost, materials []Item) (gold, mileageMaterial uint64, err error) { + want := make(map[[2]uint64]uint64, len(costs)) + for _, cost := range costs { + want[[2]uint64{cost.Type, cost.ID}] += cost.Count + } + got := make(map[[2]uint64]uint64, len(materials)) + for _, item := range materials { + got[[2]uint64{item.Type, item.ID}] += item.Count + if item.Type == 4 { + gold += item.Count + } else { + mileageMaterial += item.Count + } + } + if len(got) != len(want) { + return 0, 0, errors.New("player: equipment smelting material kinds mismatch") + } + for key, count := range want { + if got[key] != count { + return 0, 0, fmt.Errorf("player: equipment smelting material %d/%d=%d want=%d", key[0], key[1], got[key], count) + } + } + return gold, mileageMaterial, nil +} + +func (s *EquipmentInventory) selectSmeltingCosts(ctx command.Context, costs []gamedata.PromotionCost, attempts uint64) ([]Item, uint64, error) { + var result []Item + var gold uint64 + for _, cost := range costs { + if attempts != 0 && cost.Count > ^uint64(0)/attempts { + return nil, 0, errors.New("player: equipment smelting cost overflow") + } + count := cost.Count * attempts + switch cost.Type { + case 4: + if cost.ID != 0 || gold != 0 || !s.wallet.CanSpendGold(count) { + return nil, 0, errors.New("player: insufficient equipment smelting gold") + } + gold = count + result = append(result, Item{Type: 4, Count: count}) + case 8: + items, err := s.inventory.SelectMutable(ctx, cost.Type, cost.ID, count) + if err != nil { + return nil, 0, err + } + result = append(result, items...) + default: + return nil, 0, fmt.Errorf("player: unsupported equipment smelting cost type %d", cost.Type) + } + } + return result, gold, nil +} + +// commitSmeltingLocked keeps refinement's three typed snapshots synchronized +// in memory. Caller holds equipment.mu; this method takes the remaining locks +// in wallet -> inventory order, calculates every candidate before writing, +// then publishes all three only after their saves succeed. The surrounding +// request transaction supplies durable all-or-none recovery across the writes. +func (s *EquipmentInventory) commitSmeltingLocked(ctx command.Context, + nextEquipment equipmentSnapshot, + consumed []Item, + gold, mileageMaterial uint64, + identity, operation string, +) (Currency, uint64, error) { + if identity == "" || mileageMaterial == 0 || s.wallet == nil || s.inventory == nil { + return Currency{}, 0, errors.New("player: invalid transactional equipment smelting") + } + + nextWallet := cloneWallet(s.wallet.state) + if nextWallet.Spent[identity] { + return Currency{}, 0, errors.New("player: equipment smelting request was already committed") + } + if nextWallet.Gold < gold { + return Currency{}, 0, errors.New("player: insufficient gold for equipment smelting") + } + nextWallet.Gold -= gold + nextWallet.Spent[identity] = true + threshold, rewardCount := s.smelting.Mileage.UseCount, s.smelting.Mileage.RewardCount + if threshold == 0 || rewardCount == 0 || nextWallet.EquipMileageExchangeGage >= threshold || + math.MaxUint64-nextWallet.EquipMileageExchangeGage < mileageMaterial { + return Currency{}, 0, errors.New("player: invalid equipment smelting gauge") + } + total := nextWallet.EquipMileageExchangeGage + mileageMaterial + exchanges := total / threshold + earned := exchanges * rewardCount + if exchanges != 0 && earned/exchanges != rewardCount || math.MaxUint64-nextWallet.EquipMileage < earned { + return Currency{}, 0, errors.New("player: equipment mileage overflow") + } + nextWallet.EquipMileageExchangeGage = total % threshold + nextWallet.EquipMileage += earned + + nextItems := cloneOwnedSnapshot(s.inventory.owned) + for _, want := range consumed { + if err := consumeOwnedItem(&nextItems, want); err != nil { + return Currency{}, 0, err + } + } + + walletData, err := json.Marshal(nextWallet) + if err != nil { + return Currency{}, 0, err + } + if err := s.store.SaveWithEntries(ctx.State, "wallet", walletData, entry("spent", identity, []byte("true"))); err != nil { + return Currency{}, 0, fmt.Errorf("player: persist equipment %s wallet: %w", operation, err) + } + if err := s.inventory.commitOwned(ctx, nextItems); err != nil { + return Currency{}, 0, fmt.Errorf("player: persist equipment %s items: %w", operation, err) + } + if err := s.persistEquipment(ctx, nextEquipment, operation); err != nil { + return Currency{}, 0, err + } + s.owned = nextEquipment + s.persisted = cloneEquipmentSnapshot(nextEquipment) + s.wallet.state = nextWallet + s.inventory.owned = nextItems + return nextWallet.Currency, earned, nil +} + +func consumeOwnedItem(next *ownedSnapshot, want Item) error { + if next == nil || want.InvenIndex == 0 || want.ID == 0 || want.Type == 0 || want.Count == 0 { + return errors.New("player: invalid item consumption") + } + for i, current := range next.Items { + if current.InvenIndex != want.InvenIndex { + continue + } + if current.ID != want.ID || current.Type != want.Type || current.Count < want.Count { + return fmt.Errorf("player: item %d consumption mismatch", want.InvenIndex) + } + current.Count -= want.Count + if current.Count == 0 { + next.Items = append(next.Items[:i], next.Items[i+1:]...) + } else { + next.Items[i] = current + } + return nil + } + return fmt.Errorf("player: item %d is not mutable-owned", want.InvenIndex) +} + +func validateOptionRerollMaterials(costs []gamedata.PromotionCost, materials []Item, conversion *gamedata.EquipmentOptionRerollConversion) (uint64, []Item, error) { + if len(costs) == 0 || len(materials) == 0 { + return 0, nil, errors.New("player: EquipOptionReRoll has no material") + } + expected := make(map[[2]uint64]uint64, len(costs)) + for _, cost := range costs { + key := [2]uint64{cost.Type, cost.ID} + if cost.Type == 0 || cost.Count == 0 || (cost.Type == 4 && cost.ID != 0) || math.MaxUint64-expected[key] < cost.Count { + return 0, nil, errors.New("player: invalid equipment option reroll cost") + } + expected[key] += cost.Count + } + actual := make(map[[2]uint64]uint64, len(materials)) + consumed := make([]Item, 0, len(materials)) + for _, material := range materials { + key := [2]uint64{material.Type, material.ID} + if math.MaxUint64-actual[key] < material.Count { + return 0, nil, errors.New("player: EquipOptionReRoll material count overflows") + } + actual[key] += material.Count + if material.Type != 4 { + consumed = append(consumed, material) + } + } + for key, want := range expected { + if conversion != nil && key == ([2]uint64{conversion.TargetType, conversion.TargetID}) && + expected[[2]uint64{conversion.SourceType, conversion.SourceID}] == 0 { + targetKey := key + sourceKey := [2]uint64{conversion.SourceType, conversion.SourceID} + targetCount := actual[targetKey] + if targetCount > want || conversion.Ratio == 0 || want-targetCount > math.MaxUint64/conversion.Ratio || actual[sourceKey] != (want-targetCount)*conversion.Ratio { + return 0, nil, fmt.Errorf("player: EquipOptionReRoll converted material %d/%d does not match cost", key[0], key[1]) + } + delete(actual, targetKey) + delete(actual, sourceKey) + continue + } + if actual[key] != want { + return 0, nil, fmt.Errorf("player: EquipOptionReRoll material %d/%d=%d want=%d", key[0], key[1], actual[key], want) + } + delete(actual, key) + } + if len(actual) != 0 { + return 0, nil, errors.New("player: EquipOptionReRoll material kinds mismatch") + } + return expected[[2]uint64{4, 0}], consumed, nil +} + +func (s *EquipmentInventory) commitOptionRerollLocked(ctx command.Context, pending *equipmentOptionRerollPending, consumed []Item, gold uint64, identity string) error { + if pending == nil || identity == "" || s.wallet == nil || s.inventory == nil { + return errors.New("player: invalid transactional equipment option reroll") + } + + nextWallet := cloneWallet(s.wallet.state) + if nextWallet.Spent[identity] { + return errors.New("player: equipment option reroll request was already committed") + } + if nextWallet.Gold < gold { + return errors.New("player: insufficient gold for equipment option reroll") + } + nextWallet.Gold -= gold + nextWallet.Spent[identity] = true + nextItems := cloneOwnedSnapshot(s.inventory.owned) + for _, want := range consumed { + if err := consumeOwnedItem(&nextItems, want); err != nil { + return err + } + } + walletData, err := json.Marshal(nextWallet) + if err != nil { + return err + } + if err := s.store.SaveWithEntries(ctx.State, "wallet", walletData, entry("spent", identity, []byte("true"))); err != nil { + return fmt.Errorf("player: persist equipment option reroll wallet: %w", err) + } + if err := s.inventory.commitOwned(ctx, nextItems); err != nil { + return fmt.Errorf("player: persist equipment option reroll items: %w", err) + } + if err := s.persistEquipmentState(ctx, s.owned, pending, true, "option reroll"); err != nil { + return err + } + s.wallet.state = nextWallet + s.inventory.owned = nextItems + s.pendingReroll = &equipmentOptionRerollPending{Equipment: cloneEquipment(pending.Equipment)} + return nil +} + +func upgradeLackItems(costs []gamedata.PromotionCost) []Item { + items := make([]Item, 0, len(costs)) + for _, cost := range costs { + items = append(items, Item{ID: cost.ID, Type: cost.Type, Count: cost.Count}) + } + return items +} + +func (s *EquipmentInventory) equipmentPositionLocked(index uint64) int { + for i := range s.owned.Equipment { + if s.owned.Equipment[i].InvenIndex == index { + return i + } + } + return -1 +} + +func (s *EquipmentInventory) selectUpgradeCosts(ctx command.Context, costs []gamedata.PromotionCost) ([]Item, uint64, error) { + var selected []Item + var gold uint64 + for _, cost := range costs { + switch cost.Type { + case 4: + if cost.ID != 0 || cost.Count == 0 || gold != 0 { + return nil, 0, errors.New("player: invalid equipment upgrade gold cost") + } + gold = cost.Count + selected = append(selected, Item{Type: 4, Count: cost.Count}) + case 8: + items, err := s.inventory.SelectMutable(ctx, cost.Type, cost.ID, cost.Count) + if err != nil { + return nil, 0, err + } + selected = append(selected, items...) + default: + return nil, 0, fmt.Errorf("player: unsupported equipment upgrade cost type %d", cost.Type) + } + } + return selected, gold, nil +} + +func (s *EquipmentInventory) attemptUpgradeLocked(ctx command.Context, index uint64, materials []Item) (Equipment, bool, uint64, []Item, error) { + if s.upgrade == nil || s.wallet == nil { + return Equipment{}, false, 0, nil, errors.New("player: equipment upgrade unavailable") + } + position := s.equipmentPositionLocked(index) + if position < 0 { + return Equipment{}, false, 0, nil, fmt.Errorf("player: unknown equipment %d", index) + } + current := s.owned.Equipment[position] + level, _, err := s.upgrade.Level(current.ID, current.Level) + if err != nil { + return Equipment{}, false, 0, nil, err + } + want := make(map[[2]uint64]uint64, len(level.Costs)) + for _, cost := range level.Costs { + want[[2]uint64{cost.Type, cost.ID}] += cost.Count + } + got := make(map[[2]uint64]uint64) + var gold uint64 + var items []Item + for _, material := range materials { + got[[2]uint64{material.Type, material.ID}] += material.Count + if material.Type == 4 { + if material.InvenIndex != 0 || material.ID != 0 || gold != 0 { + return Equipment{}, false, 0, nil, errors.New("player: invalid equipment upgrade currency") + } + gold = material.Count + } else { + items = append(items, material) + } + } + if len(got) != len(want) { + return Equipment{}, false, 0, nil, fmt.Errorf("player: equipment upgrade material kinds mismatch") + } + for key, count := range want { + if got[key] != count { + return Equipment{}, false, 0, nil, fmt.Errorf("player: equipment upgrade material %d/%d=%d want=%d", key[0], key[1], got[key], count) + } + } + if gold != 0 && !s.wallet.CanSpendGold(gold) { + return Equipment{}, false, 0, nil, errors.New("player: insufficient gold for equipment upgrade") + } + if len(items) != 0 { + if s.inventory == nil { + return Equipment{}, false, 0, nil, errors.New("player: equipment upgrade inventory unavailable") + } + if err := s.inventory.CanConsume(ctx, items); err != nil { + return Equipment{}, false, 0, nil, err + } + } + // Draw before any cross-store write: RNG failure must never charge the + // player. Ordinary enhancement also unlocks an official grade at the + // +3/+6/+9 pivots; smelting can later change those grades, but is separate. + success, err := s.upgrade.Roll(level.SuccessRatio) + if err != nil { + return Equipment{}, false, 0, nil, fmt.Errorf("player: roll equipment upgrade: %w", err) + } + var rankSlot, rankValue uint64 + if success { + nextLevel := current.Level + 1 + switch nextLevel { + case 3: + rankSlot = 1 + case 6: + rankSlot = 2 + case 9: + rankSlot = 3 + } + if rankSlot != 0 { + if len(current.Rank) != 3 || current.Rank[rankSlot-1] != 0 { + return Equipment{}, false, 0, nil, fmt.Errorf("player: equipment %d invalid rank state at +%d", index, nextLevel) + } + rankValue, err = s.upgrade.RollRank(current.ID, rankSlot) + if err != nil { + return Equipment{}, false, 0, nil, fmt.Errorf("player: roll equipment rank: %w", err) + } + } + } + attempt := current.UpgradeAttempts + 1 + identity := "equip-upgrade:" + strconv.FormatUint(index, 10) + ":" + strconv.FormatUint(attempt, 10) + if gold != 0 { + if _, err := s.wallet.SpendGoldOnce(ctx, identity, gold); err != nil { + return Equipment{}, false, 0, nil, fmt.Errorf("player: spend equipment upgrade gold: %w", err) + } + } + if len(items) != 0 { + if err := s.inventory.Consume(ctx, items); err != nil { + return Equipment{}, false, 0, nil, fmt.Errorf("player: consume equipment upgrade items: %w", err) + } + } + next := cloneEquipmentSnapshot(s.owned) + next.Equipment[position].UpgradeAttempts = attempt + if success { + next.Equipment[position].Level++ + if rankSlot != 0 { + next.Equipment[position].Rank[rankSlot-1] = rankValue + } + } + if err := s.commitLocked(ctx, next, "upgrade"); err != nil { + return Equipment{}, false, 0, nil, err + } + return next.Equipment[position], success, gold, materials, nil +} + +func (s *EquipmentInventory) equippedCharacter(ctx command.Context, entry Equipment) (EquipmentCharacter, bool) { + if entry.UseChar == 0 || s.characters == nil { + return EquipmentCharacter{}, false + } + return s.characters.EquipmentCharacter(ctx, entry.UseChar) +} + +// EquipmentInfo.MakeStringCustomMarkData emits either "~Login complete. You can return to the game.
`) +} +func (s *Service) poll(w http.ResponseWriter, r *http.Request) { + id := r.PathValue("id") + if !s.limits.allow("poll:"+remoteIP(r.RemoteAddr)+":"+id, s.identity.Now(), time.Minute, 60) { + w.Header().Set("Retry-After", "2") + http.Error(w, "poll rate exceeded", http.StatusTooManyRequests) + return + } + auth := r.Header.Get("Authorization") + if !strings.HasPrefix(auth, "Device ") { + http.Error(w, "invalid device transaction", http.StatusForbidden) + return + } + result, err := s.identity.Poll(id, strings.TrimPrefix(auth, "Device ")) + if err != nil { + writeFailure(w, err, "login result unavailable") + return + } + switch result.Status { + case "pending": + writeJSON(w, http.StatusAccepted, map[string]any{"status": "pending", "retry_after": 2}) + case "failed": + writeJSON(w, http.StatusForbidden, map[string]string{"status": "failed", "error": result.ErrorCode}) + case "complete": + writeJSON(w, http.StatusOK, tokenResponse(result.Tokens)) + } +} +func (s *Service) refresh(w http.ResponseWriter, r *http.Request) { + var request struct { + RefreshToken string `json:"refresh_token"` + AttemptID string `json:"attempt_id"` + } + if !decodeJSON(w, r, &request) { + return + } + tokens, err := s.identity.Refresh(request.RefreshToken, request.AttemptID) + if err != nil { + writeFailure(w, err, "refresh unavailable") + return + } + writeJSON(w, http.StatusOK, tokenResponse(tokens)) +} +func (s *Service) revoke(w http.ResponseWriter, r *http.Request) { + auth := r.Header.Get("Authorization") + if !strings.HasPrefix(auth, "Bearer ") || strings.TrimPrefix(auth, "Bearer ") == "" { + http.Error(w, "access token required", http.StatusUnauthorized) + return + } + if err := s.identity.Revoke(strings.TrimPrefix(auth, "Bearer ")); err != nil { + writeFailure(w, err, "revocation unavailable") + return + } + w.WriteHeader(http.StatusNoContent) +} +func (s *Service) ValidateAccess(token string) (string, error) { + return s.identity.ValidateAccess(token) +} +func (s *Service) AuthenticateLogin(request []byte) (string, error) { + token, found, err := wire.Bytes(request, 2) + if err != nil || !found { + return "", identity.ErrUnauthorized + } + return s.identity.ValidateAccess(string(token)) +} + +type providerFailure struct { + Provider string + Stage string + Reason string + HTTPStatus int + OAuthError string +} + +func (e *providerFailure) Error() string { + return fmt.Sprintf("provider=%s stage=%s reason=%s status=%d oauth_error=%s", e.Provider, e.Stage, e.Reason, e.HTTPStatus, e.OAuthError) +} + +func networkProviderFailure(ctx context.Context, provider, stage string, err error) error { + reason := "network_error" + if errors.Is(ctx.Err(), context.DeadlineExceeded) || errors.Is(err, context.DeadlineExceeded) { + reason = "timeout" + } else if errors.Is(ctx.Err(), context.Canceled) || errors.Is(err, context.Canceled) { + reason = "cancelled" + } else { + var networkError net.Error + if errors.As(err, &networkError) && networkError.Timeout() { + reason = "timeout" + } + } + return &providerFailure{Provider: provider, Stage: stage, Reason: reason} +} + +func rejectedProviderFailure(provider, stage string, response *http.Response) error { + failure := &providerFailure{Provider: provider, Stage: stage, Reason: "http_rejected", HTTPStatus: response.StatusCode} + var body struct { + Error string `json:"error"` + } + decoder := json.NewDecoder(io.LimitReader(response.Body, 8<<10)) + if decoder.Decode(&body) == nil { + failure.OAuthError = safeOAuthError(body.Error) + } + return failure +} + +func invalidProviderResponse(provider, stage string) error { + return &providerFailure{Provider: provider, Stage: stage, Reason: "invalid_response", HTTPStatus: http.StatusOK} +} + +func safeOAuthError(value string) string { + switch value { + case "invalid_request", "invalid_client", "invalid_grant", "unauthorized_client", + "unsupported_grant_type", "invalid_scope", "access_denied", "server_error", "temporarily_unavailable": + return value + default: + return "unknown" + } +} + +func (s *Service) exchangeIdentity(ctx context.Context, provider, code, verifier, nonce string) (identity.ProviderIdentity, error) { + values := url.Values{"client_id": {s.config.Providers[provider].ClientID}, "client_secret": {s.config.ProviderSecrets[provider]}, "grant_type": {"authorization_code"}, "code": {code}, "redirect_uri": {s.redirectURL(provider)}, "code_verifier": {verifier}} + request, _ := http.NewRequestWithContext(ctx, http.MethodPost, providerTokenURL(provider), strings.NewReader(values.Encode())) + request.Header.Set("Content-Type", "application/x-www-form-urlencoded") + response, err := s.client.Do(request) + if err != nil { + return identity.ProviderIdentity{}, networkProviderFailure(ctx, provider, "token_exchange", err) + } + defer func() { _ = response.Body.Close() }() + if response.StatusCode != http.StatusOK { + return identity.ProviderIdentity{}, rejectedProviderFailure(provider, "token_exchange", response) + } + var token struct { + AccessToken string `json:"access_token"` + IDToken string `json:"id_token"` + } + if err := decodeProviderJSON(response.Body, &token); err != nil || token.AccessToken == "" { + return identity.ProviderIdentity{}, invalidProviderResponse(provider, "token_exchange") + } + if provider == "google" { + if token.IDToken == "" { + return identity.ProviderIdentity{}, invalidProviderResponse(provider, "token_exchange") + } + identity, err := s.verifyGoogleIDToken(ctx, token.IDToken, nonce) + token.AccessToken, token.IDToken = "", "" + return identity, err + } + userinfo, _ := http.NewRequestWithContext(ctx, http.MethodGet, providerUserURL(provider), nil) + userinfo.Header.Set("Authorization", "Bearer "+token.AccessToken) + response, err = s.client.Do(userinfo) + token.AccessToken = "" + if err != nil { + return identity.ProviderIdentity{}, networkProviderFailure(ctx, provider, "userinfo", err) + } + defer func() { _ = response.Body.Close() }() + if response.StatusCode != http.StatusOK { + return identity.ProviderIdentity{}, rejectedProviderFailure(provider, "userinfo", response) + } + var user struct { + ID string `json:"id"` + Sub string `json:"sub"` + } + if err := decodeProviderJSON(response.Body, &user); err != nil { + return identity.ProviderIdentity{}, invalidProviderResponse(provider, "userinfo") + } + if provider == "discord" && user.ID != "" { + return identity.ProviderIdentity{Issuer: "https://discord.com", Subject: user.ID}, nil + } + return identity.ProviderIdentity{}, invalidProviderResponse(provider, "userinfo") +} + +// verifyGoogleIDToken delegates signature and standard-claim verification to +// Google's HTTPS tokeninfo endpoint, then independently verifies this server's +// audience, nonce and expiry. The raw ID token is never persisted or logged. +func (s *Service) verifyGoogleIDToken(ctx context.Context, idToken, nonce string) (identity.ProviderIdentity, error) { + endpoint := "https://oauth2.googleapis.com/tokeninfo?id_token=" + url.QueryEscape(idToken) + request, _ := http.NewRequestWithContext(ctx, http.MethodGet, endpoint, nil) + response, err := s.client.Do(request) + if err != nil { + return identity.ProviderIdentity{}, networkProviderFailure(ctx, "google", "id_token_verify", err) + } + defer func() { _ = response.Body.Close() }() + if response.StatusCode != http.StatusOK { + return identity.ProviderIdentity{}, rejectedProviderFailure("google", "id_token_verify", response) + } + var claims struct { + Issuer string `json:"iss"` + Audience string `json:"aud"` + Subject string `json:"sub"` + Nonce string `json:"nonce"` + Expires string `json:"exp"` + } + if err := decodeProviderJSON(response.Body, &claims); err != nil { + return identity.ProviderIdentity{}, invalidProviderResponse("google", "id_token_verify") + } + expires, err := strconv.ParseInt(claims.Expires, 10, 64) + + if err != nil || !s.identity.ValidateGoogleClaims(claims.Issuer, claims.Audience, claims.Subject, claims.Nonce, nonce, expires) { + return identity.ProviderIdentity{}, &providerFailure{Provider: "google", Stage: "id_token_verify", Reason: "invalid_claims"} + } + return identity.ProviderIdentity{Issuer: "https://accounts.google.com", Subject: claims.Subject}, nil +} + +func decodeProviderJSON(reader io.Reader, target any) error { + data, err := io.ReadAll(io.LimitReader(reader, 1<<20+1)) + if err != nil { + return err + } + if len(data) > 1<<20 { + return errors.New("auth: provider response is too large") + } + decoder := json.NewDecoder(bytes.NewReader(data)) + if err := decoder.Decode(target); err != nil { + return err + } + var trailing any + if err := decoder.Decode(&trailing); !errors.Is(err, io.EOF) { + return errors.New("auth: provider response has trailing JSON") + } + return nil +} + +func (l *requestLimiter) allow(key string, now time.Time, duration time.Duration, maximum int) bool { + l.mu.Lock() + defer l.mu.Unlock() + if l.windows == nil { + l.windows = make(map[string]limitWindow) + } + if l.lastSweep.IsZero() || now.Sub(l.lastSweep) >= time.Minute { + for candidate, window := range l.windows { + if now.Sub(window.started) >= duration { + delete(l.windows, candidate) + } + } + l.lastSweep = now + } + window, exists := l.windows[key] + if !exists || now.Sub(window.started) >= duration { + if !exists && len(l.windows) >= 4096 { + return false + } + l.windows[key] = limitWindow{started: now, count: 1} + return true + } + if window.count >= maximum { + return false + } + window.count++ + l.windows[key] = window + return true +} + +func remoteIP(remoteAddr string) string { + host, _, err := net.SplitHostPort(remoteAddr) + if err == nil && host != "" { + return host + } + return remoteAddr +} + +func (s *Service) redirectURL(provider string) string { + return s.config.PublicURL + "/auth/" + provider + "/callback" +} +func providerScope(provider string) string { + if provider == "discord" { + return "identify" + } + return "openid" +} +func providerAuthorizeURL(provider string) string { + if provider == "discord" { + return "https://discord.com/oauth2/authorize" + } + return "https://accounts.google.com/o/oauth2/v2/auth" +} +func providerTokenURL(provider string) string { + if provider == "discord" { + return "https://discord.com/api/v10/oauth2/token" + } + return "https://oauth2.googleapis.com/token" +} +func providerUserURL(provider string) string { + return "https://discord.com/api/v10/users/@me" +} + +func (s *Service) AuthorizeAdministrator(token string) (string, error) { + return s.identity.AuthorizeAdministrator(token) +} diff --git a/go/internal/server/authconfig/config.go b/go/internal/server/gateway/authconfig/config.go similarity index 99% rename from go/internal/server/authconfig/config.go rename to go/internal/server/gateway/authconfig/config.go index eb374ad..f111f80 100644 --- a/go/internal/server/authconfig/config.go +++ b/go/internal/server/gateway/authconfig/config.go @@ -13,7 +13,7 @@ import ( "strings" "time" - "bd2server/internal/server/configfile" + "bd2server/internal/server/platform/configfile" ) const FileName = "authentication.json" diff --git a/go/internal/server/bootstrap/bootstrap.go b/go/internal/server/gateway/bootstrap/bootstrap.go similarity index 98% rename from go/internal/server/bootstrap/bootstrap.go rename to go/internal/server/gateway/bootstrap/bootstrap.go index 6c623e9..d8dbb10 100644 --- a/go/internal/server/bootstrap/bootstrap.go +++ b/go/internal/server/gateway/bootstrap/bootstrap.go @@ -8,7 +8,7 @@ import ( "strings" "time" - "bd2server/internal/server/wire" + "bd2server/internal/server/protocol/wire" ) type Config struct { diff --git a/go/internal/server/gateway/session/routing_test.go b/go/internal/server/gateway/session/routing_test.go new file mode 100644 index 0000000..d62ae3d --- /dev/null +++ b/go/internal/server/gateway/session/routing_test.go @@ -0,0 +1,264 @@ +package session_test + +import ( + "bd2server/internal/server/gateway/session" + "bd2server/internal/server/gateway/transport" + "bd2server/internal/server/runtime/player" + "bytes" + "context" + "crypto/aes" + "crypto/cipher" + "encoding/base64" + "encoding/binary" + "encoding/json" + "errors" + "sync" + "sync/atomic" + "testing" + "time" +) + +// This client implements the documented base64/protobuf/AES transport itself. +// It obtains each key from LoginUser, never through an internal test key API. +func clientEncrypt(t *testing.T, plain, key []byte, payload bool) []byte { + t.Helper() + if payload { + plain = []byte(base64.StdEncoding.EncodeToString(plain)) + } + block, e := aes.NewCipher(key) + if e != nil { + t.Fatal(e) + } + pad := aes.BlockSize - len(plain)%aes.BlockSize + plain = append(plain, bytes.Repeat([]byte{byte(pad)}, pad)...) + out := make([]byte, len(plain)) + cipher.NewCBCEncrypter(block, make([]byte, aes.BlockSize)).CryptBlocks(out, plain) + return []byte(base64.StdEncoding.EncodeToString(out)) +} +func clientDecrypt(t *testing.T, encoded string, key []byte) []byte { + t.Helper() + raw, e := base64.StdEncoding.DecodeString(encoded) + if e != nil { + t.Fatal(e) + } + block, e := aes.NewCipher(key) + if e != nil { + t.Fatal(e) + } + plain := make([]byte, len(raw)) + cipher.NewCBCDecrypter(block, make([]byte, aes.BlockSize)).CryptBlocks(plain, raw) + pad := int(plain[len(plain)-1]) + plain = plain[:len(plain)-pad] + decoded, e := base64.StdEncoding.DecodeString(string(plain)) + if e != nil { + t.Fatal(e) + } + return decoded +} +func scalar(n int, v uint64) []byte { + return binary.AppendUvarint(binary.AppendUvarint(nil, uint64(n<<3)), v) +} +func message(n int, b []byte) []byte { + out := binary.AppendUvarint(nil, uint64(n<<3|2)) + out = binary.AppendUvarint(out, uint64(len(b))) + return append(out, b...) +} +func field(proto []byte, wanted int) []byte { + for len(proto) > 0 { + k, n := binary.Uvarint(proto) + if n <= 0 { + return nil + } + proto = proto[n:] + if k&7 == 0 { + _, n = binary.Uvarint(proto) + if n <= 0 { + return nil + } + proto = proto[n:] + continue + } + if k&7 != 2 { + return nil + } + size, n := binary.Uvarint(proto) + if n <= 0 || size > uint64(len(proto)-n) { + return nil + } + value := proto[n : n+int(size)] + if int(k>>3) == wanted { + return value + } + proto = proto[n+int(size):] + } + return nil +} + +type authenticator struct{} + +func (authenticator) AuthenticateLogin(p []byte) (string, error) { + id := string(field(p, 2)) + if id != "A" && id != "B" { + return "", errors.New("unknown identity") + } + return id, nil +} + +type executor struct { + account string + blocked chan struct{} + release chan struct{} + once sync.Once + executions atomic.Int32 +} + +func (e *executor) Recover(context.Context) error { return nil } + +func (e *executor) Execute(ctx context.Context, c player.Command) (player.Reply, error) { + if c.Requests[0].Path == "/LoginUser" { + user := message(3, c.LoginSessionKey) + return player.Reply{Responses: []player.Response{{PacketCode: 3, Body: message(1, user)}}}, nil + } + if e.account == "A" { + e.once.Do(func() { close(e.blocked); <-e.release }) + } + if ctx.Err() != nil { + return player.Reply{}, ctx.Err() + } + e.executions.Add(1) + var reply player.Reply + for _, r := range c.Requests { + reply.Responses = append(reply.Responses, player.Response{PacketCode: 1, Body: message(1, []byte(e.account+r.Path)), Notification: message(1, []byte(r.Path))}) + } + return reply, nil +} + +type router map[string]*player.Runtime + +func (r router) Acquire(_ context.Context, id string) (*player.Runtime, func(), error) { + owner := r[id] + if owner == nil { + return nil, nil, errors.New("unknown player") + } + return owner, func() {}, nil +} +func TestCancelledClientRetryAndTwoAccountsDoNotShareExecution(t *testing.T) { + a := &executor{account: "A", blocked: make(chan struct{}), release: make(chan struct{})} + b := &executor{account: "B"} + owners := router{} + for id, e := range map[string]*executor{"A": a, "B": b} { + owner, err := player.New(id, e, player.Limits{}) + if err != nil { + t.Fatal(err) + } + owners[id] = owner + defer func() { + ctx, cancel := context.WithTimeout(context.Background(), time.Second) + defer cancel() + _ = owner.Close(ctx) + }() + } + server, err := session.NewServer(owners, authenticator{}) + if err != nil { + t.Fatal(err) + } + fixed := []byte("abcdefghijkrstuv024680wxyzlmnopq") + login := func(id string) (string, []byte) { + t.Helper() + request := append(scalar(1, 1), message(2, []byte(id))...) + reply, err := server.DispatchRaw(context.Background(), "/LoginUser", clientEncrypt(t, request, fixed, true), "") + if err != nil { + t.Fatal(err) + } + var envelope struct{ Data string } + if err = json.Unmarshal(reply.Body, &envelope); err != nil { + t.Fatal(err) + } + key := field(field(clientDecrypt(t, envelope.Data, fixed), 1), 3) + if len(key) != 32 { + t.Fatal("client received invalid login key") + } + return "s=" + reply.Cookie, key + } + aCookie, aKey := login("A") + bCookie, bKey := login("B") + if bytes.Equal(aKey, bKey) { + t.Fatal("accounts share encryption key") + } + request := scalar(1, 2) + aBody := clientEncrypt(t, request, aKey, true) + ctx, cancel := context.WithCancel(context.Background()) + pending := make(chan error, 1) + go func() { _, e := server.DispatchRaw(ctx, "/ItemInfo", aBody, aCookie); pending <- e }() + select { + case <-a.blocked: + case <-time.After(time.Second): + t.Fatal("A did not enter owner") + } + bDone := make(chan error, 1) + go func() { + _, e := server.DispatchRaw(context.Background(), "/ItemInfo", clientEncrypt(t, request, bKey, true), bCookie) + bDone <- e + }() + select { + case e := <-bDone: + if e != nil { + t.Fatal(e) + } + case <-time.After(time.Second): + t.Fatal("blocked A serialized unrelated B") + } + cancel() + select { + case e := <-pending: + if !errors.Is(e, context.Canceled) { + t.Fatal("HTTP cancellation not propagated", e) + } + case <-time.After(time.Second): + t.Fatal("cancelled client stayed blocked") + } + close(a.release) + if _, err = server.DispatchRaw(context.Background(), "/ItemInfo", aBody, aCookie); err != nil { + t.Fatal(err) + } + if a.executions.Load() != 1 || b.executions.Load() != 1 { + t.Fatal("retry duplicated accepted operation") + } + changed := clientEncrypt(t, append(request, scalar(2, 99)...), aKey, true) + if _, err = server.DispatchRaw(context.Background(), "/ItemInfo", changed, aCookie); !errors.Is(err, player.ErrIdentityConflict) { + t.Fatal("same operation identity accepted modified asset request", err) + } + if _, err = server.DispatchRaw(context.Background(), "/ItemInfo", aBody, bCookie); err == nil { + t.Fatal("one player's ciphertext accepted with another cookie") + } + // A restored actor has no unfinished battle/preview state. Old encrypted + // traffic must reconnect instead of interpreting a cached transient reply. + if err := owners["B"].Recover(context.Background()); err != nil { + t.Fatal(err) + } + if _, err = server.DispatchRaw(context.Background(), "/ItemInfo", clientEncrypt(t, scalar(1, 8), bKey, true), bCookie); !errors.Is(err, transport.ErrGameSessionExpired) { + t.Fatal("recovered owner accepted old game session", err) + } + bCookie, bKey = login("B") + oldCookie, oldKey := bCookie, bKey + bCookie, bKey = login("B") + if _, err = server.DispatchRaw(context.Background(), "/ItemInfo", clientEncrypt(t, scalar(1, 9), oldKey, true), oldCookie); !errors.Is(err, transport.ErrGameSessionExpired) { + t.Fatal("superseded login retained mutable player session", err) + } + if _, err = server.DispatchRaw(context.Background(), "/ItemInfo", clientEncrypt(t, scalar(1, 9), bKey, true), bCookie); err != nil { + t.Fatal("new game login rejected", err) + } + if err := owners["B"].Close(context.Background()); err != nil { + t.Fatal(err) + } + replacement, err := player.New("B", b, player.Limits{}) + if err != nil { + t.Fatal(err) + } + defer func() { _ = replacement.Close(context.Background()) }() + owners["B"] = replacement + if _, err = server.DispatchRaw(context.Background(), "/ItemInfo", clientEncrypt(t, scalar(1, 10), bKey, true), bCookie); !errors.Is(err, transport.ErrGameSessionExpired) { + t.Fatal("unloaded actor's key reused by fresh state generation", err) + } + +} diff --git a/go/internal/server/gateway/session/server.go b/go/internal/server/gateway/session/server.go new file mode 100644 index 0000000..8c5d73a --- /dev/null +++ b/go/internal/server/gateway/session/server.go @@ -0,0 +1,366 @@ +// Package session authenticates game sessions and routes commands to their account owner. +package session + +import ( + "bd2server/internal/server/domain/command" + "bd2server/internal/server/gateway/transport" + "bd2server/internal/server/protocol" + "bd2server/internal/server/protocol/cryptox" + "bd2server/internal/server/protocol/wire" + "bd2server/internal/server/runtime/player" + "context" + "crypto/rand" + "crypto/sha256" + "encoding/binary" + "encoding/hex" + "encoding/json" + "errors" + "fmt" + "strings" + "sync" + "time" +) + +const ( + gameSessionTTL = 24 * time.Hour + maxGameSessions = 1024 + maxCookieHeaderLen = 8 << 10 +) + +var errSessionRequired = errors.New("session login required") + +type LoginAuthenticator interface{ AuthenticateLogin([]byte) (string, error) } +type RuntimeRouter interface { + Acquire(context.Context, string) (*player.Runtime, func(), error) +} +type Handler interface { + Handle(command.Context, string, []byte) (int, []byte, bool, error) +} +type ResponseObserver interface { + BeforeDispatch(command.Context, string, []byte) error + AfterDispatch(command.Context, string, []byte, []byte) ([]byte, error) +} +type LoginService interface { + Login(command.Context, []byte, []byte) ([]byte, error) +} +type gameSession struct { + key []byte + accountID, id string + expiresAt, lastUsed time.Time + generation uint64 +} +type Server struct { + mu sync.Mutex + sessions map[[sha256.Size]byte]*gameSession + router RuntimeRouter + auth LoginAuthenticator + now func() time.Time + sessionTTL time.Duration + maxSessions int +} + +func NewServer(router RuntimeRouter, auth LoginAuthenticator) (*Server, error) { + if router == nil { + return nil, errors.New("session runtime router is nil") + } + return &Server{sessions: map[[sha256.Size]byte]*gameSession{}, router: router, auth: auth, now: time.Now, sessionTTL: gameSessionTTL, maxSessions: maxGameSessions}, nil +} +func (s *Server) DispatchRaw(ctx context.Context, path string, body []byte, cookie string) (reply transport.RawReply, err error) { + started := time.Now() + timing := requestTiming{} + completed := false + defer func() { timing.log(path, time.Since(started), len(reply.Body), err, completed) }() + reply, err = s.dispatchRaw(ctx, path, body, cookie, &timing) + completed = true + return reply, err +} +func (s *Server) dispatchRaw(ctx context.Context, path string, body []byte, cookie string, timing *requestTiming) (transport.RawReply, error) { + if err := ctx.Err(); err != nil { + return transport.RawReply{}, err + } + if path == "/LoginUser" { + return s.login(ctx, body, timing) + } + started := time.Now() + game, err := s.authorize(cookie) + timing.auth += time.Since(started) + if err != nil { + return transport.RawReply{}, err + } + defer clear(game.key) + started = time.Now() + var requests []player.Request + if path == "/BatchRequest" { + batch, decoded, err := protocol.DecodeBatchRequest(body, game.key) + if err != nil { + return transport.RawReply{}, err + } + for i, item := range batch { + requests = append(requests, player.Request{Path: item.Path, Body: decoded[i]}) + } + } else { + plain, err := cryptox.DecryptBase64Payload(string(body), game.key) + if err != nil { + return transport.RawReply{}, fmt.Errorf("%s decrypt: %w", path, err) + } + requests = []player.Request{{Path: path, Body: plain}} + } + timing.decode += time.Since(started) + timing.items = len(requests) + identity, digest, err := requestIdentity(game, requests) + if err != nil { + return transport.RawReply{}, err + } + response, _, err := s.execute(ctx, player.Command{Identity: identity, Digest: digest, Requests: requests}, game.generation, timing) + if err != nil { + return transport.RawReply{}, err + } + if len(response.Responses) != len(requests) { + return transport.RawReply{}, errors.New("player response count does not match command") + } + started = time.Now() + defer func() { timing.encode += time.Since(started) }() + if path != "/BatchRequest" { + item := response.Responses[0] + encoded, err := protocol.EncodeWithNotify(item.PacketCode, item.Body, game.key, s.now().UnixMilli(), item.Notification) + return transport.RawReply{Body: encoded}, err + } + items := make([]protocol.BatchResponse, 0, len(requests)) + for i, item := range response.Responses { + envelope, err := protocol.EnvelopeWithNotify(item.PacketCode, item.Body, game.key, s.now().UnixMilli(), item.Notification) + if err != nil { + return transport.RawReply{}, err + } + items = append(items, protocol.BatchResponse{Path: requests[i].Path, ResponseData: envelope}) + } + encoded, err := json.Marshal(items) + return transport.RawReply{Body: encoded}, err +} +func (s *Server) execute(ctx context.Context, c player.Command, expectedGeneration uint64, timing *requestTiming) (player.Reply, uint64, error) { + owner, release, err := s.router.Acquire(ctx, c.Identity.AccountID) + if err != nil { + return player.Reply{}, 0, err + } + if release == nil { + return player.Reply{}, 0, errors.New("player runtime lease has no release") + } + defer release() + generation := owner.Generation() + if expectedGeneration != 0 && generation != expectedGeneration { + return player.Reply{}, 0, transport.ErrGameSessionExpired + } + c.ExpectedGeneration = expectedGeneration + future, err := owner.Submit(ctx, c) + if err != nil { + if errors.Is(err, player.ErrGenerationExpired) { + return player.Reply{}, 0, transport.ErrGameSessionExpired + } + return player.Reply{}, 0, err + } + response, err := future.Wait(ctx) + timing.queue += response.Timing.Queue + timing.execute += response.Timing.Execute + timing.observer += response.Timing.Observer + timing.begin += response.Timing.Begin + timing.commit += response.Timing.Commit + timing.rollback += response.Timing.Rollback + if expectedGeneration != 0 { + var recovery interface{ RequiresRecovery() bool } + if errors.Is(err, player.ErrGenerationExpired) || owner.Generation() != generation || (errors.As(err, &recovery) && recovery.RequiresRecovery()) { + return player.Reply{}, 0, fmt.Errorf("%w: player state generation changed", transport.ErrGameSessionExpired) + } + } + return response, response.Generation, err +} +func (s *Server) login(ctx context.Context, body []byte, timing *requestTiming) (transport.RawReply, error) { + started := time.Now() + request, err := cryptox.DecryptBase64Payload(string(body), cryptox.Key()) + timing.decode += time.Since(started) + if err != nil { + return transport.RawReply{}, fmt.Errorf("LoginUser decrypt: %w", err) + } + accountID := "local-owner" + if s.auth != nil { + started = time.Now() + accountID, err = s.auth.AuthenticateLogin(request) + timing.auth += time.Since(started) + if err != nil { + return transport.RawReply{}, fmt.Errorf("%w: %v", transport.ErrAccessCredentialInvalid, err) + } + if accountID == "" { + return transport.RawReply{}, errors.New("LoginUser authentication returned an empty account ID") + } + } + game, token, err := newGameSession(accountID, s.now(), s.sessionTTL) + if err != nil { + return transport.RawReply{}, err + } + requests := []player.Request{{Path: "/LoginUser", Body: request}} + identity, digest, err := requestIdentity(game, requests) + if err != nil { + clear(game.key) + return transport.RawReply{}, err + } + response, generation, err := s.execute(ctx, player.Command{Identity: identity, Digest: digest, Requests: requests, LoginSessionKey: game.key}, 0, timing) + if err != nil { + clear(game.key) + return transport.RawReply{}, err + } + if len(response.Responses) != 1 { + clear(game.key) + return transport.RawReply{}, errors.New("login player response count invalid") + } + started = time.Now() + encoded, err := protocol.Encode(3, response.Responses[0].Body, cryptox.Key(), s.now().UnixMilli()) + timing.encode += time.Since(started) + if err != nil { + clear(game.key) + return transport.RawReply{}, err + } + s.mu.Lock() + s.pruneSessions(s.now()) + for existingToken, existing := range s.sessions { + if existing.accountID == accountID { + s.deleteSession(existingToken, existing) + } + } + s.makeSessionRoom() + game.generation = generation + s.sessions[sessionTokenKey(token)] = game + s.mu.Unlock() + return transport.RawReply{Body: encoded, Cookie: token}, nil +} +func requestIdentity(game *gameSession, requests []player.Request) (command.Identity, [32]byte, error) { + sequences := sha256.New() + content := sha256.New() + var size [8]byte + for _, req := range requests { + seq, present, err := wire.Varint(req.Body, 1) + if err != nil || !present || seq == 0 { + return command.Identity{}, [32]byte{}, fmt.Errorf("%s has no request sequence", req.Path) + } + binary.BigEndian.PutUint64(size[:], uint64(len(req.Path))) + _, _ = sequences.Write(size[:]) + _, _ = sequences.Write([]byte(req.Path)) + binary.BigEndian.PutUint64(size[:], seq) + _, _ = sequences.Write(size[:]) + binary.BigEndian.PutUint64(size[:], uint64(len(req.Path))) + _, _ = content.Write(size[:]) + _, _ = content.Write([]byte(req.Path)) + binary.BigEndian.PutUint64(size[:], uint64(len(req.Body))) + _, _ = content.Write(size[:]) + _, _ = content.Write(req.Body) + } + var digest [32]byte + copy(digest[:], content.Sum(nil)) + return command.Identity{AccountID: game.accountID, SessionID: game.id, RequestID: hex.EncodeToString(sequences.Sum(nil))}, digest, nil +} +func (s *Server) authorize(cookie string) (*gameSession, error) { + token, err := parseSessionCookie(cookie) + if err != nil { + if errors.Is(err, errSessionRequired) { + return nil, fmt.Errorf("%w: %v", transport.ErrGameSessionExpired, err) + } + return nil, err + } + s.mu.Lock() + defer s.mu.Unlock() + now := s.now() + s.pruneSessions(now) + game, ok := s.sessions[sessionTokenKey(token)] + if !ok { + return nil, fmt.Errorf("%w: cookie no longer names a live session", transport.ErrGameSessionExpired) + } + game.lastUsed = now + return &gameSession{key: append([]byte(nil), game.key...), accountID: game.accountID, id: game.id, expiresAt: game.expiresAt, lastUsed: game.lastUsed, generation: game.generation}, nil +} +func newGameSession(accountID string, now time.Time, ttl time.Duration) (*gameSession, string, error) { + keyBytes := make([]byte, 16) + if _, err := rand.Read(keyBytes); err != nil { + return nil, "", fmt.Errorf("create game session key: %w", err) + } + tokenBytes := make([]byte, 24) + if _, err := rand.Read(tokenBytes); err != nil { + return nil, "", fmt.Errorf("create game session token: %w", err) + } + idBytes := make([]byte, 12) + if _, err := rand.Read(idBytes); err != nil { + return nil, "", fmt.Errorf("create login request identity: %w", err) + } + return &gameSession{ + key: []byte(hex.EncodeToString(keyBytes)), + accountID: accountID, + id: hex.EncodeToString(idBytes), + expiresAt: now.Add(ttl), + lastUsed: now, + }, hex.EncodeToString(tokenBytes) + "|1", nil +} + +func sessionTokenKey(token string) [sha256.Size]byte { return sha256.Sum256([]byte(token)) } + +func (s *Server) pruneSessions(now time.Time) { + for token, game := range s.sessions { + if !now.Before(game.expiresAt) { + s.deleteSession(token, game) + } + } +} + +func (s *Server) makeSessionRoom() { + for len(s.sessions) >= s.maxSessions { + var oldestToken [sha256.Size]byte + var oldest *gameSession + for token, game := range s.sessions { + if oldest == nil || game.lastUsed.Before(oldest.lastUsed) { + oldestToken, oldest = token, game + } + } + if oldest == nil { + return + } + s.deleteSession(oldestToken, oldest) + } +} + +func (s *Server) deleteSession(token [sha256.Size]byte, game *gameSession) { + delete(s.sessions, token) + for i := range game.key { + game.key[i] = 0 + } + game.accountID = "" + game.id = "" +} + +func parseSessionCookie(cookie string) (string, error) { + if cookie == "" { + return "", errSessionRequired + } + if len(cookie) > maxCookieHeaderLen { + return "", errors.New("game session cookie header is too large") + } + var token string + seen := false + for value := range strings.SplitSeq(cookie, ";") { + name, candidate, found := strings.Cut(strings.TrimSpace(value), "=") + if !found || name != "s" { + continue + } + if seen { + return "", errors.New("duplicate game session cookie") + } + seen = true + token = candidate + } + if !seen { + return "", errSessionRequired + } + if len(token) != 50 || token[48:] != "|1" { + return "", errors.New("invalid game session cookie") + } + for _, char := range token[:48] { + if !(char >= '0' && char <= '9' || char >= 'a' && char <= 'f') { //nolint:staticcheck // QF1001 + return "", errors.New("invalid game session cookie") + } + } + return token, nil +} diff --git a/go/internal/server/gateway/session/timing.go b/go/internal/server/gateway/session/timing.go new file mode 100644 index 0000000..9276411 --- /dev/null +++ b/go/internal/server/gateway/session/timing.go @@ -0,0 +1,33 @@ +package session + +import ( + "log/slog" + "time" +) + +type requestTiming struct { + queue, decode, auth, execute, observer time.Duration + begin, commit, rollback, encode time.Duration + items int +} + +func (t *requestTiming) log(path string, total time.Duration, responseBytes int, err error, completed bool) { + status := "ok" + if !completed { + status = "panic" + } else if err != nil { + status = "error" + } + other := total - t.queue - t.decode - t.auth - t.execute - t.observer - t.begin - t.commit - t.rollback - t.encode + slog.Info("session request phases", "path", path, "status", status, + "queue_ms", milliseconds(t.queue), "decode_ms", milliseconds(t.decode), + "auth_ms", milliseconds(t.auth), "execute_ms", milliseconds(t.execute), + "observer_ms", milliseconds(t.observer), "transaction_begin_ms", milliseconds(t.begin), + "commit_ms", milliseconds(t.commit), "rollback_ms", milliseconds(t.rollback), + "encode_ms", milliseconds(t.encode), "other_ms", milliseconds(other), + "duration_ms", milliseconds(total), "items", t.items, "response_bytes", responseBytes) +} + +func milliseconds(duration time.Duration) float64 { + return float64(duration) / float64(time.Millisecond) +} diff --git a/go/internal/server/transport/commerce.go b/go/internal/server/gateway/transport/commerce.go similarity index 100% rename from go/internal/server/transport/commerce.go rename to go/internal/server/gateway/transport/commerce.go diff --git a/go/internal/server/transport/http.go b/go/internal/server/gateway/transport/http.go similarity index 92% rename from go/internal/server/transport/http.go rename to go/internal/server/gateway/transport/http.go index 26f86a8..95346ad 100644 --- a/go/internal/server/transport/http.go +++ b/go/internal/server/gateway/transport/http.go @@ -2,6 +2,7 @@ package transport import ( + "context" "encoding/base64" "encoding/json" "errors" @@ -12,10 +13,11 @@ import ( "strings" "time" - "bd2server/internal/server/authconfig" - "bd2server/internal/server/bootstrap" - "bd2server/internal/server/resourcepolicy" - "bd2server/internal/server/stateio" + "bd2server/internal/server/gateway/authconfig" + "bd2server/internal/server/gateway/bootstrap" + "bd2server/internal/server/resources/policy" + "bd2server/internal/server/runtime/player" + "bd2server/internal/server/storage/stateio" ) type Envelope struct { @@ -50,7 +52,7 @@ type RawReply struct { // RawDispatcher handles encrypted/session packets whose outer representation // is not always an Envelope (BatchRequest returns a JSON array). type RawDispatcher interface { - DispatchRaw(path string, wireBody []byte, cookie string) (RawReply, error) + DispatchRaw(ctx context.Context, path string, wireBody []byte, cookie string) (RawReply, error) } type Availability interface { @@ -280,10 +282,22 @@ func (h HTTP) game(w http.ResponseWriter, r *http.Request) { http.Error(w, "authenticated packet not yet implemented", http.StatusNotImplemented) return } - reply, err := h.Raw.DispatchRaw(path, body, r.Header.Get("Cookie")) + reply, err := h.Raw.DispatchRaw(r.Context(), path, body, r.Header.Get("Cookie")) if err != nil { + if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) { + return + } + if errors.Is(err, player.ErrMailboxFull) { + w.Header().Set("Retry-After", "1") + http.Error(w, "player command capacity exhausted", http.StatusServiceUnavailable) + return + } + if errors.Is(err, player.ErrClosed) || errors.Is(err, player.ErrUnavailable) { + h.reconnect(w, "player-unavailable") + return + } if errors.Is(err, stateio.ErrStateRecoveryRequired) { - h.logger().Error("account state requires process recovery", "path", path, "duration_ms", elapsedMilliseconds(started), "error", err) + h.logger().Error("account state isolated for recovery", "path", path, "duration_ms", elapsedMilliseconds(started), "error", err) h.reconnect(w, "state-recovery") return } diff --git a/go/internal/server/hunting/dispatch.go b/go/internal/server/hunting/dispatch.go deleted file mode 100644 index 1980cf4..0000000 --- a/go/internal/server/hunting/dispatch.go +++ /dev/null @@ -1,402 +0,0 @@ -package hunting - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" - "bytes" - "crypto/rand" - "encoding/json" - "fmt" - "math/big" - "sort" - "strconv" - "strings" - "time" -) - -type dispatchReceipt struct { - Request, Response []byte - Code int -} -type dispatchJob struct { - Group, ID, Count, Start, End, Free, Bonus uint64 - Rewards []gamedata.BattleReward - Runs [][]gamedata.BattleReward -} -type dispatchState struct { - Version string `json:"version"` - Jobs map[string]dispatchJob `json:"jobs"` - Receipts map[string]dispatchReceipt `json:"receipts"` -} - -func (s *Service) AttachDispatchEligibility(check func(*gamedata.DispatchDesign) error) { - s.mu.Lock() - defer s.mu.Unlock() - s.dispatchEligibility = check -} -func (s *Service) HandleSession(path string, req []byte, session string) (int, []byte, bool, error) { - if strings.HasPrefix(path, "/HuntDispatch") { - return s.handleDispatch(path, req, session) - } - return s.handleGroundSession(path, req, session) -} -func (s *Service) handleDispatch(path string, req []byte, session string) (int, []byte, bool, error) { - code := map[string]int{"/HuntDispatchInfo": 189, "/HuntDispatchStart": 190, "/HuntDispatchEnd": 191, "/HuntDispatchRewardPreview": 194, "/HuntDispatch": 0}[path] - if path != "/HuntDispatchInfo" && path != "/HuntDispatchStart" && path != "/HuntDispatchEnd" && path != "/HuntDispatchRewardPreview" && path != "/HuntDispatch" { - return 0, nil, false, nil - } - s.mu.Lock() - defer s.mu.Unlock() - if e := s.refreshAP(); e != nil { - return code, nil, true, e - } - seq, found, err := wire.Varint(req, 1) - if err != nil || !found || seq == 0 { - return code, nil, true, fmt.Errorf("hunting: missing sequence") - } - ds := dispatchState{Version: versionconfig.State(), Jobs: map[string]dispatchJob{}, Receipts: map[string]dispatchReceipt{}} - raw, err := s.storage.Load("huntdispatch") - if err != nil { - return code, nil, true, err - } - if raw != nil { - if err = stateio.RequireExactJSONObject(raw, "version", "jobs", "receipts"); err != nil { - return code, nil, true, err - } - decoder := json.NewDecoder(bytes.NewReader(raw)) - decoder.DisallowUnknownFields() - if err = decoder.Decode(&ds); err != nil { - return code, nil, true, err - } - if ds.Version != versionconfig.State() || ds.Jobs == nil || ds.Receipts == nil { - return code, nil, true, fmt.Errorf("hunting: invalid dispatch state") - } - } - if path == "/HuntDispatchInfo" { - var out []byte - keys := make([]string, 0, len(ds.Jobs)) - for k := range ds.Jobs { - keys = append(keys, k) - } - sort.Strings(keys) - for _, k := range keys { - out = wire.AppendBytes(out, 1, dispatchJobWire(ds.Jobs[k])) - } - // A client without a remembered dispatch pack needs a default. This - // server selects the smallest cleared ordinary hunting pack. - var minimum uint64 - for key, state := range s.state.Packs { - if state.Highest == 0 { - continue - } - pack, err := strconv.ParseUint(key, 10, 32) - if err != nil || pack == 0 { - return code, nil, true, fmt.Errorf("hunting: invalid saved dispatch pack") - } - if minimum == 0 || pack < minimum { - minimum = pack - } - } - if minimum != 0 { - out = wire.AppendVarint(out, 2, minimum) - } - return code, out, true, nil - } - for k, j := range ds.Jobs { - if k != fmt.Sprintf("%d:%d", j.Group, j.ID) || j.Group == 0 || j.ID == 0 || j.Count == 0 || j.Count > 20 || j.Start == 0 || j.End < j.Start || len(j.Runs) != int(j.Count) || j.Free > 2147483647 || j.Bonus > 2147483647 { - return code, nil, true, fmt.Errorf("hunting: invalid saved dispatch job") - } - } - for k, r := range ds.Receipts { - if k == "" || len(r.Request) == 0 || r.Response == nil { - return code, nil, true, fmt.Errorf("hunting: invalid saved dispatch receipt") - } - } - key := session + ":" + strconv.FormatUint(seq, 10) - if r, ok := ds.Receipts[key]; ok { - if !bytes.Equal(r.Request, append([]byte(path), req...)) { - return code, nil, true, fmt.Errorf("hunting: sequence reused with different dispatch request") - } - return r.Code, r.Response, true, nil - } - group, _, e := wire.Varint(req, 2) - id, _, e2 := wire.Varint(req, 3) - if e != nil || e2 != nil || group == 0 || id == 0 { - return code, nil, true, fmt.Errorf("hunting: invalid dispatch ID") - } - jobKey := fmt.Sprintf("%d:%d", group, id) - job, exists := ds.Jobs[jobKey] - d, err := s.dispatchLoad(group, id) - if err != nil { - return code, nil, true, err - } - var out []byte - if path == "/HuntDispatchRewardPreview" { - if !exists { - return code, nil, true, fmt.Errorf("hunting: dispatch not started") - } - played := dispatchPlayed(job, d.ClearTime) - out = wire.AppendVarint(out, 1, played) - preview := dispatchCompletedRewards(job, played) - remaining := d.AP * (job.Count - played) - bonus := min(remaining, job.Bonus) - free := remaining - bonus - if free > 0 { - preview = append(preview, gamedata.BattleReward{Type: 21, Count: free}) - } - if bonus > 0 { - preview = append(preview, gamedata.BattleReward{Type: 23, Count: bonus}) - } - bundle := dispatchPreview(preview) - for _, r := range preview { - v := wire.AppendVarint(nil, 2, r.ID) - v = wire.AppendVarint(v, 3, r.Type) - v = wire.AppendVarint(v, 4, r.Count) - bundle = wire.AppendBytes(bundle, 1, v) - } - out = wire.AppendBytes(out, 2, bundle) - return code, out, true, nil - } - if path == "/HuntDispatchEnd" { - if !exists { - return code, nil, true, fmt.Errorf("hunting: dispatch not started") - } - played := dispatchPlayed(job, d.ClearTime) - completed := dispatchCompletedRewards(job, played) - remaining := d.AP * (job.Count - played) - refundBonus := min(remaining, job.Bonus) - refundFree := remaining - refundBonus - next := s.clone() - next.Free += refundFree - next.Bonus += refundBonus - if e := s.persist(next); e != nil { - return code, nil, true, e - } - if refundFree > 0 { - completed = append(completed, gamedata.BattleReward{Type: 21, Count: refundFree}) - } - if refundBonus > 0 { - completed = append(completed, gamedata.BattleReward{Type: 23, Count: refundBonus}) - } - bundle, e := s.dispatchGrant("dispatch:"+key, completed) - if e != nil { - return code, nil, true, e - } - out = wire.AppendBytes(out, 1, bundle) - delete(ds.Jobs, jobKey) - } else { - count, _, e := wire.Varint(req, 4) - if e != nil || count == 0 || count > 20 { - return code, nil, true, fmt.Errorf("hunting: dispatch count must be 1..20") - } - if exists { - return code, nil, true, fmt.Errorf("hunting: dispatch already running") - } - for _, active := range ds.Jobs { - design, e := s.dispatchLoad(active.Group, active.ID) - if e != nil { - return code, nil, true, e - } - if design.TypeGroup == d.TypeGroup { - return code, nil, true, fmt.Errorf("hunting: dispatch category already running") - } - } - if s.dispatchEligibility != nil { - if e = s.dispatchEligibility(d); e != nil { - return code, nil, true, e - } - } else { - if d.TypeGroup != 0 { - return code, nil, true, fmt.Errorf("hunting: SkyWay dispatch eligibility unavailable") - } - st := s.state.Packs[strconv.Itoa(int(d.Pack))] - if st.Highest < d.GroundID { - return code, nil, true, fmt.Errorf("hunting: dispatch requires cleared difficulty") - } - } - if d.AP > uint64(2147483647)/count { - return code, nil, true, fmt.Errorf("hunting: dispatch AP overflow") - } - cost := d.AP * count - if s.state.Free < cost && s.state.Bonus < cost-s.state.Free { - return code, nil, true, fmt.Errorf("hunting: insufficient hunting AP") - } - rewards, e := d.Roll(count, func(n uint64) (uint64, error) { - v, e := rand.Int(rand.Reader, new(big.Int).SetUint64(n)) - if e != nil { - return 0, e - } - return v.Uint64(), nil - }) - if e != nil { - return code, nil, true, e - } - next := s.clone() - free := min(cost, next.Free) - bonus := cost - free - next.Free -= free - next.Bonus -= bonus - if path == "/HuntDispatchStart" { - now := uint64(time.Now().UnixMilli()) - runs := make([][]gamedata.BattleReward, count) - for i := range runs { - runs[i], e = d.Roll(1, func(n uint64) (uint64, error) { - v, e := rand.Int(rand.Reader, new(big.Int).SetUint64(n)) - if e != nil { - return 0, e - } - return v.Uint64(), nil - }) - if e != nil { - return code, nil, true, e - } - } - job = dispatchJob{Group: group, ID: id, Count: count, Start: now, End: now + d.ClearTime*count*1000, Free: free, Bonus: bonus, Rewards: rewards, Runs: runs} - ds.Jobs[jobKey] = job - out = wire.AppendBytes(out, 1, dispatchJobWire(job)) - } else { - beforeFree, beforeBonus := s.state.Free, s.state.Bonus - bundle, e := s.dispatchGrant("dispatch:"+key, rewards) - if e != nil { - return code, nil, true, e - } - out = wire.AppendBytes(out, 1, bundle) - if s.state.Free >= beforeFree { - next.Free += s.state.Free - beforeFree - } else { - delta := beforeFree - s.state.Free - if next.Free < delta { - return code, nil, true, fmt.Errorf("hunting: AP grant conflict") - } - next.Free -= delta - } - if s.state.Bonus >= beforeBonus { - next.Bonus += s.state.Bonus - beforeBonus - } else { - delta := beforeBonus - s.state.Bonus - if next.Bonus < delta { - return code, nil, true, fmt.Errorf("hunting: AP grant conflict") - } - next.Bonus -= delta - } - } - if e = s.persist(next); e != nil { - return code, nil, true, e - } - } - ds.Receipts[key] = dispatchReceipt{append([]byte(path), req...), out, code} - b, e := json.Marshal(ds) - if e != nil { - return code, nil, true, e - } - if e = s.storage.Save("huntdispatch", b); e != nil { - return code, nil, true, e - } - return code, out, true, nil -} -func dispatchPlayed(j dispatchJob, seconds uint64) uint64 { - if seconds == 0 { - return j.Count - } - now := uint64(time.Now().UnixMilli()) - if now <= j.Start { - return 0 - } - n := min((now-j.Start)/(seconds*1000), j.Count) - return n -} -func dispatchJobWire(j dispatchJob) []byte { - var out []byte - for n, v := range map[int]uint64{1: j.Group, 2: j.ID, 3: j.Count, 4: j.Start, 5: j.End, 6: j.Free, 7: j.Bonus} { - out = wire.AppendVarint(out, n, v) - } - return out -} -func dispatchPreview(rs []gamedata.BattleReward) []byte { - var out []byte - for _, r := range rs { - v := wire.AppendVarint(nil, 2, r.ID) - v = wire.AppendVarint(v, 3, r.Type) - v = wire.AppendVarint(v, 4, r.Count) - out = wire.AppendBytes(out, 6, v) - } - return out -} -func (s *Service) dispatchGrant(identity string, rs []gamedata.BattleReward) ([]byte, error) { - if s.grant != nil { - var rewards []gamedata.Reward - var ap []gamedata.BattleReward - for _, r := range rs { - if r.Type == 21 || r.Type == 23 { - ap = append(ap, r) - } else { - rewards = append(rewards, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 - } - } - s.mu.Unlock() - bundle, err := s.grant(identity, rewards) - s.mu.Lock() - if err != nil { - return nil, err - } - bundle = append(bundle, dispatchPreview(ap)...) - for _, r := range ap { - v := wire.AppendVarint(nil, 3, r.Type) - v = wire.AppendVarint(v, 4, r.Count) - bundle = wire.AppendBytes(bundle, 1, v) - } - return bundle, nil - } - - var currency []gamedata.Reward - var items []gamedata.BattleReward - for _, r := range rs { - switch r.Type { - case 21, 23: - continue - case 2, 3, 4, 12, 20: - currency = append(currency, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 - default: - items = append(items, r) - } - } - if _, err := s.wallet.GrantQuestOnce(identity+":currency", currency); err != nil { - return nil, err - } - granted, err := s.inventory.GrantOnce(identity+":items", items) - if err != nil { - return nil, err - } - if len(granted) == 0 { - granted = s.inventory.GrantedItems(identity + ":items") - } - out := dispatchPreview(rs) - for _, r := range rs { - if r.Type == 21 || r.Type == 23 { - v := wire.AppendVarint(nil, 3, r.Type) - v = wire.AppendVarint(v, 4, r.Count) - out = wire.AppendBytes(out, 1, v) - } - } - for _, r := range currency { - v := wire.AppendVarint(nil, 2, r.ID) - v = wire.AppendVarint(v, 3, r.Type) - v = wire.AppendVarint(v, 4, r.Count) - out = wire.AppendBytes(out, 1, v) - } - for _, item := range granted { - out = wire.AppendBytes(out, 1, player.ItemWire(item)) - } - return out, nil -} - -func dispatchCompletedRewards(j dispatchJob, n uint64) []gamedata.BattleReward { - var out []gamedata.BattleReward - for i := uint64(0); i < n && i < uint64(len(j.Runs)); i++ { - out = append(out, j.Runs[i]...) - } - return out -} diff --git a/go/internal/server/hunting/dispatch_test.go b/go/internal/server/hunting/dispatch_test.go deleted file mode 100644 index 6bbbbde..0000000 --- a/go/internal/server/hunting/dispatch_test.go +++ /dev/null @@ -1,124 +0,0 @@ -package hunting - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "testing" -) - -func TestDispatchSweepReplayAndValidation(t *testing.T) { - store := stateio.NewMemory() - inv, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(store, player.Currency{}) - if err != nil { - t.Fatal(err) - } - s, err := Open(store, "", "", inv, wallet, func() (int, error) { return 1, nil }, 100, 10) - if err != nil { - t.Fatal(err) - } - s.dispatchLoad = func(g, id uint64) (*gamedata.DispatchDesign, error) { - return &gamedata.DispatchDesign{Group: g, ID: id, Pack: 1, GroundID: 1, AP: 6, ClearTime: 1200, Rewards: []gamedata.BattleReward{{Type: 4, Count: 10}, {Type: 8, ID: 1000, Count: 1}}}, nil - } - s.state.Packs["1"] = packState{Highest: 1} - req := wire.AppendVarint(nil, 1, 1) - req = wire.AppendVarint(req, 2, 1) - req = wire.AppendVarint(req, 3, 1) - req = wire.AppendVarint(req, 4, 2) - _, a, _, err := s.HandleSession("/HuntDispatch", req, "s1") - if err != nil { - t.Fatal(err) - } - _, b, _, err := s.HandleSession("/HuntDispatch", req, "s1") - if err != nil || !bytes.Equal(a, b) { - t.Fatalf("replay %v", err) - } - free, bonus, _ := s.HuntingAP() - if free != 88 || bonus != 10 || wallet.Snapshot().Gold != 20 { - t.Fatalf("AP/currency %d/%d %+v", free, bonus, wallet.Snapshot()) - } - req = wire.AppendVarint(req, 4, 3) - if _, _, _, err = s.HandleSession("/HuntDispatch", req, "s1"); err == nil { - t.Fatal("sequence conflict accepted") - } - s.state.Packs["1"] = packState{} - if _, _, _, err = s.HandleSession("/HuntDispatch", req, "s2"); err == nil { - t.Fatal("uncleared ground accepted") - } -} -func TestDispatchCancelRefundsUnplayedAP(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, _ := Open(store, "", "", inv, wallet, func() (int, error) { return 1, nil }, 10, 10) - s.dispatchLoad = func(g, id uint64) (*gamedata.DispatchDesign, error) { - return &gamedata.DispatchDesign{Group: g, ID: id, Pack: 1, GroundID: 1, AP: 6, ClearTime: 1200, Rewards: []gamedata.BattleReward{{Type: 4, Count: 10}}}, nil - } - s.state.Packs["1"] = packState{Highest: 1} - req := wire.AppendVarint(nil, 1, 1) - req = wire.AppendVarint(req, 2, 1) - req = wire.AppendVarint(req, 3, 1) - req = wire.AppendVarint(req, 4, 2) - if _, _, _, e := s.HandleSession("/HuntDispatchStart", req, "session"); e != nil { - t.Fatal(e) - } - end := wire.AppendVarint(nil, 1, 2) - end = wire.AppendVarint(end, 2, 1) - end = wire.AppendVarint(end, 3, 1) - if _, _, _, e := s.HandleSession("/HuntDispatchEnd", end, "session"); e != nil { - t.Fatal(e) - } - free, bonus, _ := s.HuntingAP() - if free != 10 || bonus != 10 || wallet.Snapshot().Gold != 0 { - t.Fatalf("cancel %d %d %+v", free, bonus, wallet.Snapshot()) - } -} - -func TestDispatchPreviewHasAPItemInfo(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, _ := Open(store, "", "", inv, wallet, func() (int, error) { return 1, nil }, 10, 10) - s.dispatchLoad = func(g, id uint64) (*gamedata.DispatchDesign, error) { - return &gamedata.DispatchDesign{Group: g, ID: id, Pack: 1, GroundID: 1, AP: 6, ClearTime: 1200, Rewards: []gamedata.BattleReward{{Type: 4, Count: 10}}}, nil - } - s.state.Packs["1"] = packState{Highest: 1} - req := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 1), 3, 1), 4, 2) - if _, _, _, e := s.HandleSession("/HuntDispatchStart", req, "session"); e != nil { - t.Fatal(e) - } - preview := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 1), 3, 1) - _, out, _, e := s.HandleSession("/HuntDispatchRewardPreview", preview, "session") - if e != nil { - t.Fatal(e) - } - totals := map[uint64]uint64{} - if err := wire.Walk(out, func(f wire.Field) error { - if f.Number == 2 { - return wire.Walk(f.Value, func(item wire.Field) error { - if item.Number == 1 { - typ, _, _ := wire.Varint(item.Value, 3) - n, _, _ := wire.Varint(item.Value, 4) - totals[typ] += n - } - return nil - }) - } - return nil - }); err != nil { - t.Fatal(err) - } - if totals[21] != 10 || totals[23] != 2 || totals[4] != 0 { - t.Fatalf("preview refund ItemInfo %+v", totals) - } - free, bonus, _ := s.HuntingAP() - if free != 0 || bonus != 8 { - t.Fatal("preview changed AP") - } -} diff --git a/go/internal/server/hunting/dispatch_transaction_test.go b/go/internal/server/hunting/dispatch_transaction_test.go deleted file mode 100644 index 55a5dfc..0000000 --- a/go/internal/server/hunting/dispatch_transaction_test.go +++ /dev/null @@ -1,96 +0,0 @@ -package hunting - -import ( - "errors" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -type failedDispatchStore struct{ stateio.Store } - -func (s failedDispatchStore) Save(name string, payload []byte) error { - if name == "huntdispatch" { - return errors.New("injected dispatch receipt write failure") - } - return s.Store.Save(name, payload) -} - -func TestDispatchReceiptFailureRollsBackAPAndRewards(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - inv, err := player.OpenInventory(repo, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(repo, player.Currency{Gold: 12}) - if err != nil { - t.Fatal(err) - } - if err = inv.EnsurePersisted(); err != nil { - t.Fatal(err) - } - if err = wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - s, err := Open(failedDispatchStore{repo}, "", "", inv, wallet, func() (int, error) { return 1, nil }, 10, 4) - if err != nil { - t.Fatal(err) - } - s.state.Packs["1"] = packState{Highest: 1} - s.dispatchLoad = func(group, id uint64) (*gamedata.DispatchDesign, error) { - return &gamedata.DispatchDesign{Group: group, ID: id, Pack: 1, GroundID: 1, AP: 2, Rewards: []gamedata.BattleReward{{Type: 4, Count: 100}, {Type: 8, ID: 987, Count: 3}}}, nil - } - operation, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - req := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 1), 3, 1), 4, 2) - if _, _, _, err = s.HandleSession("/HuntDispatch", req, "session-A"); err == nil { - _ = operation.Rollback() - t.Fatal("receipt failure accepted") - } - _ = operation.Rollback() - if err = repo.Close(); err != nil { - t.Fatal(err) - } - reopened, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := reopened.Close(); err != nil { - t.Error(err) - } - }() - for _, name := range []string{"hunting", "huntdispatch"} { - raw, err := reopened.Load(name) - if err != nil || raw != nil { - t.Fatalf("%s survived rollback: %s %v", name, raw, err) - } - } - inv, err = player.OpenInventory(reopened, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err = player.OpenWallet(reopened, player.Currency{}) - if err != nil { - t.Fatal(err) - } - if len(inv.All()) != 0 || wallet.Snapshot().Gold != 12 { - t.Fatalf("dispatch rewards survived rollback: %+v %+v", inv.All(), wallet.Snapshot()) - } -} diff --git a/go/internal/server/hunting/hunting.go b/go/internal/server/hunting/hunting.go deleted file mode 100644 index fba117e..0000000 --- a/go/internal/server/hunting/hunting.go +++ /dev/null @@ -1,516 +0,0 @@ -// Package hunting owns ordinary HuntingGround progress and repeatable rewards. -package hunting - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" - "encoding/binary" - "encoding/json" - "fmt" - "maps" - "math" - "slices" - "strconv" - "sync" - "time" -) - -const BattleMode uint64 = 5 - -type packState struct { - Current, Highest uint64 - Auto bool - Defeated []uint64 -} -type snapshot struct { - Version string `json:"version"` - Free uint64 `json:"free"` - Bonus uint64 `json:"bonus"` - Packs map[string]packState `json:"packs"` - Receipts map[string]bool `json:"receipts"` -} -type Service struct { - mu sync.Mutex - storage stateio.Store - inventory *player.Inventory - wallet *player.Wallet - state snapshot - currentPack func() (int, error) - load func(int) (*gamedata.HuntingPack, error) - dispatchLoad func(uint64, uint64) (*gamedata.DispatchDesign, error) - dispatchEligibility func(*gamedata.DispatchDesign) error - eligibility func(int, uint64) error - grant func(string, []gamedata.Reward) ([]byte, error) - apDesign *gamedata.HuntingAPDesign - now func() time.Time -} - -func Open(store stateio.Store, root, version string, inventory *player.Inventory, wallet *player.Wallet, currentPack func() (int, error), free, bonus uint64) (*Service, error) { - if store == nil || inventory == nil || wallet == nil || currentPack == nil { - return nil, fmt.Errorf("hunting: invalid configuration") - } - s := &Service{storage: store, inventory: inventory, wallet: wallet, currentPack: currentPack, state: snapshot{versionconfig.State(), free, bonus, map[string]packState{}, map[string]bool{}}} - s.dispatchLoad = func(group, id uint64) (*gamedata.DispatchDesign, error) { - return gamedata.LoadDispatchDesign(root, version, group, id) - } - s.load = func(pack int) (*gamedata.HuntingPack, error) { return gamedata.LoadHuntingPack(root, version, pack) } - b, err := store.Load("hunting") - if err != nil { - return nil, err - } - if b != nil { - if err := stateio.RequireExactJSONObject(b, "version", "free", "bonus", "packs", "receipts"); err != nil { - return nil, err - } - if err := json.Unmarshal(b, &s.state); err != nil { - return nil, err - } - if s.state.Version != versionconfig.State() || s.state.Packs == nil || s.state.Receipts == nil { - return nil, fmt.Errorf("hunting: incompatible state") - } - } - if s.state.Free > math.MaxInt32 || s.state.Bonus > math.MaxInt32 { - return nil, fmt.Errorf("hunting: AP exceeds protocol range") - } - for key, st := range s.state.Packs { - pack, err := strconv.Atoi(key) - if err != nil || pack <= 0 || strconv.Itoa(pack) != key { - return nil, fmt.Errorf("hunting: invalid saved pack") - } - d, err := s.load(pack) - if err != nil { - return nil, err - } - if _, ok := ground(d, st.Current); !ok { - return nil, fmt.Errorf("hunting: saved current ground missing from GameData") - } - if st.Highest != 0 { - if _, ok := ground(d, st.Highest); !ok { - return nil, fmt.Errorf("hunting: saved highest ground missing from GameData") - } - } - g, _ := ground(d, st.Current) - seen := map[uint64]bool{} - for _, id := range st.Defeated { - member := id == g.BossID - for _, candidate := range g.Monsters { - member = member || id == candidate - } - if !member || seen[id] { - return nil, fmt.Errorf("hunting: invalid defeated monster") - } - seen[id] = true - } - } - for key, claimed := range s.state.Receipts { - if key == "" || !claimed { - return nil, fmt.Errorf("hunting: invalid receipt") - } - } - return s, nil -} -func (s *Service) HuntingAP() (uint64, uint64, error) { - s.mu.Lock() - defer s.mu.Unlock() - if e := s.refreshAP(); e != nil { - return 0, 0, e - } - return s.state.Free, s.state.Bonus, nil -} -func (s *Service) Handle(path string, req []byte) (int, []byte, bool, error) { - if path == "/HuntDispatch" || path == "/HuntDispatchInfo" || path == "/HuntDispatchStart" || path == "/HuntDispatchEnd" || path == "/HuntDispatchRewardPreview" { - return s.handleDispatch(path, req, "local") - } - if path != "/HuntingGroundInfo" && path != "/HuntingGroundInfoList" && path != "/HuntingGroundEnter" { - return 0, nil, false, nil - } - s.mu.Lock() - defer s.mu.Unlock() - seq, found, err := wire.Varint(req, 1) - if err != nil || !found || seq == 0 { - return 0, nil, true, fmt.Errorf("hunting: missing sequence") - } - if path == "/HuntingGroundInfoList" { - ids, err := packed(req, 2) - if err != nil { - return 387, nil, true, err - } - var response []byte - seen := map[uint64]bool{} - for _, id := range ids { - if id == 0 || id > math.MaxInt32 || seen[id] { - return 387, nil, true, fmt.Errorf("hunting: invalid pack list") - } - seen[id] = true - d, err := s.load(int(id)) - if err != nil { - return 387, nil, true, err - } - if len(d.Grounds) == 0 { - response = wire.AppendBytes(response, 1, wire.AppendVarint(nil, 5, id)) - continue - } - info, err := s.info(int(id), d) - if err != nil { - return 387, nil, true, err - } - response = wire.AppendBytes(response, 1, info) - } - return 387, response, true, nil - } - pack, found, err := wire.Varint(req, 2) - if err != nil || !found || pack == 0 || pack > math.MaxInt32 { - return 134, nil, true, fmt.Errorf("hunting: invalid pack") - } - d, err := s.load(int(pack)) - if err != nil { - return 134, nil, true, err - } - if path == "/HuntingGroundInfo" { - // A catalog entry is not an account's active hunting run. The detail - // endpoint preserves an explicitly present empty message until entry. - if s.state.Packs[strconv.Itoa(int(pack))].Current == 0 { - return 134, wire.AppendBytes(nil, 1, nil), true, nil - } - b, err := s.info(int(pack), d) - return 134, wire.AppendBytes(nil, 1, b), true, err - } - if len(d.Grounds) == 0 { - return 110, nil, true, fmt.Errorf("hunting: pack has no hunting ground") - } - current, err := s.currentPack() - if err != nil || current != int(pack) { - return 110, nil, true, fmt.Errorf("hunting: enter pack is not current") - } - id, _, err := wire.Varint(req, 3) - if err != nil { - return 110, nil, true, err - } - auto, _, err := wire.Varint(req, 4) - if err != nil || auto > 1 { - return 110, nil, true, fmt.Errorf("hunting: invalid auto flag") - } - st := s.state.Packs[strconv.Itoa(int(pack))] - g, ok := ground(d, id) - if !ok { - return 110, nil, true, fmt.Errorf("hunting: unknown ground") - } - if s.eligibility != nil { - if err := s.eligibility(int(pack), g.Difficulty); err != nil { - return 110, nil, true, err - } - } else if id != d.Grounds[0].ID { - return 110, nil, true, fmt.Errorf("hunting: main quest difficulty eligibility unavailable") - } - if st.Current != id { - st.Defeated = nil - } - st.Current, st.Auto = id, auto != 0 - next := s.clone() - next.Packs[strconv.Itoa(int(pack))] = st - if err := s.persist(next); err != nil { - return 110, nil, true, err - } - var out []byte - for _, m := range monstersWithState(d, g, st.Defeated) { - out = wire.AppendBytes(out, 1, m) - } - return 110, out, true, nil -} -func (s *Service) info(pack int, d *gamedata.HuntingPack) ([]byte, error) { - st := s.state.Packs[strconv.Itoa(pack)] - initial := st.Current == 0 - if st.Current == 0 { - st.Current = d.Grounds[0].ID - } - g, ok := ground(d, st.Current) - if !ok { - return nil, fmt.Errorf("hunting: saved ground missing from GameData") - } - b := wire.AppendVarint(nil, 2, st.Current) - if st.Highest != 0 { - b = wire.AppendVarint(b, 3, st.Highest) - } - b = wire.AppendVarint(b, 5, uint64(pack)) - if st.Auto { - b = wire.AppendVarint(b, 1, 1) - } - var monsterInfos [][]byte - if initial { - // The initial catalog lists ordinary monsters only. A boss is not an - // active account encounter simply because its design row exists. - for _, id := range g.Monsters { - monsterInfos = append(monsterInfos, monsterWire(d.Monsters[id], true)) - } - } else { - monsterInfos = monstersWithState(d, g, st.Defeated) - } - for _, m := range monsterInfos { - b = wire.AppendBytes(b, 4, m) - } - return b, nil -} -func ground(d *gamedata.HuntingPack, id uint64) (gamedata.HuntingGround, bool) { - for _, g := range d.Grounds { - if g.ID == id { - return g, true - } - } - return gamedata.HuntingGround{}, false -} -func monsterWire(m gamedata.HuntingMonster, active bool) []byte { - b := wire.AppendVarint(nil, 1, m.ID) - b = wire.AppendVarint(b, 2, m.Decks[0]) - if active { - b = wire.AppendVarint(b, 6, 1) - } - return b -} -func monstersWithState(d *gamedata.HuntingPack, g gamedata.HuntingGround, defeated []uint64) [][]byte { - dead := map[uint64]bool{} - for _, id := range defeated { - dead[id] = true - } - boss := len(g.Monsters) == 0 - all := true - for _, id := range g.Monsters { - all = all && dead[id] - } - boss = boss || all - out := make([][]byte, 0, len(g.Monsters)+1) - for _, id := range g.Monsters { - out = append(out, monsterWire(d.Monsters[id], !dead[id])) - } - if boss { - out = append(out, monsterWire(d.Monsters[g.BossID], true)) - } - return out -} -func (s *Service) ValidateBattle(pack int, mode, monster, deck uint64) error { - if mode != BattleMode { - return nil - } - s.mu.Lock() - defer s.mu.Unlock() - if e := s.refreshAP(); e != nil { - return e - } - _, _, _, err := s.validate(pack, monster, deck) - return err -} -func (s *Service) validate(pack int, monster, deck uint64) (*gamedata.HuntingPack, gamedata.HuntingGround, gamedata.HuntingMonster, error) { - d, err := s.load(pack) - if err != nil { - return nil, gamedata.HuntingGround{}, gamedata.HuntingMonster{}, err - } - st := s.state.Packs[strconv.Itoa(pack)] - g, ok := ground(d, st.Current) - if !ok { - return nil, g, gamedata.HuntingMonster{}, fmt.Errorf("hunting: battle before ground enter") - } - member := monster == g.BossID - for _, id := range g.Monsters { - member = member || id == monster - } - m := d.Monsters[monster] - if monster == g.BossID { - for _, id := range g.Monsters { - done := false - for _, dead := range st.Defeated { - done = done || dead == id - } - if !done { - return nil, g, m, fmt.Errorf("hunting: boss locked until ordinary monsters are defeated") - } - } - } - if slices.Contains(st.Defeated, monster) { - return nil, g, m, fmt.Errorf("hunting: defeated monster requires reentry") - } - known := false - for _, id := range m.Decks { - known = known || id == deck - } - if !member || !known { - return nil, g, m, fmt.Errorf("hunting: monster/deck does not belong to selected ground") - } - cost := d.NormalAP - if m.Type == 1 { - cost = d.BossAP - } - if s.state.Free < cost && s.state.Bonus < cost-s.state.Free { - return nil, g, m, fmt.Errorf("hunting: insufficient hunting AP") - } - return d, g, m, nil -} -func (s *Service) CompleteBattle(pack int, mode, monster, deck uint64, receipt string) ([]byte, [][]byte, error) { - if mode != BattleMode { - return nil, nil, nil - } - s.mu.Lock() - defer s.mu.Unlock() - if receipt == "" { - return nil, nil, fmt.Errorf("hunting: missing battle receipt") - } - fingerprint := fmt.Sprintf("%d:%d:%d:%d", pack, mode, monster, deck) - ledger, err := s.loadBattleReceipts() - if err != nil { - return nil, nil, err - } - if saved, ok := ledger[receipt]; ok { - if saved.Fingerprint != fingerprint { - return nil, nil, fmt.Errorf("hunting: battle receipt conflict") - } - return saved.Bundle, saved.Monsters, nil - } - d, g, m, err := s.validate(pack, monster, deck) - if err != nil { - return nil, nil, err - } - rewards := m.Rewards[deck] - identity := "hunting:" + receipt - currency := make([]gamedata.Reward, 0) - stack := make([]gamedata.BattleReward, 0) - for _, r := range rewards { - if r.Type == 2 || r.Type == 3 || r.Type == 4 || r.Type == 12 || r.Type == 20 { - currency = append(currency, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 - } else { - stack = append(stack, r) - } - } - var grantedBundle []byte - var items []player.Item - if s.grant != nil { - rs := make([]gamedata.Reward, len(rewards)) - for i, r := range rewards { - rs[i] = gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count} //nolint:staticcheck // S1016 - } - s.mu.Unlock() - grantedBundle, err = s.grant(identity, rs) - s.mu.Lock() - if err != nil { - return nil, nil, err - } - } else { - if _, err := s.wallet.GrantQuestOnce(identity+":currency", currency); err != nil { - return nil, nil, err - } - items, err = s.inventory.GrantOnce(identity+":items", stack) - if err != nil { - return nil, nil, err - } - if len(items) == 0 { - items = s.inventory.GrantedItems(identity + ":items") - } - } - next := s.clone() - cost := d.NormalAP - if m.Type == 1 { - cost = d.BossAP - } - if next.Free >= cost { - next.Free -= cost - } else { - next.Bonus -= cost - next.Free - next.Free = 0 - } - st := next.Packs[strconv.Itoa(pack)] - if monster == g.BossID { - if st.Highest < g.ID { - st.Highest = g.ID - } - st.Defeated = nil - } else { - st.Defeated = append(append([]uint64(nil), st.Defeated...), monster) - } - next.Packs[strconv.Itoa(pack)] = st - next.Receipts[receipt] = true - var bundle []byte - if s.grant != nil { - bundle = grantedBundle - } - for _, item := range items { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(item)) - view := wire.AppendVarint(nil, 2, item.ID) - view = wire.AppendVarint(view, 3, item.Type) - view = wire.AppendVarint(view, 4, item.Count) - bundle = wire.AppendBytes(bundle, 6, view) - } - for _, r := range currency { - if s.grant != nil { - break - } - b := wire.AppendVarint(nil, 3, r.Type) - b = wire.AppendVarint(b, 4, r.Count) - bundle = wire.AppendBytes(bundle, 1, b) - } - updates := [][]byte{monsterWire(m, false)} - if monster == g.BossID { - for _, id := range g.Monsters { - updates = append(updates, monsterWire(d.Monsters[id], true)) - } - } else if len(st.Defeated) == len(g.Monsters) { - updates = append(updates, monsterWire(d.Monsters[g.BossID], true)) - } - if err := s.persist(next); err != nil { - return nil, nil, err - } - latest, err := s.loadBattleReceipts() - if err != nil { - return nil, nil, err - } - maps.Copy(ledger, latest) - ledger[receipt] = battleReceipt{Fingerprint: fingerprint, Bundle: bundle, Monsters: updates} - if err := s.saveBattleReceipts(ledger); err != nil { - return nil, nil, err - } - return bundle, updates, nil -} -func (s *Service) clone() snapshot { - next := s.state - next.Packs = map[string]packState{} - maps.Copy(next.Packs, s.state.Packs) - next.Receipts = map[string]bool{} - maps.Copy(next.Receipts, s.state.Receipts) - return next -} -func (s *Service) persist(next snapshot) error { - b, err := json.Marshal(next) - if err != nil { - return err - } - if err := s.storage.Save("hunting", b); err != nil { - return err - } - s.state = next - return nil -} -func packed(req []byte, n int) ([]uint64, error) { - var out []uint64 - err := wire.Walk(req, func(f wire.Field) error { - if f.Number != n { - return nil - } - if f.Type == 0 { - v, _ := binary.Uvarint(f.Value) - out = append(out, v) - return nil - } - if f.Type != 2 { - return fmt.Errorf("hunting: invalid repeated integer") - } - for raw := f.Value; len(raw) > 0; { - v, n := binary.Uvarint(raw) - if n <= 0 { - return fmt.Errorf("hunting: invalid packed integer") - } - out = append(out, v) - raw = raw[n:] - } - return nil - }) - return out, err -} diff --git a/go/internal/server/hunting/hunting_test.go b/go/internal/server/hunting/hunting_test.go deleted file mode 100644 index a226198..0000000 --- a/go/internal/server/hunting/hunting_test.go +++ /dev/null @@ -1,311 +0,0 @@ -package hunting - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "fmt" - "testing" - "time" -) - -func TestHuntingCatalogPlaceholdersAndDetailRequireEnteredRun(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, err := Open(store, "", "", inv, wallet, func() (int, error) { return 73, nil }, 20, 5) - if err != nil { - t.Fatal(err) - } - d := &gamedata.HuntingPack{Grounds: []gamedata.HuntingGround{{ID: 9, MapID: 123, BossID: 41, Monsters: []uint64{31, 32, 33, 34, 35}}}, Monsters: map[uint64]gamedata.HuntingMonster{}} - for _, id := range []uint64{31, 32, 33, 34, 35, 41} { - d.Monsters[id] = gamedata.HuntingMonster{ID: id, Decks: []uint64{id + 100}} - } - s.load = func(pack int) (*gamedata.HuntingPack, error) { - switch pack { - case 73: - return d, nil - case 84, 85: - return &gamedata.HuntingPack{}, nil - default: - return nil, fmt.Errorf("unknown pack") - } - } - list := wire.AppendVarint(nil, 1, 1) - for _, id := range []uint64{84, 73, 85} { - list = wire.AppendVarint(list, 2, id) - } - code, response, ok, err := s.Handle("/HuntingGroundInfoList", list) - if err != nil || !ok || code != 387 { - t.Fatalf("list %d %v %v", code, ok, err) - } - var entries [][]byte - if err := wire.Walk(response, func(f wire.Field) error { - if f.Number == 1 { - entries = append(entries, f.Value) - } - return nil - }); err != nil { - t.Fatal(err) - } - if len(entries) != 3 || !bytes.Equal(entries[0], wire.AppendVarint(nil, 5, 84)) || !bytes.Equal(entries[2], wire.AppendVarint(nil, 5, 85)) { - t.Fatalf("catalog lost ordered pack placeholders: %x", entries) - } - current, _, _ := wire.Varint(entries[1], 2) - monsters := 0 - _ = wire.Walk(entries[1], func(f wire.Field) error { - if f.Number == 4 { - monsters++ - } - return nil - }) - if current != 9 || monsters != 5 { - t.Fatalf("catalog current=%d monsters=%d", current, monsters) - } - for _, id := range []uint64{73, 84} { - req := wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, id) - _, detail, _, err := s.Handle("/HuntingGroundInfo", req) - if err != nil || !bytes.Equal(detail, wire.AppendBytes(nil, 1, nil)) { - t.Fatalf("unentered pack %d returned active state: %x %v", id, detail, err) - } - } - if len(s.state.Packs) != 0 { - t.Fatal("query created hunting progress") - } - enter := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 3), 2, 73), 3, 9) - if _, _, _, err := s.Handle("/HuntingGroundEnter", enter); err != nil { - t.Fatal(err) - } - _, detail, _, err := s.Handle("/HuntingGroundInfo", wire.AppendVarint(wire.AppendVarint(nil, 1, 4), 2, 73)) - active, found, _ := wire.Bytes(detail, 1) - if err != nil || !found || len(active) == 0 { - t.Fatalf("entered detail missing: %x %v", detail, err) - } -} - -func TestHuntingRepeatEnterReceiptAPAndDefeated(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, err := Open(store, "", "", inv, wallet, func() (int, error) { return 73, nil }, 20, 5) - if err != nil { - t.Fatal(err) - } - d := &gamedata.HuntingPack{NormalAP: 3, BossAP: 7, Grounds: []gamedata.HuntingGround{{ID: 9, MapID: 123, BossID: 41, Monsters: []uint64{31}}, {ID: 15, MapID: 123, BossID: 51, Monsters: []uint64{32}}}, Monsters: map[uint64]gamedata.HuntingMonster{31: {ID: 31, Decks: []uint64{81}, Rewards: map[uint64][]gamedata.BattleReward{81: {{Type: 8, ID: 999, Count: 2}, {Type: 4, Count: 10}}}}, 41: {ID: 41, Type: 1, Decks: []uint64{91}, Rewards: map[uint64][]gamedata.BattleReward{91: {{Type: 8, ID: 998, Count: 4}}}}}} - s.load = func(int) (*gamedata.HuntingPack, error) { return d, nil } - enter := func(id uint64) error { - r := wire.AppendVarint(nil, 1, 1) - r = wire.AppendVarint(r, 2, 73) - r = wire.AppendVarint(r, 3, id) - _, _, _, err := s.Handle("/HuntingGroundEnter", r) - return err - } - if err := enter(15); err == nil { - t.Fatal("higher difficulty unlocked") - } - if err := enter(9); err != nil { - t.Fatal(err) - } - if err := s.ValidateBattle(73, 5, 31, 91); err == nil { - t.Fatal("wrong deck accepted") - } - if _, _, err := s.CompleteBattle(73, 5, 31, 81, "sessionA:2"); err != nil { - t.Fatal(err) - } - if err := s.ValidateBattle(73, 5, 31, 81); err == nil { - t.Fatal("defeated monster replay accepted") - } - if _, _, err := s.CompleteBattle(73, 5, 31, 81, "sessionA:2"); err != nil { - t.Fatal(err) - } - free, bonus, _ := s.HuntingAP() - if free != 17 || bonus != 5 { - t.Fatalf("AP=%d/%d", free, bonus) - } - info, err := s.info(73, d) - if err != nil { - t.Fatal(err) - } - var inactive bool - if err := wire.Walk(info, func(f wire.Field) error { - if f.Number == 4 { - id, _, _ := wire.Varint(f.Value, 1) - active, _, _ := wire.Varint(f.Value, 6) - if id == 31 { - inactive = active == 0 - } - } - return nil - }); err != nil { - t.Fatal(err) - } - if !inactive { - t.Fatal("info resurrected defeated monster") - } - if err := enter(9); err != nil { - t.Fatal(err) - } - if _, _, err := s.CompleteBattle(73, 5, 31, 81, "sessionB:2"); err == nil { - t.Fatal("same difficulty entry reset round") - } - if _, _, err := s.CompleteBattle(73, 5, 41, 91, "sessionB:boss"); err != nil { - t.Fatal(err) - } - if _, _, err := s.CompleteBattle(73, 5, 31, 81, "sessionB:2"); err != nil { - t.Fatal(err) - } - var normal uint64 - for _, item := range inv.All() { - if item.ID == 999 { - normal += item.Count - } - } - if normal != 4 { - t.Fatalf("repeated rewards=%+v", inv.All()) - } - free, _, _ = s.HuntingAP() - if free != 7 { - t.Fatalf("repeat AP=%d", free) - } - if _, _, err := s.CompleteBattle(73, 5, 41, 91, "sessionB:3"); err != nil { - t.Fatal(err) - } - if s.state.Packs["73"].Highest != 9 { - t.Fatal("boss did not unlock next ground") - } -} - -func TestHuntingStageLoopAndBattleReplay(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, _ := Open(store, "", "", inv, wallet, func() (int, error) { return 1, nil }, 100, 0) - d := &gamedata.HuntingPack{NormalAP: 1, BossAP: 1, Grounds: []gamedata.HuntingGround{{ID: 1, MapID: 13, BossID: 3, Monsters: []uint64{1, 2}}}, Monsters: map[uint64]gamedata.HuntingMonster{}} - for _, id := range []uint64{1, 2, 3} { - typ := uint64(0) - if id == 3 { - typ = 1 - } - d.Monsters[id] = gamedata.HuntingMonster{ID: id, Type: typ, Decks: []uint64{id}, Rewards: map[uint64][]gamedata.BattleReward{id: {{Type: 4, Count: 10}}}} - } - s.load = func(int) (*gamedata.HuntingPack, error) { return d, nil } - if _, e := s.EnsureForPack(1); e != nil { - t.Fatal(e) - } - if e := s.ValidateBattle(1, 5, 3, 3); e == nil { - t.Fatal("early boss accepted") - } - if _, _, e := s.CompleteBattle(1, 5, 1, 1, "a:1"); e != nil { - t.Fatal(e) - } - req := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 1), 3, 1) - if _, _, _, e := s.HandleSession("/HuntingGroundEnter", req, "a"); e != nil { - t.Fatal(e) - } - if e := s.ValidateBattle(1, 5, 1, 1); e == nil { - t.Fatal("enter reset partial stage") - } - bundle, updates, e := s.CompleteBattle(1, 5, 2, 2, "a:3") - if e != nil || len(updates) != 2 { - t.Fatalf("boss transition %v %+v", e, updates) - } - again, replay, e := s.CompleteBattle(1, 5, 2, 2, "a:3") - if e != nil || !bytes.Equal(bundle, again) || len(replay) != 2 { - t.Fatal("battle replay lost response") - } - if _, _, e = s.CompleteBattle(1, 5, 3, 3, "a:3"); e == nil { - t.Fatal("battle receipt conflict accepted") - } - _, updates, e = s.CompleteBattle(1, 5, 3, 3, "a:4") - if e != nil || len(updates) != 3 || len(s.state.Packs["1"].Defeated) != 0 { - t.Fatalf("round reset %v %v", e, updates) - } - if e = s.ValidateBattle(1, 5, 1, 1); e != nil { - t.Fatal(e) - } - free, _, _ := s.HuntingAP() - if free != 97 || wallet.Snapshot().Gold != 30 { - t.Fatal("duplicate AP/reward") - } -} -func TestHuntingAPExchangeReplayAndOverflow(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, _ := Open(store, "", "", inv, wallet, func() (int, error) { return 1, nil }, 10, 2) - cost := []gamedata.Reward{{Type: 21, Count: 4}} - reward := []gamedata.Reward{{Type: 23, Count: 3}} - if e := s.ExchangeAPOnce("x", cost, reward); e != nil { - t.Fatal(e) - } - if e := s.ExchangeAPOnce("x", cost, reward); e != nil { - t.Fatal(e) - } - free, bonus, _ := s.HuntingAP() - if free != 6 || bonus != 5 { - t.Fatal("AP replay") - } - if e := s.ExchangeAPOnce("x", nil, reward); e == nil { - t.Fatal("identity conflict") - } - if e := s.CanExchangeAP(nil, []gamedata.Reward{{Type: 21, Count: 2147483647}}); e == nil { - t.Fatal("AP overflow") - } -} - -func TestHuntingRewardAPCallbackDoesNotDeadlock(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, _ := Open(store, "", "", inv, wallet, func() (int, error) { return 1, nil }, 10, 0) - s.load = func(int) (*gamedata.HuntingPack, error) { - return &gamedata.HuntingPack{NormalAP: 1, Grounds: []gamedata.HuntingGround{{ID: 1, BossID: 2, Monsters: []uint64{1}}}, Monsters: map[uint64]gamedata.HuntingMonster{1: {ID: 1, Decks: []uint64{1}, Rewards: map[uint64][]gamedata.BattleReward{1: {{Type: 21, Count: 3}}}}, 2: {ID: 2, Decks: []uint64{2}}}}, nil - } - s.AttachRewards(func(id string, _ []gamedata.Reward) ([]byte, error) { - return nil, s.ExchangeAPOnce(id, nil, []gamedata.Reward{{Type: 21, Count: 3}}) - }) - if _, err := s.EnsureForPack(1); err != nil { - t.Fatal(err) - } - done := make(chan error, 1) - go func() { _, _, e := s.CompleteBattle(1, 5, 1, 1, "apreward"); done <- e }() - select { - case e := <-done: - if e != nil { - t.Fatal(e) - } - case <-time.After(2 * time.Second): - t.Fatal("reward callback deadlocked") - } - free, _, _ := s.HuntingAP() - if free != 12 { - t.Fatalf("AP grant lost %d", free) - } -} - -func TestHuntingAPDailyRefreshPreservesBonus(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - s, _ := Open(store, "", "", inv, wallet, func() (int, error) { return 1, nil }, 4, 7) - if e := s.AttachAPRefresh(gamedata.HuntingAPDesign{Max: 90, ResetSeconds: 9 * 3600}); e != nil { - t.Fatal(e) - } - now := time.Date(2026, 10, 4, 23, 59, 0, 0, time.UTC) - s.now = func() time.Time { return now } - if _, _, err := s.HuntingAP(); err != nil { - t.Fatal(err) - } - now = now.Add(2 * time.Minute) - free, bonus, e := s.HuntingAP() - if e != nil || free != 90 || bonus != 7 { - t.Fatalf("refresh %d %d %v", free, bonus, e) - } - s.state.Free = 12 - free, _, _ = s.HuntingAP() - if free != 12 { - t.Fatal("refreshed same day twice") - } -} diff --git a/go/internal/server/hunting/progress.go b/go/internal/server/hunting/progress.go deleted file mode 100644 index dc00a01..0000000 --- a/go/internal/server/hunting/progress.go +++ /dev/null @@ -1,224 +0,0 @@ -package hunting - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" - "bytes" - "encoding/json" - "fmt" - "strconv" -) - -type battleReceipt struct { - Fingerprint string - Bundle []byte - Monsters [][]byte -} -type battleLedger struct { - Version string `json:"version"` - Receipts map[string]battleReceipt `json:"receipts"` -} - -func (s *Service) loadBattleReceipts() (map[string]battleReceipt, error) { - b, e := s.storage.Load("huntingbattle") - if e != nil { - return nil, e - } - if b == nil { - return map[string]battleReceipt{}, nil - } - if e = stateio.RequireExactJSONObject(b, "version", "receipts"); e != nil { - return nil, e - } - var l battleLedger - d := json.NewDecoder(bytes.NewReader(b)) - d.DisallowUnknownFields() - if e = d.Decode(&l); e != nil { - return nil, e - } - if l.Version != versionconfig.State() || l.Receipts == nil { - return nil, fmt.Errorf("hunting: invalid battle ledger") - } - return l.Receipts, nil -} -func (s *Service) saveBattleReceipts(rs map[string]battleReceipt) error { - b, e := json.Marshal(battleLedger{versionconfig.State(), rs}) - if e != nil { - return e - } - return s.storage.Save("huntingbattle", b) -} -func (s *Service) AttachEligibility(check func(int, uint64) error) { - s.mu.Lock() - defer s.mu.Unlock() - s.eligibility = check -} -func (s *Service) AttachRewards(grant func(string, []gamedata.Reward) ([]byte, error)) { - s.mu.Lock() - defer s.mu.Unlock() - s.grant = grant -} - -// EnsureForPack initializes the ordinary stage when the world loads a pack. -func (s *Service) EnsureForPack(pack int) ([]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() - d, e := s.load(pack) - if e != nil { - return nil, e - } - if len(d.Grounds) == 0 { - return nil, nil - } - st := s.state.Packs[strconv.Itoa(pack)] - if st.Current == 0 { - st.Current = d.Grounds[0].ID - next := s.clone() - next.Packs[strconv.Itoa(pack)] = st - if e = s.persist(next); e != nil { - return nil, e - } - } - return s.info(pack, d) -} -func (s *Service) SnapshotForPack(pack int) ([]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() - d, e := s.load(pack) - if e != nil { - return nil, e - } - if len(d.Grounds) == 0 { - return nil, nil - } - return s.info(pack, d) -} - -// Ground mutations have their own persisted session+sequence replay ledger. -func (s *Service) handleGroundSession(path string, req []byte, session string) (int, []byte, bool, error) { - if path != "/HuntingGroundEnter" { - return s.Handle(path, req) - } - seq, found, e := wire.Varint(req, 1) - if e != nil || !found || seq == 0 { - return 110, nil, true, fmt.Errorf("hunting: missing sequence") - } - key := "enter:" + session + ":" + strconv.FormatUint(seq, 10) - rs, e := s.loadBattleReceipts() - if e != nil { - return 110, nil, true, e - } - fp := fmt.Sprintf("%x", req) - if r, ok := rs[key]; ok { - if r.Fingerprint != fp { - return 110, nil, true, fmt.Errorf("hunting: enter sequence conflict") - } - return 110, r.Bundle, true, nil - } - code, out, ok, e := s.Handle(path, req) - if e != nil || !ok { - return code, out, ok, e - } - rs[key] = battleReceipt{Fingerprint: fp, Bundle: out} - if e = s.saveBattleReceipts(rs); e != nil { - return code, nil, true, e - } - return code, out, ok, nil -} -func (s *Service) reload() error { - b, e := s.storage.Load("hunting") - if e != nil { - return e - } - if b == nil { - return nil - } - var st snapshot - if e = json.Unmarshal(b, &st); e != nil { - return e - } - s.state = st - return nil -} -func (s *Service) CanExchangeAP(costs, rewards []gamedata.Reward) error { - s.mu.Lock() - defer s.mu.Unlock() - if e := s.reload(); e != nil { - return e - } - if e := s.refreshAP(); e != nil { - return e - } - _, e := s.exchangeAP(costs, rewards) - return e -} -func (s *Service) exchangeAP(costs, rewards []gamedata.Reward) (snapshot, error) { - st := s.clone() - apply := func(rs []gamedata.Reward, cost bool) error { - for _, r := range rs { - if (r.Type != 21 && r.Type != 23) || r.ID != 0 || r.Count == 0 { - return fmt.Errorf("hunting: invalid AP exchange") - } - p := &st.Free - if r.Type == 23 { - p = &st.Bonus - } - if cost { - if *p < r.Count { - return fmt.Errorf("hunting: insufficient AP") - } - *p -= r.Count - } else { - if r.Count > 2147483647-*p { - return fmt.Errorf("hunting: AP overflow") - } - *p += r.Count - } - } - return nil - } - if e := apply(costs, true); e != nil { - return st, e - } - return st, apply(rewards, false) -} -func (s *Service) ExchangeAPOnce(identity string, costs, rewards []gamedata.Reward) error { - s.mu.Lock() - defer s.mu.Unlock() - if identity == "" { - return fmt.Errorf("hunting: missing AP identity") - } - if e := s.reload(); e != nil { - return e - } - key := "apexchange:" + identity - fpb, e := json.Marshal([][]gamedata.Reward{costs, rewards}) - if e != nil { - return e - } - rs, e := s.loadBattleReceipts() - if e != nil { - return e - } - fp := string(fpb) - if r, ok := rs[key]; ok { - if r.Fingerprint != fp { - return fmt.Errorf("hunting: AP identity conflict") - } - return nil - } - if e := s.refreshAP(); e != nil { - return e - } - st, e := s.exchangeAP(costs, rewards) - if e != nil { - return e - } - if e = s.persist(st); e != nil { - return e - } - rs[key] = battleReceipt{Fingerprint: fp} - return s.saveBattleReceipts(rs) -} diff --git a/go/internal/server/hunting/transaction_test.go b/go/internal/server/hunting/transaction_test.go deleted file mode 100644 index 3984ba2..0000000 --- a/go/internal/server/hunting/transaction_test.go +++ /dev/null @@ -1,103 +0,0 @@ -package hunting - -import ( - "encoding/json" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" -) - -func TestHuntingSettlementRollsBackAPWalletInventoryAndProgress(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - inv, err := player.OpenInventory(repo, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(repo, player.Currency{Gold: 12}) - if err != nil { - t.Fatal(err) - } - if err := inv.EnsurePersisted(); err != nil { - t.Fatal(err) - } - if err := wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - load := func(int) (*gamedata.HuntingPack, error) { - return &gamedata.HuntingPack{NormalAP: 1, BossAP: 2, - Grounds: []gamedata.HuntingGround{{ID: 1, MapID: 13, BossID: 11}}, - Monsters: map[uint64]gamedata.HuntingMonster{11: {ID: 11, Type: 1, Decks: []uint64{21}, Rewards: map[uint64][]gamedata.BattleReward{21: {{Type: 4, Count: 100}, {Type: 8, ID: 987, Count: 3}}}}}}, nil - } - s, err := Open(repo, "", "", inv, wallet, func() (int, error) { return 1, nil }, 10, 4) - if err != nil { - t.Fatal(err) - } - s.load = load - enter := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 1), 3, 1) - if _, _, _, err := s.Handle("/HuntingGroundEnter", enter); err != nil { - t.Fatal(err) - } - before, err := repo.Load("hunting") - if err != nil { - t.Fatal(err) - } - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, err := s.CompleteBattle(1, BattleMode, 11, 21, "login-A:2"); err != nil { - _ = op.Rollback() - t.Fatal(err) - } - // Model a later domain failure in the same request/batch. A dirty rollback - // fences in-memory domains, requiring restart before they can serve again. - _ = op.Rollback() - if repo.Check() == nil { - t.Fatal("dirty rollback did not fence stale domains") - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - reopened, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := reopened.Close(); err != nil { - t.Error(err) - } - }() - after, err := reopened.Load("hunting") - if err != nil || string(after) != string(before) { - t.Fatalf("hunting progress survived rollback: %s %v", after, err) - } - inv, err = player.OpenInventory(reopened, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err = player.OpenWallet(reopened, player.Currency{}) - if err != nil { - t.Fatal(err) - } - var persisted snapshot - if err := json.Unmarshal(after, &persisted); err != nil { - t.Fatal(err) - } - free, bonus := persisted.Free, persisted.Bonus - if err != nil || free != 10 || bonus != 4 || wallet.Snapshot().Gold != 12 || len(inv.All()) != 0 { - t.Fatalf("settlement survived rollback: AP=%d/%d wallet=%+v items=%+v err=%v", free, bonus, wallet.Snapshot(), inv.All(), err) - } -} diff --git a/go/internal/server/lifecycle/gate_test.go b/go/internal/server/lifecycle/gate_test.go deleted file mode 100644 index 1d40823..0000000 --- a/go/internal/server/lifecycle/gate_test.go +++ /dev/null @@ -1,31 +0,0 @@ -package lifecycle - -import ( - "context" - "testing" - "time" -) - -func TestDrainRejectsNewRequestsAndWaitsForAdmittedRequest(t *testing.T) { - gate := NewGate() - done, ok := gate.BeginRequest() - if !ok || !gate.Ready() { - t.Fatal("ready gate rejected request") - } - gate.Drain() - if gate.Ready() { - t.Fatal("draining gate remained ready") - } - if _, ok := gate.BeginRequest(); ok { - t.Fatal("draining gate admitted new request") - } - ctx, cancel := context.WithTimeout(context.Background(), 20*time.Millisecond) - defer cancel() - if err := gate.Wait(ctx); err == nil { - t.Fatal("drain completed before admitted request") - } - done() - if err := gate.Wait(context.Background()); err != nil { - t.Fatal(err) - } -} diff --git a/go/internal/server/logging/logging_test.go b/go/internal/server/logging/logging_test.go deleted file mode 100644 index 83c96b9..0000000 --- a/go/internal/server/logging/logging_test.go +++ /dev/null @@ -1,140 +0,0 @@ -package logging - -import ( - "bytes" - "context" - "io" - "log/slog" - "os" - "strings" - "sync" - "testing" -) - -func TestLevelsAndColors(t *testing.T) { - for _, test := range []struct { - name, ansi string - level slog.Level - }{ - {"TRACE", "90", LevelTrace}, {"DEBUG", "36", slog.LevelDebug}, - {"INFO", "32", slog.LevelInfo}, {"WARN", "33", slog.LevelWarn}, {"ERROR", "31", slog.LevelError}, - } { - t.Run(test.name, func(t *testing.T) { - var out bytes.Buffer - h, err := NewHandler(&out, Options{Level: LevelTrace, Color: ColorAlways}) - if err != nil { - t.Fatal(err) - } - slog.New(h).Log(context.Background(), test.level, "hello", "level", "ERROR", "text", "a\nb") - want := "level=\x1b[" + test.ansi + "m" + test.name + "\x1b[0m msg=hello level=ERROR text=\"a\\nb\"" - if !strings.Contains(out.String(), want) { - t.Fatalf("output=%q want fragment=%q", out.String(), want) - } - }) - } -} - -func TestFilterAndDynamicLevel(t *testing.T) { - var out bytes.Buffer - var level slog.LevelVar - h, _ := NewHandler(&out, Options{Level: &level, Color: ColorNever}) - logger := slog.New(h) - logger.Log(context.Background(), LevelTrace, "hidden") - logger.Debug("hidden") - logger.Info("visible") - if strings.Contains(out.String(), "hidden") { - t.Fatal(out.String()) - } - level.Set(LevelTrace) - logger.Log(context.Background(), LevelTrace, "trace visible") - if !strings.Contains(out.String(), "level=TRACE msg=\"trace visible\"") { - t.Fatal(out.String()) - } -} - -func TestGroupsAndConcurrentDerivedLoggers(t *testing.T) { - var out bytes.Buffer - h, _ := NewHandler(&out, Options{Color: ColorAlways}) - logger := slog.New(h).With("service", "server").WithGroup("request").With("id", 7) - var workers sync.WaitGroup - for range 50 { - workers.Go(func() { logger.Info("handled", slog.Group("result", "ok", true)) }) - } - workers.Wait() - lines := strings.Split(strings.TrimSpace(out.String()), "\n") - if len(lines) != 50 { - t.Fatalf("lines=%d", len(lines)) - } - for _, line := range lines { - if !strings.Contains(line, "service=server request.id=7 request.result.ok=true") || strings.Count(line, "\x1b[0m") != 1 { - t.Fatalf("damaged record %q", line) - } - } -} - -func TestAutoRedirectedOutputIsPlain(t *testing.T) { - file, err := os.CreateTemp(t.TempDir(), "log") - if err != nil { - t.Fatal(err) - } - defer func() { - if err := file.Close(); err != nil { - t.Error(err) - } - }() - h, _ := NewHandler(file, Options{}) - slog.New(h).Warn("redirected") - if _, err := file.Seek(0, 0); err != nil { - t.Fatal(err) - } - data, err := io.ReadAll(file) - if err != nil { - t.Fatal(err) - } - if bytes.Contains(data, []byte("\x1b")) { - t.Fatalf("ANSI in redirected log %q", data) - } - var out bytes.Buffer - h, _ = NewHandler(&out, Options{}) - slog.New(h).Info("buffer") - if strings.Contains(out.String(), "\x1b") { - t.Fatal(out.String()) - } -} - -func TestEnvironmentAndValidation(t *testing.T) { - t.Setenv("BD2_LOG_LEVEL", "trace") - t.Setenv("BD2_LOG_COLOR", "never") - options, err := OptionsFromEnv() - if err != nil || options.Level.Level() != LevelTrace || options.Color != ColorNever { - t.Fatalf("options=%+v err=%v", options, err) - } - t.Setenv("BD2_LOG_LEVEL", "invalid") - if _, err := OptionsFromEnv(); err == nil { - t.Fatal("invalid level accepted") - } - if _, err := ParseColorMode("invalid"); err == nil { - t.Fatal("invalid color accepted") - } - if _, err := NewHandler(nil, Options{}); err == nil { - t.Fatal("nil writer accepted") - } -} - -func TestAutoColorEnvironment(t *testing.T) { - t.Setenv("TERM", "xterm-256color") - t.Setenv("NO_COLOR", "") - if autoColorAllowed() { - t.Fatal("NO_COLOR presence must suppress auto color") - } - if err := os.Unsetenv("NO_COLOR"); err != nil { - t.Fatal(err) - } - if !autoColorAllowed() { - t.Fatal("ordinary terminal must allow auto color") - } - t.Setenv("TERM", "dumb") - if autoColorAllowed() { - t.Fatal("dumb terminal must suppress auto color") - } -} diff --git a/go/internal/server/mail/attendance_test.go b/go/internal/server/mail/attendance_test.go deleted file mode 100644 index 4059fb5..0000000 --- a/go/internal/server/mail/attendance_test.go +++ /dev/null @@ -1,156 +0,0 @@ -package mail - -import ( - "bytes" - "testing" - "time" - - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -type attendanceDesign struct{ selections int } - -func (*attendanceDesign) Character(uint64) (gamedata.CharacterDesign, bool) { - return gamedata.CharacterDesign{}, false -} -func (*attendanceDesign) RollOptions(uint64) ([]gamedata.EquipmentOptionChoice, []gamedata.EquipmentOptionChoice, *gamedata.EquipmentOptionChoice, error) { - return nil, nil, nil, nil -} -func (d *attendanceDesign) Resolve(r []gamedata.BattleReward) ([]gamedata.BattleReward, error) { - return d.ResolveGranted(r) -} -func (d *attendanceDesign) ResolveGranted(r []gamedata.BattleReward) ([]gamedata.BattleReward, error) { - var out []gamedata.BattleReward - for _, reward := range r { - if reward.Type == 9 && reward.ID == 100 { - d.selections++ - out = append(out, gamedata.BattleReward{Type: 4, Count: 10 * reward.Count}, gamedata.BattleReward{Type: 8, ID: 1000, Count: 2 * reward.Count}, gamedata.BattleReward{Type: 9, ID: 200, Count: reward.Count}) - } else { - out = append(out, reward) - } - } - return out, nil -} - -func attendanceMailFixture(t *testing.T, store stateio.Store, seed *Starter, d *attendanceDesign) (*Service, *player.Inventory, *player.Wallet) { - t.Helper() - items, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(store, player.Currency{}) - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(store, nil) - if err != nil { - t.Fatal(err) - } - equipment, err := player.OpenEquipmentInventory(store) - if err != nil { - t.Fatal(err) - } - for _, persist := range []func() error{items.EnsurePersisted, wallet.EnsurePersisted, collection.EnsurePersisted, equipment.EnsurePersisted} { - if err := persist(); err != nil { - t.Fatal(err) - } - } - economy, err := events.NewEconomy(store, items, wallet, collection, equipment, d, d, d, nil) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(store, seed, items, wallet) - if err != nil { - t.Fatal(err) - } - if err := service.AttachAttendanceRewardEconomy(economy); err != nil { - t.Fatal(err) - } - return service, items, wallet -} - -func TestAttendanceMailDefersRewardsAndReplaysAcrossRestart(t *testing.T) { - store := stateio.NewMemory() - seed := &Starter{Version: "2.35.10", MailCount: 1, MaxMailID: 100} - design := &attendanceDesign{} - s, items, wallet := attendanceMailFixture(t, store, seed, design) - now := time.Date(2026, 10, 5, 12, 0, 0, 0, time.UTC) - s.now = func() time.Time { return now } - reward := []gamedata.Reward{{Type: 9, ID: 100, Count: 1}} - if err := s.IssueAttachmentsOnce("daily:1", "签到奖励", "请领取", reward, now); err != nil { - t.Fatal(err) - } - if err := s.IssueAttachmentsOnce("daily:1", "签到奖励", "请领取", reward, now.Add(time.Minute)); err != nil { - t.Fatal(err) - } - if err := s.IssueAttachmentsOnce("daily:1", "签到奖励", "请领取", []gamedata.Reward{{Type: 4, Count: 999}}, now); err == nil { - t.Fatal("identity accepted changed attachments") - } - if len(s.dynamic) != 1 || wallet.Snapshot().Gold != 0 || len(items.All()) != 0 || design.selections != 0 { - t.Fatal("issuing mail granted rewards") - } - if err := s.IssueAttachmentsOnce("daily:2", "签到奖励", "请领取", reward, now); err != nil { - t.Fatal(err) - } - if s.issued["attendance:daily:2"] != 102 { - t.Fatal("mail ID did not increase") - } - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendVarint(request, 2, 101) - request = wire.AppendVarint(request, 2, 102) - _, response, _, err := s.Handle("/MailOpen", request) - if err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Gold != 20 || design.selections != 2 || len(items.All()) != 4 || len(s.history) != 2 { - t.Fatalf("claim missing domains/history: gold=%d selections=%d items=%d", wallet.Snapshot().Gold, design.selections, len(items.All())) - } - for _, item := range items.All() { - if item.Type == 9 && item.ID != 200 { - t.Fatal("OPEN wrapper stored instead of DIRECT child") - } - } - reopened, items, wallet := attendanceMailFixture(t, store, seed, design) - reopened.now = s.now - _, replay, _, err := reopened.Handle("/MailOpen", request) - if err != nil || !bytes.Equal(response, replay) || design.selections != 2 || wallet.Snapshot().Gold != 20 || len(items.All()) != 4 { - t.Fatal("restart replay changed or repeated grants", err) - } - if err := reopened.IssueAttachmentsOnce("expired", "签到奖励", "请领取", reward, now.Add(-31*24*time.Hour)); err != nil { - t.Fatal(err) - } - expired := wire.AppendVarint(nil, 1, 2) - expired = wire.AppendVarint(expired, 2, 103) - if _, _, _, err := reopened.Handle("/MailOpen", expired); err == nil || design.selections != 2 || len(reopened.history) != 2 { - t.Fatal("expired mail granted") - } -} - -func TestAttendanceEconomyDoesNotRegrantExistingOpenedMail(t *testing.T) { - store := stateio.NewMemory() - seed := &Starter{Version: "2.35.10", MailCount: 1, MaxMailID: 100} - design := &attendanceDesign{} - s, _, _ := attendanceMailFixture(t, store, seed, design) - now := time.Date(2026, 10, 5, 12, 0, 0, 0, time.UTC) - s.now = func() time.Time { return now } - if err := s.EnqueueCompensation("existing:grant", "奖励", "请领取", []gamedata.Reward{{Type: 4, Count: 7}}, now); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendVarint(request, 2, 101) - if _, _, _, err := s.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - reopened, _, wallet := attendanceMailFixture(t, store, seed, design) - reopened.now = s.now - if _, _, _, err := reopened.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Gold != 7 || design.selections != 0 { - t.Fatal("existing opened mail routed to new attendance economy") - } -} diff --git a/go/internal/server/mail/cash.go b/go/internal/server/mail/cash.go deleted file mode 100644 index 8b29d55..0000000 --- a/go/internal/server/mail/cash.go +++ /dev/null @@ -1,111 +0,0 @@ -package mail - -import ( - "errors" - "fmt" - "maps" - "math" - "reflect" - "sort" - "strings" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -// Cash attachments are already resolved at purchase, so manual boxes must not -// be expanded again and selected random rewards must not be rolled on claim. -type CashRewardEconomy interface { - ApplyResolved(string, []gamedata.Reward, []gamedata.Reward) ([]byte, error) -} - -func (s *Service) AttachCashRewards(e CashRewardEconomy, templates map[uint64]bool) error { - if e == nil || len(templates) == 0 { - return errors.New("mail: missing cash economy or templates") - } - s.cashEconomy = e - s.cashTemplates = make(map[uint64]bool, len(templates)) - maps.Copy(s.cashTemplates, templates) - return nil -} -func (s *Service) IssueCashOnce(identity string, template uint64, rewards []gamedata.Reward, sentAt time.Time) error { - s.mu.Lock() - defer s.mu.Unlock() - if s.cashEconomy == nil || !s.cashTemplates[template] { - return fmt.Errorf("mail: cash template %d or economy unavailable", template) - } - if identity == "" || strings.TrimSpace(identity) != identity || len(identity) > 450 || sentAt.UnixMilli() <= 0 || sentAt.UnixMilli() >= 253402300799000 { - return errors.New("mail: invalid cash mail identity or time") - } - if err := validateCashAttachments(rewards); err != nil { - return err - } - key := "cash:" + identity - if id, ok := s.issued[key]; ok { - previous := s.dynamic[id] - if !previous.IsCash || previous.TemplateID != template || !reflect.DeepEqual(mailAttachments(previous), rewards) { - return errors.New("mail: cash identity reused with different attachments") - } - return nil - } - return s.enqueueCompensations([]compensation{{identity: key, templateID: template, isCash: true, rewards: rewards, sentAt: sentAt}}) -} -func validateCashAttachments(rewards []gamedata.Reward) error { - if len(rewards) == 0 { - return errors.New("mail: empty cash attachments") - } - for _, r := range rewards { - if r.Type < 2 || r.Type > 72 || r.Count == 0 || r.Count > math.MaxInt32 || r.ID > math.MaxInt32 { - return fmt.Errorf("mail: invalid cash attachment %+v", r) - } - } - return nil -} -func (s *Service) cashInfo(request []byte) ([]byte, error) { - start, _, err := wire.Varint(request, 2) - if err != nil || start > math.MaxInt64 { - return nil, errors.New("mail: invalid cash cursor") - } - count, present, err := wire.Varint(request, 3) - if err != nil || !present || count == 0 || count > math.MaxInt32 { - return nil, errors.New("mail: invalid cash select count") - } - if count > mailHistoryPageMax { - count = mailHistoryPageMax - } - entries := map[uint64]MailDBInfo{} - for _, entry := range s.Starter.Mails { - if entry.IsCash && !containsID(s.state.Opened, entry.MailID) { - entries[entry.MailID] = entry - } - } - for id, entry := range s.dynamic { - if entry.IsCash && !containsID(s.state.Opened, id) { - entries[id] = entry - } - } - ids := make([]uint64, 0, len(entries)) - var max uint64 - for id := range entries { - ids = append(ids, id) - if id > max { - max = id - } - } - sort.Slice(ids, func(i, j int) bool { return ids[i] > ids[j] }) - var result []byte - var selected uint64 - for _, id := range ids { - if start != 0 && id >= start { - continue - } - if selected >= count { - break - } - result = wire.AppendBytes(result, 1, entries[id].encode()) - selected++ - } - result = wire.AppendVarint(result, 2, uint64(len(entries))) - return wire.AppendVarint(result, 3, max), nil -} diff --git a/go/internal/server/mail/content_ticket_test.go b/go/internal/server/mail/content_ticket_test.go deleted file mode 100644 index a791515..0000000 --- a/go/internal/server/mail/content_ticket_test.go +++ /dev/null @@ -1,21 +0,0 @@ -package mail - -import ( - "bd2server/internal/server/gamedata" - "testing" -) - -func TestContentTicketsUseCatalogAndRequireSingleUse(t *testing.T) { - s := &Service{} - if err := s.AttachContentTickets(&gamedata.GachaContentTicketDesign{IDs: map[uint64]bool{660003: true}}); err != nil { - t.Fatal(err) - } - for _, tc := range []struct { - id, count uint64 - want bool - }{{660003, 1, true}, {450030, 1, false}, {660003, 2, false}, {660003, 0, false}} { - if got := s.supportedItemDBInfoReward(gamedata.Reward{Type: 19, ID: tc.id, Count: tc.count}); got != tc.want { - t.Fatalf("ticket=%+v supported=%v", tc, got) - } - } -} diff --git a/go/internal/server/mail/grant_spool_test.go b/go/internal/server/mail/grant_spool_test.go deleted file mode 100644 index ca3a7e7..0000000 --- a/go/internal/server/mail/grant_spool_test.go +++ /dev/null @@ -1,475 +0,0 @@ -package mail - -import ( - "bd2server/internal/server/gamedata" - "encoding/json" - "errors" - "math" - "os" - "path/filepath" - "strings" - "testing" - "time" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func spoolTestService(t *testing.T, store stateio.Store) (*Service, *player.Inventory, *player.Wallet) { - t.Helper() - inventory, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(store, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(store, &Starter{Version: "2.35.10", MailCount: 1, MaxMailID: 100}, inventory, wallet) - if err != nil { - t.Fatal(err) - } - return service, inventory, wallet -} - -func spoolGrant(identity string, rewards ...GrantReward) Grant { - return Grant{Identity: identity, Title: "系统奖励", Body: "请领取邮件附件。", SentAt: time.Date(2026, 10, 3, 1, 0, 0, 0, time.UTC).UnixMilli(), Rewards: rewards} -} - -func writeSpool(t *testing.T, path string, grants ...Grant) { - t.Helper() - data, err := json.Marshal(GrantSpool{Version: 1, Grants: grants}) - if err != nil { - t.Fatal(err) - } - if err := os.WriteFile(path, data, 0o600); err != nil { - t.Fatal(err) - } -} - -func TestGrantSpoolImportsDynamicMailOnceAcrossOpenAndRestart(t *testing.T) { - store := stateio.NewMemory() - service, _, wallet := spoolTestService(t, store) - path := filepath.Join(t.TempDir(), "grants.json") - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - if _, _, _, err := service.Handle("/MailInfo", request); err != nil { - t.Fatal(err) - } - writeSpool(t, path, - spoolGrant("dev-mail:a", GrantReward{Type: 4, Count: 123}), - spoolGrant("dev-mail:b", GrantReward{Type: 8, ID: 9, Count: 10}), - ) - // Spool attachment never changes the versioned starter mailbox. - if len(service.Starter.Mails) != 0 { - t.Fatal("starter changed") - } - for range 2 { - code, response, handled, err := service.Handle("/MailInfo", request) - if err != nil || !handled || code != packetCode { - t.Fatalf("code=%d handled=%v err=%v", code, handled, err) - } - count, _, _ := wire.Varint(response, 2) - max, _, _ := wire.Varint(response, 3) - if count != 3 || max != 102 { - t.Fatalf("count=%d max=%d", count, max) - } - } - if wallet.Snapshot().Gold != 0 { - t.Fatal("unopened mail granted currency") - } - open := wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 101) - if _, _, _, err := service.Handle("/MailOpen", open); err != nil { - t.Fatal(err) - } - reopened, inventory, wallet := spoolTestService(t, store) - if err := reopened.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - _, response, _, err := reopened.Handle("/MailInfo", request) - if err != nil { - t.Fatal(err) - } - if count, _, _ := wire.Varint(response, 2); count != 2 { - t.Fatalf("count=%d", count) - } - if len(reopened.dynamic) != 2 || reopened.state.NextDynamicMailID != 103 { - t.Fatalf("reissued mail: %+v", reopened.state) - } - open = wire.AppendVarint(wire.AppendVarint(nil, 1, 3), 2, 102) - if _, _, _, err := reopened.Handle("/MailOpen", open); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Gold != 123 || len(inventory.All()) != 1 || inventory.All()[0].Count != 10 { - t.Fatalf("wallet=%+v inventory=%+v", wallet.Snapshot(), inventory.All()) - } - if _, _, _, err := reopened.Handle("/MailInfo", request); err != nil { - t.Fatal(err) - } - if reopened.state.NextDynamicMailID != 103 { - t.Fatal("opened grants were reissued") - } -} - -func TestGrantSpoolPersistsIssuedIdentityInSQLiteRequestTransaction(t *testing.T) { - dir := t.TempDir() - database := filepath.Join(dir, "state.db") - path := filepath.Join(dir, "grants.json") - writeSpool(t, path, spoolGrant("dev-mail:sqlite", GrantReward{Type: 4, Count: 123})) - repository, err := accountstate.Open(database) - if err != nil { - t.Fatal(err) - } - service, _, _ := spoolTestService(t, repository) - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - operation, err := repository.BeginOperation() - if err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - if _, _, _, err := service.Handle("/MailInfo", request); err != nil { - t.Fatal(err) - } - if err := operation.Commit(); err != nil { - t.Fatal(err) - } - if err := repository.Close(); err != nil { - t.Fatal(err) - } - repository, err = accountstate.Open(database) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repository.Close(); err != nil { - t.Error(err) - } - }() - service, _, _ = spoolTestService(t, repository) - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - operation, err = repository.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", request); err != nil { - t.Fatal(err) - } - if err := operation.Commit(); err != nil { - t.Fatal(err) - } - if len(service.dynamic) != 1 || service.issued["dev-mail:sqlite"] != 101 || service.state.NextDynamicMailID != 102 { - t.Fatalf("SQLite restart reissued grant: state=%+v issued=%+v", service.state, service.issued) - } -} - -func TestGrantSpoolPaidJewelryAndDrawTicketsClaimAndRestart(t *testing.T) { - store := stateio.NewMemory() - service, _, _ := spoolTestService(t, store) - path := filepath.Join(t.TempDir(), "grants.json") - writeSpool(t, path, spoolGrant("paid-and-draw-tickets", - GrantReward{Type: 2, Count: 100000000}, - GrantReward{Type: 8, ID: 1000, Count: 100000000}, - GrantReward{Type: 8, ID: 1104, Count: 100000000}, - GrantReward{Type: 4, Count: 1000000000}, - )) - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 1)); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 101) - _, response, _, err := service.Handle("/MailOpen", request) - if err != nil { - t.Fatal(err) - } - bundle, found, err := wire.Bytes(response, 1) - if err != nil || !found { - t.Fatalf("reward bundle missing: %v", err) - } - items := map[[2]uint64]uint64{} - views := map[[2]uint64]uint64{} - if err := wire.Walk(bundle, func(field wire.Field) error { - if field.Number != 1 && field.Number != 6 { - t.Fatalf("unexpected reward bundle field %d", field.Number) - } - id, _, err := wire.Varint(field.Value, 2) - if err != nil { - return err - } - typ, _, err := wire.Varint(field.Value, 3) - if err != nil { - return err - } - count, _, err := wire.Varint(field.Value, 4) - if err != nil { - return err - } - index, _, err := wire.Varint(field.Value, 1) - if err != nil { - return err - } - if field.Number == 1 { - if typ == 8 && index == 0 || typ != 8 && index != 0 { - t.Fatalf("incorrect item instance for type=%d index=%d", typ, index) - } - items[[2]uint64{typ, id}] = count - } else { - views[[2]uint64{typ, id}] = count - } - return nil - }); err != nil { - t.Fatal(err) - } - if len(items) != 4 || items[[2]uint64{2, 0}] != 100000000 || items[[2]uint64{4, 0}] != 1000000000 || - items[[2]uint64{8, 1000}] != 100000000 || items[[2]uint64{8, 1104}] != 100000000 || - len(views) != 2 || views[[2]uint64{8, 1000}] != 100000000 || views[[2]uint64{8, 1104}] != 100000000 { - t.Fatalf("items=%v views=%v", items, views) - } - service, inventory, wallet := spoolTestService(t, store) - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 3)); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - if got := wallet.Snapshot(); got.Jewelry != 100000000 || got.Gold != 1000000000 || len(inventory.All()) != 2 || service.state.NextDynamicMailID != 102 { - t.Fatalf("grant duplicated after restart: wallet=%+v items=%+v state=%+v", got, inventory.All(), service.state) - } -} - -func TestGrantSpoolEnforcesMailInt32IDAndCountBoundaries(t *testing.T) { - for _, reward := range []GrantReward{ - {Type: 8, ID: math.MaxInt32, Count: math.MaxInt32}, - {Type: 2, Count: math.MaxInt32}, - {Type: 19, ID: 450030, Count: 1}, - } { - input, err := json.Marshal(GrantSpool{Version: 1, Grants: []Grant{spoolGrant("boundary", reward)}}) - if err != nil { - t.Fatal(err) - } - if _, err := decodeGrantSpool(input); err != nil { - t.Fatalf("valid boundary reward rejected: %+v: %v", reward, err) - } - } - for _, reward := range []GrantReward{ - {Type: 8, ID: math.MaxInt32 + 1, Count: 1}, - {Type: 8, ID: 1000, Count: math.MaxInt32 + 1}, - {Type: 2, ID: 1, Count: 1}, - {Type: 2, Count: math.MaxInt32 + 1}, - {Type: 19, ID: 450030, Count: 0}, - {Type: 19, ID: 450030, Count: 2}, - } { - input, err := json.Marshal(GrantSpool{Version: 1, Grants: []Grant{spoolGrant("boundary", reward)}}) - if err != nil { - t.Fatal(err) - } - if _, err := decodeGrantSpool(input); err == nil { - t.Fatalf("invalid boundary reward accepted: %+v", reward) - } - } -} - -func TestGrantSpoolOneUseContentTicketItemDBInfoPersistsAndIsIdempotent(t *testing.T) { - // In 2.35.10 CommonPacket.AddItemInfo dispatches ElementType 19 to - // AddContentTicketItem; it consumes a normal ItemDBInfo with an instance ID. - store := stateio.NewMemory() - service, inventory, _ := spoolTestService(t, store) - if err := service.AttachContentTickets(&gamedata.GachaContentTicketDesign{IDs: map[uint64]bool{450030: true, 660003: true}}); err != nil { - t.Fatal(err) - } - path := filepath.Join(t.TempDir(), "grants.json") - writeSpool(t, path, spoolGrant("full-moon-one-use", GrantReward{Type: 19, ID: 450030, Count: 1})) - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 1)); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 101) - _, response, _, err := service.Handle("/MailOpen", request) - if err != nil { - t.Fatal(err) - } - bundle, found, err := wire.Bytes(response, 1) - if err != nil || !found { - t.Fatalf("reward bundle missing: %v", err) - } - item, found, err := wire.Bytes(bundle, 1) - if err != nil || !found { - t.Fatalf("ItemDBInfo missing: %v", err) - } - for field, want := range map[int]uint64{2: 450030, 3: 19, 4: 1} { - if got, present, err := wire.Varint(item, field); err != nil || !present || got != want { - t.Fatalf("ItemDBInfo field %d=%d present=%v err=%v", field, got, present, err) - } - } - index, _, err := wire.Varint(item, 1) - if err != nil || index == 0 { - t.Fatalf("content ticket instance=%d err=%v", index, err) - } - if got := inventory.All(); len(got) != 1 || got[0].Type != 19 || got[0].ID != 450030 || got[0].Count != 1 || got[0].InvenIndex != index { - t.Fatalf("content ticket not stored: %+v", got) - } - service, inventory, _ = spoolTestService(t, store) - if err := service.AttachContentTickets(&gamedata.GachaContentTicketDesign{IDs: map[uint64]bool{450030: true, 660003: true}}); err != nil { - t.Fatal(err) - } - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 3)); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - if got := inventory.All(); len(got) != 1 || got[0].Count != 1 || got[0].InvenIndex != index || service.state.NextDynamicMailID != 102 { - t.Fatalf("content ticket reissued: items=%+v state=%+v", got, service.state) - } -} - -func TestGrantSpoolValidatesEntireBatchBeforeIssuing(t *testing.T) { - store := stateio.NewMemory() - service, _, _ := spoolTestService(t, store) - path := filepath.Join(t.TempDir(), "grants.json") - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - good := spoolGrant("a", GrantReward{Type: 4, Count: 10}) - bad := spoolGrant("b", GrantReward{Type: 6, ID: 1, Count: 1}) - writeSpool(t, path, good, bad) - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 1)); err == nil { - t.Fatal("accepted unsupported second grant") - } - assertNoSpoolMail(t, service, store) - writeSpool(t, path, good, good) - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 2)); err == nil { - t.Fatal("accepted duplicate identity") - } - assertNoSpoolMail(t, service, store) - writeSpool(t, path, good) - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 3)); err != nil { - t.Fatal(err) - } - if service.issued["a"] != 101 { - t.Fatal("valid replacement did not import") - } -} - -func assertNoSpoolMail(t *testing.T, service *Service, store stateio.EntryStore) { - t.Helper() - if len(service.dynamic) != 0 || len(service.issued) != 0 || service.state.NextDynamicMailID != 101 { - t.Fatalf("partial memory update: %+v", service.state) - } - for _, bucket := range []string{"dynamic", "issued"} { - entries, err := store.ListEntries("mail", bucket) - if err != nil || len(entries) != 0 { - t.Fatalf("partial storage %s: entries=%v err=%v", bucket, entries, err) - } - } -} - -type failingSpoolStore struct { - *stateio.Memory - fail bool - writes int -} - -func (s *failingSpoolStore) SaveWithEntries(domain string, core []byte, changes []stateio.EntryMutation) error { - if domain == "mail" && len(changes) != 0 { - s.writes++ - if s.fail { - return errors.New("injected spool save failure") - } - } - return s.Memory.SaveWithEntries(domain, core, changes) -} - -func TestGrantSpoolBatchWriteFailureLeavesMemoryAndStorageUnchanged(t *testing.T) { - store := &failingSpoolStore{Memory: stateio.NewMemory(), fail: true} - service, _, _ := spoolTestService(t, store) - path := filepath.Join(t.TempDir(), "grants.json") - writeSpool(t, path, spoolGrant("a", GrantReward{Type: 4, Count: 10}), spoolGrant("b", GrantReward{Type: 4, Count: 20})) - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - if _, _, _, err := service.Handle("/MailInfo", request); err == nil { - t.Fatal("save failure ignored") - } - assertNoSpoolMail(t, service, store) - if store.writes != 1 { - t.Fatalf("batch used %d writes", store.writes) - } - store.fail = false - if _, _, _, err := service.Handle("/MailInfo", request); err != nil { - t.Fatal(err) - } - if len(service.dynamic) != 2 || service.state.NextDynamicMailID != 103 { - t.Fatal("retry did not issue complete batch") - } -} - -func TestGrantSpoolBatchIDExhaustionDoesNotIssueFirstMail(t *testing.T) { - store := stateio.NewMemory() - service, _, _ := spoolTestService(t, store) - service.state.NextDynamicMailID = ^uint64(0) - 1 - path := filepath.Join(t.TempDir(), "grants.json") - writeSpool(t, path, spoolGrant("a", GrantReward{Type: 4, Count: 10}), spoolGrant("b", GrantReward{Type: 4, Count: 20})) - if err := service.AttachGrantSpoolPath(path); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 1)); err == nil { - t.Fatal("accepted overflowing batch") - } - if len(service.dynamic) != 0 || service.state.NextDynamicMailID != ^uint64(0)-1 { - t.Fatal("partially imported overflowing batch") - } -} - -func TestGrantSpoolRejectsMalformedSchema(t *testing.T) { - valid := `{"version":1,"grants":[{"identity":"a","title":"t","body":"b","sent_at":1790989200000,"rewards":[{"type":4,"id":0,"count":10}]}]}` - if _, err := decodeGrantSpool([]byte(valid)); err != nil { - t.Fatal(err) - } - invalid := []string{ - `null`, `{}`, `{"version":1,"grants":null}`, `{"version":1,"grants":[],"extra":0}`, - strings.Replace(valid, `"version":1`, `"version":2`, 1), - strings.Replace(valid, `"version":1`, `"version":1,"version":1`, 1), - strings.Replace(valid, `"title":"t"`, `"title":"t","title":"u"`, 1), - strings.Replace(valid, `"title":"t"`, `"Title":"t"`, 1), - strings.Replace(valid, `"body":"b"`, `"body":null`, 1), - strings.Replace(valid, `"identity":"a"`, `"identity":" "`, 1), - strings.Replace(valid, `"sent_at":1790989200000`, `"sent_at":0`, 1), - strings.Replace(valid, `"sent_at":1790989200000`, `"sent_at":9223372036854775807`, 1), - strings.Replace(valid, `"count":10`, `"count":0`, 1), - strings.Replace(valid, `"count":10`, `"count":1.5`, 1), - strings.Replace(valid, `"count":10`, `"count":null`, 1), - strings.Replace(valid, `"count":10`, `"count":10,"count":10`, 1), - strings.Replace(valid, `"count":10`, `"count":10,"extra":0`, 1), - strings.Replace(valid, `"id":0,`, ``, 1), - strings.Replace(valid, `"id":0`, `"id":1`, 1), - strings.Replace(valid, `"type":4`, `"type":8`, 1), - valid + `{}`, - } - for _, input := range invalid { - if _, err := decodeGrantSpool([]byte(input)); err == nil { - t.Errorf("accepted malformed spool: %s", input) - } - } - if _, err := decodeGrantSpool([]byte(`{"version":1,"grants":[]}`)); err != nil { - t.Fatal(err) - } -} diff --git a/go/internal/server/mail/mail_test.go b/go/internal/server/mail/mail_test.go deleted file mode 100644 index 39cd026..0000000 --- a/go/internal/server/mail/mail_test.go +++ /dev/null @@ -1,750 +0,0 @@ -package mail - -import ( - "bytes" - "os" - "path/filepath" - "testing" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func TestStarterAnswersMailInfoWithoutCapture(t *testing.T) { - seed, err := Load(filepath.Join("..", "..", "..", "seed", "v2_35_10", "mail.json")) - if err != nil { - t.Fatal(err) - } - code, proto, ok, err := seed.Handle("/MailInfo", wire.AppendVarint(nil, 1, 249)) - if err != nil || !ok || code != packetCode || len(proto) == 0 { - t.Fatalf("MailInfo: code=%d bytes=%d ok=%t err=%v", code, len(proto), ok, err) - } - if _, _, ok, err := seed.Handle("/MailOpen", wire.AppendVarint(nil, 1, 1)); ok || err != nil { - t.Fatal("seed claimed an endpoint it does not own") - } - if _, _, ok, err := seed.Handle("/MailInfo", nil); !ok || err == nil { - t.Fatal("missing sequence was accepted") - } -} - -func TestDynamicCompensationMailPersistsAndIsIdempotent(t *testing.T) { - seed := &Starter{Version: "2.35.10", MailCount: 1, MaxMailID: 100} - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - now := time.Date(2026, 9, 24, 0, 0, 0, 0, time.UTC) - rewards := []gamedata.Reward{{Type: 4, Count: 123}} - if err := service.EnqueueCompensation("daily:2026-09-23:1", "日常任务到期补发", "任务已完成但未领取。", rewards, now); err != nil { - t.Fatal(err) - } - if err := service.EnqueueCompensation("daily:2026-09-23:1", "日常任务到期补发", "任务已完成但未领取。", rewards, now); err != nil { - t.Fatal(err) - } - reopened, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - if len(reopened.dynamic) != 1 || reopened.issued["daily:2026-09-23:1"] != 101 { - t.Fatalf("dynamic=%+v issued=%+v", reopened.dynamic, reopened.issued) - } - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendVarint(request, 2, 101) - if _, _, _, err := reopened.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Gold != 123 { - t.Fatalf("gold=%d", wallet.Snapshot().Gold) - } -} - -func TestMailOpenGrantsItemsAndCurrencyAndPersists(t *testing.T) { - seed := &Starter{Version: "2.35.10", MailCount: 3, MaxMailID: 12, Mails: []MailDBInfo{ - {MailID: 11, MailType: 2, ExpiresAt: 100, SentAt: 10, RewardTypes: []uint64{3, 4}, RewardIDs: []uint64{0, 0}, RewardCounts: []uint64{70, 123456789}}, - {MailID: 12, MailType: 2, ExpiresAt: 100, SentAt: 10, RewardTypes: []uint64{8}, RewardIDs: []uint64{9}, RewardCounts: []uint64{10}}, - }} - starter := &player.Starter{Version: "2.35.10"} - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, starter) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{Gold: 321}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendBytes(request, 2, packed([]uint64{11, 12})) - code, response, ok, err := service.Handle("/MailOpen", request) - if err != nil || !ok || code != 132 { - t.Fatalf("code=%d ok=%v err=%v", code, ok, err) - } - if wallet.Snapshot().FreeJewelry != 70 { - t.Fatalf("wallet=%+v", wallet.Snapshot()) - } - if wallet.Snapshot().Gold != 123457110 { - t.Fatalf("gold did not stack into wallet: %+v", wallet.Snapshot()) - } - if len(inv.All()) != 1 || inv.All()[0].ID != 9 || inv.All()[0].Count != 10 { - t.Fatalf("items=%+v", inv.All()) - } - bundle, found, _ := wire.Bytes(response, 1) - if !found || len(bundle) == 0 { - t.Fatal("reward bundle missing") - } - infoReq := wire.AppendVarint(nil, 1, 2) - _, info, _, err := service.Handle("/MailInfo", infoReq) - if err != nil { - t.Fatal(err) - } - if _, found, _ := wire.Bytes(info, 1); found { - t.Fatal("opened mail remained visible") - } - count, found, _ := wire.Varint(info, 2) - if !found || count != 1 { - t.Fatalf("mail count=%d found=%v", count, found) - } - service, err = OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - _, info, _, _ = service.Handle("/MailInfo", infoReq) - if _, found, _ := wire.Bytes(info, 1); found { - t.Fatal("opened mail was not persisted") - } -} - -func TestMailHistoryPersistsPagesAndKeepsFirstOpenTime(t *testing.T) { - seed := &Starter{Version: "2.35.10", MailCount: 4, MaxMailID: 13, Mails: []MailDBInfo{ - {MailID: 11, MailType: 2, Title: "first", ExpiresAt: 100, SentAt: 10, RewardTypes: []uint64{4}, RewardIDs: []uint64{0}, RewardCounts: []uint64{1}}, - {MailID: 12, MailType: 2, Title: "second", ExpiresAt: 200, SentAt: 20, RewardTypes: []uint64{4}, RewardIDs: []uint64{0}, RewardCounts: []uint64{2}}, - {MailID: 13, MailType: 2, Title: "third", ExpiresAt: 300, SentAt: 30, RewardTypes: []uint64{4}, RewardIDs: []uint64{0}, RewardCounts: []uint64{3}}, - }} - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - openedAt := time.Date(2026, 10, 3, 5, 6, 7, 8_000_000, time.UTC) - service.now = func() time.Time { return openedAt } - open := wire.AppendVarint(nil, 1, 1) - open = wire.AppendBytes(open, 2, packed([]uint64{11, 12, 13})) - if _, _, _, err := service.Handle("/MailOpen", open); err != nil { - t.Fatal(err) - } - - request := wire.AppendVarint(nil, 1, 2) - request = wire.AppendVarint(request, 2, 0) - request = wire.AppendVarint(request, 3, 2) - code, response, handled, err := service.Handle("/MailHistoryInfo", request) - if err != nil || !handled || code != historyPacketCode { - t.Fatalf("code=%d handled=%v err=%v", code, handled, err) - } - entries := historyEntries(t, response) - if len(entries) != 2 || historyID(t, entries[0]) != 13 || historyID(t, entries[1]) != 12 { - t.Fatalf("first page IDs=%v", historyIDs(t, entries)) - } - total, found, err := wire.Varint(response, 2) - if err != nil || !found || total != 3 { - t.Fatalf("total=%d found=%v err=%v", total, found, err) - } - if isOpen, found, _ := wire.Varint(entries[0], 11); !found || isOpen != 1 { - t.Fatalf("history is_open=%d found=%v", isOpen, found) - } - if got, found, _ := wire.Varint(entries[0], 12); !found || got != uint64(openedAt.UnixMilli()) { - t.Fatalf("open_time=%d found=%v", got, found) - } - if got, found, _ := wire.Varint(entries[0], 14); !found || got != uint64(openedAt.Add(mailHistoryPeriod).UnixMilli()) { - t.Fatalf("history_delete_time=%d found=%v", got, found) - } - - next := wire.AppendVarint(nil, 1, 3) - next = wire.AppendVarint(next, 2, 12) - next = wire.AppendVarint(next, 3, 2) - _, response, _, err = service.Handle("/MailHistoryInfo", next) - if err != nil { - t.Fatal(err) - } - entries = historyEntries(t, response) - if len(entries) != 1 || historyID(t, entries[0]) != 11 { - t.Fatalf("second page IDs=%v", historyIDs(t, entries)) - } - - service.now = func() time.Time { return openedAt.Add(24 * time.Hour) } - if _, _, _, err := service.Handle("/MailOpen", wire.AppendVarint(wire.AppendVarint(nil, 1, 4), 2, 13)); err != nil { - t.Fatal(err) - } - reopened, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - reopened.now = service.now - _, response, _, err = reopened.Handle("/MailHistoryInfo", request) - if err != nil { - t.Fatal(err) - } - entries = historyEntries(t, response) - if got, _, _ := wire.Varint(entries[0], 12); got != uint64(openedAt.UnixMilli()) { - t.Fatalf("retry changed first open time to %d", got) - } - - reopened.now = func() time.Time { return openedAt.Add(mailHistoryPeriod) } - _, response, _, err = reopened.Handle("/MailHistoryInfo", request) - if err != nil { - t.Fatal(err) - } - if entries := historyEntries(t, response); len(entries) != 0 { - t.Fatalf("expired history remains: %v", historyIDs(t, entries)) - } - if total, _, _ := wire.Varint(response, 2); total != 0 { - t.Fatalf("expired total=%d", total) - } -} - -func TestMailHistoryRejectsInvalidPagination(t *testing.T) { - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, &Starter{Version: "2.35.10", MailCount: 1, MaxMailID: 10}, inv, wallet) - if err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - if _, _, handled, err := service.Handle("/MailHistoryInfo", request); !handled || err == nil { - t.Fatalf("handled=%v err=%v", handled, err) - } -} - -type mailCostumeDesign map[uint64]gamedata.CharacterDesign - -func (d mailCostumeDesign) Character(costumeID uint64) (gamedata.CharacterDesign, bool) { - value, ok := d[costumeID] - return value, ok -} - -func TestMailCostumeRewardCreatesEnhancementFiveIdempotently(t *testing.T) { - seed := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 11, Mails: []MailDBInfo{{ - MailID: 11, MailType: 2, ExpiresAt: 100, SentAt: 10, - RewardTypes: []uint64{11}, RewardIDs: []uint64{206}, RewardCounts: []uint64{6}, - }}} - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(storage, nil) - if err != nil { - t.Fatal(err) - } - if err := collection.BindBaseCharacters(nil); err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - if err := service.AttachCostumeRewards(collection, mailCostumeDesign{206: { - ID: 20, HP: 100, CostumeMaxLevel: 5, OverflowItemType: 20, OverflowItemCount: 10, - }}); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 11) - _, response, _, err := service.Handle("/MailOpen", request) - if err != nil { - t.Fatal(err) - } - if got := collection.Costumes(); len(got) != 1 || got[0].ID != 206 || got[0].Level != 5 { - t.Fatalf("costumes=%+v", got) - } - bundle, found, err := wire.Bytes(response, 1) - if err != nil || !found { - t.Fatalf("bundle: found=%v err=%v", found, err) - } - var characters, costumes, upgrades int - if err := wire.Walk(bundle, func(field wire.Field) error { - switch field.Number { - case 2: - characters++ - case 3: - costumes++ - case 9: - upgrades++ - } - return nil - }); err != nil { - t.Fatal(err) - } - if characters != 1 || costumes != 1 || upgrades != 5 { - t.Fatalf("characters=%d costumes=%d upgrades=%d", characters, costumes, upgrades) - } - if _, _, _, err := service.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - if got := collection.Costumes(); len(got) != 1 || got[0].Level != 5 { - t.Fatalf("retry costumes=%+v", got) - } -} - -func TestEnsureStarterLimitedCostumesIsDurablyIdempotent(t *testing.T) { - storage := stateio.NewMemory() - service, _, _ := spoolTestService(t, storage) - design := mailCostumeDesign{ - 206: {ID: 20, HP: 100, CostumeMaxLevel: 5, OverflowItemType: 20, OverflowItemCount: 10}, - 306: {ID: 30, HP: 100, CostumeMaxLevel: 5, OverflowItemType: 20, OverflowItemCount: 10}, - } - collection, err := player.OpenCollectionStore(storage, nil) - if err != nil { - t.Fatal(err) - } - if err := service.AttachCostumeRewards(collection, design); err != nil { - t.Fatal(err) - } - firstTime := time.Date(2026, 10, 3, 8, 0, 0, 0, time.UTC) - if err := service.EnsureStarterLimitedCostumes([]uint64{206}, firstTime); err != nil { - t.Fatal(err) - } - firstID := service.issued[starterLimitedCostumeIdentity] - if firstID == 0 || len(service.dynamic) != 1 { - t.Fatalf("first issue: id=%d dynamic=%v", firstID, service.dynamic) - } - // A retry after the schedule changed must preserve the originally frozen - // mail rather than allocating a second ID or replacing its attachments. - if err := service.EnsureStarterLimitedCostumes([]uint64{306}, firstTime.Add(24*time.Hour)); err != nil { - t.Fatal(err) - } - if service.issued[starterLimitedCostumeIdentity] != firstID || len(service.dynamic) != 1 || service.dynamic[firstID].RewardIDs[0] != 206 { - t.Fatalf("in-process retry changed gift: issued=%v dynamic=%v", service.issued, service.dynamic) - } - - reopened, _, _ := spoolTestService(t, storage) - if err := reopened.AttachCostumeRewards(collection, design); err != nil { - t.Fatal(err) - } - if err := reopened.EnsureStarterLimitedCostumes([]uint64{306}, firstTime.Add(48*time.Hour)); err != nil { - t.Fatal(err) - } - if reopened.issued[starterLimitedCostumeIdentity] != firstID || len(reopened.dynamic) != 1 || reopened.dynamic[firstID].RewardIDs[0] != 206 { - t.Fatalf("restart retry changed gift: issued=%v dynamic=%v", reopened.issued, reopened.dynamic) - } -} - -func TestStarterPrestigeSkinMailClaimsAndPersistsOwnershipOnce(t *testing.T) { - storage := stateio.NewMemory() - service, _, _ := spoolTestService(t, storage) - sentAt := time.Date(2026, 10, 5, 8, 0, 0, 0, time.UTC) - skins := []uint64{101, 202} - if err := service.EnsureStarterPrestigeSkins(skins, sentAt); err != nil { - t.Fatal(err) - } - id := service.issued[starterPrestigeSkinIdentity] - if id == 0 || len(service.dynamic) != 1 { - t.Fatalf("gift missing: issued=%v dynamic=%v", service.issued, service.dynamic) - } - original := append([]byte(nil), service.dynamic[id].encode()...) - if err := service.EnsureStarterPrestigeSkins(skins, sentAt.Add(time.Hour)); err != nil { - t.Fatal(err) - } - service, inventory, _ := spoolTestService(t, storage) - if service.issued[starterPrestigeSkinIdentity] != id || len(service.dynamic) != 1 || !bytes.Equal(original, service.dynamic[id].encode()) { - t.Fatal("retry or restart changed the unclaimed gift") - } - request := wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, id) - code, response, handled, err := service.Handle("/MailOpen", request) - if err != nil || !handled || code != 132 { - t.Fatalf("MailOpen: code=%d handled=%v err=%v", code, handled, err) - } - bundle, found, err := wire.Bytes(response, 1) - if err != nil || !found { - t.Fatalf("reward bundle: found=%v err=%v", found, err) - } - // Check the actual client-visible ItemDBInfo, including the inventory - // indices which must remain stable when this mail is claimed again. - assertSkins := func(payload []byte) map[uint64]uint64 { - t.Helper() - indices := make(map[uint64]uint64) - for _, item := range historyEntries(t, payload) { - index, _, _ := wire.Varint(item, 1) - design, _, _ := wire.Varint(item, 2) - typ, _, _ := wire.Varint(item, 3) - count, _, _ := wire.Varint(item, 4) - if index == 0 || typ != 45 || count != 1 || (design != skins[0] && design != skins[1]) || indices[design] != 0 { - t.Fatalf("invalid or duplicate skin ItemDBInfo: index=%d design=%d type=%d count=%d", index, design, typ, count) - } - indices[design] = index - } - if len(indices) != len(skins) { - t.Fatalf("skin ownership=%v", indices) - } - return indices - } - granted := assertSkins(bundle) - assertOwnership := func(inv *player.Inventory) { - t.Helper() - _, info, handled, err := inv.Handle("/ItemInfo", wire.AppendVarint(nil, 1, 2)) - if err != nil || !handled { - t.Fatalf("ItemInfo: handled=%v err=%v", handled, err) - } - owned := assertSkins(info) - for design, index := range granted { - if owned[design] != index { - t.Fatalf("skin %d inventory index changed: got=%d want=%d", design, owned[design], index) - } - } - } - assertOwnership(inventory) - if _, _, _, err := service.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - assertOwnership(inventory) - if err := service.EnsureStarterPrestigeSkins([]uint64{303}, sentAt.Add(24*time.Hour)); err != nil { - t.Fatal(err) - } - service, inventory, _ = spoolTestService(t, storage) - if err := service.EnsureStarterPrestigeSkins([]uint64{404}, sentAt.Add(48*time.Hour)); err != nil { - t.Fatal(err) - } - if service.issued[starterPrestigeSkinIdentity] != id || len(service.dynamic) != 1 || !bytes.Equal(original, service.dynamic[id].encode()) || !containsID(service.state.Opened, id) { - t.Fatal("restart lost the claimed gift ledger or replaced its frozen attachments") - } - if _, _, _, err := service.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - assertOwnership(inventory) -} - -func historyEntries(t *testing.T, response []byte) [][]byte { - t.Helper() - var entries [][]byte - if err := wire.Walk(response, func(field wire.Field) error { - if field.Number == 1 { - entries = append(entries, append([]byte(nil), field.Value...)) - } - return nil - }); err != nil { - t.Fatal(err) - } - return entries -} - -func historyID(t *testing.T, entry []byte) uint64 { - t.Helper() - id, found, err := wire.Varint(entry, 1) - if err != nil || !found { - t.Fatalf("history ID: found=%v err=%v", found, err) - } - return id -} - -func historyIDs(t *testing.T, entries [][]byte) []uint64 { - t.Helper() - ids := make([]uint64, 0, len(entries)) - for _, entry := range entries { - ids = append(ids, historyID(t, entry)) - } - return ids -} - -func TestStarterContentTicketOnlyAllowsAuditedSingleUseReward(t *testing.T) { - for _, reward := range []struct { - id, count uint64 - valid bool - }{{450030, 1, true}, {450029, 1, true}, {450030, 0, false}, {450030, 2, false}} { - seed := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 13, Mails: []MailDBInfo{{ - MailID: 13, MailType: 2, ExpiresAt: 100, SentAt: 10, - RewardTypes: []uint64{19}, RewardIDs: []uint64{reward.id}, RewardCounts: []uint64{reward.count}, - }}} - if err := seed.Validate(); (err == nil) != reward.valid { - t.Fatalf("id=%d count=%d valid=%v err=%v", reward.id, reward.count, reward.valid, err) - } - } -} - -func TestMailOpenGrantsNonResourceItemDBInfoType(t *testing.T) { - seed := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 13, Mails: []MailDBInfo{{ - MailID: 13, MailType: 2, ExpiresAt: 100, SentAt: 10, - RewardTypes: []uint64{14}, RewardIDs: []uint64{1}, RewardCounts: []uint64{2}, - }}} - starter := &player.Starter{Version: "2.35.10"} - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, starter) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendBytes(request, 2, packed([]uint64{13})) - if _, _, _, err := service.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - items := inv.All() - if len(items) != 1 || items[0].ID != 1 || items[0].Type != 14 || items[0].Count != 2 { - t.Fatalf("items=%+v", items) - } -} - -func TestMailOpenGrantsCatalystAndMileageCurrenciesAndPersists(t *testing.T) { - seed := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 14, Mails: []MailDBInfo{{ - MailID: 14, MailType: 2, ExpiresAt: 100, SentAt: 10, - RewardTypes: []uint64{12, 20}, RewardIDs: []uint64{0, 0}, RewardCounts: []uint64{250, 300}, - }}} - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{Catalyst: 10, Mileage: 20}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, seed, inv, wallet) - if err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendBytes(request, 2, packed([]uint64{14})) - _, response, _, err := service.Handle("/MailOpen", request) - if err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Catalyst != 260 || wallet.Snapshot().Mileage != 320 || len(inv.All()) != 0 { - t.Fatalf("wallet=%+v items=%+v", wallet.Snapshot(), inv.All()) - } - bundle, found, _ := wire.Bytes(response, 1) - if !found { - t.Fatal("reward bundle missing") - } - gotCurrencies := map[uint64]uint64{} - if err := wire.Walk(bundle, func(field wire.Field) error { - if field.Number != 1 { - return nil - } - typ, _, err := wire.Varint(field.Value, 3) - if err != nil { - return err - } - count, _, err := wire.Varint(field.Value, 4) - if err != nil { - return err - } - gotCurrencies[typ] = count - return nil - }); err != nil { - t.Fatal(err) - } - if gotCurrencies[12] != 250 || gotCurrencies[20] != 300 { - t.Fatalf("currency rewards=%v", gotCurrencies) - } - if _, _, _, err := service.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Catalyst != 260 || wallet.Snapshot().Mileage != 320 { - t.Fatalf("replay wallet=%+v", wallet.Snapshot()) - } - reopened, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - if reopened.Snapshot().Catalyst != 260 || reopened.Snapshot().Mileage != 320 { - t.Fatalf("persisted wallet=%+v", reopened.Snapshot()) - } -} - -func TestWatchedSeedReloadsOnlyValidAtomicReplacement(t *testing.T) { - dir := t.TempDir() - seedPath := filepath.Join(dir, "seed.json") - first := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 11, Mails: []MailDBInfo{{MailID: 11, MailType: 2, ExpiresAt: 100, SentAt: 10}}} - if err := first.Write(seedPath); err != nil { - t.Fatal(err) - } - starter, err := Load(seedPath) - if err != nil { - t.Fatal(err) - } - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, starter, inv, wallet) - if err != nil { - t.Fatal(err) - } - if err := service.AttachSeedPath(seedPath); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 1) - if _, response, _, err := service.Handle("/MailInfo", request); err != nil { - t.Fatal(err) - } else if id, _, _ := wire.Varint(mustFirstMail(t, response), 1); id != 11 { - t.Fatalf("initial mail ID=%d", id) - } - second := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 12, Mails: []MailDBInfo{{MailID: 12, MailType: 2, ExpiresAt: 100, SentAt: 10}}} - if err := second.Write(seedPath); err != nil { - t.Fatal(err) - } - if _, response, _, err := service.Handle("/MailInfo", request); err != nil { - t.Fatal(err) - } else if id, _, _ := wire.Varint(mustFirstMail(t, response), 1); id != 12 { - t.Fatalf("reloaded mail ID=%d", id) - } - if err := os.WriteFile(seedPath, []byte("not json"), 0o600); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", request); err == nil { - t.Fatal("accepted malformed watched seed") - } - // The malformed file did not replace the last known-good mailbox. - if service.Starter.Mails[0].MailID != 12 { - t.Fatalf("last known-good seed lost: %+v", service.Starter.Mails) - } -} - -func TestExpandedWatchedSeedAdvancesDynamicMailAllocator(t *testing.T) { - dir := t.TempDir() - seedPath := filepath.Join(dir, "seed.json") - first := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 11, Mails: []MailDBInfo{{MailID: 11, MailType: 2, ExpiresAt: 100, SentAt: 10}}} - if err := first.Write(seedPath); err != nil { - t.Fatal(err) - } - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := OpenService(storage, first, inv, wallet) - if err != nil { - t.Fatal(err) - } - if err := service.EnsurePersisted(); err != nil { - t.Fatal(err) - } - if err := service.AttachSeedPath(seedPath); err != nil { - t.Fatal(err) - } - second := &Starter{Version: "2.35.10", MailCount: 3, MaxMailID: 101, Mails: []MailDBInfo{ - {MailID: 11, MailType: 2, ExpiresAt: 100, SentAt: 10}, - {MailID: 101, MailType: 2, ExpiresAt: 100, SentAt: 10}, - }} - if err := second.Write(seedPath); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/MailInfo", wire.AppendVarint(nil, 1, 1)); err != nil { - t.Fatal(err) - } - if service.state.NextDynamicMailID != 102 { - t.Fatalf("next dynamic mail ID=%d", service.state.NextDynamicMailID) - } - reopened, err := OpenService(storage, second, inv, wallet) - if err != nil { - t.Fatal(err) - } - if reopened.state.NextDynamicMailID != 102 { - t.Fatalf("reopened next dynamic mail ID=%d", reopened.state.NextDynamicMailID) - } -} - -func mustFirstMail(t *testing.T, response []byte) []byte { - t.Helper() - entry, found, err := wire.Bytes(response, 1) - if err != nil || !found { - t.Fatalf("missing first mail: found=%v err=%v", found, err) - } - return entry -} - -func TestMailOpenAcceptsOfficialPackedRequest(t *testing.T) { - request := []byte{0x08, 0xef, 0x01, 0x12, 0x0f, 0x85, 0xc1, 0xe1, 0xce, 0x30, 0x88, 0xc1, 0xe1, 0xce, 0x30, 0x8a, 0xc1, 0xe1, 0xce, 0x30} - ids, err := requestMailIDs(request) - if err != nil { - t.Fatal(err) - } - want := []uint64{13050077317, 13050077320, 13050077322} - if len(ids) != len(want) { - t.Fatalf("ids=%v", ids) - } - for i := range want { - if ids[i] != want[i] { - t.Fatalf("ids=%v", ids) - } - } -} - -func TestStarterMailProtocolEncodingAndRoundTrip(t *testing.T) { - seed := &Starter{Version: "2.35.10", MailCount: 2, MaxMailID: 7, Mails: []MailDBInfo{{ - MailID: 7, MailType: 2, Title: "Welcome", Body: "Rewards", ExpiresAt: 1000, SentAt: 10, - RewardTypes: []uint64{3, 8}, RewardIDs: []uint64{0, 17}, RewardCounts: []uint64{5, 300}, - }}} - path := filepath.Join(t.TempDir(), "mail.json") - if err := seed.Write(path); err != nil { - t.Fatal(err) - } - loaded, err := Load(path) - if err != nil { - t.Fatal(err) - } - // MailInfoResponse contains MailDBInfo field 1 plus count and maximum ID. - // Reward arrays are packed protobuf varints; default ID 0 remains a slot. - wantMail := []byte{0x08, 7, 0x10, 2, 0x2a, 7, 'W', 'e', 'l', 'c', 'o', 'm', 'e', 0x32, 7, 'R', 'e', 'w', 'a', 'r', 'd', 's', 0x38, 0xe8, 7, 0x42, 2, 3, 8, 0x4a, 2, 0, 17, 0x52, 3, 5, 0xac, 2, 0x68, 10} - want := append([]byte{0x0a, byte(len(wantMail))}, wantMail...) - want = append(want, 0x10, 2, 0x18, 7) - code, actual, handled, err := loaded.Handle("/MailInfo", wire.AppendVarint(nil, 1, 249)) - if err != nil || !handled || code != 131 || !bytes.Equal(actual, want) { - t.Fatalf("code=%d proto=%x handled=%v err=%v", code, actual, handled, err) - } -} - -func TestValidateRejectsRewardLengthMismatch(t *testing.T) { - seed := &Starter{Version: "2.35.10", Mails: []MailDBInfo{{MailID: 1, MailType: 2, ExpiresAt: 1, SentAt: 1, RewardTypes: []uint64{8}, RewardIDs: []uint64{1}}}, MailCount: 2, MaxMailID: 1} - if seed.Validate() == nil { - t.Fatal("accepted bad rewards") - } -} diff --git a/go/internal/server/mail/notifications.go b/go/internal/server/mail/notifications.go deleted file mode 100644 index 403002a..0000000 --- a/go/internal/server/mail/notifications.go +++ /dev/null @@ -1,29 +0,0 @@ -package mail - -import "bd2server/internal/server/wire" - -// BeforeDispatch and AfterDispatch observe newly allocated mail inside the -// account request transaction, including mail issued by other domains. -func (s *Service) BeforeDispatch(string, []byte) error { - s.mu.Lock() - defer s.mu.Unlock() - s.beforeMailID = s.state.NextDynamicMailID - return nil -} - -func (s *Service) AfterDispatch(string, []byte, []byte) ([]byte, error) { - s.mu.Lock() - defer s.mu.Unlock() - // Allocation is monotonic. Without a new ID, no existing row can match - // the notification window, so read-only packets need not scan the inbox. - if s.state.NextDynamicMailID == s.beforeMailID { - return nil, nil - } - now := uint64(s.now().UnixMilli()) - for id, entry := range s.dynamic { - if id >= s.beforeMailID && !containsID(s.state.Opened, id) && entry.ExpiresAt > now { - return wire.AppendVarint(nil, 1, 1), nil // Notify.IsNewMail - } - } - return nil, nil -} diff --git a/go/internal/server/mail/notifications_test.go b/go/internal/server/mail/notifications_test.go deleted file mode 100644 index 5c98b09..0000000 --- a/go/internal/server/mail/notifications_test.go +++ /dev/null @@ -1,34 +0,0 @@ -package mail - -import ( - "testing" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func TestNewMailNotificationDoesNotRepeatOnUnchangedRequests(t *testing.T) { - s, _, _ := attendanceMailFixture(t, stateio.NewMemory(), &Starter{Version: "2.35.10", MailCount: 1}, &attendanceDesign{}) - now := time.Now().UTC() - if err := s.BeforeDispatch("/Attendance", nil); err != nil { - t.Fatal(err) - } - if err := s.IssueAttachmentsOnce("today", "Attendance Rewards", "Claim your rewards.", []gamedata.Reward{{Type: 9, ID: 100, Count: 1}}, now); err != nil { - t.Fatal(err) - } - notice, err := s.AfterDispatch("/Attendance", nil, nil) - if v, ok, parseErr := wire.Varint(notice, 1); err != nil || parseErr != nil || !ok || v != 1 { - t.Fatal("new mail did not notify", err, parseErr) - } - if err := s.BeforeDispatch("/read", nil); err != nil { - t.Fatal(err) - } - if err := s.IssueAttachmentsOnce("today", "Attendance Rewards", "Claim your rewards.", []gamedata.Reward{{Type: 9, ID: 100, Count: 1}}, now); err != nil { - t.Fatal(err) - } - if notice, err := s.AfterDispatch("/read", nil, nil); err != nil || len(notice) != 0 { - t.Fatal("unchanged mail notified again", err) - } -} diff --git a/go/internal/server/missions/achievement_progress_test.go b/go/internal/server/missions/achievement_progress_test.go deleted file mode 100644 index 22e9304..0000000 --- a/go/internal/server/missions/achievement_progress_test.go +++ /dev/null @@ -1,128 +0,0 @@ -package missions - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "testing" -) - -type testAchievementProgress uint64 - -func (p *testAchievementProgress) AchievementValue(uint64) (uint64, error) { return uint64(*p), nil } -func TestAchievementClearRequiresProgressBeforeAnyBatchReward(t *testing.T) { - store := stateio.NewMemory() - inv, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - design := &gamedata.MissionDesign{Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{{GroupID: 987, ID: 1}: {Target: 5, CounterGroup: 987, AddExp: 7, Rewards: []gamedata.Reward{{Type: 8, ID: 123, Count: 1}}}, {GroupID: 987, ID: 2}: {Target: 10, CounterGroup: 987, AddExp: 9, Rewards: []gamedata.Reward{{Type: 8, ID: 123, Count: 2}}}}} - s, err := Open(store, design, inv) - if err != nil { - t.Fatal(err) - } - progress := testAchievementProgress(5) - if err := s.AttachAchievementProgress(&progress); err != nil { - t.Fatal(err) - } - info := wire.AppendVarint(nil, 1, 987) - info = wire.AppendVarint(info, 2, 1) - info = wire.AppendVarint(info, 2, 2) - req := wire.AppendBytes(wire.AppendVarint(nil, 1, 1), 3, info) - if _, _, _, err := s.Handle("/AchievementClear", req); err == nil { - t.Fatal("unearned batch claimed") - } - if len(inv.All()) != 0 || len(s.state.Claimed) != 0 { - t.Fatal("earlier batch reward mutated") - } - progress = 10 - if _, _, _, err := s.Handle("/AchievementClear", req); err != nil { - t.Fatal(err) - } - if exp, err := s.AchievementExperience(); err != nil || exp != 16 { - t.Fatalf("exp=%d err=%v", exp, err) - } - if len(inv.All()) != 2 || inv.All()[0].Count+inv.All()[1].Count != 3 { - t.Fatalf("rewards=%+v", inv.All()) - } - if _, _, _, err := s.Handle("/AchievementClear", req); err != nil { - t.Fatal(err) - } - if len(inv.All()) != 2 || inv.All()[0].Count+inv.All()[1].Count != 3 { - t.Fatal("replay granted twice") - } -} - -func TestAchievementCurrencyRewardsCreditWalletAndReplay(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - wallet, _ := player.OpenWallet(store, player.Currency{}) - d := &gamedata.MissionDesign{Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{{GroupID: 987, ID: 1}: {Target: 1, AddExp: 9, Rewards: []gamedata.Reward{{Type: 3, Count: 50}, {Type: 4, Count: 100}}}}} - s, err := Open(store, d, inv) - if err != nil { - t.Fatal(err) - } - if err := s.AttachWallet(wallet); err != nil { - t.Fatal(err) - } - p := testAchievementProgress(1) - if err := s.AttachAchievementProgress(&p); err != nil { - t.Fatal(err) - } - req := wire.AppendBytes(wire.AppendVarint(nil, 1, 1), 3, wire.AppendVarint(wire.AppendVarint(nil, 1, 987), 2, 1)) - for range 2 { - if _, _, _, err := s.Handle("/AchievementClear", req); err != nil { - t.Fatal(err) - } - } - if c := wallet.Snapshot(); c.Gold != 100 || c.FreeJewelry != 50 { - t.Fatalf("wallet=%+v", c) - } - if len(inv.All()) != 0 { - t.Fatal("currency stored in items") - } -} - -func TestAchievementTierOrderUsesActualIDsAndAllowsUnorderedBatch(t *testing.T) { - for _, content := range []uint64{0, 1} { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - first, last := uint64(17), uint64(93) - if content == 1 { - first, last = 1017, 1093 - } - d := &gamedata.MissionDesign{Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{{ContentsGroup: content, GroupID: 987, ID: first}: {Target: 1, AddExp: 7}, {ContentsGroup: content, GroupID: 987, ID: last}: {Target: 2, AddExp: 9}}} - s, err := Open(store, d, inv) - if err != nil { - t.Fatal(err) - } - p := testAchievementProgress(2) - if err := s.AttachAchievementProgress(&p); err != nil { - t.Fatal(err) - } - request := func(ids ...uint64) []byte { - info := wire.AppendVarint(nil, 1, 987) - for _, id := range ids { - info = wire.AppendVarint(info, 2, id) - } - return wire.AppendBytes(wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, content), 3, info) - } - if _, _, _, err := s.Handle("/AchievementClear", request(last)); err == nil { - t.Fatal("skipped earlier tier") - } - if len(s.state.Claimed) != 0 { - t.Fatal("invalid jump mutated") - } - if _, _, _, err := s.Handle("/AchievementClear", request(last, first)); err != nil { - t.Fatal(err) - } - if exp, err := s.AchievementExperience(); err != nil || exp != 16 { - t.Fatalf("exp=%d err=%v", exp, err) - } - s.state.Claimed = append(s.state.Claimed, "achievement:0/12345/1") - if _, err := s.AchievementExperience(); err == nil { - t.Fatal("unknown historical claim omitted") - } - } -} diff --git a/go/internal/server/missions/events_test.go b/go/internal/server/missions/events_test.go deleted file mode 100644 index c693afa..0000000 --- a/go/internal/server/missions/events_test.go +++ /dev/null @@ -1,86 +0,0 @@ -package missions - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "testing" - "time" -) - -func TestEventsFollowConditionsAndLoginPersistsAcrossRestart(t *testing.T) { - daily := gamedata.MissionKey{GroupType: 0, GroupID: 42, ID: 9876} - weekly := gamedata.MissionKey{GroupType: 1, GroupID: 66, ID: 7654} - gacha := gamedata.MissionKey{GroupType: 0, GroupID: 42, ID: 8765} - locked := gamedata.MissionKey{GroupType: 0, GroupID: 42, ID: 8654} - design := &gamedata.MissionDesign{Missions: map[gamedata.MissionKey][]gamedata.Reward{daily: nil, weekly: nil, gacha: nil, locked: nil}, Conditions: map[gamedata.MissionKey]gamedata.MissionCondition{ - daily: {Type: ConditionConnect, TargetValue: 1}, weekly: {Type: ConditionConnect, TargetValue: 3}, gacha: {Type: ConditionGachaBuy, SubType: 5, SubTypeComparison: 1, TargetValue: 7}, locked: {Type: ConditionGachaBuy, UnlockPack: 21, UnlockQuest: 4, TargetValue: 1}, - }} - store := stateio.NewMemory() - inv, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - s, err := Open(store, design, inv) - if err != nil { - t.Fatal(err) - } - now := time.Now().UTC() - s.now = func() time.Time { return now } - if err := s.RecordLogin(nil); err != nil { - t.Fatal(err) - } - s, err = Open(store, design, inv) - if err != nil { - t.Fatal(err) - } - s.now = func() time.Time { return now } - if err := s.RecordLogin(nil); err != nil { - t.Fatal(err) - } - if s.state.Progress[missionName(weekly)] != 1 { - t.Fatal("reconnect incremented weekly login") - } - if err := s.RecordEvent(ConditionGachaBuy, 4, 20, nil); err != nil { - t.Fatal(err) - } - if s.state.Progress[missionName(gacha)] != 0 || s.state.Progress[missionName(locked)] != 0 { - t.Fatal("subtype/locked mission matched") - } - if err := s.RecordEvent(ConditionGachaBuy, 6, 3, nil); err != nil { - t.Fatal(err) - } - if s.state.Progress[missionName(gacha)] != 3 { - t.Fatal("event did not use count") - } - if err := s.RecordEvent(ConditionGachaBuy, 6, 100, func(pack, quest uint64) bool { return pack == 21 && quest == 4 }); err != nil { - t.Fatal(err) - } - if s.state.Progress[missionName(gacha)] != 7 || s.state.Progress[missionName(locked)] != 1 { - t.Fatal("target cap/unlock did not apply") - } - wallet, err := player.OpenWallet(store, player.Currency{}) - if err != nil { - t.Fatal(err) - } - mailbox, err := mail.OpenService(store, &mail.Starter{Version: "2.35.10", MaxMailID: 100, MailCount: 1}, inv, wallet) - if err != nil { - t.Fatal(err) - } - if err := s.AttachMail(mailbox); err != nil { - t.Fatal(err) - } - now = now.AddDate(0, 0, 1) - if err := s.RecordLogin(nil); err != nil { - t.Fatal(err) - } - // Monday resets weekly; other days increment its connection count. - want := uint64(2) - if now.Weekday() == time.Monday { - want = 1 - } - if s.state.Progress[missionName(weekly)] != want { - t.Fatalf("weekly progress=%d want=%d", s.state.Progress[missionName(weekly)], want) - } -} diff --git a/go/internal/server/missions/missions.go b/go/internal/server/missions/missions.go deleted file mode 100644 index d076f58..0000000 --- a/go/internal/server/missions/missions.go +++ /dev/null @@ -1,1181 +0,0 @@ -// Package missions owns local completion eligibility and claim state for the -// three regular (non-scheduled-event) mission endpoints. Static definitions -// come from GameData; this package never replays an HTTP capture. -package missions - -import ( - "encoding/json" - "errors" - "fmt" - "maps" - "slices" - "sort" - "strconv" - "strings" - "sync" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" -) - -const ( - missionClearPacket = 120 - missionSectionPacket = 122 - achievementClearPacket = 168 -) - -var ErrInvalidRequest = errors.New("missions: invalid request") - -// Service is deliberately given completion signals by authoritative gameplay -// code. A request cannot manufacture a mission completion merely by naming a -// table row. AchievementClear is different: its request carries the exact -// client-calculated completed achievement ids, as in the official protocol. -type AchievementProgressSource interface { - AchievementValue(groupID uint64) (uint64, error) -} - -func (s *Service) AttachAchievementProgress(source AchievementProgressSource) error { - if source == nil { - return errors.New("missions: nil achievement progress") - } - s.achievementProgress = source - return nil -} - -type Service struct { - eventHandler interface { - Handle(string, []byte) (int, []byte, bool, error) - } - levelRewards *gamedata.AchievementLevelDesign - levelReward uint64 - achievementProgress AchievementProgressSource - mu sync.Mutex - storage stateio.Store - design *gamedata.MissionDesign - inventory *player.Inventory - wallet *player.Wallet - mail *mail.Service - state snapshot - now func() time.Time -} - -func (s *Service) AttachWallet(wallet *player.Wallet) error { - if wallet == nil { - return errors.New("missions: nil wallet") - } - s.wallet = wallet - return nil -} - -type snapshot struct { - Version string `json:"version"` - DailyPeriod string `json:"daily_period"` - WeeklyPeriod string `json:"weekly_period"` - Completed []string `json:"completed"` - Claimed []string `json:"claimed"` - Progress map[string]uint64 `json:"progress"` -} - -func Open(storage stateio.Store, design *gamedata.MissionDesign, inventory *player.Inventory) (*Service, error) { - if storage == nil || design == nil || inventory == nil { - return nil, errors.New("missions: invalid service configuration") - } - now := time.Now().UTC() - s := &Service{storage: storage, design: design, inventory: inventory, now: time.Now, state: snapshot{ - Version: versionconfig.State(), DailyPeriod: dailyPeriod(now), WeeklyPeriod: weeklyPeriod(now), Progress: map[string]uint64{}, - }} - b, err := storage.Load("missions") - if err != nil { - return nil, fmt.Errorf("missions: load state: %w", err) - } - if b == nil { - return s, nil - } - if err := stateio.RequireExactJSONObject(b, "version", "daily_period", "weekly_period", "completed", "claimed", "progress"); err != nil { - return nil, fmt.Errorf("missions: incompatible state layout: %w", err) - } - if err := json.Unmarshal(b, &s.state); err != nil || s.state.Version != versionconfig.State() || s.state.DailyPeriod == "" || s.state.WeeklyPeriod == "" { - return nil, errors.New("missions: malformed state") - } - if s.state.Progress == nil { - return nil, errors.New("missions: progress must be an object") - } - return s, nil -} - -func (s *Service) AttachMail(mailbox *mail.Service) error { - if mailbox == nil { - return errors.New("missions: nil compensation mailbox") - } - s.mu.Lock() - defer s.mu.Unlock() - s.mail = mailbox - return nil -} - -func (s *Service) EnsurePersisted() error { - s.mu.Lock() - defer s.mu.Unlock() - b, err := s.storage.Load("missions") - if err != nil { - return err - } - if b != nil { - return nil - } - return s.persist(cloneSnapshot(s.state)) -} - -// CompleteMission is the only way normal mission eligibility enters this -// package. It is persistent and idempotent. Gameplay/event handlers should -// call it only after they have independently verified the table condition. -func (s *Service) CompleteMission(key gamedata.MissionKey) error { - s.mu.Lock() - defer s.mu.Unlock() - if err := s.rolloverLocked(); err != nil { - return err - } - if _, ok := s.design.Missions[key]; !ok { - return fmt.Errorf("missions: unknown mission %+v", key) - } - condition := s.design.Conditions[key] - value := condition.TargetValue - if value == 0 { - value = 1 - } - return s.setProgressLocked(key, value) -} - -func (s *Service) SetProgress(key gamedata.MissionKey, value uint64) error { - s.mu.Lock() - defer s.mu.Unlock() - if err := s.rolloverLocked(); err != nil { - return err - } - return s.setProgressLocked(key, value) -} - -func (s *Service) setProgressLocked(key gamedata.MissionKey, value uint64) error { - if _, ok := s.design.Missions[key]; !ok { - return fmt.Errorf("missions: unknown mission %+v", key) - } - name := missionName(key) - if s.state.Progress[name] >= value { - return nil - } - next := cloneSnapshot(s.state) - next.Progress[name] = value - condition := s.design.Conditions[key] - target := condition.TargetValue - if target == 0 { - target = 1 - } - if s.state.Progress[name] < target && value >= target { - s.applyCompletionDependencies(&next, key) - } - return s.commit(next) -} - -func (s *Service) applyCompletionDependencies(next *snapshot, completed gamedata.MissionKey) { - queue := []gamedata.MissionKey{completed} - seen := map[string]bool{} - for len(queue) != 0 { - current := queue[0] - queue = queue[1:] - currentName := missionName(current) - if seen[currentName] { - continue - } - seen[currentName] = true - for key, condition := range s.design.Conditions { - if key.GroupType == 2 { - continue - } - matched := condition.Type == 30 && key.GroupType == current.GroupType - matched = matched || (condition.Type == 31 && condition.SubType == current.GroupType) - if !matched { - continue - } - name := missionName(key) - target := condition.TargetValue - if target == 0 { - target = 1 - } - before := next.Progress[name] - if before >= target { - continue - } - next.Progress[name] = min(before+1, target) - if before < target && next.Progress[name] >= target { - queue = append(queue, key) - } - } - } -} - -func (s *Service) Handle(path string, request []byte) (int, []byte, bool, error) { - if s.eventHandler != nil && path == "/MissionUpdate" { - regular, event, err := splitMissionUpdates(request) - if err != nil { - return 119, nil, true, err - } - if len(event) > 0 { - if len(regular) > 0 { - s.mu.Lock() - err = s.validateRegularUpdates(regular) - s.mu.Unlock() - if err != nil { - return 119, nil, true, err - } - } - if _, _, handled, err := s.eventHandler.Handle(path, event); err != nil || !handled { - if err == nil { - err = ErrInvalidRequest - } - return 119, nil, true, err - } - if len(regular) == 0 { - return 119, nil, true, nil - } - request = regular - } - } - if s.eventHandler != nil && path == "/MissionClear" { - if code, response, handled, err := s.eventHandler.Handle(path, request); handled { - return code, response, handled, err - } - } - s.mu.Lock() - defer s.mu.Unlock() - if err := s.rolloverLocked(); err != nil { - return 0, nil, true, err - } - switch path { - case "/UserLevelReward": - response, err := s.userLevelReward(request) - return 73, response, true, err - case "/MissionInfo": - if err := requireSeq(request); err != nil { - return 118, nil, true, err - } - return 118, s.missionInfo(), true, nil - case "/MissionUpdate": - if err := s.update(request); err != nil { - return 119, nil, true, err - } - return 119, nil, true, nil - case "/AchievementInfo": - if err := requireSeq(request); err != nil { - return 166, nil, true, err - } - return 166, s.achievementInfo(), true, nil - case "/MissionClear": - response, err := s.clearMission(request) - return missionClearPacket, response, true, err - case "/MissionSectionReward": - response, err := s.clearSection(request) - return missionSectionPacket, response, true, err - case "/AchievementClear": - response, err := s.clearAchievements(request) - return achievementClearPacket, response, true, err - default: - return 0, nil, false, nil - } -} - -// Each update carries its own event identity; a client can acknowledge regular -// and scheduled missions together in one request. -func splitMissionUpdates(request []byte) (regular, event []byte, err error) { - if err = requireSeq(request); err != nil { - return - } - seq, _, _ := wire.Varint(request, 1) - err = wire.Walk(request, func(f wire.Field) error { - if f.Number != 2 { - return nil - } - if f.Type != 2 { - return ErrInvalidRequest - } - uid, e := scalar(f.Value, 4) - if e != nil { - return e - } - if uid == 0 { - if len(regular) == 0 { - regular = wire.AppendVarint(nil, 1, seq) - } - regular = wire.AppendBytes(regular, 2, f.Value) - } else { - if len(event) == 0 { - event = wire.AppendVarint(nil, 1, seq) - } - event = wire.AppendBytes(event, 2, f.Value) - } - return nil - }) - return -} - -func (s *Service) validateRegularUpdates(request []byte) error { - return wire.Walk(request, func(f wire.Field) error { - if f.Number != 2 { - return nil - } - if f.Type != 2 { - return ErrInvalidRequest - } - group, err := scalar(f.Value, 1) - if err != nil || group == 0 { - return ErrInvalidRequest - } - id, err := scalar(f.Value, 2) - if err != nil || id == 0 { - return ErrInvalidRequest - } - if _, err = scalar(f.Value, 3); err != nil { - return ErrInvalidRequest - } - matches := 0 - for key := range s.design.Missions { - if key.GroupType != 2 && key.GroupID == group && key.ID == id { - matches++ - } - } - if matches != 1 { - return ErrInvalidRequest - } - return nil - }) -} - -// AttachEventHandler delegates scheduled event operations before regular -// mission processing. The event service validates calendar and player state. -func (s *Service) AttachEventHandler(handler interface { - Handle(string, []byte) (int, []byte, bool, error) -}) error { - if handler == nil { - return errors.New("missions: nil event handler") - } - s.eventHandler = handler - return nil -} - -func dailyPeriod(now time.Time) string { return now.UTC().Format("2006-01-02") } - -func weeklyPeriod(now time.Time) string { - now = now.UTC() - daysSinceMonday := (int(now.Weekday()) + 6) % 7 - return now.AddDate(0, 0, -daysSinceMonday).Format("2006-01-02") -} - -func (s *Service) rolloverLocked() error { - if s.now == nil { - return errors.New("missions: missing clock") - } - now := s.now().UTC() - daily, weekly := dailyPeriod(now), weeklyPeriod(now) - if daily == s.state.DailyPeriod && weekly == s.state.WeeklyPeriod { - return nil - } - if s.mail == nil { - return errors.New("missions: compensation mailbox is not attached") - } - next := cloneSnapshot(s.state) - if daily != next.DailyPeriod { - if err := s.compensatePeriodLocked(0, next.DailyPeriod, now); err != nil { - return err - } - clearMissionPeriod(&next, 0) - next.DailyPeriod = daily - } - if weekly != next.WeeklyPeriod { - if err := s.compensatePeriodLocked(1, next.WeeklyPeriod, now); err != nil { - return err - } - clearMissionPeriod(&next, 1) - next.WeeklyPeriod = weekly - } - return s.commit(next) -} - -func (s *Service) compensatePeriodLocked(groupType uint64, period string, now time.Time) error { - label := "日常" - if groupType == 1 { - label = "周常" - } - for _, key := range s.completedUnclaimedForType(groupType) { - rewards := s.design.Missions[key] - if len(rewards) == 0 { - continue - } - identity := fmt.Sprintf("expired-mission:%d:%s:%s", groupType, period, missionName(key)) - title := label + "任务到期补发" - body := fmt.Sprintf("%s周期 %s 已结束。任务 %d/%d 已完成但未领取，奖励由系统自动补发。", label, period, key.GroupID, key.ID) - if err := s.mail.EnqueueCompensation(identity, title, body, rewards, now); err != nil { - return fmt.Errorf("missions: enqueue expired mission %+v: %w", key, err) - } - } - for _, key := range s.eligibleUnclaimedSectionsForType(groupType) { - rewards := s.design.Sections[key].Rewards - if len(rewards) == 0 { - continue - } - identity := fmt.Sprintf("expired-section:%d:%s:%d", groupType, period, key.ID) - title := label + "阶段奖励到期补发" - body := fmt.Sprintf("%s周期 %s 已结束。阶段奖励 %d 已达成但未领取，奖励由系统自动补发。", label, period, key.ID) - if err := s.mail.EnqueueCompensation(identity, title, body, rewards, now); err != nil { - return fmt.Errorf("missions: enqueue expired section %+v: %w", key, err) - } - } - return nil -} - -func (s *Service) completedUnclaimedForType(groupType uint64) []gamedata.MissionKey { - var result []gamedata.MissionKey - for key, condition := range s.design.Conditions { - if key.GroupType != groupType { - continue - } - target := condition.TargetValue - if target == 0 { - target = 1 - } - if s.state.Progress[missionName(key)] >= target && !contains(s.state.Claimed, "mission:"+missionName(key)) { - result = append(result, key) - } - } - sort.Slice(result, func(i, j int) bool { - if result[i].GroupID != result[j].GroupID { - return result[i].GroupID < result[j].GroupID - } - return result[i].ID < result[j].ID - }) - return result -} - -func (s *Service) eligibleUnclaimedSectionsForType(groupType uint64) []gamedata.SectionRewardKey { - completed := uint64(len(s.completedForType(groupType))) - var result []gamedata.SectionRewardKey - for key, section := range s.design.Sections { - identity := fmt.Sprintf("section:%d/%d", key.GroupType, key.ID) - if key.GroupType == groupType && section.SectionValue != 0 && completed >= section.SectionValue && !contains(s.state.Claimed, identity) { - result = append(result, key) - } - } - sort.Slice(result, func(i, j int) bool { return result[i].ID < result[j].ID }) - return result -} - -func (s *Service) completedForType(groupType uint64) []gamedata.MissionKey { - var result []gamedata.MissionKey - for key, condition := range s.design.Conditions { - if key.GroupType != groupType { - continue - } - target := condition.TargetValue - if target == 0 { - target = 1 - } - if s.state.Progress[missionName(key)] >= target { - result = append(result, key) - } - } - return result -} - -func clearMissionPeriod(state *snapshot, groupType uint64) { - prefix := strconv.FormatUint(groupType, 10) + "/" - for name := range state.Progress { - if strings.HasPrefix(name, prefix) { - delete(state.Progress, name) - } - } - state.Completed = filterPeriodIdentities(state.Completed, groupType) - state.Claimed = filterPeriodIdentities(state.Claimed, groupType) -} - -func filterPeriodIdentities(values []string, groupType uint64) []string { - missionPrefix := "mission:" + strconv.FormatUint(groupType, 10) + "/" - sectionPrefix := "section:" + strconv.FormatUint(groupType, 10) + "/" - result := values[:0] - for _, value := range values { - if !strings.HasPrefix(value, missionPrefix) && !strings.HasPrefix(value, sectionPrefix) { - result = append(result, value) - } - } - return result -} - -func (s *Service) missionInfo() []byte { - var response []byte - for _, key := range s.progressMissionKeys() { - entry := wire.AppendVarint(nil, 1, key.GroupID) - entry = wire.AppendVarint(entry, 2, key.ID) - if key.GroupType != 0 { - entry = wire.AppendVarint(entry, 3, key.GroupType) - } - entry = wire.AppendVarint(entry, 4, s.state.Progress[missionName(key)]) - if contains(s.state.Claimed, "mission:"+missionName(key)) { - entry = wire.AppendVarint(entry, 5, 1) - } - response = wire.AppendBytes(response, 1, entry) - } - for _, identity := range s.state.Claimed { - var groupType, id uint64 - if _, err := fmt.Sscanf(identity, "section:%d/%d", &groupType, &id); err != nil { - continue - } - entry := wire.AppendVarint(nil, 1, groupType) - entry = wire.AppendVarint(entry, 2, id) - response = wire.AppendBytes(response, 2, entry) - } - now := s.now().UTC() - daily := time.Date(now.Year(), now.Month(), now.Day()+1, 0, 0, 0, 0, time.UTC) - daysUntilMonday := (8 - int(now.Weekday())) % 7 - if daysUntilMonday == 0 { - daysUntilMonday = 7 - } - weekly := time.Date(now.Year(), now.Month(), now.Day()+daysUntilMonday, 0, 0, 0, 0, time.UTC) - response = wire.AppendVarint(response, 3, uint64(daily.UnixMilli())) - response = wire.AppendVarint(response, 4, uint64(weekly.UnixMilli())) - return response -} - -func (s *Service) update(request []byte) error { - if err := requireSeq(request); err != nil { - return err - } - var updates int - err := wire.Walk(request, func(field wire.Field) error { - if field.Number != 2 { - return nil - } - if field.Type != 2 { - return ErrInvalidRequest - } - groupID, err := scalar(field.Value, 1) - if err != nil || groupID == 0 { - return ErrInvalidRequest - } - id, err := scalar(field.Value, 2) - if err != nil || id == 0 { - return ErrInvalidRequest - } - value, err := scalar(field.Value, 3) - if err != nil { - return ErrInvalidRequest - } - var key gamedata.MissionKey - found := false - for candidate := range s.design.Missions { - if candidate.GroupID == groupID && candidate.ID == id && candidate.GroupType != 2 { - if found { - return ErrInvalidRequest - } - key, found = candidate, true - } - } - if !found { - return ErrInvalidRequest - } - if err := s.setProgressLocked(key, value); err != nil { - return err - } - updates++ - return nil - }) - if err != nil { - return err - } - if updates == 0 { - return ErrInvalidRequest - } - return nil -} - -func (s *Service) achievementInfo() []byte { - // AchievementInfo contains progress and last completed IDs. This local - // account currently records acknowledgement/claim state through the - // inventory idempotency key, but deliberately does not fabricate progress. - return nil -} - -func (s *Service) clearMission(request []byte) ([]byte, error) { - if err := requireSeq(request); err != nil { - return nil, err - } - all, err := boolean(request, 2) - if err != nil { - return nil, err - } - groupType, err := scalar(request, 3) - if err != nil { - return nil, err - } - groupID, err := scalar(request, 4) - if err != nil { - return nil, err - } - id, err := scalar(request, 5) - if err != nil { - return nil, err - } - if groupType == 2 { - return nil, errors.New("missions: scheduled event missions require an event service") - } - var keys []gamedata.MissionKey - if all { - if groupType != 0 || groupID != 0 || id != 0 { - return nil, fmt.Errorf("%w: bulk MissionClear has identifiers", ErrInvalidRequest) - } - keys = s.completedMissionKeys() - } else { - if groupID == 0 || id == 0 { - return nil, fmt.Errorf("%w: MissionClear identity", ErrInvalidRequest) - } - keys = []gamedata.MissionKey{{GroupType: groupType, GroupID: groupID, ID: id}} - } - items, err := s.claimMissions(keys) - if err != nil { - return nil, err - } - bundle := rewardBundle(items) - for _, key := range keys { - for _, reward := range s.design.Missions[key] { - if reward.Type == 3 || reward.Type == 4 { - entry := wire.AppendVarint(nil, 3, reward.Type) - entry = wire.AppendVarint(entry, 4, reward.Count) - bundle = wire.AppendBytes(bundle, 1, entry) - } - } - } - return bundle, nil -} - -func (s *Service) clearSection(request []byte) ([]byte, error) { - if err := requireSeq(request); err != nil { - return nil, err - } - groupType, err := scalar(request, 2) - if err != nil { - return nil, fmt.Errorf("%w: section group type", ErrInvalidRequest) - } - all, err := boolean(request, 3) - if err != nil { - return nil, err - } - id, err := scalar(request, 4) - if err != nil { - return nil, err - } - var keys []gamedata.SectionRewardKey - if all { - if id != 0 { - return nil, fmt.Errorf("%w: bulk section id", ErrInvalidRequest) - } - keys = s.eligibleSections(groupType) - } else { - if id == 0 { - return nil, fmt.Errorf("%w: section id", ErrInvalidRequest) - } - key := gamedata.SectionRewardKey{GroupType: groupType, ID: id} - if !s.sectionEligible(key) { - return nil, fmt.Errorf("%w: section is not complete", ErrInvalidRequest) - } - keys = []gamedata.SectionRewardKey{key} - } - items, err := s.claimSections(keys) - if err != nil { - return nil, err - } - var response []byte - for _, item := range items { - response = wire.AppendBytes(response, 1, player.ItemWire(item)) - } - for _, key := range keys { - for _, reward := range s.design.Sections[key].Rewards { - if reward.Type == 3 || reward.Type == 4 { - entry := wire.AppendVarint(nil, 3, reward.Type) - entry = wire.AppendVarint(entry, 4, reward.Count) - response = wire.AppendBytes(response, 1, entry) - } - } - } - return response, nil -} - -func (s *Service) clearAchievements(request []byte) ([]byte, error) { - if err := requireSeq(request); err != nil { - return nil, err - } - contents, err := scalar(request, 2) - if err != nil { - return nil, fmt.Errorf("%w: achievement contents group", ErrInvalidRequest) - } - claims, err := achievementClaims(request) - if err != nil || len(claims) == 0 { - return nil, fmt.Errorf("%w: achievement clear info", ErrInvalidRequest) - } - requested := map[gamedata.AchievementKey]bool{} - for _, claim := range claims { - for _, id := range claim.IDs { - key, _, ok := s.achievementDesign(contents, claim.GroupID, id) - if !ok { - return nil, fmt.Errorf("%w: unknown achievement", ErrInvalidRequest) - } - requested[key] = true - } - } - for key := range requested { - if contains(s.state.Claimed, "achievement:"+achievementName(key)) { - continue - } - for earlier := range s.design.Achievements { - if earlier.ContentsGroup != key.ContentsGroup || earlier.GroupID != key.GroupID || earlier.ID >= key.ID { - continue - } - if !requested[earlier] && !contains(s.state.Claimed, "achievement:"+achievementName(earlier)) { - return nil, fmt.Errorf("missions: achievement tier %v requires earlier tier %v to be claimed", key, earlier) - } - } - } - // Validate the whole batch before granting any reward. - for _, claim := range claims { - for _, id := range claim.IDs { - key, d, ok := s.achievementDesign(contents, claim.GroupID, id) - if !ok { - return nil, fmt.Errorf("%w: unknown achievement", ErrInvalidRequest) - } - if contains(s.state.Claimed, "achievement:"+achievementName(key)) { - continue - } - if d.Target > 0 { - if s.achievementProgress == nil { - return nil, errors.New("missions: achievement progress unavailable") - } - group := d.CounterGroup - if group == 0 { - group = key.GroupID - } - value, err := s.achievementProgress.AchievementValue(group) - if err != nil { - return nil, err - } - if float64(value) < d.Target { - return nil, fmt.Errorf("missions: achievement %v requires %g progress, got %d", key, d.Target, value) - } - } - } - } - var allItems []player.Item - var addExp uint64 - var currencyRewards []gamedata.Reward - next := cloneSnapshot(s.state) - for _, claim := range claims { - for _, id := range claim.IDs { - key, design, ok := s.achievementDesign(contents, claim.GroupID, id) - if !ok { - return nil, fmt.Errorf("%w: unknown achievement %+v", ErrInvalidRequest, key) - } - identity := "achievement:" + achievementName(key) - if contains(next.Claimed, identity) { - continue - } - items, err := s.grantRewards(identity, design.Rewards) - if err != nil { - return nil, err - } - if ^uint64(0)-addExp < design.AddExp { - return nil, errors.New("missions: achievement exp overflow") - } - addExp += design.AddExp - for _, reward := range design.Rewards { - if reward.Type == 2 || reward.Type == 3 || reward.Type == 4 || reward.Type == 12 || reward.Type == 20 { - currencyRewards = append(currencyRewards, reward) - } - } - allItems = append(allItems, items...) - next.Claimed = append(next.Claimed, identity) - } - } - if addExp > uint64(^uint(0)>>1) { - return nil, errors.New("missions: achievement exp overflow") - } - if err := s.commit(next); err != nil { - return nil, err - } - response := wire.AppendVarint(nil, 1, addExp) - bundle := rewardBundle(allItems) - for _, reward := range currencyRewards { - item := wire.AppendVarint(nil, 3, reward.Type) - item = wire.AppendVarint(item, 4, reward.Count) - bundle = wire.AppendBytes(bundle, 1, item) - } - return wire.AppendBytes(response, 2, bundle), nil -} - -// The official all-clear request omits ContentsGroup (protobuf value zero). -// In that form GroupID+ID must identify exactly one static row; ambiguity is -// rejected rather than resolved by map iteration order. -func (s *Service) achievementDesign(contents, groupID, id uint64) (gamedata.AchievementKey, gamedata.AchievementDesign, bool) { - if contents != 0 { - key := gamedata.AchievementKey{ContentsGroup: contents, GroupID: groupID, ID: id} - design, ok := s.design.Achievements[key] - return key, design, ok - } - var foundKey gamedata.AchievementKey - var found gamedata.AchievementDesign - matched := false - for key, design := range s.design.Achievements { - if key.GroupID != groupID || key.ID != id { - continue - } - if matched { - return gamedata.AchievementKey{}, gamedata.AchievementDesign{}, false - } - foundKey, found, matched = key, design, true - } - return foundKey, found, matched -} - -func (s *Service) progressMissionKeys() []gamedata.MissionKey { - result := make([]gamedata.MissionKey, 0, len(s.state.Progress)) - for name, value := range s.state.Progress { - if value == 0 { - continue - } - for key := range s.design.Missions { - if missionName(key) == name { - result = append(result, key) - break - } - } - } - sort.Slice(result, func(i, j int) bool { - a, b := result[i], result[j] - if a.GroupType != b.GroupType { - return a.GroupType < b.GroupType - } - if a.GroupID != b.GroupID { - return a.GroupID < b.GroupID - } - return a.ID < b.ID - }) - return result -} - -func (s *Service) completedMissionKeys() []gamedata.MissionKey { - var result []gamedata.MissionKey - for key, condition := range s.design.Conditions { - target := condition.TargetValue - if target == 0 { - target = 1 - } - name := missionName(key) - if s.state.Progress[name] >= target && !contains(s.state.Claimed, "mission:"+name) { - result = append(result, key) - } - } - sort.Slice(result, func(i, j int) bool { - a, b := result[i], result[j] - if a.GroupType != b.GroupType { - return a.GroupType < b.GroupType - } - if a.GroupID != b.GroupID { - return a.GroupID < b.GroupID - } - return a.ID < b.ID - }) - return result -} - -func (s *Service) claimMissions(keys []gamedata.MissionKey) ([]player.Item, error) { - next := cloneSnapshot(s.state) - var result []player.Item - for _, key := range keys { - condition := s.design.Conditions[key] - target := condition.TargetValue - if target == 0 { - target = 1 - } - if s.state.Progress[missionName(key)] < target { - return nil, fmt.Errorf("%w: mission is not complete", ErrInvalidRequest) - } - rewards, ok := s.design.Missions[key] - if !ok { - return nil, fmt.Errorf("%w: unknown mission %+v", ErrInvalidRequest, key) - } - identity := "mission:" + missionName(key) - items, err := s.grantRewards(identity, rewards) - if err != nil { - return nil, err - } - result = append(result, items...) - if !contains(next.Claimed, identity) { - next.Claimed = append(next.Claimed, identity) - } - } - if err := s.commit(next); err != nil { - return nil, err - } - return result, nil -} - -func (s *Service) eligibleSections(groupType uint64) []gamedata.SectionRewardKey { - var result []gamedata.SectionRewardKey - for key := range s.design.Sections { - // Proto3 omits group_type for the official bulk request. Zero means - // all mission groups there, not an invalid group. - if (groupType == 0 || key.GroupType == groupType) && s.sectionEligible(key) { - result = append(result, key) - } - } - sort.Slice(result, func(i, j int) bool { return result[i].ID < result[j].ID }) - return result -} - -// Section reward thresholds are authoritative static values. The completed -// set, rather than the claimed set, is deliberately counted as the client does. -func (s *Service) sectionEligible(key gamedata.SectionRewardKey) bool { - section, ok := s.design.Sections[key] - if !ok { - return false - } - if section.SectionValue == 0 { - return false - } - var count uint64 - for _, identity := range s.state.Claimed { - var group, missionGroup, missionID uint64 - if _, err := fmt.Sscanf(identity, "mission:%d/%d/%d", &group, &missionGroup, &missionID); err == nil && group == key.GroupType { - count++ - } - } - return count >= section.SectionValue -} - -func (s *Service) claimSections(keys []gamedata.SectionRewardKey) ([]player.Item, error) { - next := cloneSnapshot(s.state) - var result []player.Item - for _, key := range keys { - section, ok := s.design.Sections[key] - if !ok || !s.sectionEligible(key) { - return nil, fmt.Errorf("%w: section is not complete", ErrInvalidRequest) - } - identity := fmt.Sprintf("section:%d/%d", key.GroupType, key.ID) - items, err := s.grantRewards(identity, section.Rewards) - if err != nil { - return nil, err - } - result = append(result, items...) - if !contains(next.Claimed, identity) { - next.Claimed = append(next.Claimed, identity) - } - } - if err := s.commit(next); err != nil { - return nil, err - } - return result, nil -} - -func (s *Service) grantRewards(identity string, rewards []gamedata.Reward) ([]player.Item, error) { - if s.wallet != nil { - if _, err := s.wallet.GrantQuestOnce(identity+":currency", rewards); err != nil { - return nil, err - } - } - var stack []gamedata.BattleReward - for _, reward := range rewards { - if reward.Type == 2 || reward.Type == 3 || reward.Type == 4 || reward.Type == 12 || reward.Type == 20 { - if s.wallet == nil { - return nil, errors.New("missions: currency reward wallet unavailable") - } - continue - } - if reward.ID == 0 || reward.Count == 0 { - // Type 12 is an account resource whose full/overflow conversion is - // server-dynamic. It is intentionally not fabricated as ItemDBInfo. - continue - } - stack = append(stack, gamedata.BattleReward{Type: reward.Type, ID: reward.ID, Count: reward.Count}) //nolint:staticcheck // S1016 - } - if len(stack) == 0 { - return nil, nil - } - items, err := s.inventory.GrantOnce(identity+":items", stack) - if err != nil { - return nil, err - } - if len(items) == 0 { - items = s.inventory.GrantedItems(identity + ":items") - } - return items, nil -} - -func rewardBundle(items []player.Item) []byte { - var bundle []byte - for _, item := range items { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(item)) - view := wire.AppendVarint(nil, 2, item.ID) - view = wire.AppendVarint(view, 3, item.Type) - view = wire.AppendVarint(view, 4, item.Count) - bundle = wire.AppendBytes(bundle, 6, view) - } - return bundle -} - -func (s *Service) commit(next snapshot) error { - sort.Strings(next.Completed) - sort.Strings(next.Claimed) - if next.DailyPeriod == s.state.DailyPeriod && next.WeeklyPeriod == s.state.WeeklyPeriod && equalStrings(next.Completed, s.state.Completed) && equalStrings(next.Claimed, s.state.Claimed) && equalProgress(next.Progress, s.state.Progress) { - return nil - } - return s.persist(next) -} - -func (s *Service) persist(next snapshot) error { - b, err := json.Marshal(next) - if err != nil { - return err - } - if err := s.storage.Save("missions", b); err != nil { - return err - } - s.state = next - return nil -} -func equalProgress(a, b map[string]uint64) bool { - if len(a) != len(b) { - return false - } - for key, value := range a { - if b[key] != value { - return false - } - } - return true -} - -func cloneSnapshot(in snapshot) snapshot { - progress := make(map[string]uint64, len(in.Progress)) - maps.Copy(progress, in.Progress) - return snapshot{Version: in.Version, DailyPeriod: in.DailyPeriod, WeeklyPeriod: in.WeeklyPeriod, Completed: append([]string(nil), in.Completed...), Claimed: append([]string(nil), in.Claimed...), Progress: progress} -} -func contains(values []string, want string) bool { - return slices.Contains(values, want) -} -func equalStrings(a, b []string) bool { - if len(a) != len(b) { - return false - } - for i := range a { - if a[i] != b[i] { - return false - } - } - return true -} -func missionName(k gamedata.MissionKey) string { - return fmt.Sprintf("%d/%d/%d", k.GroupType, k.GroupID, k.ID) -} -func achievementName(k gamedata.AchievementKey) string { - return fmt.Sprintf("%d/%d/%d", k.ContentsGroup, k.GroupID, k.ID) -} - -func requireSeq(b []byte) error { - seq, err := scalar(b, 1) - if err != nil || seq == 0 { - return ErrInvalidRequest - } - return nil -} -func scalar(b []byte, number int) (uint64, error) { - value, found, err := wire.Varint(b, number) - if err != nil { - return 0, ErrInvalidRequest - } - if !found { - return 0, nil - } - return value, nil -} -func boolean(b []byte, number int) (bool, error) { - value, err := scalar(b, number) - if err != nil || value > 1 { - return false, ErrInvalidRequest - } - return value == 1, nil -} - -type achievementClaim struct { - GroupID uint64 - IDs []uint64 -} - -func achievementClaims(b []byte) ([]achievementClaim, error) { - var claims []achievementClaim - err := wire.Walk(b, func(field wire.Field) error { - if field.Number != 3 { - return nil - } - if field.Type != 2 { - return ErrInvalidRequest - } - group, err := scalar(field.Value, 1) - if err != nil || group == 0 { - return ErrInvalidRequest - } - ids, err := packed(field.Value, 2) - if err != nil || len(ids) == 0 { - return ErrInvalidRequest - } - claims = append(claims, achievementClaim{group, ids}) - return nil - }) - return claims, err -} -func packed(b []byte, number int) ([]uint64, error) { - var values []uint64 - err := wire.Walk(b, func(field wire.Field) error { - if field.Number != number { - return nil - } - if field.Type == 0 { - value, _ := decode(field.Value) - values = append(values, value) - return nil - } - if field.Type != 2 { - return ErrInvalidRequest - } - for remaining := field.Value; len(remaining) > 0; { - value, n := decode(remaining) - if n == 0 { - return ErrInvalidRequest - } - values = append(values, value) - remaining = remaining[n:] - } - return nil - }) - return values, err -} -func decode(b []byte) (uint64, int) { - var value uint64 - for i, x := range b { - value |= uint64(x&127) << (7 * i) - if x < 128 { - return value, i + 1 - } - if i == 9 { - return 0, 0 - } - } - return 0, 0 -} diff --git a/go/internal/server/missions/missions_test.go b/go/internal/server/missions/missions_test.go deleted file mode 100644 index c12208d..0000000 --- a/go/internal/server/missions/missions_test.go +++ /dev/null @@ -1,213 +0,0 @@ -package missions - -import ( - "testing" - "time" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/mail" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func TestAchievementClaimsAcceptPackedAndUnpackedIDs(t *testing.T) { - info := wire.AppendVarint(nil, 1, 7) - packed := []byte{1, 2, 3} - info = wire.AppendBytes(info, 2, packed) - request := wire.AppendVarint(nil, 1, 1) - request = wire.AppendVarint(request, 2, 1) - request = wire.AppendBytes(request, 3, info) - claims, err := achievementClaims(request) - if err != nil { - t.Fatal(err) - } - if len(claims) != 1 || claims[0].GroupID != 7 || len(claims[0].IDs) != 3 || claims[0].IDs[2] != 3 { - t.Fatalf("claims=%+v", claims) - } -} - -func TestDailyRolloverMailsCompletedMissionAndSectionRewards(t *testing.T) { - mission := gamedata.MissionKey{GroupType: 0, GroupID: 1, ID: 101} - section := gamedata.SectionRewardKey{GroupType: 0, ID: 10} - design := &gamedata.MissionDesign{ - Missions: map[gamedata.MissionKey][]gamedata.Reward{mission: {{Type: 4, Count: 100}}}, - Conditions: map[gamedata.MissionKey]gamedata.MissionCondition{mission: {TargetValue: 1}}, - Sections: map[gamedata.SectionRewardKey]gamedata.SectionRewardDesign{section: { - SectionValue: 1, Rewards: []gamedata.Reward{{Type: 3, Count: 20}}, - }}, - Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{}, - } - storage := stateio.NewMemory() - starter := &player.Starter{Version: "2.35.10"} - inv, err := player.OpenInventory(storage, starter) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - mailbox, err := mail.OpenService(storage, &mail.Starter{Version: "2.35.10", MailCount: 1, MaxMailID: 100}, inv, wallet) - if err != nil { - t.Fatal(err) - } - service, err := Open(storage, design, inv) - if err != nil { - t.Fatal(err) - } - if err := service.AttachWallet(wallet); err != nil { - t.Fatal(err) - } - if err := service.AttachMail(mailbox); err != nil { - t.Fatal(err) - } - service.state.DailyPeriod = "2026-09-23" - service.state.WeeklyPeriod = "2026-09-21" - service.now = func() time.Time { return time.Date(2026, 9, 24, 0, 1, 0, 0, time.UTC) } - service.state.Progress[missionName(mission)] = 1 - if _, _, _, err := service.Handle("/MissionInfo", wire.AppendVarint(nil, 1, 1)); err != nil { - t.Fatal(err) - } - _, mailInfo, _, err := mailbox.Handle("/MailInfo", wire.AppendVarint(nil, 1, 2)) - if err != nil { - t.Fatal(err) - } - mailCount := 0 - if err := wire.Walk(mailInfo, func(field wire.Field) error { - if field.Number == 1 { - mailCount++ - } - return nil - }); err != nil { - t.Fatal(err) - } - if service.state.DailyPeriod != "2026-09-24" || len(service.state.Progress) != 0 || mailCount != 2 { - t.Fatalf("mission=%+v mailCount=%d", service.state, mailCount) - } - for id := uint64(101); id <= 102; id++ { - request := wire.AppendVarint(nil, 1, id) - request = wire.AppendVarint(request, 2, id) - if _, _, _, err := mailbox.Handle("/MailOpen", request); err != nil { - t.Fatal(err) - } - } - if got := wallet.Snapshot(); got.Gold != 100 || got.FreeJewelry != 20 { - t.Fatalf("wallet=%+v", got) - } -} - -func TestRequireSeqRejectsMissingSequence(t *testing.T) { - if err := requireSeq(nil); err == nil { - t.Fatal("expected missing sequence error") - } -} - -func TestOfficialBulkAchievementRequestMayOmitContentsGroup(t *testing.T) { - design := &gamedata.MissionDesign{ - Missions: map[gamedata.MissionKey][]gamedata.Reward{}, Sections: map[gamedata.SectionRewardKey]gamedata.SectionRewardDesign{}, - Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{ - {ContentsGroup: 7, GroupID: 1, ID: 1}: {AddExp: 3}, - }, - } - starter := &player.Starter{Version: "2.35.10"} - inventory, err := player.OpenInventory(stateio.NewMemory(), starter) - if err != nil { - t.Fatal(err) - } - service, err := Open(stateio.NewMemory(), design, inventory) - if err != nil { - t.Fatal(err) - } - info := wire.AppendVarint(nil, 1, 1) - info = wire.AppendVarint(info, 2, 1) - request := wire.AppendVarint(nil, 1, 242) - request = wire.AppendBytes(request, 3, info) - code, response, handled, err := service.Handle("/AchievementClear", request) - if err != nil { - t.Fatal(err) - } - if !handled || code != 168 { - t.Fatalf("handled=%v code=%d", handled, code) - } - exp, found, err := wire.Varint(response, 1) - if err != nil || !found || exp != 3 { - t.Fatalf("exp=%d found=%v err=%v", exp, found, err) - } -} - -func TestOfficialBulkSectionRequestAcceptsDefaultGroupType(t *testing.T) { - design := &gamedata.MissionDesign{ - Missions: map[gamedata.MissionKey][]gamedata.Reward{}, Sections: map[gamedata.SectionRewardKey]gamedata.SectionRewardDesign{}, - Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{}, - } - starter := &player.Starter{Version: "2.35.10"} - inventory, err := player.OpenInventory(stateio.NewMemory(), starter) - if err != nil { - t.Fatal(err) - } - service, err := Open(stateio.NewMemory(), design, inventory) - if err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(nil, 1, 233) - request = wire.AppendVarint(request, 3, 1) - code, _, handled, err := service.Handle("/MissionSectionReward", request) - if err != nil { - t.Fatal(err) - } - if !handled || code != 122 { - t.Fatalf("handled=%v code=%d", handled, code) - } -} - -func TestMissionProgressIsNotClaimedAndDailyTypeZeroIsValid(t *testing.T) { - key := gamedata.MissionKey{GroupType: 0, GroupID: 1, ID: 101} - design := &gamedata.MissionDesign{ - Missions: map[gamedata.MissionKey][]gamedata.Reward{key: {{Type: 4, Count: 1000}}}, - Conditions: map[gamedata.MissionKey]gamedata.MissionCondition{key: {TargetValue: 1}}, - Sections: map[gamedata.SectionRewardKey]gamedata.SectionRewardDesign{}, Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{}, - } - starter := &player.Starter{Version: "2.35.10"} - storage := stateio.NewMemory() - inv, err := player.OpenInventory(storage, starter) - if err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(storage, player.Currency{}) - if err != nil { - t.Fatal(err) - } - service, err := Open(storage, design, inv) - if err != nil { - t.Fatal(err) - } - if err := service.AttachWallet(wallet); err != nil { - t.Fatal(err) - } - if err := service.CompleteMission(key); err != nil { - t.Fatal(err) - } - _, response, _, err := service.Handle("/MissionInfo", wire.AppendVarint(nil, 1, 1)) - if err != nil { - t.Fatal(err) - } - entry, found, _ := wire.Bytes(response, 1) - if !found { - t.Fatal("mission missing") - } - value, _, _ := wire.Varint(entry, 4) - claimed, claimedFound, _ := wire.Varint(entry, 5) - if value != 1 || claimedFound || claimed != 0 { - t.Fatalf("value=%d claimed=%d/%v", value, claimed, claimedFound) - } - clear := wire.AppendVarint(nil, 1, 2) - clear = wire.AppendVarint(clear, 4, 1) - clear = wire.AppendVarint(clear, 5, 101) - if _, _, _, err := service.Handle("/MissionClear", clear); err != nil { - t.Fatal(err) - } - if got := wallet.Snapshot().Gold; got != 1000 { - t.Fatalf("gold=%d", got) - } -} diff --git a/go/internal/server/missions/mixed_update_test.go b/go/internal/server/missions/mixed_update_test.go deleted file mode 100644 index 7e8c492..0000000 --- a/go/internal/server/missions/mixed_update_test.go +++ /dev/null @@ -1,36 +0,0 @@ -package missions - -import ( - "bd2server/internal/server/wire" - "testing" -) - -func TestSplitMixedMissionUpdates(t *testing.T) { - regular := wire.AppendVarint(nil, 1, 1) - regular = wire.AppendVarint(regular, 2, 101) - regular = wire.AppendVarint(regular, 3, 2) - event := wire.AppendVarint(nil, 1, 9) - event = wire.AppendVarint(event, 2, 10) - event = wire.AppendVarint(event, 3, 1) - event = wire.AppendVarint(event, 4, 77) - req := wire.AppendVarint(nil, 1, 40) - req = wire.AppendBytes(req, 2, regular) - req = wire.AppendBytes(req, 2, event) - r, e, err := splitMissionUpdates(req) - if err != nil { - t.Fatal(err) - } - for _, part := range [][]byte{r, e} { - seq, _, _ := wire.Varint(part, 1) - if seq != 40 { - t.Fatal("lost sequence") - } - } - rr, _, _ := wire.Bytes(r, 2) - ee, _, _ := wire.Bytes(e, 2) - rid, _, _ := wire.Varint(rr, 2) - eid, _, _ := wire.Varint(ee, 4) - if rid != 101 || eid != 77 { - t.Fatal("mixed update was not split by event identity") - } -} diff --git a/go/internal/server/missions/user_level_reward.go b/go/internal/server/missions/user_level_reward.go deleted file mode 100644 index a137f56..0000000 --- a/go/internal/server/missions/user_level_reward.go +++ /dev/null @@ -1,148 +0,0 @@ -package missions - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" - "encoding/json" - "fmt" -) - -type levelRewardSnapshot struct { - Version string `json:"version"` - LevelReward uint64 `json:"level_reward"` -} - -// AttachUserLevelRewards reads the dedicated claim bucket in the same store as missions. -func (s *Service) AttachUserLevelRewards(d *gamedata.AchievementLevelDesign) error { - if err := d.Validate(); err != nil { - return err - } - s.mu.Lock() - defer s.mu.Unlock() - entries, ok := s.storage.(stateio.EntryStore) - if !ok { - return fmt.Errorf("missions: user level rewards require entry storage") - } - raw, _, err := entries.LoadEntry("missions", "user_level_rewards", "state") - if err != nil { - return err - } - var state levelRewardSnapshot - if raw != nil { - if err := stateio.RequireExactJSONObject(raw, "version", "level_reward"); err != nil { - return err - } - if err := json.Unmarshal(raw, &state); err != nil { - return err - } - if state.Version != versionconfig.State() { - return fmt.Errorf("missions: incompatible user level reward version") - } - if state.LevelReward != 0 { - found := false - for _, l := range d.Levels { - found = found || l.ID == state.LevelReward - } - if !found { - return fmt.Errorf("missions: claimed user level missing from GameData") - } - } - } - s.levelRewards = d - s.levelReward = state.LevelReward - return nil -} -func (s *Service) LevelRewardCount() (uint64, error) { - s.mu.Lock() - defer s.mu.Unlock() - if s.levelRewards == nil { - return 0, fmt.Errorf("missions: user level rewards unavailable") - } - return s.levelReward, nil -} - -func (s *Service) userLevelReward(request []byte) ([]byte, error) { - if s.levelRewards == nil { - return nil, fmt.Errorf("missions: user level rewards unavailable") - } - if err := requireSeq(request); err != nil { - return nil, err - } - ids, err := packed(request, 2) - if err != nil || len(ids) == 0 { - return nil, ErrInvalidRequest - } - exp, err := s.achievementExperienceLocked() - if err != nil { - return nil, err - } - level := s.levelRewards.Level(exp) - next := s.levelReward - var fresh []gamedata.AchievementLevel - var previous uint64 - for _, id := range ids { - if id == 0 || id <= previous { - return nil, ErrInvalidRequest - } - previous = id - index := -1 - for i, l := range s.levelRewards.Levels { - if l.ID == id { - index = i - break - } - } - if index < 0 || id > level { - return nil, ErrInvalidRequest - } - if id <= s.levelReward { - continue - } - expected := uint64(0) - for _, l := range s.levelRewards.Levels { - if l.ID > next { - expected = l.ID - break - } - } - if id != expected { - return nil, fmt.Errorf("%w: user level reward requires preceding claims", ErrInvalidRequest) - } - fresh = append(fresh, s.levelRewards.Levels[index]) - next = id - } - var items []player.Item - var currencies []gamedata.Reward - for _, l := range fresh { - granted, err := s.grantRewards(fmt.Sprintf("user-level:%d", l.ID), l.Rewards) - if err != nil { - return nil, err - } - items = append(items, granted...) - for _, r := range l.Rewards { - if r.Type == 2 || r.Type == 3 || r.Type == 4 || r.Type == 12 || r.Type == 20 { - currencies = append(currencies, r) - } - } - } - if next != s.levelReward { - raw, err := json.Marshal(levelRewardSnapshot{versionconfig.State(), next}) - if err != nil { - return nil, err - } - if err := s.storage.(stateio.EntryStore).PutEntry("missions", "user_level_rewards", "state", raw); err != nil { - return nil, err - } - s.levelReward = next - } - bundle := rewardBundle(items) - for _, r := range currencies { - item := wire.AppendVarint(nil, 3, r.Type) - item = wire.AppendVarint(item, 4, r.Count) - bundle = wire.AppendBytes(bundle, 1, item) - } - return wire.AppendBytes(nil, 1, bundle), nil -} diff --git a/go/internal/server/missions/user_level_reward_test.go b/go/internal/server/missions/user_level_reward_test.go deleted file mode 100644 index 71ebcd6..0000000 --- a/go/internal/server/missions/user_level_reward_test.go +++ /dev/null @@ -1,62 +0,0 @@ -package missions - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "testing" -) - -func TestUserLevelRewardEligibilityBatchPersistenceAndRetry(t *testing.T) { - store := stateio.NewMemory() - inv, _ := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - d := &gamedata.MissionDesign{Achievements: map[gamedata.AchievementKey]gamedata.AchievementDesign{{GroupID: 9, ID: 1}: {AddExp: 10}}} - s, _ := Open(store, d, inv) - levels := &gamedata.AchievementLevelDesign{Levels: []gamedata.AchievementLevel{{ID: 17, NeedEXP: 4, Rewards: []gamedata.Reward{{Type: 8, ID: 987, Count: 3}}}, {ID: 23, NeedEXP: 8, Rewards: []gamedata.Reward{{Type: 8, ID: 988, Count: 7}}}, {ID: 41, NeedEXP: 10, Rewards: []gamedata.Reward{{Type: 8, ID: 989, Count: 9}}}}} - if err := s.AttachUserLevelRewards(levels); err != nil { - t.Fatal(err) - } - req := func(ids ...uint64) []byte { - r := wire.AppendVarint(nil, 1, 1) - for _, id := range ids { - r = wire.AppendVarint(r, 2, id) - } - return r - } - for _, ids := range [][]uint64{{23}, {17, 23}, {17, 17}, {99}} { - if _, _, _, err := s.Handle("/UserLevelReward", req(ids...)); err == nil { - t.Fatalf("accepted %v", ids) - } - } - if len(inv.All()) != 0 { - t.Fatal("invalid batch granted rewards") - } - s.state.Claimed = []string{"achievement:" + achievementName(gamedata.AchievementKey{GroupID: 9, ID: 1})} - if _, _, _, err := s.Handle("/UserLevelReward", req(17, 23, 41)); err == nil { - t.Fatal("accepted unearned level") - } - if _, _, _, err := s.Handle("/UserLevelReward", req(17, 23)); err != nil { - t.Fatal(err) - } - if count, _ := s.LevelRewardCount(); count != 23 { - t.Fatalf("count=%d", count) - } - if len(inv.All()) != 2 || inv.All()[0].Count+inv.All()[1].Count != 10 { - t.Fatalf("items=%+v", inv.All()) - } - reopened, _ := Open(store, d, inv) - if err := reopened.AttachUserLevelRewards(levels); err != nil { - t.Fatal(err) - } - reopened.state.Claimed = s.state.Claimed - if _, _, _, err := reopened.Handle("/UserLevelReward", req(17, 23)); err != nil { - t.Fatal(err) - } - if len(inv.All()) != 2 { - t.Fatal("retry granted twice") - } - if n, _ := reopened.LevelRewardCount(); n != 23 { - t.Fatalf("reopened=%d", n) - } -} diff --git a/go/internal/server/monsterhunt/calendar_test.go b/go/internal/server/monsterhunt/calendar_test.go deleted file mode 100644 index 300c6cc..0000000 --- a/go/internal/server/monsterhunt/calendar_test.go +++ /dev/null @@ -1,87 +0,0 @@ -package monsterhunt - -import ( - "testing" - "time" - - "bd2server/internal/server/readonly" - "bd2server/internal/server/wire" -) - -func TestFutureSeasonDoesNotReplaceActiveOrRewardSeason(t *testing.T) { - rows := []season{ - {ID: 5, Start: 100, End: 200}, - {ID: 3, Start: 300, End: 400}, - {ID: 999, Start: 500, End: 600}, - } - for _, tc := range []struct{ now, want uint64 }{ - {50, 5}, {100, 5}, {199, 5}, {200, 5}, {299, 5}, - {300, 3}, {400, 3}, {500, 999}, - } { - if got := selectSeason(rows, tc.now); got.ID != tc.want { - t.Fatalf("time=%d season=%d want=%d", tc.now, got.ID, tc.want) - } - } -} - -func TestSchedulePreservesAllCalendarsAndPlacesSelectedCategoriesLast(t *testing.T) { - field := func(n int, v uint64) readonly.Field { return readonly.Field{Number: n, Type: 0, Varint: v} } - row := func(id uint64, independent bool) readonly.Field { - flag := uint64(0) - if independent { - flag = 1 - } - return readonly.Field{Number: 1, Type: 2, Fields: []readonly.Field{ - {Number: 1, Type: 2, Fields: []readonly.Field{field(1, id)}}, field(6, flag), - }} - } - s := &Service{now: func() time.Time { return time.UnixMilli(150) }, - seasons: []season{{ID: 1, Start: 100, End: 200}, {ID: 2, Start: 300, End: 400}, {ID: 3, Start: 100, End: 200, Independent: true}, {ID: 4, Start: 300, End: 400, Independent: true}}, - seed: &readonly.Seed{Responses: map[string]readonly.Response{"/MonsterHuntScheduleInfo": { - Fields: []readonly.Field{row(1, false), row(2, false), row(3, true), row(4, true), field(2, 5), {Number: 3, Type: 2, Fields: []readonly.Field{field(1, 77)}}}, - }}}, - } - for _, now := range []int64{50, 150, 250, 350, 450} { - s.now = func() time.Time { return time.UnixMilli(now) } - _, response, _, err := s.scheduleInfo(wire.AppendVarint(nil, 1, 1)) - if err != nil { - t.Fatal(err) - } - var ids []uint64 - var lastRegular, lastIndependent uint64 - history := 0 - if err := wire.Walk(response, func(f wire.Field) error { - if f.Number == 1 { - nested, _, _ := wire.Bytes(f.Value, 1) - id, _, _ := wire.Varint(nested, 1) - ids = append(ids, id) - flag, _, _ := wire.Varint(f.Value, 6) - if flag == 0 { - lastRegular = id - } else { - lastIndependent = id - } - } - if f.Number == 3 { - history++ - } - return nil - }); err != nil { - t.Fatal(err) - } - want := uint64(1) - if now >= 300 { - want = 2 - } - seen := map[uint64]bool{} - for _, id := range ids { - seen[id] = true - } - if len(ids) != 4 || len(seen) != 4 || !seen[1] || !seen[2] || !seen[3] || !seen[4] || lastRegular != want || lastIndependent != want+2 || history != 1 { - t.Fatalf("time=%d rows=%v history=%d", now, ids, history) - } - if len(s.seed.Responses["/MonsterHuntScheduleInfo"].Fields) != 6 { - t.Fatal("projection mutated published calendars") - } - } -} diff --git a/go/internal/server/monsterhunt/monsterhunt_test.go b/go/internal/server/monsterhunt/monsterhunt_test.go deleted file mode 100644 index 84d99d1..0000000 --- a/go/internal/server/monsterhunt/monsterhunt_test.go +++ /dev/null @@ -1,196 +0,0 @@ -package monsterhunt - -import ( - "bd2server/internal/server/calendar" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/readonly" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "os" - "path/filepath" - "strconv" - "testing" - "time" -) - -func installed(t *testing.T) (*Service, *stateio.Memory) { - t.Helper() - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("BD2_REAL_GAMEDATA not configured") - } - seed, e := readonly.Load(filepath.Join("..", "..", "..", "seed", "v2_35_10", "readonly.json")) - if e != nil { - t.Fatal(e) - } - calendars, e := calendar.LoadDirectory(filepath.Join("..", "..", "..", "..", "schedules"), "2.35.10", "20260923193640") - if e != nil { - t.Fatal(e) - } - seed, e = calendars.ApplyReadonly(seed) - if e != nil { - t.Fatal(e) - } - store := stateio.NewMemory() - inv, e := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if e != nil { - t.Fatal(e) - } - wallet, e := player.OpenWallet(store, player.Currency{Gold: 100000}) - if e != nil { - t.Fatal(e) - } - s, e := Open(store, root, "20260923193640", seed, inv, wallet) - if e != nil { - t.Fatal(e) - } - c := s.current() - s.now = func() time.Time { return time.UnixMilli(int64(c.Start + 1000)) } - return s, store -} -func battleRequest(id, deck, mode uint64) []byte { - b := wire.AppendVarint(nil, 1, 1) - b = wire.AppendVarint(b, 4, deck) - b = wire.AppendVarint(b, 5, mode) - b = wire.AppendVarint(b, 6, id) - return wire.AppendVarint(b, 10, 1) -} -func TestBattleSessionProgressRewardsAndPersistentRetry(t *testing.T) { - s, store := installed(t) - c := s.current() - d, e := s.load(c.Hunt) - if e != nil { - t.Fatal(e) - } - req := battleRequest(c.Hunt, d.DeckID, BattleMode) - if _, e = s.EnterBattle(req, "session:1"); e != nil { - t.Fatal(e) - } - s.BeginSession("session") - end := wire.AppendVarint(nil, 1, 2) - end = wire.AppendVarint(end, 2, 1) - out, e := s.CompleteBattle(end, "session:1") - if e != nil { - t.Fatal(e) - } - if _, ok, _ := wire.Bytes(out, 7); !ok { - t.Fatal("clear reward missing") - } - u := s.state.Users[strconv.FormatUint(c.ID, 10)] - if u.ClearLevel != 1 || u.Level != 2 || !u.Played { - t.Fatalf("progress=%+v", u) - } - retry, e := s.CompleteBattle(end, "session:1") - if e != nil || !bytes.Equal(out, retry) { - t.Fatalf("retry=%x err=%v", retry, e) - } - bad := wire.AppendVarint(end, 7, 1) - if _, e = s.CompleteBattle(bad, "session:1"); e == nil { - t.Fatal("changed retry accepted") - } - restored, e := Open(store, s.root, s.version, s.seed, s.inventory, s.wallet) - if e != nil { - t.Fatal(e) - } - retry, e = restored.CompleteBattle(end, "session:1") - if e != nil || !bytes.Equal(out, retry) { - t.Fatalf("restored retry failed %v", e) - } -} -func TestPracticeHasNoRewardsAndRejectsInvalidHP(t *testing.T) { - s, _ := installed(t) - c := s.current() - d, e := s.load(c.Hunt) - if e != nil { - t.Fatal(e) - } - if _, e = s.EnterBattle(battleRequest(c.Hunt, d.DeckID, PracticeMode), "p:1"); e != nil { - t.Fatal(e) - } - hp, _ := d.HP(1) - bad := wire.AppendVarint(nil, 7, hp+1) - if _, e = s.CompleteBattle(bad, "p:1"); e == nil { - t.Fatal("impossible HP accepted") - } - if b, e := s.CompleteBattle(nil, "p:1"); e != nil || len(b) != 0 { - t.Fatalf("practice=%x err=%v", b, e) - } - if len(s.state.Users) != 0 { - t.Fatal("practice changed competitive progress") - } -} -func TestPresetSlotCurrencyAndIdempotency(t *testing.T) { - s, _ := installed(t) - req := wire.AppendVarint(nil, 1, 41) - req = wire.AppendVarint(req, 2, 1) - before := s.state.Slots - for range 2 { - if _, _, _, e := s.HandleSession("/MonsterHuntPresetSlotAdd", req, "session"); e != nil { - t.Fatal(e) - } - } - if s.state.Slots != before+1 { - t.Fatal("slot retry doubled purchase") - } - changed := wire.AppendVarint(nil, 1, 41) - changed = wire.AppendVarint(changed, 2, 2) - if _, _, _, e := s.HandleSession("/MonsterHuntPresetSlotAdd", changed, "session"); e == nil { - t.Fatal("changed slot retry accepted") - } -} -func TestDeckShapeRejectsRepeatedPositionAndAcceptsTeamThree(t *testing.T) { - s := &Service{} - char := wire.AppendVarint(nil, 1, 1) - char = wire.AppendVarint(char, 3, 1) - deck := wire.AppendVarint(nil, 1, 3) - deck = wire.AppendBytes(deck, 2, char) - if e := s.validateDecks([][]byte{deck}); e != nil { - t.Fatal(e) - } - dup := wire.AppendVarint(nil, 1, 2) - dup = wire.AppendVarint(dup, 3, 2) - deck = wire.AppendBytes(deck, 2, dup) - if e := s.validateDecks([][]byte{deck}); e == nil { - t.Fatal("duplicate position accepted") - } -} -func TestRankRewardUsesActualLocalPercentThreshold(t *testing.T) { - s := &Service{} - d := &gamedata.MonsterHunt{Ranks: map[uint64][]gamedata.MonsterHuntRankReward{1: {{Type: 1, Ranking: 10, Rewards: []gamedata.BattleReward{{Count: 10}}}, {Type: 1, Ranking: 100, Rewards: []gamedata.BattleReward{{Count: 1}}}}}} - got := s.rankRewards(d, 1) - if len(got) != 1 || got[0].Count != 1 { - t.Fatalf("rewards=%+v", got) - } -} - -func TestDailyRewardUpgradePaysDifference(t *testing.T) { - d := &gamedata.MonsterHunt{Rewards: map[uint64]gamedata.MonsterHuntRewards{1: {Daily: []gamedata.BattleReward{{Type: 4, Count: 100}}}, 2: {Daily: []gamedata.BattleReward{{Type: 4, Count: 150}, {Type: 8, ID: 1, Count: 2}}}}} - if got := dailyDifference(d, 1, 1); len(got) != 0 { - t.Fatalf("same level repeated payout=%v", got) - } - got := dailyDifference(d, 1, 2) - if len(got) != 2 || got[0].Count != 50 || got[1].Count != 2 { - t.Fatalf("upgrade=%v", got) - } -} -func TestSessionRetryRejectsChangedBodyWithoutGameData(t *testing.T) { - s := &Service{state: snapshot{Replies: map[string]reply{"sid:/MonsterHuntDeckSave:2": {Request: wire.AppendVarint(nil, 1, 2), Response: []byte{}}}}} - req := wire.AppendVarint(nil, 1, 2) - if _, _, ok, e := s.HandleSession("/MonsterHuntDeckSave", req, "sid"); e != nil || !ok { - t.Fatal(e) - } - changed := wire.AppendVarint(req, 4, 1) - if _, _, _, e := s.HandleSession("/MonsterHuntDeckSave", changed, "sid"); e == nil { - t.Fatal("changed same-sequence request accepted") - } -} -func TestRequestMalformedAndUnsignedWrapRejected(t *testing.T) { - s := &Service{} - for _, req := range [][]byte{{0xff}, wire.AppendVarint(nil, 1, ^uint64(0))} { - if _, _, ok, e := s.HandleSession("/MonsterHuntUserInfo", req, "sid"); e == nil || !ok { - t.Fatalf("bad request accepted %x", req) - } - } -} diff --git a/go/internal/server/monsterhunt/preset.go b/go/internal/server/monsterhunt/preset.go deleted file mode 100644 index ad4b9c9..0000000 --- a/go/internal/server/monsterhunt/preset.go +++ /dev/null @@ -1,124 +0,0 @@ -package monsterhunt - -import ( - "bd2server/internal/server/player" - "bd2server/internal/server/wire" - "fmt" -) - -func (s *Service) presetBindings(p []byte) (map[uint64]uint64, []player.PresetEquipmentBinding, error) { - assignments := map[uint64]uint64{} - var bindings []player.PresetEquipmentBinding - entries, e := messages(p, 5) - if e != nil { - return nil, nil, e - } - for _, entry := range entries { - base, _, e := wire.Bytes(entry, 1) - if e != nil { - return nil, nil, e - } - index, _ := scalar(base, 1) - costume, _ := scalar(entry, 2) - if _, duplicate := assignments[index]; duplicate { - return nil, nil, fmt.Errorf("monsterhunt: duplicate preset character") - } - assignments[index] = costume - if costume != 0 && s.collection != nil { - c, found := s.collection.CostumeByIndex(costume) - if !found || c.UseChar != index { - return nil, nil, fmt.Errorf("monsterhunt: preset costume not owned by character") - } - } - binding := player.PresetEquipmentBinding{CharacterIndex: index, Equipment: make([]uint64, 5)} - equipment, e := messages(entry, 3) - if e != nil { - return nil, nil, e - } - seen := map[uint64]bool{} - for _, item := range equipment { - t, _ := scalar(item, 1) - id, _ := scalar(item, 2) - if t >= 5 || seen[t] { - return nil, nil, fmt.Errorf("monsterhunt: invalid preset equipment slot") - } - seen[t] = true - binding.Equipment[t] = id - } - bindings = append(bindings, binding) - } - if len(bindings) > 0 && s.equipment != nil { - if e = s.equipment.ValidatePresetEquipment(bindings); e != nil { - return nil, nil, e - } - } - return assignments, bindings, nil -} -func (s *Service) applyPreset(p []byte) ([]byte, error) { - if s.characters == nil || s.equipment == nil || s.collection == nil { - return nil, fmt.Errorf("monsterhunt: preset ownership runtime unavailable") - } - assignments, bindings, e := s.presetBindings(p) - if e != nil { - return nil, e - } - if _, e = s.characters.ApplyPresetCostumes(assignments); e != nil { - return nil, e - } - if len(bindings) > 0 { - if _, e = s.equipment.ApplyPresetEquipment(bindings); e != nil { - return nil, e - } - } - var out []byte - for _, binding := range bindings { - c, ok := s.characters.Find(binding.CharacterIndex) - if ok { - out = wire.AppendBytes(out, 2, player.CharacterWire(c)) - } - b := wire.AppendVarint(nil, 1, binding.CharacterIndex) - for _, id := range binding.Equipment { - b = wire.AppendVarint(b, 2, id) - } - out = wire.AppendBytes(out, 3, b) - } - return out, nil -} - -func (s *Service) validateSettings(settings [][]byte) error { - for _, setting := range settings { - index, e := scalar(setting, 1) - if e != nil || index == 0 { - return fmt.Errorf("monsterhunt: invalid costume setting character") - } - if s.characters != nil { - if _, ok := s.characters.Find(index); !ok { - return fmt.Errorf("monsterhunt: setting character not owned") - } - } - seq, e := messages(setting, 2) - if e != nil { - return e - } - mode, _ := scalar(setting, 3) - if mode != BattleMode && mode != PracticeMode { - return fmt.Errorf("monsterhunt: invalid setting battle mode") - } - for _, item := range seq { - costume, _, e := wire.Varint(item, 1) - if e != nil { - return e - } - if costume == 0 || costume == ^uint64(0) { - continue - } - if s.collection != nil { - c, ok := s.collection.CostumeByIndex(costume) - if !ok || c.UseChar != index { - return fmt.Errorf("monsterhunt: setting costume not owned") - } - } - } - } - return nil -} diff --git a/go/internal/server/monsterhunt/rewards.go b/go/internal/server/monsterhunt/rewards.go deleted file mode 100644 index 0dbbcb9..0000000 --- a/go/internal/server/monsterhunt/rewards.go +++ /dev/null @@ -1,99 +0,0 @@ -package monsterhunt - -import ( - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" -) - -// Daily payout is the best reward achieved that UTC day. Improvements pay -// only positive per-item differences from the already paid level. -func dailyDifference(d *gamedata.MonsterHunt, previous, next uint64) []gamedata.BattleReward { - paid := map[[2]uint64]uint64{} - for _, r := range d.Rewards[previous].Daily { - paid[[2]uint64{r.Type, r.ID}] += r.Count - } - var out []gamedata.BattleReward - for _, r := range d.Rewards[next].Daily { - n := paid[[2]uint64{r.Type, r.ID}] - if r.Count > n { - r.Count -= n - out = append(out, r) - } - } - return out -} - -// This server has one persistent account state: a submitted participant ranks -// first among the local participants. No official leaderboard is imported. -func (s *Service) score(u user) float64 { - progress := float64(0) - if u.HighestHP > 0 { - progress = float64(u.HighestHP-u.StartHP) / float64(u.HighestHP) * 10000 - } - return float64(u.ClearLevel*1000000) + progress -} -func (s *Service) rankWire(u user) []byte { - b := wire.AppendVarint(nil, 7, 1) - b = wire.AppendDouble(b, 8, s.score(u)) - return wire.AppendDouble(b, 10, 100) -} -func (s *Service) rankRewards(d *gamedata.MonsterHunt, group uint64) []gamedata.BattleReward { - var selected *gamedata.MonsterHuntRankReward - for _, r := range d.Ranks[group] { - threshold := float64(1) - if r.Type == 1 { - threshold = 100 - } - if r.Ranking >= threshold && (selected == nil || r.Type < selected.Type || r.Type == selected.Type && r.Ranking < selected.Ranking) { - selected = &r - } - } - if selected == nil { - return nil - } - return selected.Rewards -} -func (s *Service) grant(identity string, rewards []gamedata.BattleReward) ([]byte, error) { - if s.rewardGrant != nil { - rs := make([]gamedata.Reward, len(rewards)) - for i, r := range rewards { - rs[i] = gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count} //nolint:staticcheck // S1016 - } - return s.rewardGrant(identity, rs) - } - var currency []gamedata.Reward - var stack []gamedata.BattleReward - for _, r := range rewards { - switch r.Type { - case 2, 3, 4, 12, 20: - currency = append(currency, gamedata.Reward{Type: r.Type, ID: r.ID, Count: r.Count}) //nolint:staticcheck // S1016 - default: - stack = append(stack, r) - } - } - if _, e := s.wallet.GrantQuestOnce(identity+":currency", currency); e != nil { - return nil, e - } - items, e := s.inventory.GrantOnce(identity+":items", stack) - if e != nil { - return nil, e - } - if len(items) == 0 { - items = s.inventory.GrantedItems(identity + ":items") - } - var bundle []byte - for _, item := range items { - bundle = wire.AppendBytes(bundle, 1, player.ItemWire(item)) - v := wire.AppendVarint(nil, 2, item.ID) - v = wire.AppendVarint(v, 3, item.Type) - v = wire.AppendVarint(v, 4, item.Count) - bundle = wire.AppendBytes(bundle, 6, v) - } - for _, r := range currency { - v := wire.AppendVarint(nil, 3, r.Type) - v = wire.AppendVarint(v, 4, r.Count) - bundle = wire.AppendBytes(bundle, 1, v) - } - return bundle, nil -} diff --git a/go/internal/server/monsterhunt/transaction_test.go b/go/internal/server/monsterhunt/transaction_test.go deleted file mode 100644 index d115f6a..0000000 --- a/go/internal/server/monsterhunt/transaction_test.go +++ /dev/null @@ -1,69 +0,0 @@ -package monsterhunt - -import ( - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" -) - -func TestPresetPurchaseRollsBackCurrencyAndSlotTogether(t *testing.T) { - s, _ := installed(t) - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - wallet, err := player.OpenWallet(repo, player.Currency{Gold: 100000}) - if err != nil { - t.Fatal(err) - } - if err = wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - s.storage = repo - s.wallet = wallet - before := s.state.Slots - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - req := wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 1) - if _, _, _, err = s.HandleSession("/MonsterHuntPresetSlotAdd", req, "session-A"); err != nil { - _ = op.Rollback() - t.Fatal(err) - } - _ = op.Rollback() - if err = repo.Close(); err != nil { - t.Fatal(err) - } - reopened, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := reopened.Close(); err != nil { - t.Error(err) - } - }() - if raw, err := reopened.Load("monsterhunt"); err != nil || raw != nil { - t.Fatalf("slot survived rollback: %s %v", raw, err) - } - wallet, err = player.OpenWallet(reopened, player.Currency{}) - if err != nil { - t.Fatal(err) - } - if wallet.Snapshot().Gold != 100000 { - t.Fatalf("gold survived rollback: %+v", wallet.Snapshot()) - } - if s.state.Slots != before+1 || repo.Check() == nil { - t.Fatal("dirty domain memory was not fenced") - } -} diff --git a/go/internal/server/monsterhunt/validation.go b/go/internal/server/monsterhunt/validation.go deleted file mode 100644 index 44c5659..0000000 --- a/go/internal/server/monsterhunt/validation.go +++ /dev/null @@ -1,37 +0,0 @@ -package monsterhunt - -import ( - "bd2server/internal/server/wire" - "encoding/binary" - "fmt" - "math" -) - -func validateRequest(req []byte) error { - return wire.Walk(req, func(f wire.Field) error { - if f.Type == 0 { - v, _ := binary.Uvarint(f.Value) - if v > math.MaxInt32 { - return fmt.Errorf("monsterhunt: integer outside protocol range") - } - } - return nil - }) -} - -func (s *Service) ValidatePack(pack int, req []byte) error { - s.mu.Lock() - defer s.mu.Unlock() - id, e := scalar(req, 6) - if e != nil { - return e - } - d, e := s.load(id) - if e != nil { - return e - } - if pack <= 0 || uint64(pack) != d.PackID { - return fmt.Errorf("monsterhunt: hunt does not belong to current pack") - } - return nil -} diff --git a/go/internal/server/npcinn/inn_test.go b/go/internal/server/npcinn/inn_test.go deleted file mode 100644 index 5cd7ed9..0000000 --- a/go/internal/server/npcinn/inn_test.go +++ /dev/null @@ -1,174 +0,0 @@ -package npcinn - -import ( - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "errors" - "path/filepath" - "testing" -) - -func innTestService(t *testing.T, gold, level, state uint64) (*Service, *player.CharacterStore, *player.Wallet, stateio.Store) { - t.Helper() - store := stateio.NewMemory() - return innTestServiceWithStore(t, store, gold, level, state) -} - -func innTestServiceWithStore(t *testing.T, store stateio.Store, gold, level, state uint64) (*Service, *player.CharacterStore, *player.Wallet, stateio.Store) { - t.Helper() - inv, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - chars, err := player.OpenCharacterStore(store, []player.Character{{InvenIndex: 71, ID: 350, Level: 1, HP: 10}, {InvenIndex: 72, ID: 360, Level: 1, HP: 10}}, inv, "", "") - if err != nil { - t.Fatal(err) - } - if err := chars.AttachMaxHealth(func(player.Character) (uint64, error) { return 100, nil }); err != nil { - t.Fatal(err) - } - if err = chars.EnsurePersisted(); err != nil { - t.Fatal(err) - } - wallet, err := player.OpenWallet(store, player.Currency{Gold: gold}) - if err != nil { - t.Fatal(err) - } - if err = wallet.EnsurePersisted(); err != nil { - t.Fatal(err) - } - s, err := New(store, chars, wallet, func(pack, npc uint64) (gamedata.InnRule, uint64, error) { - return gamedata.InnRule{NPCID: 7, MapID: 33, ItemCount: 2, Currency: 4, FreeSquadLevel: 20, GoodDiscount: 10}, state, nil - }, func() (uint64, error) { return level, nil }, func() bool { return false }) - if err != nil { - t.Fatal(err) - } - s.BeginSession("test") - return s, chars, wallet, store -} - -type innReceiptFailStore struct{ stateio.Store } - -func (s innReceiptFailStore) Save(name string, b []byte) error { - if name == "npcinn" { - return errors.New("receipt write failed") - } - return s.Store.Save(name, b) -} - -func TestInnReceiptFailureRollsBackGoldAndHealthTogether(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - s, _, _, _ := innTestServiceWithStore(t, repo, 10, 21, 1) - s.store = innReceiptFailStore{repo} - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/CharAllRevival", recoveryRequest(1, 71)); err == nil { - t.Fatal("receipt failure ignored") - } - if err = op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - next, chars, wallet, _ := innTestServiceWithStore(t, repo, 0, 21, 1) - hp, _ := chars.CurrentHealth(71) - if hp != 10 || wallet.Snapshot().Gold != 10 { - t.Fatal("failed inn request committed health or charge") - } - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = next.Handle("/CharAllRevival", recoveryRequest(1, 71)); err != nil { - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - hp, _ = chars.CurrentHealth(71) - if hp != 100 || wallet.Snapshot().Gold != 5 { - t.Fatal("retry did not commit recovery once") - } -} -func recoveryRequest(seq uint64, indices ...uint64) []byte { - b := wire.AppendVarint(nil, 1, seq) - b = wire.AppendVarint(b, 2, 1) - b = wire.AppendVarint(b, 3, 7) - for _, i := range indices { - b = wire.AppendVarint(b, 4, i) - } - return b -} - -func TestInnPartialRecoveryAndReplayUsePersistedHealth(t *testing.T) { - s, chars, wallet, store := innTestService(t, 2, 21, 1) - b := recoveryRequest(1, 71, 72) - code, response, handled, err := s.Handle("/CharAllRevival", b) - if err != nil || !handled || code != 11 { - t.Fatal(err) - } - hp, _ := chars.CurrentHealth(71) - second, _ := chars.CurrentHealth(72) - if hp != 50 || second != 10 || wallet.Snapshot().Gold != 0 { - t.Fatalf("partial recovery hp=%d second=%d wallet=%+v", hp, second, wallet.Snapshot()) - } - next, err := New(store, chars, wallet, s.context, s.level, s.battle) - if err != nil { - t.Fatal(err) - } - next.BeginSession("test") - _, replay, _, err := next.Handle("/CharAllRevival", b) - if err != nil || !bytes.Equal(response, replay) { - t.Fatal("receipt lost on reopening") - } - changed := recoveryRequest(1, 72) - if _, _, _, err = next.Handle("/CharAllRevival", changed); err == nil { - t.Fatal("changed recovery replay accepted") - } -} - -func TestInnFreeLevelDeadRevivalAndDiscountPrice(t *testing.T) { - s, chars, wallet, _ := innTestService(t, 0, 20, 1) - if err := chars.SetCurrentHealth(72, 0); err != nil { - t.Fatal(err) - } - if _, _, _, err := s.Handle("/CharAllRevival", recoveryRequest(1, 71, 72)); err != nil { - t.Fatal(err) - } - a, _ := chars.CurrentHealth(71) - b, _ := chars.CurrentHealth(72) - if a != 100 || b != 1 || wallet.Snapshot().Gold != 0 { - t.Fatal("free squad recovery or free 1 HP resurrection incorrect") - } - s, chars, wallet, _ = innTestService(t, 10, 21, 2) - if _, _, _, err := s.Handle("/CharAllRevival", recoveryRequest(1, 71)); err != nil { - t.Fatal(err) - } - a, _ = chars.CurrentHealth(71) - if a != 100 || wallet.Snapshot().Gold != 5 { - t.Fatal("discount recovery must ceil 90 * 2 /40 *90%=4.05 to 5 gold") - } - if _, _, _, err := s.Handle("/CharAllRevival", recoveryRequest(2, 71, 71)); err == nil { - t.Fatal("duplicate targets accepted") - } -} diff --git a/go/internal/server/npcshop/sale_validation_test.go b/go/internal/server/npcshop/sale_validation_test.go deleted file mode 100644 index 3268e8a..0000000 --- a/go/internal/server/npcshop/sale_validation_test.go +++ /dev/null @@ -1,263 +0,0 @@ -package npcshop - -import ( - "bytes" - "encoding/json" - "errors" - "path/filepath" - "strconv" - "testing" - "time" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" -) - -func saleLine(index, product, n, rate uint64) []byte { - b := wire.AppendVarint(nil, 1, index) - b = wire.AppendVarint(b, 2, product) - b = wire.AppendVarint(b, 3, n) - return wire.AppendVarint(b, 4, rate) -} -func saleRequest(seq uint64, rows ...[]byte) []byte { - b := wire.AppendVarint(wire.AppendVarint(nil, 1, seq), 2, 71) - for _, row := range rows { - b = wire.AppendBytes(b, 3, row) - } - return b -} - -func TestSaleAllClientTypesAndWholeBatchValidation(t *testing.T) { - s, e, store := fixture(t) - s.design.Sell[23] = gamedata.NPCProduct{ID: 23, Reward: gamedata.Reward{Type: 8, ID: 401, Count: 1}, Price: gamedata.Reward{Type: 4, Count: 9}} - items, err := s.items.GrantOnce("sale-types", []gamedata.BattleReward{{Type: 5, ID: 400, Count: 8}, {Type: 8, ID: 401, Count: 10}}) - if err != nil { - t.Fatal(err) - } - food, resource := items[0], items[1] - good := saleLine(resource.InvenIndex, 23, 4, 100) - for _, bad := range [][]byte{ - saleLine(resource.InvenIndex, 19, 1, 100), - saleLine(resource.InvenIndex, 23, 0, 100), - saleLine(resource.InvenIndex, 23, 11, 100), - saleLine(resource.InvenIndex, 23, 1<<31, 100), - saleLine(resource.InvenIndex, 23, 1, 0), - wire.AppendVarint(good, 3, 1), - wire.AppendBytes(good, 4, nil), - } { - if _, _, _, err = s.Handle("/ShopSell", saleRequest(1, saleLine(food.InvenIndex, 19, 3, 100), bad)); err == nil || e.calls != 0 { - t.Fatalf("invalid batch mutated economy: %x %v", bad, err) - } - } - // KeepFlag is the actual ItemDBInfo protection flag; these two types have - // no binding/equipment LockFlag or UseChar fields in the current protocol. - resource.KeepFlag = 1 - payload, _ := json.Marshal(resource) - if err = store.SaveWithEntries("items", nil, []stateio.EntryMutation{{Bucket: "items", Key: strconv.FormatUint(resource.InvenIndex, 10), Payload: payload}}); err != nil { - t.Fatal(err) - } - s.items, err = player.OpenInventory(store, &player.Starter{Version: versionconfig.State()}) - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/ShopSell", saleRequest(2, good)); err == nil || e.calls != 0 { - t.Fatal("protected resource sold", err) - } - resource.KeepFlag = 0 - payload, _ = json.Marshal(resource) - if err = store.SaveWithEntries("items", nil, []stateio.EntryMutation{{Bucket: "items", Key: strconv.FormatUint(resource.InvenIndex, 10), Payload: payload}}); err != nil { - t.Fatal(err) - } - s.items, err = player.OpenInventory(store, &player.Starter{Version: versionconfig.State()}) - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/ShopSell", saleRequest(3, saleLine(food.InvenIndex, 19, 3, 100), good)); err != nil { - t.Fatal(err) - } - if e.calls != 1 || len(e.rewards) != 2 || e.rewards[0].Count != 21 || e.rewards[1].Count != 36 { - t.Fatal("wrong mixed sale", e) - } - remaining := s.items.All() - if len(remaining) != 2 || remaining[0].Count != 5 || remaining[1].Count != 6 { - t.Fatal("wrong mixed inventory debit", remaining) - } -} - -func TestSaleQuoteSeedRemainsUntilShopInfoRefresh(t *testing.T) { - s, _, _ := fixture(t) - p := gamedata.NPCProduct{Reward: gamedata.Reward{Type: 8, ID: 401, Count: 1}, Discount: 20, Premium: 19} - // Independent vectors from the client's ulong WELL512 operations. - for _, row := range []struct{ seed, rate uint64 }{{0, 117}, {20731, 80}, {20732, 101}} { - s.quotedSeed = &row.seed - if got := s.rate(p, 71, 2); got != row.rate { - t.Fatal("client WELL512 mismatch", row, got) - } - } - s.quotedSeed = nil - s.now = func() time.Time { return time.Unix(20731*86400, 0) } - _, info, _, err := s.Handle("/ShopInfo", wire.AppendVarint(nil, 1, 1)) - seed, _, _ := wire.Varint(info, 2) - if err != nil || seed != 20731 { - t.Fatal(seed, err) - } - s.now = func() time.Time { return time.Unix(20732*86400, 0) } - s.BeginSession("login") // The router activates the same session on each request. - if s.rate(p, 71, 2) != 80 { - t.Fatal("clock change invalidated displayed sale price") - } - if _, _, _, err = s.Handle("/ShopInfo", wire.AppendVarint(nil, 1, 2)); err != nil || s.rate(p, 71, 2) != 101 { - t.Fatal("new quote failed to refresh seed", err) - } -} - -func TestShopPriceUsesUnsignedWELLMaskForFoodResourceAndBuy(t *testing.T) { - s, _, _ := fixture(t) - seed := uint64(20732) - s.quotedSeed = &seed - for _, row := range []struct { - typ, item, tab, expected uint64 - }{ - // Current SellItemTable[2], also observed in the client's request: - // ShopInfo seed=20732; shop=1; Food=5; item=2002; Sell=2; Rate=98. - {5, 2002, 2, 98}, - {8, 101, 2, 93}, - {5, 2002, 1, 110}, - } { - p := gamedata.NPCProduct{Reward: gamedata.Reward{Type: row.typ, ID: row.item, Count: 1}, Discount: 20, Premium: 19} - if got := s.rate(p, 1, row.tab); got != row.expected { - t.Fatalf("unsigned WELL mask type=%d item=%d tab=%d got=%d want=%d", row.typ, row.item, row.tab, got, row.expected) - } - } -} - -func TestSaleClientObservedFoodRateCreditsDerivedGold(t *testing.T) { - s, economy, _ := fixture(t) - s.design.Shops[1] = gamedata.NPCShop{ID: 1, PackID: 91} - s.design.Sell[2] = gamedata.NPCProduct{ID: 2, Reward: gamedata.Reward{Type: 5, ID: 2002, Count: 1}, Price: gamedata.Reward{Type: 4, Count: 28}, Discount: 20, Premium: 19, PremiumPriceType: 1, HighDay: 1, HighShop: 14, HighPremium: 20} - seed := uint64(20732) - s.quotedSeed = &seed - items, err := s.items.GrantOnce("observed-food", []gamedata.BattleReward{{Type: 5, ID: 2002, Count: 2}}) - if err != nil { - t.Fatal(err) - } - request := func(rate uint64) []byte { - b := wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 1) - return wire.AppendBytes(b, 3, saleLine(items[0].InvenIndex, 2, 1, rate)) - } - if _, _, _, err = s.Handle("/ShopSell", request(93)); err == nil || economy.calls != 0 { - t.Fatal("incorrect sign-extended rate accepted", err) - } - if _, _, _, err = s.Handle("/ShopSell", request(98)); err != nil { - t.Fatal("actual client rate rejected", err) - } - if economy.calls != 1 || len(economy.rewards) != 1 || economy.rewards[0].Type != 4 || economy.rewards[0].Count != 27 || s.items.All()[0].Count != 1 { - t.Fatal("wrong derived sale credit/debit", economy.rewards, s.items.All()) - } -} - -type saleWalletEconomy struct{ wallet *player.Wallet } - -func (e saleWalletEconomy) Apply(identity string, costs, rewards []gamedata.Reward) ([]byte, error) { - if err := e.wallet.ExchangeOnce(identity, costs, rewards); err != nil { - return nil, err - } - return wire.AppendVarint(nil, 1, e.wallet.Snapshot().Gold), nil -} - -func TestSQLiteMixedSaleWalletInventoryReceiptRollbackAndRestartRetry(t *testing.T) { - base, _, _ := fixture(t) - base.design.Sell[23] = gamedata.NPCProduct{ID: 23, Reward: gamedata.Reward{Type: 8, ID: 401, Count: 1}, Price: gamedata.Reward{Type: 4, Count: 9}} - path := filepath.Join(t.TempDir(), "state.db") - starter := &player.Starter{Version: versionconfig.State()} - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { _ = repo.Close() }() - items, err := player.OpenInventory(repo, starter) - if err != nil { - t.Fatal(err) - } - granted, err := items.GrantOnce("initial", []gamedata.BattleReward{{Type: 5, ID: 400, Count: 8}, {Type: 8, ID: 401, Count: 10}}) - if err != nil { - t.Fatal(err) - } - req := saleRequest(1, saleLine(granted[0].InvenIndex, 19, 3, 100), saleLine(granted[1].InvenIndex, 23, 4, 100)) - var wallet *player.Wallet - var response []byte - for phase := range 3 { - items, err = player.OpenInventory(repo, starter) - if err != nil { - t.Fatal(err) - } - wallet, err = player.OpenWallet(repo, player.Currency{Gold: 100}) - if err != nil { - t.Fatal(err) - } - var store stateio.Store = repo - if phase == 0 { - store = failedReceiptStore{repo} - } - s, err := New(base.design, store, saleWalletEconomy{wallet}, items, base.available) - if err != nil { - t.Fatal(err) - } - s.now, s.session = base.now, base.session - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - _, body, _, handleErr := s.Handle("/ShopSell", req) - if phase == 0 { - if handleErr == nil { - t.Fatal("injected receipt failure accepted") - } - if err = op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - } else { - if handleErr != nil { - t.Fatal(handleErr) - } - if phase == 1 { - response = body - } else if !bytes.Equal(response, body) { - t.Fatal("restart replay changed response") - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - } - if err = repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - checkItems, err := player.OpenInventory(repo, starter) - if err != nil { - t.Fatal(err) - } - checkWallet, err := player.OpenWallet(repo, player.Currency{Gold: 100}) - if err != nil { - t.Fatal(err) - } - wantGold, wantFood, wantResource := uint64(157), uint64(5), uint64(6) - if phase == 0 { - wantGold, wantFood, wantResource = 100, 8, 10 - if raw, err := repo.Load("npc_shop"); err != nil || raw != nil { - t.Fatal("sale receipt survived rollback", err) - } - } - rows := checkItems.All() - if checkWallet.Snapshot().Gold != wantGold || len(rows) != 2 || rows[0].Count != wantFood || rows[1].Count != wantResource { - t.Fatal("SQLite atomic/replay state mismatch", phase, checkWallet.Snapshot(), rows) - } - } -} diff --git a/go/internal/server/npcshop/shop_test.go b/go/internal/server/npcshop/shop_test.go deleted file mode 100644 index 439a959..0000000 --- a/go/internal/server/npcshop/shop_test.go +++ /dev/null @@ -1,243 +0,0 @@ -package npcshop - -import ( - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" - "bytes" - "errors" - "fmt" - "path/filepath" - "testing" - "time" -) - -type economy struct { - calls int - costs, rewards []gamedata.Reward - fail bool -} -type failedReceiptStore struct{ stateio.Store } - -func (s failedReceiptStore) Save(name string, b []byte) error { - if name == "npc_shop" { - return fmt.Errorf("receipt disk failure") - } - return s.Store.Save(name, b) -} - -type storedEconomy struct{ store stateio.Store } - -func (e storedEconomy) Apply(_ string, _ []gamedata.Reward, r []gamedata.Reward) ([]byte, error) { - if err := e.store.Save("npc_test_grants", []byte{1}); err != nil { - return nil, err - } - return []byte{8, 1}, nil -} -func TestFailedSaleReceiptRollsBackConcreteInventoryAndReward(t *testing.T) { - base, _, _ := fixture(t) - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - starter := &player.Starter{Version: versionconfig.State()} - items, err := player.OpenInventory(repo, starter) - if err != nil { - t.Fatal(err) - } - granted, err := items.GrantOnce("initial-food", []gamedata.BattleReward{{Type: 5, ID: 400, Count: 8}}) - if err != nil { - t.Fatal(err) - } - s, err := New(base.design, failedReceiptStore{repo}, storedEconomy{repo}, items, base.available) - if err != nil { - t.Fatal(err) - } - s.now = base.now - s.SetSessionSource(base.session) - line := wire.AppendVarint(nil, 1, granted[0].InvenIndex) - line = wire.AppendVarint(line, 2, 19) - line = wire.AppendVarint(line, 3, 3) - line = wire.AppendVarint(line, 4, 100) - req := wire.AppendBytes(wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 71), 3, line) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/ShopSell", req); err == nil { - t.Fatal("expected failed receipt persistence") - } - if err = op.Rollback(); err != nil && !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatal(err) - } - if err = repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - reopened, err := player.OpenInventory(repo, starter) - if err != nil { - t.Fatal(err) - } - if rows := reopened.All(); len(rows) != 1 || rows[0].Count != 8 { - t.Fatal("inventory removal survived rollback", rows) - } - for _, name := range []string{"npc_shop", "npc_test_grants"} { - raw, err := repo.Load(name) - if err != nil || raw != nil { - t.Fatal("reward or receipt survived rollback", name, err) - } - } -} - -func (e *economy) Apply(_ string, c, r []gamedata.Reward) ([]byte, error) { - if e.fail { - return nil, fmt.Errorf("insufficient balance") - } - e.calls++ - e.costs = c - e.rewards = r - return []byte{8, 1}, nil -} -func fixture(t *testing.T) (*Service, *economy, *stateio.Memory) { - t.Helper() - store := stateio.NewMemory() - items, err := player.OpenInventory(store, &player.Starter{Version: versionconfig.State()}) - if err != nil { - t.Fatal(err) - } - d := gamedata.NPCShopDesign{Shops: map[uint64]gamedata.NPCShop{71: {ID: 71, PackID: 91, ResetType: 1}}, Products: map[uint64]map[uint64]gamedata.NPCProduct{71: {3: {ID: 3, GroupID: 71, MaxCount: 3, Reward: gamedata.Reward{Type: 5, ID: 400, Count: 2}, Price: gamedata.Reward{Type: 4, Count: 25}}}}, Sell: map[uint64]gamedata.NPCProduct{19: {ID: 19, Reward: gamedata.Reward{Type: 5, ID: 400, Count: 1}, Price: gamedata.Reward{Type: 4, Count: 7}}}} - e := &economy{} - s, err := New(d, store, e, items, func(pack uint64) bool { return pack == 91 }) - if err != nil { - t.Fatal(err) - } - s.now = func() time.Time { return time.Date(2026, 10, 5, 12, 0, 0, 0, time.UTC) } - s.SetSessionSource(func() string { return "login" }) - return s, e, store -} -func buy(seq, n, rate uint64) []byte { - b := wire.AppendVarint(nil, 1, seq) - g := wire.AppendVarint(nil, 1, 71) - p := wire.AppendVarint(nil, 2, 3) - p = wire.AppendVarint(p, 3, n) - p = wire.AppendVarint(p, 4, rate) - return wire.AppendBytes(b, 2, wire.AppendBytes(g, 2, p)) -} -func TestPurchaseReceiptLimitsResetAndWholeBatchValidation(t *testing.T) { - s, e, store := fixture(t) - request := buy(1, 2, 100) - code, response, ok, err := s.Handle("/ShopBuy", request) - if err != nil || !ok || code != 54 { - t.Fatal(code, ok, err) - } - if e.calls != 1 || len(e.costs) != 1 || e.costs[0].Count != 50 || e.rewards[0].Count != 4 { - t.Fatal(e) - } - reopened, err := New(s.design, store, e, s.items, s.available) - if err != nil { - t.Fatal(err) - } - reopened.now = s.now - reopened.SetSessionSource(s.session) - _, replay, _, err := reopened.Handle("/ShopBuy", request) - if err != nil || !bytes.Equal(response, replay) || e.calls != 1 { - t.Fatal("retry did not preserve receipt", err) - } - for _, bad := range [][]byte{buy(1, 1, 100), buy(2, 2, 100), buy(3, 1, 1)} { - if _, _, _, err = s.Handle("/ShopBuy", bad); err == nil { - t.Fatalf("invalid purchase accepted %x", bad) - } - } - batch := buy(4, 1, 100) - g := wire.AppendVarint(nil, 1, 71) - p := wire.AppendVarint(wire.AppendVarint(nil, 2, 999), 3, 1) - batch = wire.AppendBytes(batch, 2, wire.AppendBytes(g, 2, p)) - if _, _, _, err = s.Handle("/ShopBuy", batch); err == nil || e.calls != 1 { - t.Fatal("partially granted invalid batch") - } - s.now = func() time.Time { return time.Date(2026, 10, 6, 12, 0, 0, 0, time.UTC) } - if _, _, _, err = s.Handle("/ShopBuy", buy(5, 3, 100)); err != nil { - t.Fatal("daily reset did not reopen stock", err) - } -} -func TestSaleConsumesTheRequestedStackAndRejectsDuplicates(t *testing.T) { - s, e, _ := fixture(t) - granted, err := s.items.GrantOnce("food", []gamedata.BattleReward{{Type: 5, ID: 400, Count: 8}}) - if err != nil { - t.Fatal(err) - } - item := granted[0] - line := wire.AppendVarint(nil, 1, item.InvenIndex) - line = wire.AppendVarint(line, 2, 19) - line = wire.AppendVarint(line, 3, 3) - line = wire.AppendVarint(line, 4, 100) - req := wire.AppendBytes(wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 71), 3, line) - duplicate := wire.AppendBytes(req, 3, line) - if _, _, _, err = s.Handle("/ShopSell", duplicate); err == nil || e.calls != 0 { - t.Fatal("duplicate stack sale accepted") - } - code, first, _, err := s.Handle("/ShopSell", req) - if err != nil || code != 55 { - t.Fatal(code, err) - } - if len(e.costs) != 0 || e.rewards[0].Count != 21 { - t.Fatal("wrong sale reward", e) - } - var remaining uint64 - for _, v := range s.items.All() { - if v.InvenIndex == item.InvenIndex { - remaining = v.Count - } - } - if remaining != 5 { - t.Fatal("wrong concrete stack consumed", remaining) - } - _, replay, _, err := s.Handle("/ShopSell", req) - if err != nil || !bytes.Equal(first, replay) || e.calls != 1 { - t.Fatal("sale retry spent twice", err) - } -} -func TestResourcePurchaseValidatesAndConsumesExactMaterialRows(t *testing.T) { - s, e, _ := fixture(t) - p := s.design.Products[71][3] - p.Price = gamedata.Reward{Type: 8, ID: 401, Count: 5} - s.design.Products[71][3] = p - granted, err := s.items.GrantOnce("resource", []gamedata.BattleReward{{Type: 8, ID: 401, Count: 18}}) - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/ShopBuy", buy(1, 2, 100)); err == nil || e.calls != 0 { - t.Fatal("missing resource payment accepted") - } - material := wire.AppendVarint(nil, 1, granted[0].InvenIndex) - material = wire.AppendVarint(material, 2, 401) - material = wire.AppendVarint(material, 3, 8) - material = wire.AppendVarint(material, 4, 10) - if _, _, _, err = s.Handle("/ShopBuy", wire.AppendBytes(buy(2, 2, 100), 3, material)); err != nil { - t.Fatal(err) - } - if len(e.costs) != 0 || e.rewards[0].Count != 4 { - t.Fatal("material cost sent to generic stack selection", e) - } - rows := s.items.All() - if len(rows) != 1 || rows[0].Count != 8 { - t.Fatal("resource stack was not spent", rows) - } -} diff --git a/go/internal/server/npcshop/talent_discount_test.go b/go/internal/server/npcshop/talent_discount_test.go deleted file mode 100644 index 4528942..0000000 --- a/go/internal/server/npcshop/talent_discount_test.go +++ /dev/null @@ -1,58 +0,0 @@ -package npcshop - -import ( - "bd2server/internal/server/wire" - "testing" -) - -func TestTalentBargainingUsesBoundNPCBasePriceAndSuppressesReputation(t *testing.T) { - s, e, _ := fixture(t) - s.design.ShopNPCs = map[uint64][]uint64{71: {8801}} - calls := 0 - s.SetTalentDiscountSource(func(pack, npc uint64) (uint64, error) { - if pack != 91 || npc != 8801 { - t.Fatal("wrong NPC binding", pack, npc) - } - calls++ - return 20, nil - }) - s.SetReputationSource(func(uint64) (uint64, uint64, error) { return 2, 10, nil }) - p := s.design.Products[71][3] - p.Discount = 40 - p.Premium = 40 - s.design.Products[71][3] = p - rate := s.rate(p, 71, 1) - _, b, _, err := s.Handle("/ShopOpen", wire.AppendVarint(nil, 1, 1)) - if err != nil { - t.Fatal(err) - } - on, _, _ := wire.Varint(b, 2) - if on != 1 { - t.Fatal("client bargaining mode not enabled") - } - if _, _, _, err = s.Handle("/ShopBuy", buy(2, 1, rate)); err != nil { - t.Fatal(err) - } - if calls == 0 || len(e.costs) != 1 || e.costs[0].Count != 20 { - t.Fatal("talent price combined with reputation/market", e.costs) - } - s.SetTalentDiscountSource(func(uint64, uint64) (uint64, error) { return 0, nil }) - if _, _, _, err = s.Handle("/ShopBuy", buy(2, 1, rate)); err != nil || e.calls != 1 { - t.Fatal("discount state altered replay price", err, e.calls) - } -} -func TestBargainingNoBargainAndInvalidDiscountRejectBeforeEconomy(t *testing.T) { - s, e, _ := fixture(t) - s.design.ShopNPCs = map[uint64][]uint64{71: {8801}} - s.SetTalentDiscountSource(func(uint64, uint64) (uint64, error) { return 20, nil }) - p := s.design.Products[71][3] - p.NoBargain = 1 - s.design.Products[71][3] = p - if _, _, _, err := s.Handle("/ShopBuy", buy(1, 1, 100)); err == nil || e.calls != 0 { - t.Fatal("NoBargain bought during talent view") - } - s.SetTalentDiscountSource(func(uint64, uint64) (uint64, error) { return 101, nil }) - if _, _, _, err := s.Handle("/ShopBuy", buy(2, 1, 100)); err == nil || e.calls != 0 { - t.Fatal("invalid discount charged") - } -} diff --git a/go/internal/server/pictorial/health_test.go b/go/internal/server/pictorial/health_test.go deleted file mode 100644 index 8d64b05..0000000 --- a/go/internal/server/pictorial/health_test.go +++ /dev/null @@ -1,127 +0,0 @@ -package pictorial - -import ( - "bytes" - "errors" - "testing" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func TestMaximumHealthIncludesOwnedEquipmentPotentialAndAwakening(t *testing.T) { - character := player.Character{InvenIndex: 77, ID: 350, Level: 1} - s := &Service{Design: &gamedata.PictorialDesign{}, Owned: &ownedState{}} - s.baseHealth.Store([2]uint64{350, 1}, float64(100)) - s.EquipmentContributions = func(got player.Character) ([]gamedata.StatContribution, error) { - if got.InvenIndex != character.InvenIndex || got.ID != character.ID || got.Level != character.Level { - t.Fatalf("wrong stat character: %+v", got) - } - return []gamedata.StatContribution{{Stat: gamedata.StatHealth, Flat: 20, Percent: .1}}, nil - } - s.PotentialContributions = func(player.Character) ([]gamedata.StatContribution, error) { - return []gamedata.StatContribution{{Stat: gamedata.StatHealth, Percent: .2}}, nil - } - s.AwakeContributions = func(player.Character) ([]gamedata.StatContribution, error) { - return []gamedata.StatContribution{{Stat: gamedata.StatHealth, Flat: 5}}, nil - } - maximum, err := s.MaxHealth(character) - if err != nil || maximum != 162 { - t.Fatalf("maximum=%d err=%v, want trunc((100+20+5)*1.3)=162", maximum, err) - } - s.EquipmentContributions = func(player.Character) ([]gamedata.StatContribution, error) { - return nil, errors.New("equipment unavailable") - } - if _, err := s.MaxHealth(character); err == nil { - t.Fatal("ignored equipment stat failure") - } -} - -func TestPotentialConnectionSwitchRecomputesMaximumWithoutHealingOrChangingCostume(t *testing.T) { - store := stateio.NewMemory() - items, err := player.OpenInventory(store, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - characters, err := player.OpenCharacterStore(store, []player.Character{{InvenIndex: 77, ID: 350, HP: 80, Level: 1, UseCostume: 1, CostumeID: 100}}, items, "", "") - if err != nil { - t.Fatal(err) - } - collection, err := player.OpenCollectionStore(store, []player.Costume{{InvenIndex: 1, ID: 100, UseChar: 77}, {InvenIndex: 2, ID: 200, UseChar: 77}}) - if err != nil { - t.Fatal(err) - } - for _, index := range []uint64{1, 2} { - if err = collection.ActivateCostumePotential(index, []uint64{1, 2}); err != nil { - t.Fatal(err) - } - } - wallet, err := player.OpenWallet(store, player.Currency{Gold: 100}) - if err != nil { - t.Fatal(err) - } - d := &gamedata.CostumePotentialDesign{CharacterTypes: map[uint64]uint64{350: 0}, CostumeActive: map[uint64]bool{100: true, 200: true}, CharacterUnique: map[uint64]uint64{350: 9}, CostumeUnique: map[uint64]uint64{100: 9, 200: 9}, Nodes: map[uint64]map[uint64]gamedata.CostumePotentialNode{ - 100: {1: {ID: 1, NodeType: 2, StatType: 1, StatValue: 10}, 2: {ID: 2, NodeType: 1, StatType: 1, StatValue: 100}}, - 200: {1: {ID: 1, NodeType: 2, StatType: 1, StatValue: 5}, 2: {ID: 2, NodeType: 1, StatType: 1, StatValue: 20}}, - }} - potential, err := player.NewCostumePotentialService(d, collection, characters, items, wallet) - if err != nil { - t.Fatal(err) - } - if err = potential.AttachConnectStore(store); err != nil { - t.Fatal(err) - } - potential.BeginSession("health") - stats := &Service{Design: &gamedata.PictorialDesign{}, Owned: &ownedState{}, PotentialContributions: potential.Contributions} - stats.baseHealth.Store([2]uint64{350, 1}, float64(100)) - if err = characters.AttachMaxHealth(stats.MaxHealth); err != nil { - t.Fatal(err) - } - request := func(seq, costume uint64) []byte { - row := wire.AppendVarint(wire.AppendVarint(nil, 1, 77), 2, costume) - return wire.AppendBytes(wire.AppendVarint(nil, 1, seq), 2, row) - } - first := request(1, 100) - if _, _, _, err = potential.Handle("/CostumePotentialConnect", first); err != nil { - t.Fatal(err) - } - if maximum, err := characters.MaxHealth(77); err != nil || maximum != 215 { - t.Fatalf("first maximum=%d err=%v", maximum, err) - } - if hp, err := characters.CurrentHealth(77); err != nil || hp != 80 { - t.Fatal("connecting potential healed damage") - } - if err = characters.SetCurrentHealth(77, 200); err != nil { - t.Fatal(err) - } - second := request(2, 200) - code, out, handled, err := potential.Handle("/CostumePotentialConnect", second) - if err != nil || code != 267 || !handled { - t.Fatal(err) - } - c, ok := characters.Find(77) - if !ok || c.ConnectPotentialCostume != 200 || c.UseCostume != 1 || c.CostumeID != 100 || c.HP != 135 { - t.Fatalf("connection changed worn costume or failed HP clamp: %+v", c) - } - if maximum, err := characters.MaxHealth(77); err != nil || maximum != 135 { - t.Fatalf("switched maximum=%d err=%v", maximum, err) - } - _, again, _, err := potential.Handle("/CostumePotentialConnect", second) - if err != nil || !bytes.Equal(out, again) { - t.Fatal("connection replay changed response") - } - if err = characters.SetCurrentHealth(77, 0); err != nil { - t.Fatal(err) - } - if _, _, _, err = potential.Handle("/CostumePotentialConnect", request(3, 100)); err != nil { - t.Fatal(err) - } - if hp, err := characters.CurrentHealth(77); err != nil || hp != 0 { - t.Fatal("potential connection revived dead character") - } - if wallet.Snapshot().Gold != 100 || len(items.All()) != 0 { - t.Fatal("free connection charged materials or currency") - } -} diff --git a/go/internal/server/pictorial/permanent_buffs.go b/go/internal/server/pictorial/permanent_buffs.go deleted file mode 100644 index 90d6a7d..0000000 --- a/go/internal/server/pictorial/permanent_buffs.go +++ /dev/null @@ -1,9 +0,0 @@ -package pictorial - -import "bd2server/internal/server/gamedata" - -// AttachPermanentBuffs makes reward-owned account buffs available to the same -// Snapshot used by AllCharRefresh, login and authoritative stat consumers. -func (s *Service) AttachPermanentBuffs(provider func() ([]gamedata.PictorialBuffStat, error)) { - s.permanentBuffs = provider -} diff --git a/go/internal/server/pictorial/permanent_buffs_test.go b/go/internal/server/pictorial/permanent_buffs_test.go deleted file mode 100644 index ee336f6..0000000 --- a/go/internal/server/pictorial/permanent_buffs_test.go +++ /dev/null @@ -1,44 +0,0 @@ -package pictorial - -import ( - "bd2server/internal/server/events" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "testing" -) - -func TestPermanentBuffsRefreshThroughAccountSnapshot(t *testing.T) { - store := stateio.NewMemory() - design := map[uint64]gamedata.PictorialBuffStat{1: {Category: 1, StatType: 9, Value: 0.02}} - rewards, err := events.OpenBuffRewards(store, design) - if err != nil { - t.Fatal(err) - } - service := &Service{Design: &gamedata.PictorialDesign{}, Owned: &ownedState{}} - service.AttachPermanentBuffs(rewards.SnapshotBuffs) - _, before, err := service.Snapshot() - if err != nil || len(before) != 0 { - t.Fatal("buff present before ownership", err) - } - if err := rewards.GrantOnce("mission", []gamedata.Reward{{Type: 63, ID: 1, Count: 1}}); err != nil { - t.Fatal(err) - } - _, after, err := service.Snapshot() - if err != nil || len(after) != 1 || after[0].Value != 0.02 { - t.Fatal("new buff missing from live stats", after, err) - } - code, body, handled, err := service.Handle("/AllCharRefresh", wire.AppendVarint(nil, 1, 1)) - if err != nil || code != 165 || !handled || len(body) == 0 { - t.Fatal("native buff refresh missing", err) - } - restored, err := events.OpenBuffRewards(store, design) - if err != nil { - t.Fatal(err) - } - service.AttachPermanentBuffs(restored.SnapshotBuffs) - _, after, err = service.Snapshot() - if err != nil || len(after) != 1 || after[0].Category != 1 { - t.Fatal("relogin buff lost", after, err) - } -} diff --git a/go/internal/server/pictorial/service_test.go b/go/internal/server/pictorial/service_test.go deleted file mode 100644 index 6326e80..0000000 --- a/go/internal/server/pictorial/service_test.go +++ /dev/null @@ -1,251 +0,0 @@ -package pictorial - -import ( - "bytes" - "encoding/hex" - "encoding/json" - "math" - "os" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/player" - "bd2server/internal/server/wire" -) - -type ownedState struct { - characters []player.Character - costumes []player.Costume - items []player.Item - equipment []player.Equipment - discovered []player.Pictorial -} - -func (s *ownedState) PictorialCharacters() []player.Character { return s.characters } -func (s *ownedState) PictorialCostumes() []player.Costume { return s.costumes } -func (s *ownedState) PictorialItems() []player.Item { return s.items } -func (s *ownedState) PictorialEquipment() []player.Equipment { return s.equipment } -func (s *ownedState) PictorialDiscovered() []player.Pictorial { return s.discovered } - -func TestOfficialPictorialProgression(t *testing.T) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("set BD2_REAL_GAMEDATA to verify installed 2.35.10 GameData") - } - design, err := gamedata.LoadPictorialDesign(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - starter, err := player.Load(filepath.Join("..", "..", "..", "seed", "v2_35_10", "starter_player.json")) - if err != nil { - t.Fatal(err) - } - var world struct { - RewardCharacter player.Character `json:"reward_character"` - RewardCostume player.Costume `json:"reward_costume"` - StoryCharacters []player.Character `json:"story_characters"` - } - data, err := os.ReadFile(filepath.Join("..", "..", "..", "seed", "v2_35_10", "world.json")) - if err != nil { - t.Fatal(err) - } - if err := json.Unmarshal(data, &world); err != nil { - t.Fatal(err) - } - owned := &ownedState{ - characters: append(append([]player.Character(nil), starter.Characters...), world.RewardCharacter), - costumes: append(append([]player.Costume(nil), starter.Costumes...), world.RewardCostume), - items: append([]player.Item(nil), starter.Items...), - discovered: append([]player.Pictorial(nil), starter.Pictorialbook...), - } - owned.items = append(owned.items, player.Item{ID: 2101, Type: 17, Count: 1}, player.Item{ID: 2102, Type: 17, Count: 1}) - service := &Service{Design: design, Owned: owned} - assertSnapshot := func(health, attack float64, equip bool, officialHex string) { - t.Helper() - entries, buffs, err := service.Snapshot() - if err != nil { - t.Fatal(err) - } - if !equip && len(entries) != 18 { - t.Fatalf("initial account pictorial entries=%+v, want 18 official entries", entries) - } - if equip && len(entries) != 20 { - t.Fatalf("quest-28 pictorial entries=%+v, want 20", entries) - } - got := map[uint64]float64{} - for _, buff := range buffs { - got[buff.StatType] = buff.Value - } - if math.Abs(got[2]-health) > 1e-9 || math.Abs(got[4]-attack) > 1e-9 { - t.Fatalf("entries=%+v buffs=%+v, want health=%v attack=%v", entries, buffs, health, attack) - } - foundEquip := false - for _, entry := range entries { - if entry.GroupID == gamedata.PictorialEquipment && entry.ID == 1 { - foundEquip = true - } - } - if foundEquip != equip { - t.Fatalf("equipment pictorial=%v, want=%v", foundEquip, equip) - } - if !equip { - _, bookResponse, handled, err := service.Handle("/PictorialBookInfo", wire.AppendVarint(nil, 1, 41)) - if err != nil || !handled { - t.Fatalf("encode pictorial entries: handled=%v err=%v", handled, err) - } - officialBook, err := hex.DecodeString("0a040801101f0a04080110200a04080110210a04080110220a04080110230a040801103c" + - "0a04080210010a04080210090a0408041001" + - "0a06080510c5ea010a05080510b5100a05080510b610" + - "0a040807101c0a040807103d0a040807103e0a040807103f0a04080710400a0408071074") - if err != nil || !bytes.Equal(bookResponse, officialBook) { - t.Fatalf("pictorial response=%x official=%x err=%v", bookResponse, officialBook, err) - } - } - _, actual, ok, err := service.Handle("/AllCharRefresh", wire.AppendVarint(nil, 1, 42)) - if err != nil || !ok { - t.Fatalf("encode account buffs: handled=%v err=%v", ok, err) - } - official, err := hex.DecodeString(officialHex) - if err != nil || !bytes.Equal(actual, official) { - t.Fatalf("account buffs proto=%x, official=%x err=%v", actual, official, err) - } - } - assertSnapshot(.015, .0096, false, "0a0b080211b81e85eb51b88e3f0a0b080411613255302aa9833f") - owned.equipment = append(owned.equipment, player.Equipment{InvenIndex: 910000001, ID: 10010}) - owned.items = append(owned.items, player.Item{ID: 2103, Type: 17, Count: 1}) - assertSnapshot(.0175, .01, true, "0a0b080211ec51b81e85eb913f0a0b0804117b14ae47e17a843f") - owned.equipment = append(owned.equipment, player.Equipment{InvenIndex: 910000002, ID: 10010}) - assertSnapshot(.0175, .01, true, "0a0b080211ec51b81e85eb913f0a0b0804117b14ae47e17a843f") // same entry does not stack - owned.characters[5].Level = 20 - maxHP, err := service.MaxHealth(owned.characters[5]) - if err != nil || maxHP != 513 { - t.Fatalf("official level-20 character HP=%d err=%v, want 513", maxHP, err) - } - code, response, ok, err := service.Handle("/AllCharRefresh", wire.AppendVarint(nil, 1, 42)) - if err != nil || !ok || code != 165 { - t.Fatalf("AllCharRefresh: code=%d handled=%v err=%v", code, ok, err) - } - var count int - if err := wire.Walk(response, func(field wire.Field) error { - if field.Number == 1 { - count++ - } - return nil - }); err != nil || count != 2 { - t.Fatalf("buff proto: count=%d err=%v", count, err) - } -} - -func TestSharedBuffIDCountsDistinctEntriesButNotDuplicateInventory(t *testing.T) { - owned := &ownedState{items: []player.Item{{ID: 2101, Type: 17, Count: 1}, {ID: 2101, Type: 17, Count: 1}, {ID: 2102, Type: 17, Count: 1}, {ID: 2103, Type: 8, Count: 99}}} - design := &gamedata.PictorialDesign{ - Items: []gamedata.ItemPictorial{{ID: 2101, ItemID: 2101, BuffID: 2002}, {ID: 2102, ItemID: 2102, BuffID: 2002}}, - Buffs: map[uint64]gamedata.PictorialBuff{2002: {ID: 2002, StatType: 2, Value: .0025}}, - } - entries, buffs, err := (&Service{Design: design, Owned: owned}).Snapshot() - if err != nil || len(entries) != 2 || len(buffs) != 1 || buffs[0].Value != .005 { - t.Fatalf("distinct-entries aggregation entries=%v buffs=%v err=%v", entries, buffs, err) - } -} - -func TestGrowthResourceIDCannotMasqueradeAsCollectionItem(t *testing.T) { - owned := &ownedState{items: []player.Item{{ID: 8, Type: 8, Count: 99}}} - design := &gamedata.PictorialDesign{ - Items: []gamedata.ItemPictorial{{ID: 8, ItemID: 8, BuffID: 2002, Category: 0}}, - Buffs: map[uint64]gamedata.PictorialBuff{2002: {ID: 2002, StatType: 2, Value: .0025}}, - } - entries, buffs, err := (&Service{Design: design, Owned: owned}).Snapshot() - if err != nil || len(entries) != 0 || len(buffs) != 0 { - t.Fatalf("resource incorrectly unlocked pictorial entries=%v buffs=%v err=%v", entries, buffs, err) - } -} - -func TestLearnedRecipePictorialAndBuffSurviveSQLiteRestart(t *testing.T) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - t.Skip("set BD2_REAL_GAMEDATA for current recipe collection regression") - } - design, err := gamedata.LoadPictorialDesign(root, "20260923193640") - if err != nil { - t.Fatal(err) - } - // Current CookingPictorialBookTable row 1 points at recipe 101; the - // client EElementType.CookingRecipe is 7, while 14 is an unrelated use item. - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { _ = repo.Close() }() - items, err := player.OpenInventory(repo, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - _, err = items.GrantOnce("unrelated-use-item", []gamedata.BattleReward{{Type: 14, ID: 101, Count: 1}}) - if err != nil { - t.Fatal(err) - } - owned := &ownedState{items: items.All()} - service := &Service{Design: design, Owned: owned} - before, _, err := service.Snapshot() - if err != nil || len(before) != 0 { - t.Fatalf("use item unlocked a recipe collection: %v %v", before, err) - } - _, err = items.GrantOnce("learned-recipe", []gamedata.BattleReward{{Type: 7, ID: 101, Count: 1}}) - if err != nil { - t.Fatal(err) - } - if err = repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - items, err = player.OpenInventory(repo, &player.Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - owned.items = items.All() - entries, buffs, err := service.Snapshot() - if err != nil || len(entries) != 1 || entries[0].GroupID != gamedata.PictorialCooking || entries[0].ID != 1 || len(buffs) != 1 { - t.Fatalf("recipe collection after restart: %v buffs=%v err=%v", entries, buffs, err) - } - for _, route := range []string{"/PictorialBookInfo", "/AllCharRefresh"} { - _, body, handled, err := service.Handle(route, wire.AppendVarint(nil, 1, 1)) - if err != nil || !handled || len(body) == 0 { - t.Fatalf("learned recipe projection %s: %x %v", route, body, err) - } - } -} - -func BenchmarkCharInfoMaxHealthLoadedDesign(b *testing.B) { - root := os.Getenv("BD2_REAL_GAMEDATA") - if root == "" { - b.Skip("BD2_REAL_GAMEDATA not configured") - } - design, err := gamedata.LoadPictorialDesign(root, "20260923193640") - if err != nil { - b.Fatal(err) - } - // A large account resolves every character through the same immutable - // design. Repeating a known valid character isolates the /CharInfo scaling - // cost from account-specific inventory fixtures while preserving its call - // shape (one maximum-health calculation per returned character). - characters := make([]player.Character, 100) - for i := range characters { - characters[i] = player.Character{InvenIndex: uint64(i + 1), ID: 350, Level: 20} - } - service := &Service{Design: design, Owned: &ownedState{characters: characters}} - b.ReportAllocs() - b.ResetTimer() - for b.Loop() { - for _, character := range characters { - if _, err := service.MaxHealth(character); err != nil { - b.Fatal(err) - } - } - } -} diff --git a/go/internal/server/configfile/default.go b/go/internal/server/platform/configfile/default.go similarity index 100% rename from go/internal/server/configfile/default.go rename to go/internal/server/platform/configfile/default.go diff --git a/go/internal/server/lifecycle/gate.go b/go/internal/server/platform/lifecycle/gate.go similarity index 100% rename from go/internal/server/lifecycle/gate.go rename to go/internal/server/platform/lifecycle/gate.go diff --git a/go/internal/server/lifecycle/signals.go b/go/internal/server/platform/lifecycle/signals.go similarity index 100% rename from go/internal/server/lifecycle/signals.go rename to go/internal/server/platform/lifecycle/signals.go diff --git a/go/internal/server/lifecycle/signals_unix.go b/go/internal/server/platform/lifecycle/signals_unix.go similarity index 100% rename from go/internal/server/lifecycle/signals_unix.go rename to go/internal/server/platform/lifecycle/signals_unix.go diff --git a/go/internal/server/lifecycle/signals_windows.go b/go/internal/server/platform/lifecycle/signals_windows.go similarity index 100% rename from go/internal/server/lifecycle/signals_windows.go rename to go/internal/server/platform/lifecycle/signals_windows.go diff --git a/go/internal/server/logging/logging.go b/go/internal/server/platform/logging/logging.go similarity index 100% rename from go/internal/server/logging/logging.go rename to go/internal/server/platform/logging/logging.go diff --git a/go/internal/server/logging/terminal_unix.go b/go/internal/server/platform/logging/terminal_unix.go similarity index 100% rename from go/internal/server/logging/terminal_unix.go rename to go/internal/server/platform/logging/terminal_unix.go diff --git a/go/internal/server/logging/terminal_windows.go b/go/internal/server/platform/logging/terminal_windows.go similarity index 100% rename from go/internal/server/logging/terminal_windows.go rename to go/internal/server/platform/logging/terminal_windows.go diff --git a/go/internal/server/versionconfig/config.go b/go/internal/server/platform/versionconfig/config.go similarity index 100% rename from go/internal/server/versionconfig/config.go rename to go/internal/server/platform/versionconfig/config.go diff --git a/go/internal/server/player/auto_recovery.go b/go/internal/server/player/auto_recovery.go deleted file mode 100644 index 0d704dc..0000000 --- a/go/internal/server/player/auto_recovery.go +++ /dev/null @@ -1,105 +0,0 @@ -package player - -import ( - "bd2server/internal/server/wire" - "fmt" - "math" - "sort" -) - -// AutoRecoveryResult uses Define_AutoReviveDisabledType's protocol values. -type AutoRecoveryResult struct { - Caster, Experience, Catalyst, Disabled uint64 - Characters []Character -} - -// AutoRecover selects a living permanent fatigue-recovery caster. The selected -// caster is preferred; another owned caster can take over when it is fatigued. -func (s *TalentUseService) AutoRecover(seq, caster uint64, targets []uint64) (AutoRecoveryResult, error) { - r := AutoRecoveryResult{Caster: caster, Catalyst: s.wallet.CatalystBalance()} - if s.context == nil { - return r, fmt.Errorf("player: automatic recovery context unavailable") - } - pack, _, battle, err := s.context() - if err != nil { - return r, err - } - if battle { - return r, fmt.Errorf("player: automatic recovery during battle") - } - all := s.characters.RawAll() - sort.SliceStable(all, func(i, j int) bool { - left, right := all[i].InvenIndex == caster, all[j].InvenIndex == caster - if left != right { - return left - } - return all[i].InvenIndex < all[j].InvenIndex - }) - var selected Character - var cost uint64 - for _, c := range all { - if IsStoryCharacter(c) || IsCharmCharacter(c) { - continue - } - meta, ok := s.design.Characters[c.ID] - if !ok || meta.BannedPacks[pack] { - continue - } - rule, ok := s.design.Rules[[2]uint64{meta.Group, c.TalentLevel}] - if !ok || rule.Class != 10 { - continue - } - hp, e := s.characters.CurrentHealth(c.InvenIndex) - if e != nil { - return r, e - } - if hp == 0 { - continue - } - selected = c - if len(targets) > 0 && rule.Catalyst > math.MaxUint64/uint64(len(targets)) { - return r, fmt.Errorf("player: automatic recovery cost overflow") - } - cost = rule.Catalyst * uint64(len(targets)) - break - } - if selected.InvenIndex == 0 { - r.Disabled = 1 - return r, nil - } - r.Caster = selected.InvenIndex - if len(targets) == 0 { - return r, nil - } - if cost > r.Catalyst { - r.Disabled = 2 - return r, nil - } - request := wire.AppendVarint(nil, 1, seq) - request = wire.AppendVarint(request, 2, r.Caster) - for _, id := range targets { - request = wire.AppendVarint(request, 3, id) - } - _, body, _, err := s.healing(request, "auto-recovery") - if err != nil { - return r, err - } - r.Experience, _, err = wire.Varint(body, 2) - if err != nil { - return r, err - } - r.Catalyst = s.wallet.CatalystBalance() - for _, id := range targets { - c, ok := s.characters.Find(id) - if !ok { - return r, fmt.Errorf("player: recovered character unavailable") - } - hp, e := s.characters.CurrentHealth(id) - if e != nil { - return r, e - } - c.HP = hp - r.Characters = append(r.Characters, c) - } - return r, nil -} diff --git a/go/internal/server/player/auto_recovery_test.go b/go/internal/server/player/auto_recovery_test.go deleted file mode 100644 index e044460..0000000 --- a/go/internal/server/player/auto_recovery_test.go +++ /dev/null @@ -1,41 +0,0 @@ -package player - -import ( - "bd2server/internal/server/stateio" - "testing" -) - -func TestAutoRecoveryRequiresLivingCasterAndChargesActualTargets(t *testing.T) { - s, chars, wallet, _ := talentTest(t, stateio.NewMemory(), 10) - chars.characters = append(chars.characters, Character{InvenIndex: 88, ID: 350, HP: 100, Level: 1, TalentLevel: 1}) - if e := chars.SetCurrentHealth(77, 0); e != nil { - t.Fatal(e) - } - r, e := s.AutoRecover(1, 77, []uint64{77}) - if e != nil || r.Caster != 88 || r.Disabled != 0 || len(r.Characters) != 1 || r.Characters[0].HP != 100 || r.Experience != 3 || r.Catalyst != 45 { - t.Fatalf("result%+v err%v", r, e) - } - if wallet.CatalystBalance() != 45 { - t.Fatal("wrong catalyst") - } - if e := chars.SetCurrentHealth(77, 0); e != nil { - t.Fatal(e) - } - if e := chars.SetCurrentHealth(88, 0); e != nil { - t.Fatal(e) - } - r, e = s.AutoRecover(2, 88, []uint64{77}) - if e != nil || r.Disabled != 1 || len(r.Characters) != 0 || wallet.CatalystBalance() != 45 { - t.Fatalf("exhaustion%+v %v", r, e) - } - if e := chars.SetCurrentHealth(88, 100); e != nil { - t.Fatal(e) - } - rule := s.design.Rules[[2]uint64{42, 1}] - rule.Catalyst = 46 - s.design.Rules[[2]uint64{42, 1}] = rule - r, e = s.AutoRecover(3, 88, []uint64{77}) - if e != nil || r.Disabled != 2 || len(r.Characters) != 0 || wallet.CatalystBalance() != 45 { - t.Fatalf("insufficient catalyst%+v %v", r, e) - } -} diff --git a/go/internal/server/player/char_awake.go b/go/internal/server/player/char_awake.go deleted file mode 100644 index e8fc075..0000000 --- a/go/internal/server/player/char_awake.go +++ /dev/null @@ -1,300 +0,0 @@ -package player - -import ( - "errors" - "fmt" - "slices" - "strconv" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -type CharAwakeService struct { - design *gamedata.CharAwakeDesign - collection *CollectionStore - characters *CharacterStore - inventory *Inventory - wallet *Wallet -} - -func NewCharAwakeService(design *gamedata.CharAwakeDesign, collection *CollectionStore, characters *CharacterStore, inventory *Inventory, wallet *Wallet) (*CharAwakeService, error) { - if design == nil || collection == nil || characters == nil || inventory == nil || wallet == nil { - return nil, errors.New("player: incomplete character awakening service") - } - for uniqueID, progress := range collection.CharAwakeStates() { - if err := design.ValidateProgress(uniqueID, progress.ImprintLevels, progress.IsAwake); err != nil { - return nil, fmt.Errorf("player: invalid saved character awakening progress: %w", err) - } - } - return &CharAwakeService{design: design, collection: collection, characters: characters, inventory: inventory, wallet: wallet}, nil -} - -func (s *CharAwakeService) Handle(path string, request []byte) (int, []byte, bool, error) { - switch path { - case "/CharAwakeInfo": - return s.info(request) - case "/CharImprintLevelUp": - return s.imprintLevelUp(request) - case "/CharAwakeActive": - return s.awakeActive(request) - default: - return 0, nil, false, nil - } -} - -func validAwakeSequence(request []byte, path string) error { - seq, found, err := wire.Varint(request, 1) - if err != nil || !found || seq == 0 { - return fmt.Errorf("player: %s missing sequence", path) - } - return nil -} - -func (s *CharAwakeService) info(request []byte) (int, []byte, bool, error) { - if err := validAwakeSequence(request, "CharAwakeInfo"); err != nil { - return 0, nil, true, err - } - states := s.collection.CharAwakeStates() - ids := make([]uint64, 0, len(states)) - for id := range states { - ids = append(ids, id) - } - slices.Sort(ids) - var response []byte - for _, id := range ids { - progress := states[id] - var entry []byte - entry = wire.AppendVarint(entry, 1, id) - for i, level := range progress.ImprintLevels { - if level != 0 { - entry = wire.AppendVarint(entry, 2+i, level) - } - } - if progress.IsAwake { - entry = wire.AppendVarint(entry, 5, 1) - } - response = wire.AppendBytes(response, 1, entry) - } - return 326, response, true, nil -} - -func decodeAwakeMaterials(request []byte, fieldNumber int) ([]Item, error) { - var materials []Item - err := wire.Walk(request, func(field wire.Field) error { - if field.Number != fieldNumber { - return nil - } - if field.Type != 2 { - return errors.New("player: invalid character awakening material field") - } - var item Item - if err := decodeVarints(field.Value, map[int]*uint64{1: &item.InvenIndex, 2: &item.ID, 3: &item.Type, 4: &item.Count, 5: &item.KeepFlag, 6: &item.TimeValue, 9: &item.SortID, 10: &item.UseCount}); err != nil { - return err - } - if item.Type == 0 || item.Count == 0 || (item.Type != 4 && (item.InvenIndex == 0 || item.ID == 0)) { - return errors.New("player: incomplete character awakening material") - } - materials = append(materials, item) - return nil - }) - return materials, err -} - -func (s *CharAwakeService) imprintLevelUp(request []byte) (int, []byte, bool, error) { - if err := validAwakeSequence(request, "CharImprintLevelUp"); err != nil { - return 0, nil, true, err - } - characterIndex, found, err := wire.Varint(request, 2) - if err != nil || !found || characterIndex == 0 { - return 0, nil, true, errors.New("player: CharImprintLevelUp missing character") - } - var targets []gamedata.CharImprintTarget - err = wire.Walk(request, func(field wire.Field) error { - if field.Number != 3 { - return nil - } - if field.Type != 2 { - return errors.New("player: invalid imprint target field") - } - var target gamedata.CharImprintTarget - if err := decodeVarints(field.Value, map[int]*uint64{1: &target.Slot, 2: &target.TargetLevel}); err != nil { - return err - } - if target.Slot == 0 || target.TargetLevel == 0 { - return errors.New("player: incomplete imprint target") - } - targets = append(targets, target) - return nil - }) - if err != nil || len(targets) == 0 { - if err == nil { - err = errors.New("player: CharImprintLevelUp has no targets") - } - return 0, nil, true, err - } - materials, err := decodeAwakeMaterials(request, 4) - if err != nil || len(materials) == 0 { - if err == nil { - err = errors.New("player: CharImprintLevelUp has no materials") - } - return 0, nil, true, err - } - character, found := s.characters.Find(characterIndex) - if !found { - return 0, nil, true, fmt.Errorf("player: unknown imprint character %d", characterIndex) - } - uniqueID, err := s.design.ValidateGrowthCompleted(character.ID, character.Level) - if err != nil { - return 0, nil, true, fmt.Errorf("player: validate imprint character growth: %w", err) - } - current, _ := s.collection.CharAwakeState(uniqueID) - costs, levels, err := s.design.ImprintCosts(uniqueID, current.ImprintLevels, targets) - if err != nil { - return 0, nil, true, fmt.Errorf("player: validate imprint GameData: %w", err) - } - items, gold, err := s.validateCosts(costs, materials) - if err != nil { - return 0, nil, true, fmt.Errorf("player: validate imprint materials: %w", err) - } - if len(items) != 0 { - if err := s.inventory.CanConsume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: validate imprint inventory: %w", err) - } - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - return 0, nil, true, errors.New("player: insufficient gold for character imprint") - } - identity := "char-imprint:" + strconv.FormatUint(uniqueID, 10) - for _, level := range levels { - identity += ":" + strconv.FormatUint(level, 10) //nolint:modernize // stringsbuilder - } - if gold != 0 { - if _, err := s.wallet.SpendGoldOnce(identity, gold); err != nil { - return 0, nil, true, fmt.Errorf("player: spend imprint gold: %w", err) - } - } - if len(items) != 0 { - if err := s.inventory.Consume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: consume imprint materials: %w", err) - } - } - next := current - next.ImprintLevels = levels - if err := s.collection.UpdateCharAwake(uniqueID, current, next); err != nil { - return 0, nil, true, fmt.Errorf("player: persist character imprint: %w", err) - } - return 327, nil, true, nil -} - -func (s *CharAwakeService) awakeActive(request []byte) (int, []byte, bool, error) { - if err := validAwakeSequence(request, "CharAwakeActive"); err != nil { - return 0, nil, true, err - } - characterIndex, found, err := wire.Varint(request, 2) - if err != nil || !found || characterIndex == 0 { - return 0, nil, true, errors.New("player: CharAwakeActive missing character") - } - materials, err := decodeAwakeMaterials(request, 3) - if err != nil || len(materials) == 0 { - if err == nil { - err = errors.New("player: CharAwakeActive has no materials") - } - return 0, nil, true, err - } - character, found := s.characters.Find(characterIndex) - if !found { - return 0, nil, true, fmt.Errorf("player: unknown awakening character %d", characterIndex) - } - uniqueID, err := s.design.ValidateGrowthCompleted(character.ID, character.Level) - if err != nil { - return 0, nil, true, fmt.Errorf("player: validate awakening character growth: %w", err) - } - current, _ := s.collection.CharAwakeState(uniqueID) - costs, err := s.design.AwakeCosts(uniqueID, current.ImprintLevels, current.IsAwake) - if err != nil { - return 0, nil, true, fmt.Errorf("player: validate awakening GameData: %w", err) - } - items, gold, err := s.validateCosts(costs, materials) - if err != nil { - return 0, nil, true, fmt.Errorf("player: validate awakening materials: %w", err) - } - if len(items) != 0 { - if err := s.inventory.CanConsume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: validate awakening inventory: %w", err) - } - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - return 0, nil, true, errors.New("player: insufficient gold for character awakening") - } - identity := "char-awake:" + strconv.FormatUint(uniqueID, 10) - if gold != 0 { - if _, err := s.wallet.SpendGoldOnce(identity, gold); err != nil { - return 0, nil, true, fmt.Errorf("player: spend awakening gold: %w", err) - } - } - if len(items) != 0 { - if err := s.inventory.Consume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: consume awakening materials: %w", err) - } - } - next := current - next.IsAwake = true - if err := s.collection.UpdateCharAwake(uniqueID, current, next); err != nil { - return 0, nil, true, fmt.Errorf("player: persist character awakening: %w", err) - } - return 328, nil, true, nil -} - -func (s *CharAwakeService) validateCosts(costs []gamedata.CharAwakeCost, materials []Item) ([]Item, uint64, error) { - want := make(map[[2]uint64]uint64, len(costs)) - for _, cost := range costs { - key := [2]uint64{cost.Type, cost.ID} - if cost.Count > ^uint64(0)-want[key] { - return nil, 0, errors.New("character awakening cost overflow") - } - want[key] += cost.Count - } - got := make(map[[2]uint64]uint64, len(materials)) - var items []Item - var gold uint64 - for _, material := range materials { - key := [2]uint64{material.Type, material.ID} - switch material.Type { - case 4: - if material.InvenIndex != 0 || material.ID != 0 || gold != 0 { - return nil, 0, errors.New("invalid character awakening currency") - } - gold = material.Count - case 8: - items = append(items, material) - default: - return nil, 0, fmt.Errorf("unsupported character awakening material type %d", material.Type) - } - if material.Count > ^uint64(0)-got[key] { - return nil, 0, errors.New("submitted character awakening material overflow") - } - got[key] += material.Count - } - if len(got) != len(want) { - return nil, 0, fmt.Errorf("material kinds mismatch: request=%v GameData=%v", got, want) - } - for key, count := range want { - if got[key] != count { - return nil, 0, fmt.Errorf("material %d/%d=%d want %d", key[0], key[1], got[key], count) - } - } - return items, gold, nil -} - -// Contributions exposes server-side derived stats for CharInfo, revival and -// battle calculators. The client independently computes the same values from -// CharAwakeInfo, so neither side relies on a persisted derived number. -func (s *CharAwakeService) Contributions(character Character) ([]gamedata.StatContribution, error) { - uniqueID, ok := s.design.CharacterUniqueID(character.ID) - if !ok { - return nil, nil - } - progress, _ := s.collection.CharAwakeState(uniqueID) - return s.design.CharAwakeContributions(uniqueID, progress.ImprintLevels, progress.IsAwake) -} diff --git a/go/internal/server/player/char_awake_test.go b/go/internal/server/player/char_awake_test.go deleted file mode 100644 index a254406..0000000 --- a/go/internal/server/player/char_awake_test.go +++ /dev/null @@ -1,140 +0,0 @@ -package player - -import ( - "path/filepath" - "testing" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -func newCharAwakeHarness(t *testing.T) (*CharAwakeService, *CollectionStore, *Inventory, *Wallet, []Item) { - t.Helper() - dir := t.TempDir() - starter := &Starter{Version: "2.35.10"} - inventory, err := OpenInventory(testStore(filepath.Join(dir, "items.json")), starter) - if err != nil { - t.Fatal(err) - } - granted, err := inventory.GrantOnce("awake-test", []gamedata.BattleReward{ - {Type: 8, ID: 701, Count: 1}, {Type: 8, ID: 702, Count: 1}, {Type: 8, ID: 703, Count: 1}, {Type: 8, ID: 705, Count: 20}, - }) - if err != nil { - t.Fatal(err) - } - wallet, err := OpenWallet(testStore(filepath.Join(dir, "wallet.json")), Currency{Gold: 1000}) - if err != nil { - t.Fatal(err) - } - collection, err := OpenCollectionStore(testStore(filepath.Join(dir, "collection.json")), nil) - if err != nil { - t.Fatal(err) - } - characters := &CharacterStore{characters: []Character{{InvenIndex: 77, ID: 354, Level: 100}}} - character := gamedata.CharAwakeCharacter{UniqueCharID: 35, Active: true} - for i := range 3 { - character.ImprintGrowth[i] = []gamedata.CharAwakeGrowth{{ - ID: uint64(i + 1), StatType: uint64(i*2 + 1), StatValue: float64(i + 1), - Costs: []gamedata.CharAwakeCost{{Type: 8, ID: uint64(701 + i), Count: 1}, {Type: 4, Count: 100}}, - }} - } - character.AwakeGrowth = []gamedata.CharAwakeGrowth{ - {ID: 100, StatType: 4, StatValue: .12, Costs: []gamedata.CharAwakeCost{{Type: 8, ID: 705, Count: 20}, {Type: 4, Count: 500}}}, - {ID: 101, StatType: 14, StatValue: .1}, - } - design := &gamedata.CharAwakeDesign{ - Characters: map[uint64]gamedata.CharAwakeCharacter{35: character}, - Stages: map[uint64]gamedata.CharAwakeCharacterStage{ - 354: {UniqueCharID: 35, Grade: 5, GrowthGrade: 5, MaximumLevel: 100}, - }, - } - service, err := NewCharAwakeService(design, collection, characters, inventory, wallet) - if err != nil { - t.Fatal(err) - } - return service, collection, inventory, wallet, granted -} - -func awakeMaterial(item Item) []byte { return ItemWire(item) } - -func TestCharImprintAndAwakePersistConsumeAndRestoreInfo(t *testing.T) { - service, collection, inventory, wallet, granted := newCharAwakeHarness(t) - request := wire.AppendVarint(nil, 1, 10) - request = wire.AppendVarint(request, 2, 77) - for slot := uint64(1); slot <= 3; slot++ { - target := wire.AppendVarint(wire.AppendVarint(nil, 1, slot), 2, 1) - request = wire.AppendBytes(request, 3, target) - request = wire.AppendBytes(request, 4, awakeMaterial(granted[slot-1])) - } - request = wire.AppendBytes(request, 4, awakeMaterial(Item{Type: 4, Count: 300})) - code, proto, handled, err := service.Handle("/CharImprintLevelUp", request) - if err != nil || !handled || code != 327 || len(proto) != 0 { - t.Fatalf("imprint code=%d proto=%x handled=%v err=%v", code, proto, handled, err) - } - progress, found := collection.CharAwakeState(35) - if !found || progress.ImprintLevels != [3]uint64{1, 1, 1} || progress.IsAwake { - t.Fatalf("imprint progress=%+v found=%v", progress, found) - } - if wallet.Snapshot().Gold != 700 || len(inventory.All()) != 1 { - t.Fatalf("post-imprint wallet=%+v items=%+v", wallet.Snapshot(), inventory.All()) - } - - request = wire.AppendVarint(nil, 1, 11) - request = wire.AppendVarint(request, 2, 77) - request = wire.AppendBytes(request, 3, awakeMaterial(granted[3])) - request = wire.AppendBytes(request, 3, awakeMaterial(Item{Type: 4, Count: 500})) - code, proto, handled, err = service.Handle("/CharAwakeActive", request) - if err != nil || !handled || code != 328 || len(proto) != 0 { - t.Fatalf("awake code=%d proto=%x handled=%v err=%v", code, proto, handled, err) - } - progress, _ = collection.CharAwakeState(35) - if !progress.IsAwake || wallet.Snapshot().Gold != 200 || len(inventory.All()) != 0 { - t.Fatalf("post-awake progress=%+v wallet=%+v items=%+v", progress, wallet.Snapshot(), inventory.All()) - } - restarted, err := OpenCollectionStore(collection.store, nil) - if err != nil { - t.Fatal(err) - } - service.collection = restarted - progress, found = restarted.CharAwakeState(35) - if !found || progress.ImprintLevels != [3]uint64{1, 1, 1} || !progress.IsAwake { - t.Fatalf("restarted awakening progress=%+v found=%v", progress, found) - } - - code, proto, handled, err = service.Handle("/CharAwakeInfo", wire.AppendVarint(nil, 1, 12)) - if err != nil || !handled || code != 326 { - t.Fatalf("info code=%d handled=%v err=%v", code, handled, err) - } - var entry []byte - if err := wire.Walk(proto, func(field wire.Field) error { - if field.Number == 1 { - entry = append([]byte(nil), field.Value...) - } - return nil - }); err != nil { - t.Fatal(err) - } - for field, want := range map[int]uint64{1: 35, 2: 1, 3: 1, 4: 1, 5: 1} { - got, present, err := wire.Varint(entry, field) - if err != nil || !present || got != want { - t.Fatalf("CharAwakeDBInfo field %d=%d present=%v err=%v want=%d proto=%x", field, got, present, err, want, entry) - } - } -} - -func TestCharImprintRejectsClientCostMismatchBeforeMutation(t *testing.T) { - service, collection, inventory, wallet, granted := newCharAwakeHarness(t) - request := wire.AppendVarint(nil, 1, 20) - request = wire.AppendVarint(request, 2, 77) - request = wire.AppendBytes(request, 3, wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 1)) - bad := granted[0] - bad.Count = 2 - request = wire.AppendBytes(request, 4, awakeMaterial(bad)) - request = wire.AppendBytes(request, 4, awakeMaterial(Item{Type: 4, Count: 100})) - if _, _, handled, err := service.Handle("/CharImprintLevelUp", request); !handled || err == nil { - t.Fatalf("mismatched costs handled=%v err=%v", handled, err) - } - if _, found := collection.CharAwakeState(35); found || wallet.Snapshot().Gold != 1000 || len(inventory.All()) != 4 { - t.Fatalf("rejected request mutated state: found=%v wallet=%+v items=%+v", found, wallet.Snapshot(), inventory.All()) - } -} diff --git a/go/internal/server/player/collection_entries_test.go b/go/internal/server/player/collection_entries_test.go deleted file mode 100644 index ac8f6f6..0000000 --- a/go/internal/server/player/collection_entries_test.go +++ /dev/null @@ -1,135 +0,0 @@ -package player - -import ( - "bytes" - "encoding/json" - "reflect" - "testing" - - "bd2server/internal/server/stateio" -) - -type collectionWriteSpy struct { - *stateio.Memory - coreWrites int - mutations [][]stateio.EntryMutation -} - -func (s *collectionWriteSpy) SaveWithEntries(domain string, core []byte, changes []stateio.EntryMutation) error { - if core != nil { - s.coreWrites++ - } - s.mutations = append(s.mutations, append([]stateio.EntryMutation(nil), changes...)) - return s.Memory.SaveWithEntries(domain, core, changes) -} - -func TestCollectionPersistsOnlyChangedEntries(t *testing.T) { - storage := &collectionWriteSpy{Memory: stateio.NewMemory()} - collection, err := OpenCollectionStore(storage, nil) - if err != nil { - t.Fatal(err) - } - if err := collection.EnsurePersisted(); err != nil { - t.Fatal(err) - } - first := cloneCollection(collection.data) - first.Characters = []Character{{InvenIndex: 920000001, ID: 6490, Level: 1}} - first.Costumes = []Costume{{InvenIndex: 930000001, ID: 64901, UseChar: 920000001}} - first.Grants["first"] = CollectionGrant{CharacterIndices: []uint64{920000001}, CostumeIndices: []uint64{930000001}} - first.GachaApplied["draw:first"] = true - if err := collection.commit(first); err != nil { - t.Fatal(err) - } - core, err := storage.Load("collection") - if err != nil { - t.Fatal(err) - } - for _, bucket := range collectionEntryBuckets { - if bytes.Contains(core, []byte(`"`+bucket+`"`)) { - t.Fatalf("collection core still contains %s", bucket) - } - } - if storage.coreWrites != 1 || len(storage.mutations[1]) != 4 { - t.Fatalf("initial entry save: core writes=%d mutations=%+v", storage.coreWrites, storage.mutations) - } - - second := cloneCollection(collection.data) - second.Grants["second"] = CollectionGrant{ViewCostumeIDs: []uint64{64902}} - if err := collection.commit(second); err != nil { - t.Fatal(err) - } - if storage.coreWrites != 1 || len(storage.mutations[2]) != 1 || storage.mutations[2][0].Bucket != "grants" || storage.mutations[2][0].Key != "second" { - t.Fatalf("entry-only update rewrote core or unrelated entries: %+v", storage.mutations[2]) - } - loaded, err := OpenCollectionStore(storage, nil) - if err != nil { - t.Fatal(err) - } - if len(loaded.Characters()) != 1 || len(loaded.Costumes()) != 1 || !loaded.data.GachaApplied["draw:first"] { - t.Fatalf("reloaded collection lost entities or applied draw: %+v", loaded.data) - } - if _, found := loaded.Grant("second"); !found { - t.Fatal("reloaded collection lost second grant") - } -} - -func TestCollectionRejectsInlineLedgerInFinalCore(t *testing.T) { - storage := stateio.NewMemory() - core := map[string]any{ - "version": "2.35.10", "next_character_index": 920000001, - "next_costume_index": 930000001, "grants": map[string]any{}, - } - encoded, err := json.Marshal(core) - if err != nil { - t.Fatal(err) - } - if err := storage.Save("collection", encoded); err != nil { - t.Fatal(err) - } - if _, err := OpenCollectionStore(storage, nil); err == nil { - t.Fatal("old inline collection ledger was accepted") - } -} - -func TestCollectionEntryBucketsRoundTripAndDelete(t *testing.T) { - storage := stateio.NewMemory() - collection, err := OpenCollectionStore(storage, nil) - if err != nil { - t.Fatal(err) - } - next := cloneCollection(collection.data) - next.Characters = []Character{{InvenIndex: 920000001, ID: 6490, Level: 1}} - next.Costumes = []Costume{{InvenIndex: 930000001, ID: 64901, UseChar: 920000001}} - next.Grants["draw:1"] = CollectionGrant{ViewCostumeIDs: []uint64{64901}} - next.GachaApplied["draw:1"] = true - next.GachaUsers["1"] = GachaUserState{GroupID: 1, Point: 20} - next.GachaFixed["1:2"] = GachaFixedState{FixedID: 1, Type: 2, Count: 3} - next.StepUpProgress["1"] = 2 - next.GachaPointExchange["exchange:1"] = GachaPointExchange{GroupID: 1, Count: 3} - next.GachaSelections["1"] = []GachaSelection{{GroupID: 1, Slot: 1, ItemID: 64901}} - next.CostumePotential["930000001"] = []uint64{2, 4} - next.CharAwake["6490"] = CharAwakeProgress{ImprintLevels: [3]uint64{1, 0, 0}} - if err := collection.commit(next); err != nil { - t.Fatal(err) - } - loaded, err := OpenCollectionStore(storage, nil) - if err != nil { - t.Fatal(err) - } - if !reflect.DeepEqual(loaded.data, next) { - t.Fatalf("collection entries changed across reload: loaded=%+v want=%+v", loaded.data, next) - } - removed := cloneCollection(loaded.data) - delete(removed.Grants, "draw:1") - delete(removed.GachaApplied, "draw:1") - removed.Characters = nil - removed.Costumes = nil - if err := loaded.commit(removed); err != nil { - t.Fatal(err) - } - for bucket, key := range map[string]string{"grants": "draw:1", "gacha_applied": "draw:1", "characters": "920000001", "costumes": "930000001"} { - if _, found, err := storage.LoadEntry("collection", bucket, key); err != nil || found { - t.Fatalf("deleted entry %s/%s found=%v err=%v", bucket, key, found, err) - } - } -} diff --git a/go/internal/server/player/collection_promotion_reward_test.go b/go/internal/server/player/collection_promotion_reward_test.go deleted file mode 100644 index 1696307..0000000 --- a/go/internal/server/player/collection_promotion_reward_test.go +++ /dev/null @@ -1,74 +0,0 @@ -package player - -import ( - "path/filepath" - "reflect" - "testing" - - "bd2server/internal/server/gamedata" -) - -func TestNewCostumeReusesPromotedCharacter(t *testing.T) { - for _, inBase := range []bool{false, true} { - t.Run(map[bool]string{false: "collection", true: "base"}[inBase], func(t *testing.T) { - path := filepath.Join(t.TempDir(), "collection.json") - store, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - owned := Character{InvenIndex: 77, ID: 888, HP: 4000, Level: 100, CostumeID: 76541, UseCostume: 123, ConnectPotentialCostume: 76541, TalentLevel: 4} - if inBase { - if err := store.BindBaseCharacters([]Character{owned}); err != nil { - t.Fatal(err) - } - } else { - next := cloneCollection(store.data) - next.Characters = []Character{owned} - if err := store.commit(next); err != nil { - t.Fatal(err) - } - } - catalog, err := gamedata.NewRegularGachaCatalog( - map[uint64]gamedata.RegularGacha{1: {ID: 1, Count: 1, PriceType: 3, Price: 1, Pool: []gamedata.WeightedCostume{{ID: 76543, Weight: 1}}}}, - map[uint64]gamedata.CharacterDesign{76543: {ID: 999, GrowthCharacterIDs: []uint64{999, 888}, HP: 100, CostumeMaxLevel: 5, OverflowItemType: 20, OverflowItemCount: 2}}, - ) - if err != nil { - t.Fatal(err) - } - before := store.data.NextCharacterIndex - grant, err := store.GrantCostumes("new-costume", []uint64{76543}, catalog) - if err != nil { - t.Fatal(err) - } - if len(grant.CharacterIndices) != 0 || len(grant.CostumeIndices) != 1 || store.data.NextCharacterIndex != before { - t.Fatalf("new character created after promotion: %+v", grant) - } - costume, ok := store.CostumeByIndex(grant.CostumeIndices[0]) - if !ok || costume.UseChar != owned.InvenIndex { - t.Fatalf("costume=%+v", costume) - } - loaded, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - if inBase { - if err := loaded.BindBaseCharacters([]Character{owned}); err != nil { - t.Fatal(err) - } - } else if got, ok := loaded.FindCharacter(owned.InvenIndex); !ok || !reflect.DeepEqual(got, owned) { - t.Fatalf("original character changed: %+v", got) - } - retry, err := loaded.GrantCostumes("new-costume", []uint64{76543}, catalog) - if err != nil || !reflect.DeepEqual(retry, grant) || len(loaded.Costumes()) != 1 { - t.Fatalf("retry=%+v err=%v", retry, err) - } - }) - } -} - -func TestCostumeRejectsAmbiguousPromotedOwnership(t *testing.T) { - _, _, err := findCharacterByDesign(nil, []Character{{InvenIndex: 77, ID: 999}, {InvenIndex: 88, ID: 888}}, gamedata.CharacterDesign{ID: 999, GrowthCharacterIDs: []uint64{999, 888}}) - if err == nil { - t.Fatal("duplicate family silently selected") - } -} diff --git a/go/internal/server/player/collection_test.go b/go/internal/server/player/collection_test.go deleted file mode 100644 index 1fd43d5..0000000 --- a/go/internal/server/player/collection_test.go +++ /dev/null @@ -1,229 +0,0 @@ -package player - -import ( - "path/filepath" - "testing" - - "bd2server/internal/server/gamedata" -) - -func TestEarnedQuestCostumeSharesPotentialLedgerAndKeepsSingleOwnedInstance(t *testing.T) { - path := filepath.Join(t.TempDir(), "collection.json") - store, err := OpenCollectionStore(testStore(path), []Costume{{InvenIndex: 123, ID: 60101, UseChar: 44}}) - if err != nil { - t.Fatal(err) - } - reward := Costume{InvenIndex: 609338889, ID: 3501, UseChar: 535607162} - if err := store.AttachRewardCostume(reward); err != nil { - t.Fatal(err) - } - if got := store.Costumes(); len(got) != 2 || got[1].InvenIndex != reward.InvenIndex { - t.Fatalf("earned costume view=%+v", got) - } - if err := store.ActivateCostumePotential(reward.InvenIndex, []uint64{1, 2}); err != nil { - t.Fatal(err) - } - got, found := store.CostumeByIndex(reward.InvenIndex) - if !found || len(got.PotentialIDs) != 2 || got.PotentialIDs[0] != 1 || got.PotentialIDs[1] != 2 { - t.Fatalf("earned costume potential=%+v found=%v", got, found) - } - restarted, err := OpenCollectionStore(testStore(path), []Costume{{InvenIndex: 123, ID: 60101, UseChar: 44}}) - if err != nil { - t.Fatal(err) - } - if err := restarted.AttachRewardCostume(reward); err != nil { - t.Fatal(err) - } - got, found = restarted.CostumeByIndex(reward.InvenIndex) - if !found || len(got.PotentialIDs) != 2 { - t.Fatalf("restarted earned costume potential=%+v found=%v", got, found) - } - if err := restarted.AttachRewardCostume(reward); err == nil { - t.Fatal("duplicate earned quest costume accepted") - } -} - -func TestUpdateCollectionCharacterAcrossPromotionChangesDesignNotInstance(t *testing.T) { - path := filepath.Join(t.TempDir(), "collection.json") - store, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - next := cloneCollection(store.data) - next.Characters = []Character{{InvenIndex: 920000054, ID: 6510, Level: 1}} - if err := store.commit(next); err != nil { - t.Fatal(err) - } - promoted := Character{InvenIndex: 920000054, ID: 6514, Level: 100} - if err := store.CanUpdateCharacter(6510, promoted); err != nil { - t.Fatal(err) - } - if err := store.UpdateCharacter(6510, promoted); err != nil { - t.Fatal(err) - } - loaded, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - if got, found := loaded.FindCharacter(920000054); !found || got.ID != 6514 || got.Level != 100 { - t.Fatalf("persisted promoted collection character=%+v found=%v", got, found) - } - if err := loaded.UpdateCharacter(6510, promoted); err == nil { - t.Fatal("stale old design accepted") - } - if err := loaded.CanUpdateCharacter(6514, Character{InvenIndex: 920000055, ID: 6514, Level: 100}); err == nil { - t.Fatal("unknown instance accepted") - } -} - -func TestMaxCostumeDuplicatePersistsExchangeAndDisplayUpgrade(t *testing.T) { - path := filepath.Join(t.TempDir(), "collection.json") - store, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - next := cloneCollection(store.data) - next.Characters = []Character{{InvenIndex: 920000001, ID: 6490, Level: 1, UseCostume: 930000001, ConnectPotentialCostume: 64901}} - next.Costumes = []Costume{{InvenIndex: 930000001, ID: 64901, Level: 5, UseChar: 920000001}} - if err := store.commit(next); err != nil { - t.Fatal(err) - } - catalog, err := gamedata.NewRegularGachaCatalog( - map[uint64]gamedata.RegularGacha{1: {ID: 1, Count: 1, PriceType: 3, Price: 1, Pool: []gamedata.WeightedCostume{{ID: 64901, Weight: 1}}}}, - map[uint64]gamedata.CharacterDesign{64901: {ID: 6490, HP: 166, CostumeMaxLevel: 5, OverflowItemType: 20, OverflowItemCount: 2}}, - ) - if err != nil { - t.Fatal(err) - } - grant, err := store.GrantRegular("draw", []uint64{64901}, catalog) - if err != nil { - t.Fatal(err) - } - if len(grant.Exchanges) != 1 || len(grant.Upgrades) != 1 { - t.Fatalf("grant=%+v", grant) - } - if got := grant.Upgrades[0]; got.InvenIndex != 930000001 || got.CostumeID != 64901 || got.Before != 5 || got.After != 5 || got.SortID != 0 { - t.Fatalf("display upgrade=%+v", got) - } - if got := store.Costumes()[0].Level; got != 5 { - t.Fatalf("costume level=%d want=5", got) - } - restored, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - persisted, ok := restored.Grant("draw") - if !ok || len(persisted.Exchanges) != 1 || len(persisted.Upgrades) != 1 || restored.Costumes()[0].Level != 5 { - t.Fatalf("persisted=%+v costumes=%+v", persisted, restored.Costumes()) - } -} - -func TestGrantDifferentCostumesForSameCharacterReusesCharacter(t *testing.T) { - store, err := OpenCollectionStore(testStore(filepath.Join(t.TempDir(), "collection.json")), nil) - if err != nil { - t.Fatal(err) - } - catalog, err := gamedata.NewRegularGachaCatalog( - map[uint64]gamedata.RegularGacha{1: {ID: 1, Count: 1, PriceType: 3, Price: 1, Pool: []gamedata.WeightedCostume{{ID: 64901, Weight: 1}}}}, - map[uint64]gamedata.CharacterDesign{ - 64901: {ID: 6490, HP: 166, CostumeMaxLevel: 5}, - 64902: {ID: 6490, HP: 166, CostumeMaxLevel: 5}, - }, - ) - if err != nil { - t.Fatal(err) - } - first, err := store.GrantRegular("first", []uint64{64901}, catalog) - if err != nil { - t.Fatal(err) - } - second, err := store.GrantRegular("second", []uint64{64902}, catalog) - if err != nil { - t.Fatal(err) - } - if len(first.CharacterIndices) != 1 || len(second.CharacterIndices) != 0 || len(store.Characters()) != 1 { - t.Fatalf("first=%+v second=%+v characters=%+v", first, second, store.Characters()) - } - costumes := store.Costumes() - if len(costumes) != 2 || costumes[0].UseChar != costumes[1].UseChar || costumes[0].UseChar != store.Characters()[0].InvenIndex { - t.Fatalf("costumes=%+v characters=%+v", costumes, store.Characters()) - } - if store.Characters()[0].CostumeID != 64901 { - t.Fatalf("initial character costume_id=%d", store.Characters()[0].CostumeID) - } -} - -func TestFirstGachaCompletionIsExplicitAndAtomic(t *testing.T) { - path := filepath.Join(t.TempDir(), "collection.json") - store, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - if store.FirstGachaCompleted() { - t.Fatal("new collection reports completed first gacha") - } - catalog, err := gamedata.NewRegularGachaCatalog( - map[uint64]gamedata.RegularGacha{20: {ID: 20, Count: 1, PriceType: 3, Price: 1, Pool: []gamedata.WeightedCostume{{ID: 64901, Weight: 1}}}}, - map[uint64]gamedata.CharacterDesign{64901: {ID: 6490, HP: 166, CostumeMaxLevel: 5}}, - ) - if err != nil { - t.Fatal(err) - } - _, err = store.GrantRegularPurchase("regular-gacha:20:seq:1", []uint64{64901}, catalog, GachaPurchase{ - Group: gamedata.GachaGroupDesign{ID: 2, GachaSubType: 3}, - }) - if err != nil { - t.Fatal(err) - } - if !store.FirstGachaCompleted() { - t.Fatal("subtype-3 purchase did not persist first-gacha completion") - } - restored, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - if !restored.FirstGachaCompleted() { - t.Fatal("first-gacha completion did not survive restart") - } -} - -func TestFirstGachaCompletionRejectsRewardPayload(t *testing.T) { - path := filepath.Join(t.TempDir(), "collection.json") - store, err := OpenCollectionStore(testStore(path), nil) - if err != nil { - t.Fatal(err) - } - next := cloneCollection(store.data) - next.Grants[FirstGachaCompletedIdentity] = CollectionGrant{ViewCostumeIDs: []uint64{1}} - if err := store.commit(next); err != nil { - t.Fatal(err) - } - if _, err := OpenCollectionStore(testStore(path), nil); err == nil { - t.Fatal("collection accepted a non-empty first-gacha marker") - } -} - -func TestGrantCostumeForBaseCharacterDoesNotCreateCharacter(t *testing.T) { - store, err := OpenCollectionStore(testStore(filepath.Join(t.TempDir(), "collection.json")), nil) - if err != nil { - t.Fatal(err) - } - base := []Character{{InvenIndex: 535604118, ID: 6490, Level: 1, UseCostume: 635604118}} - if err := store.BindBaseCharacters(base); err != nil { - t.Fatal(err) - } - catalog, err := gamedata.NewRegularGachaCatalog( - map[uint64]gamedata.RegularGacha{1: {ID: 1, Count: 1, PriceType: 3, Price: 1, Pool: []gamedata.WeightedCostume{{ID: 64902, Weight: 1}}}}, - map[uint64]gamedata.CharacterDesign{64902: {ID: 6490, HP: 166, CostumeMaxLevel: 5}}, - ) - if err != nil { - t.Fatal(err) - } - grant, err := store.GrantRegular("base-costume", []uint64{64902}, catalog) - if err != nil { - t.Fatal(err) - } - if len(grant.CharacterIndices) != 0 || len(store.Characters()) != 0 || len(store.Costumes()) != 1 || store.Costumes()[0].UseChar != base[0].InvenIndex { - t.Fatalf("grant=%+v characters=%+v costumes=%+v", grant, store.Characters(), store.Costumes()) - } -} diff --git a/go/internal/server/player/commerce_items_test.go b/go/internal/server/player/commerce_items_test.go deleted file mode 100644 index 86ee91e..0000000 --- a/go/internal/server/player/commerce_items_test.go +++ /dev/null @@ -1,28 +0,0 @@ -package player - -import ( - "bd2server/internal/server/stateio" - "testing" -) - -func TestCommerceTimedGrantRenewalAndRetry(t *testing.T) { - inv, err := OpenInventory(stateio.NewMemory(), &Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - first, err := inv.GrantCommerceOnce("a", []Item{{Type: 19, ID: 38, Count: 1, ExpiryTime: 1000}}) - if err != nil || len(first) != 1 { - t.Fatalf("%+v %v", first, err) - } - second, err := inv.GrantCommerceOnce("b", []Item{{Type: 19, ID: 38, Count: 2, ExpiryTime: 2000}}) - if err != nil || len(second) != 1 || second[0].InvenIndex != first[0].InvenIndex || inv.ContentTicketExpiry(38) != 2000 { - t.Fatalf("renewal %+v %v", second, err) - } - retry, err := inv.GrantCommerceOnce("b", nil) - if err != nil || len(retry) != 1 || len(inv.owned.Items) != 1 { - t.Fatalf("retry %+v %v", retry, err) - } - if _, err = inv.GrantCommerceOnce("bad", []Item{{Type: 19, ID: 38, Count: 1}}); err == nil { - t.Fatal("untimed ticket accepted") - } -} diff --git a/go/internal/server/player/content_open_test.go b/go/internal/server/player/content_open_test.go deleted file mode 100644 index 4a82d29..0000000 --- a/go/internal/server/player/content_open_test.go +++ /dev/null @@ -1,261 +0,0 @@ -package player - -import ( - "bytes" - "errors" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" -) - -func contentOpenHarness(t *testing.T, store stateio.Store, prerequisite bool, level uint64) (*ContentOpenService, *Inventory) { - t.Helper() - starter := &Starter{Version: versionconfig.State()} - if prerequisite { - starter.Items = []Item{{InvenIndex: 100, Type: 19, ID: 41, Count: 1}} - } - inventory, err := OpenInventory(store, starter) - if err != nil { - t.Fatal(err) - } - design := &gamedata.ContentOpeningDesign{Prerequisite: gamedata.ContentOpenRule{TicketID: 41, SquadLevel: 3}, Completion: gamedata.ContentOpenRule{TicketID: 42, SquadLevel: 2}} - service, err := NewContentOpenService(design, inventory, store, func() (uint64, error) { return level, nil }) - if err != nil { - t.Fatal(err) - } - service.BeginSession("authenticated-session") - return service, inventory -} - -func contentOpenRequest(seq, kind uint64) []byte { - return wire.AppendVarint(wire.AppendVarint(nil, 1, seq), 2, kind) -} - -func contentOpenTicketCount(inventory *Inventory, id uint64) uint64 { - var count uint64 - for _, item := range inventory.All() { - if item.Type == 19 && item.ID == id { - count += item.Count - } - } - return count -} - -func assertContentOpenResponse(t *testing.T, body []byte) { - t.Helper() - bundle, present, err := wire.Bytes(body, 1) - if err != nil || !present { - t.Fatalf("missing RewardDBInfoBundle: %x %v", body, err) - } - item, present, err := wire.Bytes(bundle, 1) - if err != nil || !present { - t.Fatalf("missing completion ItemDBInfo: %x %v", bundle, err) - } - for field, want := range map[int]uint64{2: 42, 3: 19, 4: 1} { - got, present, err := wire.Varint(item, field) - if err != nil || !present || got != want { - t.Fatalf("ItemDBInfo field %d=%d, want %d", field, got, want) - } - } -} - -func TestContentOpenTicketOnceAcrossSequencesAndSessions(t *testing.T) { - store := stateio.NewMemory() - service, inventory := contentOpenHarness(t, store, true, 3) - request := contentOpenRequest(1, 1) - code, body, handled, err := service.Handle("/ContentOpen", request) - if err != nil || !handled || code != 622 { - t.Fatalf("open: %d %v %v", code, handled, err) - } - assertContentOpenResponse(t, body) - for _, tc := range []struct { - session string - seq uint64 - }{{"authenticated-session", 1}, {"authenticated-session", 2}, {"another-session", 1}} { - service.BeginSession(tc.session) - _, next, _, err := service.Handle("/ContentOpen", contentOpenRequest(tc.seq, 1)) - if err != nil { - t.Fatal(err) - } - if !bytes.Equal(next, body) { - t.Fatal("ticket repair/retry returned a different item") - } - } - if contentOpenTicketCount(inventory, 41) != 1 || contentOpenTicketCount(inventory, 42) != 1 { - t.Fatal("prerequisite consumed or completion duplicated") - } - data, err := store.Load("content_open") - if err != nil || bytes.Contains(data, []byte("authenticated-session")) || bytes.Contains(data, []byte("another-session")) { - t.Fatal("receipt contains session credentials") - } - service.BeginSession("authenticated-session") - changed := wire.AppendVarint(append([]byte(nil), request...), 3, 7) - if _, _, _, err := service.Handle("/ContentOpen", changed); err == nil { - t.Fatal("same sequence accepted different request") - } -} - -func TestContentOpenRequiresSessionAndRepairsOwnedTicket(t *testing.T) { - store := stateio.NewMemory() - service, inventory := contentOpenHarness(t, store, false, 0) - service.BeginSession("") - if _, _, _, err := service.Handle("/ContentOpen", contentOpenRequest(1, 1)); err == nil { - t.Fatal("missing authenticated session accepted") - } - if _, err := inventory.GrantOnce("existing-completion", []gamedata.BattleReward{{ID: 42, Type: 19, Count: 1}}); err != nil { - t.Fatal(err) - } - service.BeginSession("repair-session") - _, body, _, err := service.Handle("/ContentOpen", contentOpenRequest(1, 1)) - if err != nil { - t.Fatal(err) - } - assertContentOpenResponse(t, body) - if contentOpenTicketCount(inventory, 42) != 1 { - t.Fatal("repair duplicated owned completion") - } -} - -func TestContentOpenRejectsCorruptReceiptAtOpenAndAfterConstruction(t *testing.T) { - store := stateio.NewMemory() - service, inventory := contentOpenHarness(t, store, true, 3) - if err := store.Save("content_open", []byte(`{"receipts":{"credentials:1":{"digest":"bad","body":"CgA="}}}`)); err != nil { - t.Fatal(err) - } - if _, _, _, err := service.Handle("/ContentOpen", contentOpenRequest(1, 1)); err == nil { - t.Fatal("corrupt reloaded receipt accepted") - } - if contentOpenTicketCount(inventory, 42) != 0 { - t.Fatal("corrupt receipts granted completion") - } - if _, err := NewContentOpenService(service.design, inventory, store, service.squadLevel); err == nil { - t.Fatal("corrupt receipt accepted during open") - } -} - -func TestContentOpenRejectsUnqualifiedAndMalformedRequests(t *testing.T) { - for _, tc := range []struct { - name string - prerequisite bool - level uint64 - request []byte - }{ - {"missing-prerequisite", false, 3, contentOpenRequest(1, 1)}, - {"level", true, 2, contentOpenRequest(1, 1)}, - {"unknown", true, 3, contentOpenRequest(1, 2)}, - {"zero-type", true, 3, contentOpenRequest(1, 0)}, - {"group-is-not-type", true, 3, contentOpenRequest(1, 23)}, - {"zero-sequence", true, 3, contentOpenRequest(0, 1)}, - {"wrong-wire", true, 3, wire.AppendBytes(wire.AppendVarint(nil, 1, 1), 2, []byte{1})}, - {"truncated-trailing-wire", true, 3, append(contentOpenRequest(1, 1), 0x1a, 0x02, 0x01)}, - {"duplicate-type", true, 3, wire.AppendVarint(contentOpenRequest(1, 1), 2, 1)}, - } { - t.Run(tc.name, func(t *testing.T) { - store := stateio.NewMemory() - service, inventory := contentOpenHarness(t, store, tc.prerequisite, tc.level) - _, _, handled, err := service.Handle("/ContentOpen", tc.request) - if err == nil || !handled { - t.Fatal("invalid request accepted") - } - if contentOpenTicketCount(inventory, 42) != 0 { - t.Fatal("rejected request granted ticket") - } - state, err := store.Load("content_open") - if err != nil || state != nil { - t.Fatal("rejected request persisted receipt") - } - }) - } -} - -type contentOpenFailStore struct { - stateio.AtomicEntryStore - fail bool -} - -func (s *contentOpenFailStore) Save(name string, payload []byte) error { - if name == "content_open" && s.fail { - return errors.New("injected receipt write failure") - } - return s.AtomicEntryStore.Save(name, payload) -} - -func TestContentOpenSQLiteReceiptRestartAndRollback(t *testing.T) { - for _, fail := range []bool{false, true} { - t.Run(map[bool]string{false: "durable-exact-retry", true: "failed-receipt-rolls-back-ticket"}[fail], func(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - store := &contentOpenFailStore{AtomicEntryStore: repo, fail: fail} - service, _ := contentOpenHarness(t, store, true, 3) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - request := contentOpenRequest(1, 1) - _, body, _, err := service.Handle("/ContentOpen", request) - if fail { - if err == nil { - t.Fatal("receipt failure ignored") - } - if err := op.Rollback(); err == nil { - t.Fatal("dirty rollback did not request account recovery") - } - } else { - if err != nil { - t.Fatal(err) - } - if err := op.Commit(); err != nil { - t.Fatal(err) - } - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - service, inventory := contentOpenHarness(t, repo, true, 3) - if fail { - if contentOpenTicketCount(inventory, 42) != 0 { - t.Fatal("uncommitted completion survived restart") - } - data, err := repo.Load("content_open") - if err != nil || data != nil { - t.Fatal("failed receipt survived restart") - } - } - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - _, replay, _, err := service.Handle("/ContentOpen", request) - if err != nil { - t.Fatal(err) - } - if err := op.Commit(); err != nil { - t.Fatal(err) - } - if !fail && !bytes.Equal(body, replay) { - t.Fatal("exact retry changed after restart") - } - assertContentOpenResponse(t, replay) - if contentOpenTicketCount(inventory, 42) != 1 || contentOpenTicketCount(inventory, 41) != 1 { - t.Fatal("ticket ownership not preserved exactly once") - } - }) - } -} diff --git a/go/internal/server/player/costume_burst.go b/go/internal/server/player/costume_burst.go deleted file mode 100644 index d3b38c3..0000000 --- a/go/internal/server/player/costume_burst.go +++ /dev/null @@ -1,206 +0,0 @@ -package player - -import ( - "crypto/sha256" - "encoding/hex" - "errors" - "fmt" - "math" - "strconv" - "sync" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -const costumeBurstPacketCode = 578 - -type costumeBurstReply struct { - Digest string - Code int - Body []byte -} - -// CostumeBurstService applies one adjacent costume-burst level from the -// authoritative GameData row. The account request transaction makes the -// wallet, inventory, and collection writes one atomic SQLite operation. -type CostumeBurstService struct { - mu sync.Mutex - design *gamedata.CostumeBurstDesign - collection *CollectionStore - inventory *Inventory - wallet *Wallet - sessionID string - replies map[string]map[uint64]costumeBurstReply -} - -func NewCostumeBurstService(design *gamedata.CostumeBurstDesign, collection *CollectionStore, inventory *Inventory, wallet *Wallet) (*CostumeBurstService, error) { - if design == nil || collection == nil || inventory == nil || wallet == nil { - return nil, errors.New("player: incomplete costume burst service") - } - return &CostumeBurstService{ - design: design, collection: collection, inventory: inventory, wallet: wallet, - replies: make(map[string]map[uint64]costumeBurstReply), - }, nil -} - -// BeginSession scopes protobuf sequence replay without forgetting older live -// sessions. Session dispatch calls it before each authenticated request. -func (s *CostumeBurstService) BeginSession(id string) { - s.mu.Lock() - defer s.mu.Unlock() - if id == "" || id == s.sessionID { - return - } - s.sessionID = id - if s.replies[id] != nil { - return - } - if len(s.replies) >= 1024 { - for oldID := range s.replies { - if oldID != id { - delete(s.replies, oldID) - break - } - } - } - s.replies[id] = make(map[uint64]costumeBurstReply) -} - -func (s *CostumeBurstService) Handle(path string, request []byte) (int, []byte, bool, error) { - if path != "/CostumeBurst" { - return 0, nil, false, nil - } - seq, found, err := wire.Varint(request, 1) - if err != nil || !found || seq == 0 || seq > math.MaxInt32 { - return 0, nil, true, errors.New("player: CostumeBurst missing sequence") - } - costumeID, found, err := wire.Varint(request, 2) - if err != nil || !found || costumeID == 0 || costumeID > math.MaxInt32 { - return 0, nil, true, errors.New("player: CostumeBurst missing costume") - } - materials, err := equipmentRequestItems(request, 3, "CostumeBurst") - if err != nil { - return 0, nil, true, err - } - for _, material := range materials { - if material.InvenIndex > math.MaxInt64 || material.ID > math.MaxInt32 || material.Type > math.MaxInt32 || material.Count > math.MaxInt32 || - material.KeepFlag > math.MaxInt32 || material.TimeValue > math.MaxInt64 || material.ExpiryTime > math.MaxInt64 || - material.SortID > math.MaxInt32 || material.UseCount > math.MaxInt32 { - return 0, nil, true, errors.New("player: CostumeBurst material exceeds protocol range") - } - } - - digestBytes := sha256.Sum256(request) - digest := hex.EncodeToString(digestBytes[:]) - s.mu.Lock() - defer s.mu.Unlock() - sessionID := s.sessionID - if sessionID == "" { - sessionID = "__direct__" - } - if s.replies[sessionID] == nil { - s.replies[sessionID] = make(map[uint64]costumeBurstReply) - } - if reply, ok := s.replies[sessionID][seq]; ok { - if reply.Digest != digest { - return 0, nil, true, errors.New("player: CostumeBurst sequence reused with different request") - } - return reply.Code, append([]byte(nil), reply.Body...), true, nil - } - - costume, found := s.collection.CostumeByID(costumeID) - if !found { - return 0, nil, true, fmt.Errorf("player: CostumeBurst costume %d is not owned", costumeID) - } - if costume.BurstLevel != 0 { - if applied, ok := s.collection.CostumeBurstReplay(costume.InvenIndex, costume.BurstLevel); ok && applied.Digest == digest { - reply := costumeBurstReply{Digest: digest, Code: applied.Code, Body: append([]byte(nil), applied.Body...)} - s.replies[sessionID][seq] = reply - return reply.Code, append([]byte(nil), reply.Body...), true, nil - } - } - rule, err := s.design.UpgradeRule(costume.ID, costume.BurstLevel) - if err != nil { - return 0, nil, true, fmt.Errorf("player: resolve CostumeBurst rule: %w", err) - } - items, gold, err := validateCostumeBurstMaterials(rule.Costs, materials) - if err != nil { - return 0, nil, true, err - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - return 0, nil, true, errors.New("player: insufficient gold for CostumeBurst") - } - if len(items) != 0 { - if err := s.inventory.CanConsume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: validate CostumeBurst items: %w", err) - } - } - - identity := "costume-burst:" + strconv.FormatUint(costume.InvenIndex, 10) + ":" + strconv.FormatUint(rule.NextLevel, 10) - if gold != 0 { - if _, err := s.wallet.SpendGoldOnce(identity, gold); err != nil { - return 0, nil, true, fmt.Errorf("player: consume CostumeBurst gold: %w", err) - } - } - if len(items) != 0 { - if err := s.inventory.Consume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: consume CostumeBurst items: %w", err) - } - } - body := wire.AppendVarint(nil, 1, rule.NextLevel) - record := CostumeBurstUpgradeRecord{CostumeID: costume.ID, Level: rule.NextLevel, Digest: digest, Code: costumeBurstPacketCode, Body: append([]byte(nil), body...)} - if err := s.collection.ApplyCostumeBurst(costume.InvenIndex, costume.BurstLevel, rule.NextLevel, record); err != nil { - return 0, nil, true, fmt.Errorf("player: persist CostumeBurst: %w", err) - } - reply := costumeBurstReply{Digest: digest, Code: costumeBurstPacketCode, Body: append([]byte(nil), body...)} - s.replies[sessionID][seq] = reply - return reply.Code, append([]byte(nil), reply.Body...), true, nil -} - -func validateCostumeBurstMaterials(costs []gamedata.PromotionCost, materials []Item) ([]Item, uint64, error) { - if len(costs) == 0 || len(materials) == 0 { - return nil, 0, errors.New("player: CostumeBurst has no material") - } - want := make(map[[2]uint64]uint64, len(costs)) - for _, cost := range costs { - if cost.Count == 0 || (cost.Type == 4 && cost.ID != 0) || (cost.Type == 8 && cost.ID == 0) || (cost.Type != 4 && cost.Type != 8) { - return nil, 0, errors.New("player: invalid CostumeBurst GameData cost") - } - key := [2]uint64{cost.Type, cost.ID} - if want[key] > math.MaxUint64-cost.Count { - return nil, 0, errors.New("player: CostumeBurst GameData cost overflow") - } - want[key] += cost.Count - } - got := make(map[[2]uint64]uint64, len(materials)) - items := make([]Item, 0, len(materials)) - var gold uint64 - for _, material := range materials { - if material.Type != 4 && material.Type != 8 { - return nil, 0, fmt.Errorf("player: unsupported CostumeBurst material type %d", material.Type) - } - key := [2]uint64{material.Type, material.ID} - if got[key] > math.MaxUint64-material.Count { - return nil, 0, errors.New("player: CostumeBurst submitted material overflow") - } - got[key] += material.Count - if material.Type == 4 { - if material.InvenIndex != 0 || material.ID != 0 || gold != 0 { - return nil, 0, errors.New("player: invalid CostumeBurst currency") - } - gold = material.Count - } else { - items = append(items, material) - } - } - if len(got) != len(want) { - return nil, 0, errors.New("player: CostumeBurst material kinds mismatch") - } - for key, count := range want { - if got[key] != count { - return nil, 0, fmt.Errorf("player: CostumeBurst material %d/%d=%d want %d", key[0], key[1], got[key], count) - } - } - return items, gold, nil -} diff --git a/go/internal/server/player/costume_burst_collection_test.go b/go/internal/server/player/costume_burst_collection_test.go deleted file mode 100644 index d498889..0000000 --- a/go/internal/server/player/costume_burst_collection_test.go +++ /dev/null @@ -1,185 +0,0 @@ -package player - -import ( - "encoding/json" - "strings" - "testing" - - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func testCostumeBurstRecord(costumeID, level uint64) CostumeBurstUpgradeRecord { - return CostumeBurstUpgradeRecord{ - CostumeID: costumeID, - Level: level, - Digest: strings.Repeat("ab", 32), - Code: 578, - Body: wire.AppendVarint(nil, 1, level), - } -} - -func TestCollectionCostumeBurstPersistsOverlayAndReplayForEveryCostumeSource(t *testing.T) { - storage := &collectionWriteSpy{Memory: stateio.NewMemory()} - base := []Costume{{InvenIndex: 11, ID: 101, BurstLevel: 0}} - collection, err := OpenCollectionStore(storage, base) - if err != nil { - t.Fatal(err) - } - if err := collection.EnsurePersisted(); err != nil { - t.Fatal(err) - } - next := cloneCollection(collection.data) - next.Costumes = []Costume{{InvenIndex: 22, ID: 202}} - if err := collection.commit(next); err != nil { - t.Fatal(err) - } - - baseRecord := testCostumeBurstRecord(101, 1) - before := len(storage.mutations) - if err := collection.ApplyCostumeBurst(11, 0, 1, baseRecord); err != nil { - t.Fatal(err) - } - if len(storage.mutations) != before+1 || len(storage.mutations[before]) != 2 { - t.Fatalf("burst level and replay were not one collection write: %+v", storage.mutations[before:]) - } - if storage.mutations[before][0].Bucket != "costume_burst_levels" || storage.mutations[before][1].Bucket != "costume_burst_upgrades" { - t.Fatalf("unexpected burst mutations: %+v", storage.mutations[before]) - } - if err := collection.ApplyCostumeBurst(22, 0, 1, testCostumeBurstRecord(202, 1)); err != nil { - t.Fatal(err) - } - if err := collection.ApplyCostumeBurst(22, 1, 2, testCostumeBurstRecord(202, 2)); err != nil { - t.Fatal(err) - } - - if got, found := collection.CostumeByID(101); !found || got.InvenIndex != 11 || got.BurstLevel != 1 { - t.Fatalf("base costume overlay=%+v found=%v", got, found) - } - if got, found := collection.CostumeByIndex(22); !found || got.ID != 202 || got.BurstLevel != 2 { - t.Fatalf("collection costume overlay=%+v found=%v", got, found) - } - replay, found := collection.CostumeBurstReplay(22, 2) - if !found || replay.CostumeID != 202 || replay.Level != 2 || replay.Code != 578 { - t.Fatalf("replay=%+v found=%v", replay, found) - } - replay.Body[0] = 0 - if again, _ := collection.CostumeBurstReplay(22, 2); len(again.Body) == 0 || again.Body[0] == 0 { - t.Fatal("CostumeBurstReplay returned an aliased response body") - } - - core, err := storage.Load("collection") - if err != nil { - t.Fatal(err) - } - if strings.Contains(string(core), "costume_burst") { - t.Fatalf("burst entries leaked into collection core: %s", core) - } - restarted, err := OpenCollectionStore(storage, base) - if err != nil { - t.Fatal(err) - } - if got, _ := restarted.CostumeByIndex(11); got.BurstLevel != 1 { - t.Fatalf("restarted base burst level=%d", got.BurstLevel) - } - if got, _ := restarted.CostumeByID(202); got.BurstLevel != 2 { - t.Fatalf("restarted collection burst level=%d", got.BurstLevel) - } - if got, ok := restarted.CostumeBurstReplay(22, 1); !ok || got.Level != 1 { - t.Fatalf("restarted first replay=%+v found=%v", got, ok) - } -} - -func TestCollectionCostumeBurstCASAndRecordValidation(t *testing.T) { - collection, err := OpenCollectionStore(stateio.NewMemory(), []Costume{{InvenIndex: 11, ID: 101}}) - if err != nil { - t.Fatal(err) - } - record := testCostumeBurstRecord(101, 1) - if err := collection.ApplyCostumeBurst(11, 0, 1, record); err != nil { - t.Fatal(err) - } - for name, call := range map[string]func() error{ - "stale current": func() error { return collection.ApplyCostumeBurst(11, 0, 1, record) }, - "skipped level": func() error { return collection.ApplyCostumeBurst(11, 1, 3, testCostumeBurstRecord(101, 3)) }, - "wrong costume": func() error { return collection.ApplyCostumeBurst(11, 1, 2, testCostumeBurstRecord(102, 2)) }, - "unknown instance": func() error { return collection.ApplyCostumeBurst(99, 0, 1, record) }, - } { - t.Run(name, func(t *testing.T) { - if err := call(); err == nil { - t.Fatal("invalid costume burst transition was accepted") - } - }) - } - if got, _ := collection.CostumeByIndex(11); got.BurstLevel != 1 { - t.Fatalf("rejected transitions mutated level to %d", got.BurstLevel) - } - if _, found := collection.CostumeBurstReplay(11, 2); found { - t.Fatal("rejected transition wrote a replay") - } -} - -func TestCollectionCostumeByIDRejectsAmbiguousOwnership(t *testing.T) { - collection, err := OpenCollectionStore(stateio.NewMemory(), []Costume{{InvenIndex: 11, ID: 101}, {InvenIndex: 12, ID: 101}}) - if err != nil { - t.Fatal(err) - } - if costume, found := collection.CostumeByID(101); found || costume.InvenIndex != 0 || costume.ID != 0 { - t.Fatalf("ambiguous logical ID resolved to %+v", costume) - } -} - -func TestCollectionRejectsMalformedCostumeBurstEntries(t *testing.T) { - validRecord, err := json.Marshal(testCostumeBurstRecord(101, 1)) - if err != nil { - t.Fatal(err) - } - tests := []struct { - name string - bucket string - key string - payload []byte - seed Costume - prepare func(*stateio.Memory) - }{ - {name: "noncanonical level key", bucket: "costume_burst_levels", key: "011", payload: []byte("1"), seed: Costume{InvenIndex: 11, ID: 101}}, - {name: "orphan level", bucket: "costume_burst_levels", key: "99", payload: []byte("1"), seed: Costume{InvenIndex: 11, ID: 101}}, - {name: "zero level", bucket: "costume_burst_levels", key: "11", payload: []byte("0"), seed: Costume{InvenIndex: 11, ID: 101}}, - {name: "level decreases seed", bucket: "costume_burst_levels", key: "11", payload: []byte("1"), seed: Costume{InvenIndex: 11, ID: 101, BurstLevel: 2}}, - {name: "noncanonical replay key", bucket: "costume_burst_upgrades", key: "011:1", payload: validRecord, seed: Costume{InvenIndex: 11, ID: 101}, prepare: func(s *stateio.Memory) { _ = s.PutEntry("collection", "costume_burst_levels", "11", []byte("1")) }}, - {name: "replay above current", bucket: "costume_burst_upgrades", key: "11:1", payload: validRecord, seed: Costume{InvenIndex: 11, ID: 101}}, - {name: "invalid replay digest", bucket: "costume_burst_upgrades", key: "11:1", payload: func() []byte { - r := testCostumeBurstRecord(101, 1) - r.Digest = strings.Repeat("zz", 32) - b, _ := json.Marshal(r) - return b - }(), seed: Costume{InvenIndex: 11, ID: 101}, prepare: func(s *stateio.Memory) { _ = s.PutEntry("collection", "costume_burst_levels", "11", []byte("1")) }}, - {name: "mismatched response", bucket: "costume_burst_upgrades", key: "11:1", payload: func() []byte { - r := testCostumeBurstRecord(101, 1) - r.Body = wire.AppendVarint(nil, 1, 2) - b, _ := json.Marshal(r) - return b - }(), seed: Costume{InvenIndex: 11, ID: 101}, prepare: func(s *stateio.Memory) { _ = s.PutEntry("collection", "costume_burst_levels", "11", []byte("1")) }}, - } - for _, test := range tests { - t.Run(test.name, func(t *testing.T) { - storage := stateio.NewMemory() - collection, err := OpenCollectionStore(storage, []Costume{test.seed}) - if err != nil { - t.Fatal(err) - } - if err := collection.EnsurePersisted(); err != nil { - t.Fatal(err) - } - if test.prepare != nil { - test.prepare(storage) - } - if err := storage.PutEntry("collection", test.bucket, test.key, test.payload); err != nil { - t.Fatal(err) - } - if _, err := OpenCollectionStore(storage, []Costume{test.seed}); err == nil { - t.Fatal("malformed costume burst storage was accepted") - } - }) - } -} diff --git a/go/internal/server/player/costume_burst_test.go b/go/internal/server/player/costume_burst_test.go deleted file mode 100644 index 263709e..0000000 --- a/go/internal/server/player/costume_burst_test.go +++ /dev/null @@ -1,260 +0,0 @@ -package player - -import ( - "math" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/wire" -) - -func testCostumeBurstDesign() *gamedata.CostumeBurstDesign { - return &gamedata.CostumeBurstDesign{Levels: map[uint64]map[uint64]gamedata.CostumeBurstLevel{ - 4202: { - 1: {CostumeID: 4202, Level: 1, Costs: []gamedata.PromotionCost{{Type: 4, Count: 100}, {Type: 8, ID: 710, Count: 2}}}, - 2: {CostumeID: 4202, Level: 2, Costs: []gamedata.PromotionCost{{Type: 4, Count: 200}, {Type: 8, ID: 710, Count: 2}}}, - }, - }} -} - -func costumeBurstRequest(seq, costumeID uint64, materials ...Item) []byte { - request := wire.AppendVarint(wire.AppendVarint(nil, 1, seq), 2, costumeID) - for _, material := range materials { - request = wire.AppendBytes(request, 3, ItemWire(material)) - } - return request -} - -func newCostumeBurstFixture(t *testing.T, storePath string, stacks []gamedata.BattleReward) (*CostumeBurstService, *CollectionStore, *Inventory, *Wallet, []Item) { - t.Helper() - store := testStore(storePath) - starter := &Starter{Version: "2.35.10"} - inventory, err := OpenInventory(store, starter) - if err != nil { - t.Fatal(err) - } - items, err := inventory.GrantOnce("costume-burst-materials", stacks) - if err != nil { - t.Fatal(err) - } - wallet, err := OpenWallet(store, Currency{Gold: 1000}) - if err != nil { - t.Fatal(err) - } - collection, err := OpenCollectionStore(store, []Costume{{InvenIndex: 88, ID: 4202, DesignID: 9911}}) - if err != nil { - t.Fatal(err) - } - service, err := NewCostumeBurstService(testCostumeBurstDesign(), collection, inventory, wallet) - if err != nil { - t.Fatal(err) - } - return service, collection, inventory, wallet, items -} - -func TestCostumeBurstConsumesExactSplitCostsPersistsAndReplays(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.json") - service, collection, inventory, wallet, stacks := newCostumeBurstFixture(t, path, []gamedata.BattleReward{ - {Type: 8, ID: 710, Count: 1}, - {Type: 8, ID: 710, Count: 1}, - }) - service.BeginSession("login-a") - request := costumeBurstRequest(7, 4202, stacks[1], Item{Type: 4, Count: 100}, stacks[0]) - code, body, handled, err := service.Handle("/CostumeBurst", request) - if err != nil || !handled || code != costumeBurstPacketCode { - t.Fatalf("CostumeBurst code=%d handled=%v body=%x err=%v", code, handled, body, err) - } - level, found, err := wire.Varint(body, 1) - if err != nil || !found || level != 1 { - t.Fatalf("response level=%d found=%v err=%v body=%x", level, found, err, body) - } - got, found := collection.CostumeByID(4202) - if !found || got.InvenIndex != 88 || got.DesignID != 9911 || got.BurstLevel != 1 { - t.Fatalf("upgraded costume=%+v found=%v", got, found) - } - if wallet.Snapshot().Gold != 900 { - t.Fatalf("gold=%d", wallet.Snapshot().Gold) - } - for _, stack := range stacks { - if err := inventory.CanConsume([]Item{stack}); err == nil { - t.Fatalf("material stack %d survived", stack.InvenIndex) - } - } - - code2, body2, handled, err := service.Handle("/CostumeBurst", request) - if err != nil || !handled || code2 != code || string(body2) != string(body) { - t.Fatalf("same-session replay code=%d handled=%v body=%x err=%v", code2, handled, body2, err) - } - if wallet.Snapshot().Gold != 900 || collection.Costumes()[0].BurstLevel != 1 { - t.Fatal("same-session replay mutated state") - } - - store := testStore(path) - reloadedInventory, err := OpenInventory(store, &Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - reloadedWallet, err := OpenWallet(store, Currency{}) - if err != nil { - t.Fatal(err) - } - reloadedCollection, err := OpenCollectionStore(store, []Costume{{InvenIndex: 88, ID: 4202, DesignID: 9911}}) - if err != nil { - t.Fatal(err) - } - restarted, err := NewCostumeBurstService(testCostumeBurstDesign(), reloadedCollection, reloadedInventory, reloadedWallet) - if err != nil { - t.Fatal(err) - } - restarted.BeginSession("after-restart") - code2, body2, handled, err = restarted.Handle("/CostumeBurst", request) - if err != nil || !handled || code2 != code || string(body2) != string(body) { - t.Fatalf("durable replay code=%d handled=%v body=%x err=%v", code2, handled, body2, err) - } - if reloadedWallet.Snapshot().Gold != 900 || reloadedCollection.Costumes()[0].BurstLevel != 1 { - t.Fatal("durable replay charged or advanced again") - } -} - -func TestCostumeBurstRequiresLogicalCostumeIDAndProtectsSessionSequence(t *testing.T) { - service, collection, inventory, wallet, stacks := newCostumeBurstFixture(t, filepath.Join(t.TempDir(), "state.json"), []gamedata.BattleReward{{Type: 8, ID: 710, Count: 4}}) - material := stacks[0] - material.Count = 2 - for _, wrong := range []uint64{88, 9911} { - request := costumeBurstRequest(wrong, wrong, material, Item{Type: 4, Count: 100}) - if _, _, handled, err := service.Handle("/CostumeBurst", request); err == nil || !handled { - t.Fatalf("non-logical costume ID %d accepted: handled=%v err=%v", wrong, handled, err) - } - } - service.BeginSession("login-a") - request := costumeBurstRequest(9, 4202, material, Item{Type: 4, Count: 100}) - if _, _, _, err := service.Handle("/CostumeBurst", request); err != nil { - t.Fatal(err) - } - different := costumeBurstRequest(9, 4202, material, Item{Type: 4, Count: 200}) - if _, _, handled, err := service.Handle("/CostumeBurst", different); err == nil || !handled { - t.Fatalf("sequence reuse accepted: handled=%v err=%v", handled, err) - } - if collection.Costumes()[0].BurstLevel != 1 || wallet.Snapshot().Gold != 900 { - t.Fatal("rejected sequence reuse mutated state") - } - remaining := material - remaining.Count = 2 - if err := inventory.CanConsume([]Item{remaining}); err != nil { - t.Fatalf("rejected request consumed remaining items: %v", err) - } -} - -func TestCostumeBurstRejectsInvalidMaterialsAndProtocolRanges(t *testing.T) { - tests := []struct { - name string - materials func(Item) []Item - }{ - {"short", func(stack Item) []Item { stack.Count = 1; return []Item{stack, {Type: 4, Count: 100}} }}, - {"extra", func(stack Item) []Item { - stack.Count = 2 - return []Item{stack, {Type: 4, Count: 100}, {InvenIndex: 900, Type: 8, ID: 711, Count: 1}} - }}, - {"wrong-gold", func(stack Item) []Item { stack.Count = 2; return []Item{stack, {Type: 4, Count: 99}} }}, - {"duplicate-overdraw", func(stack Item) []Item { stack.Count = 1; return []Item{stack, stack, {Type: 4, Count: 100}} }}, - {"range", func(stack Item) []Item { - stack.Count = 2 - stack.ExpiryTime = math.MaxInt64 + 1 - return []Item{stack, {Type: 4, Count: 100}} - }}, - } - for _, test := range tests { - t.Run(test.name, func(t *testing.T) { - stackCount := uint64(2) - if test.name == "duplicate-overdraw" { - stackCount = 1 - } - service, collection, inventory, wallet, stacks := newCostumeBurstFixture(t, filepath.Join(t.TempDir(), "state.json"), []gamedata.BattleReward{{Type: 8, ID: 710, Count: stackCount}}) - request := costumeBurstRequest(1, 4202, test.materials(stacks[0])...) - if _, _, handled, err := service.Handle("/CostumeBurst", request); err == nil || !handled { - t.Fatalf("invalid request accepted: handled=%v err=%v", handled, err) - } - if collection.Costumes()[0].BurstLevel != 0 || wallet.Snapshot().Gold != 1000 { - t.Fatal("invalid request mutated level or wallet") - } - if err := inventory.CanConsume([]Item{stacks[0]}); err != nil { - t.Fatalf("invalid request consumed material: %v", err) - } - }) - } -} - -func TestCostumeBurstSQLiteRollbackIsAtomic(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repository, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - starter := &Starter{Version: "2.35.10"} - inventory, err := OpenInventory(repository, starter) - if err != nil { - t.Fatal(err) - } - stacks, err := inventory.GrantOnce("rollback-material", []gamedata.BattleReward{{Type: 8, ID: 710, Count: 2}}) - if err != nil { - t.Fatal(err) - } - wallet, err := OpenWallet(repository, Currency{Gold: 1000}) - if err != nil { - t.Fatal(err) - } - collection, err := OpenCollectionStore(repository, []Costume{{InvenIndex: 88, ID: 4202}}) - if err != nil { - t.Fatal(err) - } - for _, ensure := range []func() error{inventory.EnsurePersisted, wallet.EnsurePersisted, collection.EnsurePersisted} { - if err := ensure(); err != nil { - t.Fatal(err) - } - } - service, _ := NewCostumeBurstService(testCostumeBurstDesign(), collection, inventory, wallet) - request := costumeBurstRequest(1, 4202, stacks[0], Item{Type: 4, Count: 100}) - operation, err := repository.BeginOperation() - if err != nil { - t.Fatal(err) - } - if code, _, handled, err := service.Handle("/CostumeBurst", request); err != nil || !handled || code != costumeBurstPacketCode { - t.Fatalf("transactional request code=%d handled=%v err=%v", code, handled, err) - } - if err := operation.Rollback(); err == nil { - t.Fatal("dirty request rollback did not fence published in-memory state") - } - if err := repository.Close(); err != nil { - t.Fatal(err) - } - - reopened, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := reopened.Close(); err != nil { - t.Error(err) - } - }() - reloadedInventory, err := OpenInventory(reopened, starter) - if err != nil { - t.Fatal(err) - } - reloadedWallet, err := OpenWallet(reopened, Currency{}) - if err != nil { - t.Fatal(err) - } - reloadedCollection, err := OpenCollectionStore(reopened, []Costume{{InvenIndex: 88, ID: 4202}}) - if err != nil { - t.Fatal(err) - } - if reloadedWallet.Snapshot().Gold != 1000 || reloadedCollection.Costumes()[0].BurstLevel != 0 { - t.Fatalf("rolled back state gold=%d costume=%+v", reloadedWallet.Snapshot().Gold, reloadedCollection.Costumes()[0]) - } - if err := reloadedInventory.CanConsume([]Item{stacks[0]}); err != nil { - t.Fatalf("rolled back inventory did not restore materials: %v", err) - } -} diff --git a/go/internal/server/player/costume_potential.go b/go/internal/server/player/costume_potential.go deleted file mode 100644 index 91121d4..0000000 --- a/go/internal/server/player/costume_potential.go +++ /dev/null @@ -1,248 +0,0 @@ -package player - -import ( - "encoding/binary" - "errors" - "fmt" - "math" - "slices" - "strconv" - "strings" - "sync" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -type CostumePotentialService struct { - design *gamedata.CostumePotentialDesign - collection *CollectionStore - characters *CharacterStore - inventory *Inventory - wallet *Wallet - connectStore stateio.Store - connectSession string - connectMu sync.Mutex -} - -func NewCostumePotentialService(design *gamedata.CostumePotentialDesign, collection *CollectionStore, characters *CharacterStore, inventory *Inventory, wallet *Wallet) (*CostumePotentialService, error) { - if design == nil || collection == nil || characters == nil || inventory == nil || wallet == nil { - return nil, errors.New("player: incomplete costume potential service") - } - return &CostumePotentialService{design: design, collection: collection, characters: characters, inventory: inventory, wallet: wallet}, nil -} - -// Contributions follows CharStatInfo.GetCostumeNodeBuffStat: type 2 nodes -// contribute across every owned costume of the same UniqueCharId, while type -// 1 nodes contribute only from CharDBInfo.ConnectPotentialCostume. Values are -// already fractions for the percent options 2, 4 and 6; they are not equipment -// option percentages and must not be divided by 100 or rounded individually. -func (s *CostumePotentialService) Contributions(character Character) ([]gamedata.StatContribution, error) { - if s == nil || s.design == nil || s.collection == nil { - return nil, errors.New("player: missing costume potential contribution source") - } - unique, ok := s.design.CharacterUnique[character.ID] - if !ok || unique == 0 { - return nil, fmt.Errorf("player: unknown costume potential character %d", character.ID) - } - var result []gamedata.StatContribution - connectedFound := character.ConnectPotentialCostume == 0 - for _, costume := range s.collection.Costumes() { - if s.design.CostumeUnique[costume.ID] != unique { - continue - } - if costume.ID == character.ConnectPotentialCostume { - connectedFound = true - } - seen := map[uint64]bool{} - for _, id := range costume.PotentialIDs { - if seen[id] { - return nil, errors.New("player: duplicate active costume potential stat node") - } - seen[id] = true - node, found := s.design.Nodes[costume.ID][id] - if !found { - return nil, fmt.Errorf("player: unknown active potential node %d/%d", costume.ID, id) - } - if node.NodeType != 2 && (node.NodeType != 1 || costume.ID != character.ConnectPotentialCostume) { - continue - } - if math.IsNaN(node.StatValue) || math.IsInf(node.StatValue, 0) || node.StatValue < 0 { - return nil, errors.New("player: invalid costume potential stat value") - } - contribution := gamedata.StatContribution{Option: node.StatType} - switch node.StatType { - case 1, 2: - contribution.Stat = gamedata.StatHealth - case 3, 4: - contribution.Stat = gamedata.StatAttack - case 5, 6: - contribution.Stat = gamedata.StatMagic - case 7: - contribution.Stat = gamedata.StatDefencePercent - case 8: - contribution.Stat = gamedata.StatMagicResistancePercent - case 9: - contribution.Stat = gamedata.StatCriticalChance - case 10: - contribution.Stat = gamedata.StatCriticalDamage - case 11, 12, 13, 14, 15, 19: - contribution.Stat = gamedata.StatElementDamage - case 16, 17, 18, 20: - contribution.Stat = gamedata.StatElementResistance - default: - return nil, fmt.Errorf("player: unsupported costume potential stat option %d", node.StatType) - } - if node.StatType == 2 || node.StatType == 4 || node.StatType == 6 { - contribution.Percent = node.StatValue - } else { - contribution.Flat = node.StatValue - } - result = append(result, contribution) - } - } - if !connectedFound { - return nil, errors.New("player: connected potential costume is not owned by this character") - } - return result, nil -} - -func (s *CostumePotentialService) Handle(path string, request []byte) (int, []byte, bool, error) { - if path == "/CostumePotentialConnect" { - return s.connect(request) - } - if path != "/CostumeNodeActivation" { - return 0, nil, false, nil - } - seq, found, err := wire.Varint(request, 1) - if err != nil || !found || seq == 0 { - return 0, nil, true, errors.New("player: CostumeNodeActivation missing sequence") - } - characterIndex, found, err := wire.Varint(request, 2) - if err != nil || !found || characterIndex == 0 { - return 0, nil, true, errors.New("player: CostumeNodeActivation missing character") - } - costumeIndex, found, err := wire.Varint(request, 3) - if err != nil || !found || costumeIndex == 0 { - return 0, nil, true, errors.New("player: CostumeNodeActivation missing costume") - } - var nodes []uint64 - var materials []Item - err = wire.Walk(request, func(field wire.Field) error { - switch field.Number { - case 4: - if field.Type != 0 && field.Type != 2 { - return errors.New("player: invalid potential node field") - } - for data := field.Value; len(data) != 0; { - id, count := binary.Uvarint(data) - if count <= 0 || id == 0 { - return errors.New("player: invalid potential node ID") - } - nodes = append(nodes, id) - data = data[count:] - } - case 5: - if field.Type != 2 { - return errors.New("player: invalid potential material field") - } - var item Item - if err := decodeVarints(field.Value, map[int]*uint64{1: &item.InvenIndex, 2: &item.ID, 3: &item.Type, 4: &item.Count, 5: &item.KeepFlag, 6: &item.TimeValue, 9: &item.SortID, 10: &item.UseCount}); err != nil { - return err - } - if item.Type == 0 || item.Count == 0 || (item.Type != 4 && (item.InvenIndex == 0 || item.ID == 0)) { - return errors.New("player: incomplete potential material") - } - materials = append(materials, item) - } - return nil - }) - if err != nil { - return 0, nil, true, err - } - if len(nodes) == 0 || len(materials) == 0 { - return 0, nil, true, errors.New("player: CostumeNodeActivation has no nodes or materials") - } - character, found := s.characters.Find(characterIndex) - if !found { - return 0, nil, true, fmt.Errorf("player: unknown potential character %d", characterIndex) - } - costume, found := s.collection.CostumeByIndex(costumeIndex) - if !found || costume.UseChar != characterIndex { - return 0, nil, true, fmt.Errorf("player: potential costume %d is not owned by character %d", costumeIndex, characterIndex) - } - if err := s.collection.ValidateCostumePotentialActivation(costumeIndex, nodes); err != nil { - return 0, nil, true, err - } - costs, err := s.design.Validate(costume.ID, character.ID, 0, costume.PotentialIDs, nodes) - if err != nil { - return 0, nil, true, fmt.Errorf("player: validate costume potential GameData: %w", err) - } - want := make(map[[2]uint64]uint64) - for _, cost := range costs { - key := [2]uint64{cost.Type, cost.ID} - if cost.Count > ^uint64(0)-want[key] { - return 0, nil, true, errors.New("player: costume potential cost overflow") - } - want[key] += cost.Count - } - got := make(map[[2]uint64]uint64) - var items []Item - var gold uint64 - for _, material := range materials { - key := [2]uint64{material.Type, material.ID} - if material.Type == 4 { - if material.InvenIndex != 0 || material.ID != 0 || gold != 0 { - return 0, nil, true, errors.New("player: invalid costume potential currency") - } - gold = material.Count - } else { - if material.Type != 8 { - return 0, nil, true, fmt.Errorf("player: unsupported costume potential material type %d", material.Type) - } - items = append(items, material) - } - if material.Count > ^uint64(0)-got[key] { - return 0, nil, true, errors.New("player: submitted costume potential material overflow") - } - got[key] += material.Count - } - if len(got) != len(want) { - return 0, nil, true, fmt.Errorf("player: costume potential material kinds mismatch: request=%v GameData=%v", got, want) - } - for key, count := range want { - if got[key] != count { - return 0, nil, true, fmt.Errorf("player: costume potential material %d/%d=%d want %d", key[0], key[1], got[key], count) - } - } - if len(items) != 0 { - if err := s.inventory.CanConsume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: validate costume potential items: %w", err) - } - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - return 0, nil, true, errors.New("player: insufficient gold for costume potential") - } - sortedNodes := append([]uint64(nil), nodes...) - slices.Sort(sortedNodes) - parts := make([]string, len(sortedNodes)) - for i, id := range sortedNodes { - parts[i] = strconv.FormatUint(id, 10) - } - identity := "costume-potential:" + strconv.FormatUint(costumeIndex, 10) + ":" + strings.Join(parts, ",") - if gold != 0 { - if _, err := s.wallet.SpendGoldOnce(identity, gold); err != nil { - return 0, nil, true, fmt.Errorf("player: spend costume potential gold: %w", err) - } - } - if len(items) != 0 { - if err := s.inventory.Consume(items); err != nil { - return 0, nil, true, fmt.Errorf("player: consume costume potential items: %w", err) - } - } - if err := s.collection.ActivateCostumePotential(costumeIndex, nodes); err != nil { - return 0, nil, true, fmt.Errorf("player: persist costume potential: %w", err) - } - return 261, nil, true, nil -} diff --git a/go/internal/server/player/costume_potential_connect.go b/go/internal/server/player/costume_potential_connect.go deleted file mode 100644 index e214666..0000000 --- a/go/internal/server/player/costume_potential_connect.go +++ /dev/null @@ -1,206 +0,0 @@ -package player - -import ( - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "crypto/sha256" - "encoding/hex" - "encoding/json" - "fmt" - "math" -) - -type potentialConnectReceipt struct { - Digest string - Body []byte -} - -func (s *CostumePotentialService) AttachConnectStore(store stateio.Store) error { - if store == nil { - return fmt.Errorf("player: missing potential connect store") - } - s.connectStore = store - return nil -} -func (s *CostumePotentialService) BeginSession(id string) { - s.connectMu.Lock() - defer s.connectMu.Unlock() - s.connectSession = id -} -func (s *CostumePotentialService) connect(request []byte) (int, []byte, bool, error) { - s.connectMu.Lock() - defer s.connectMu.Unlock() - fail := func(e error) (int, []byte, bool, error) { return 267, nil, true, e } - seq, ok, err := wire.Varint(request, 1) - if err != nil || !ok || seq == 0 || seq > math.MaxInt32 || s.connectStore == nil || s.connectSession == "" { - return fail(fmt.Errorf("player: invalid potential connection session/sequence")) - } - var receipts map[string]potentialConnectReceipt - b, err := s.connectStore.Load("potentialconnect") - if err != nil { - return fail(err) - } - if b != nil { - if err = json.Unmarshal(b, &receipts); err != nil || receipts == nil { - return fail(fmt.Errorf("player: invalid potential connection receipts")) - } - } else { - receipts = map[string]potentialConnectReceipt{} - } - for _, r := range receipts { - hash, e := hex.DecodeString(r.Digest) - if e != nil || len(hash) != sha256.Size { - return fail(fmt.Errorf("player: malformed potential connection receipt digest")) - } - if e = wire.Walk(r.Body, func(wire.Field) error { return nil }); e != nil { - return fail(e) - } - } - seqFields := 0 - if err = wire.Walk(request, func(f wire.Field) error { - if f.Number == 1 { - seqFields++ - if f.Type != 0 { - return fmt.Errorf("player: invalid potential sequence wire") - } - } - return nil - }); err != nil { - return fail(err) - } - if seqFields != 1 { - return fail(fmt.Errorf("player: duplicate potential sequence")) - } - key := fmt.Sprintf("%s:%d", s.connectSession, seq) - digest := fmt.Sprintf("%x", sha256.Sum256(request)) - if prior, ok := receipts[key]; ok { - if prior.Digest != digest { - return fail(fmt.Errorf("player: changed potential connection replay")) - } - return 267, prior.Body, true, nil - } - var characters []Character - seen := map[uint64]bool{} - oldHP := map[uint64]uint64{} - err = wire.Walk(request, func(f wire.Field) error { - if f.Number != 2 { - return nil - } - if f.Type != 2 || len(characters) >= 4096 { - return fmt.Errorf("player: invalid potential connection list") - } - fields := map[int]bool{} - if e := wire.Walk(f.Value, func(field wire.Field) error { - if field.Number == 1 || field.Number == 2 { - if field.Type != 0 || fields[field.Number] { - return fmt.Errorf("player: duplicate or invalid connection scalar") - } - fields[field.Number] = true - } - return nil - }); e != nil { - return e - } - index, ok, e := wire.Varint(f.Value, 1) - if e != nil || !ok || index == 0 || index > math.MaxInt64 || seen[index] { - return fmt.Errorf("player: duplicate or invalid potential character") - } - costume, _, e := wire.Varint(f.Value, 2) - if e != nil || costume == 0 || costume > math.MaxInt32 { - return fmt.Errorf("player: potential connection requires a costume design ID") - } - c, owned := s.characters.Find(index) - if !owned || IsStoryCharacter(c) || IsCharmCharacter(c) { - return fmt.Errorf("player: potential character is not permanent owned") - } - unique := s.design.CharacterUnique[c.ID] - characterType, knownType := s.design.CharacterTypes[c.ID] - if unique == 0 || !knownType || characterType != 0 || !s.design.CostumeActive[costume] || s.design.CostumeUnique[costume] != unique || len(s.design.Nodes[costume]) == 0 { - return fmt.Errorf("player: incompatible or unavailable potential costume") - } - ownedCostume := false - for _, v := range s.collection.Costumes() { - if v.ID == costume && v.UseChar == index { - ownedCostume = true - for _, id := range v.PotentialIDs { - if _, ok := s.design.Nodes[costume][id]; !ok { - return fmt.Errorf("player: invalid saved potential node") - } - } - break - } - } - if !ownedCostume { - return fmt.Errorf("player: potential costume not owned by requested character") - } - hp, e := s.characters.CurrentHealth(index) - if e != nil { - return e - } - oldHP[index] = hp - c.ConnectPotentialCostume = costume - characters = append(characters, c) - seen[index] = true - return nil - }) - if err != nil { - return fail(err) - } - if len(characters) == 0 { - return fail(fmt.Errorf("player: empty potential connection list")) - } - // All links are validated before writes. The parent account transaction - // includes both character ownership domains, HP and the response receipt. - for _, c := range characters { - if err = s.characters.setPotentialConnection(c.InvenIndex, c.ConnectPotentialCostume); err != nil { - return fail(err) - } - } - var out []byte - for _, c := range characters { - maximum, e := s.characters.MaxHealth(c.InvenIndex) - if e != nil { - return fail(e) - } - hp := min(oldHP[c.InvenIndex], maximum) - if e = s.characters.SetCurrentHealth(c.InvenIndex, hp); e != nil { - return fail(e) - } - c.HP = hp - out = wire.AppendBytes(out, 1, CharacterWire(c)) - } - receipts[key] = potentialConnectReceipt{Digest: digest, Body: out} - b, err = json.Marshal(receipts) - if err != nil { - return fail(err) - } - if err = s.connectStore.Save("potentialconnect", b); err != nil { - return fail(err) - } - return 267, out, true, nil -} -func (s *CharacterStore) setPotentialConnection(index, costume uint64) error { - s.mu.Lock() - for i, c := range s.characters { - if c.InvenIndex == index { - next := append([]Character(nil), s.characters...) - next[i].ConnectPotentialCostume = costume - if err := s.persist(next); err != nil { - s.mu.Unlock() - return err - } - s.characters = next - s.mu.Unlock() - return nil - } - } - collection := s.collection - s.mu.Unlock() - if collection != nil { - if c, ok := collection.FindCharacter(index); ok { - c.ConnectPotentialCostume = costume - return collection.UpdateCharacter(c.ID, c) - } - } - return fmt.Errorf("player: unknown potential connection character") -} diff --git a/go/internal/server/player/costume_potential_connect_test.go b/go/internal/server/player/costume_potential_connect_test.go deleted file mode 100644 index 82f0325..0000000 --- a/go/internal/server/player/costume_potential_connect_test.go +++ /dev/null @@ -1,249 +0,0 @@ -package player - -import ( - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" - "bytes" - "errors" - "path/filepath" - "testing" -) - -func connectTestService(t *testing.T, store stateio.Store) (*CostumePotentialService, *CharacterStore) { - t.Helper() - inv, e := OpenInventory(store, &Starter{Version: "2.35.10"}) - if e != nil { - t.Fatal(e) - } - chars, e := OpenCharacterStore(store, []Character{{InvenIndex: 77, ID: 91, Level: 1, HP: 80}}, inv, "", "") - if e != nil { - t.Fatal(e) - } - if e = chars.EnsurePersisted(); e != nil { - t.Fatal(e) - } - collection, e := OpenCollectionStore(store, nil) - if e != nil { - t.Fatal(e) - } - if len(collection.Costumes()) == 0 { - v := cloneCollection(collection.data) - v.Costumes = []Costume{{InvenIndex: 88, ID: 100, UseChar: 77}, {InvenIndex: 89, ID: 200, UseChar: 77}, {InvenIndex: 90, ID: 300, UseChar: 77}} - if e = collection.commit(v); e != nil { - t.Fatal(e) - } - } - wallet, e := OpenWallet(store, Currency{Gold: 100}) - if e != nil { - t.Fatal(e) - } - d := &gamedata.CostumePotentialDesign{CharacterUnique: map[uint64]uint64{91: 9}, CostumeUnique: map[uint64]uint64{100: 9, 200: 9, 300: 10}, Nodes: map[uint64]map[uint64]gamedata.CostumePotentialNode{100: {1: {ID: 1}}, 200: {1: {ID: 1}}, 300: {1: {ID: 1}}}} - s, e := NewCostumePotentialService(d, collection, chars, inv, wallet) - d.CharacterTypes = map[uint64]uint64{91: 0} - d.CostumeActive = map[uint64]bool{100: true, 200: true, 300: true} - if e != nil { - t.Fatal(e) - } - if err := s.AttachConnectStore(store); err != nil { - t.Fatal(err) - } - s.BeginSession("test") - if err := chars.AttachMaxHealth(func(c Character) (uint64, error) { - if c.ConnectPotentialCostume == 200 { - return 50, nil - } - return 100, nil - }); err != nil { - t.Fatal(err) - } - return s, chars -} -func connectRequest(seq uint64, rows ...[2]uint64) []byte { - b := wire.AppendVarint(nil, 1, seq) - for _, r := range rows { - v := wire.AppendVarint(wire.AppendVarint(nil, 1, r[0]), 2, r[1]) - b = wire.AppendBytes(b, 2, v) - } - return b -} -func TestPotentialConnectFreeZeroNodesSwitchClampsAndReplaysAcrossRestart(t *testing.T) { - store := stateio.NewMemory() - s, chars := connectTestService(t, store) - if _, _, _, e := s.Handle("/CostumePotentialConnect", connectRequest(1, [2]uint64{77, 100})); e != nil { - t.Fatal(e) - } - c, _ := chars.Find(77) - if c.ConnectPotentialCostume != 100 || c.HP != 80 || s.wallet.Snapshot().Gold != 100 { - t.Fatal("auto connection charged or healed") - } - b := connectRequest(2, [2]uint64{77, 200}) - _, reply, _, e := s.Handle("/CostumePotentialConnect", b) - if e != nil { - t.Fatal(e) - } - c, _ = chars.Find(77) - if c.HP != 50 { - t.Fatal("maxHP decrease not clamped") - } - next, _ := connectTestService(t, store) - _, again, _, e := next.Handle("/CostumePotentialConnect", b) - if e != nil || !bytes.Equal(reply, again) { - t.Fatal("restart lost exact connection replay") - } - if _, _, _, e = next.Handle("/CostumePotentialConnect", connectRequest(3, [2]uint64{77, 100})); e != nil { - t.Fatal(e) - } - c, _ = next.characters.Find(77) - if c.HP != 50 { - t.Fatal("maxHP increase healed player") - } - if e = next.characters.SetCurrentHealth(77, 0); e != nil { - t.Fatal(e) - } - if _, _, _, e = next.Handle("/CostumePotentialConnect", connectRequest(4, [2]uint64{77, 200})); e != nil { - t.Fatal(e) - } - c, _ = next.characters.Find(77) - if c.HP != 0 { - t.Fatal("connection revived dead character") - } -} -func TestPotentialConnectRejectsUnownedOtherCharacterAndDuplicateBatch(t *testing.T) { - s, chars := connectTestService(t, stateio.NewMemory()) - for _, b := range [][]byte{connectRequest(1, [2]uint64{77, 300}), connectRequest(2, [2]uint64{77, 88}), connectRequest(3, [2]uint64{77, 0}), connectRequest(4, [2]uint64{77, 100}, [2]uint64{77, 200}), connectRequest(5, [2]uint64{77, 100}, [2]uint64{999, 200})} { - if _, _, _, e := s.Handle("/CostumePotentialConnect", b); e == nil { - t.Fatal("invalid potential connection accepted") - } - } - c, _ := chars.Find(77) - if c.ConnectPotentialCostume != 0 { - t.Fatal("invalid batch partially updated character") - } - s.design.CostumeActive[100] = false - if _, _, _, e := s.Handle("/CostumePotentialConnect", connectRequest(6, [2]uint64{77, 100})); e == nil { - t.Fatal("inactive potential costume connected") - } - s.design.CostumeActive[100] = true - s.design.CharacterTypes[91] = 1 - if _, _, _, e := s.Handle("/CostumePotentialConnect", connectRequest(7, [2]uint64{77, 100})); e == nil { - t.Fatal("temporary design character connected") - } - s.design.CharacterTypes[91] = 0 - duplicate := wire.AppendVarint(connectRequest(8, [2]uint64{77, 100}), 1, 9) - if _, _, _, e := s.Handle("/CostumePotentialConnect", duplicate); e == nil { - t.Fatal("duplicate sequence accepted") - } - row := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 77), 2, 100), 2, 200) - duplicate = wire.AppendBytes(wire.AppendVarint(nil, 1, 10), 2, row) - if _, _, _, e := s.Handle("/CostumePotentialConnect", duplicate); e == nil { - t.Fatal("duplicate nested costume scalar accepted") - } -} - -type connectFailStore struct{ stateio.Store } - -func (s connectFailStore) Save(name string, b []byte) error { - if name == "potentialconnect" { - return errors.New("receipt failed") - } - return s.Store.Save(name, b) -} -func TestPotentialConnectSQLiteFailureRollsBackLinkAndHealth(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, e := accountstate.Open(path) - if e != nil { - t.Fatal(e) - } - s, _ := connectTestService(t, repo) - if err := s.AttachConnectStore(connectFailStore{repo}); err != nil { - t.Fatal(err) - } - op, e := repo.BeginOperation() - if e != nil { - t.Fatal(e) - } - b := connectRequest(1, [2]uint64{77, 200}) - if _, _, _, e = s.Handle("/CostumePotentialConnect", b); e == nil { - t.Fatal("receipt failure ignored") - } - if e = op.Rollback(); e != nil && !errors.Is(e, stateio.ErrStateRecoveryRequired) { - t.Fatal(e) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, e = accountstate.Open(path) - if e != nil { - t.Fatal(e) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - next, chars := connectTestService(t, repo) - c, _ := chars.Find(77) - if c.ConnectPotentialCostume != 0 || c.HP != 80 { - t.Fatal("failed connection persisted link/HP") - } - op, e = repo.BeginOperation() - if e != nil { - t.Fatal(e) - } - if _, _, _, e = next.Handle("/CostumePotentialConnect", b); e != nil { - t.Fatal(e) - } - if e = op.Commit(); e != nil { - t.Fatal(e) - } -} - -func TestPotentialConnectionCollectionCharacterFallbackPersistsAcrossRestart(t *testing.T) { - store := stateio.NewMemory() - s, chars := connectTestService(t, store) - v := cloneCollection(s.collection.data) - v.Characters = []Character{{InvenIndex: 78, ID: 92, Level: 1, HP: 40, CostumeID: 201, UseCostume: 91}} - v.Costumes = append(v.Costumes, Costume{InvenIndex: 91, ID: 201, UseChar: 78}) - if e := s.collection.commit(v); e != nil { - t.Fatal(e) - } - if e := chars.AttachCollection(s.collection); e != nil { - t.Fatal(e) - } - s.design.CharacterUnique[92] = 9 - s.design.CharacterTypes[92] = 0 - s.design.CostumeUnique[201] = 9 - s.design.CostumeActive[201] = true - s.design.Nodes[201] = map[uint64]gamedata.CostumePotentialNode{1: {ID: 1}} - b := connectRequest(1, [2]uint64{78, 201}) - _, response, _, e := s.Handle("/CostumePotentialConnect", b) - if e != nil { - t.Fatal(e) - } - c, ok := chars.Find(78) - if !ok || c.ConnectPotentialCostume != 201 || c.UseCostume != 91 || c.CostumeID != 201 || c.HP != 40 { - t.Fatal("collection connection changed appearance or HP") - } - collection, e := OpenCollectionStore(store, nil) - if e != nil { - t.Fatal(e) - } - c, ok = collection.FindCharacter(78) - if !ok || c.ConnectPotentialCostume != 201 { - t.Fatal("collection connection lost after restart") - } - s.collection = collection - if err := chars.AttachCollection(collection); err != nil { - t.Fatal(err) - } - if err := s.AttachConnectStore(store); err != nil { - t.Fatal(err) - } - s.BeginSession("test") - _, again, _, e := s.Handle("/CostumePotentialConnect", b) - if e != nil || !bytes.Equal(response, again) { - t.Fatal("collection reconnect replay lost") - } -} diff --git a/go/internal/server/player/costume_potential_test.go b/go/internal/server/player/costume_potential_test.go deleted file mode 100644 index fec04d9..0000000 --- a/go/internal/server/player/costume_potential_test.go +++ /dev/null @@ -1,164 +0,0 @@ -package player - -import ( - "encoding/binary" - "math" - "path/filepath" - "testing" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func TestCostumePotentialContributionsIncludePublicAndConnectedNodes(t *testing.T) { - storage := stateio.NewMemory() - collection, err := OpenCollectionStore(storage, []Costume{{InvenIndex: 1, ID: 100}, {InvenIndex: 2, ID: 200}, {InvenIndex: 3, ID: 300}}) - if err != nil { - t.Fatal(err) - } - for _, index := range []uint64{1, 2, 3} { - if err := collection.ActivateCostumePotential(index, []uint64{1, 2, 3}); err != nil { - t.Fatal(err) - } - } - design := &gamedata.CostumePotentialDesign{CharacterUnique: map[uint64]uint64{91: 9, 101: 10}, CostumeUnique: map[uint64]uint64{100: 9, 200: 9, 300: 10}, Nodes: map[uint64]map[uint64]gamedata.CostumePotentialNode{ - 100: {1: {ID: 1, NodeType: 2, StatType: 1, StatValue: 7}, 2: {ID: 2, NodeType: 1, StatType: 2, StatValue: .085}, 3: {ID: 3, NodeType: 4}}, - 200: {1: {ID: 1, NodeType: 2, StatType: 2, StatValue: .015}, 2: {ID: 2, NodeType: 1, StatType: 1, StatValue: 20}, 3: {ID: 3, NodeType: 2, StatType: 3, StatValue: 3}}, - 300: {1: {ID: 1, NodeType: 2, StatType: 1, StatValue: 1000}, 2: {ID: 2, NodeType: 1, StatType: 1, StatValue: 1000}, 3: {ID: 3, NodeType: 4}}, - }} - service := &CostumePotentialService{design: design, collection: collection} - character := Character{ID: 91, ConnectPotentialCostume: 100} - contributions, err := service.Contributions(character) - if err != nil { - t.Fatal(err) - } - stats := gamedata.AggregateStats(gamedata.BaseStats{Health: 100, Attack: 10}, contributions) - if stats.Health != 117 || stats.Attack != 13 { - t.Fatalf("public plus connected stats=%+v contributions=%+v", stats, contributions) - } - character.ConnectPotentialCostume = 200 - contributions, err = service.Contributions(character) - if err != nil { - t.Fatal(err) - } - stats = gamedata.AggregateStats(gamedata.BaseStats{Health: 100}, contributions) - if stats.Health != 128 { - t.Fatalf("connection switch health=%f", stats.Health) - } - character.ConnectPotentialCostume = 0 - contributions, err = service.Contributions(character) - if err != nil { - t.Fatal(err) - } - stats = gamedata.AggregateStats(gamedata.BaseStats{Health: 100}, contributions) - if stats.Health != 108 { - t.Fatalf("unconnected public health=%f", stats.Health) - } - character.ConnectPotentialCostume = 300 - if _, err := service.Contributions(character); err == nil { - t.Fatal("connected another character's costume") - } - character.ConnectPotentialCostume = 100 - reopened, err := OpenCollectionStore(storage, []Costume{{InvenIndex: 1, ID: 100}, {InvenIndex: 2, ID: 200}, {InvenIndex: 3, ID: 300}}) - if err != nil { - t.Fatal(err) - } - service.collection = reopened - contributions, err = service.Contributions(character) - if err != nil { - t.Fatal(err) - } - if stats := gamedata.AggregateStats(gamedata.BaseStats{Health: 100}, contributions); stats.Health != 117 { - t.Fatalf("restart contribution health=%f", stats.Health) - } - design.Nodes[100][1] = gamedata.CostumePotentialNode{ID: 1, NodeType: 2, StatType: 1, StatValue: math.NaN()} - if _, err := service.Contributions(character); err == nil { - t.Fatal("nonfinite stat accepted") - } -} - -func TestCostumeNodeActivationSupportsSingleAndOneClickSets(t *testing.T) { - dir := t.TempDir() - inventory, err := OpenInventory(testStore(filepath.Join(dir, "items.json")), &Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - materials, err := inventory.GrantOnce("potential", []gamedata.BattleReward{{Type: 8, ID: 114, Count: 5}}) - if err != nil { - t.Fatal(err) - } - wallet, err := OpenWallet(testStore(filepath.Join(dir, "wallet.json")), Currency{Gold: 300}) - if err != nil { - t.Fatal(err) - } - characters, err := OpenCharacterStore(testStore(filepath.Join(dir, "characters.json")), []Character{{InvenIndex: 77, ID: 6514, Level: 100}}, inventory, "", "") - if err != nil { - t.Fatal(err) - } - collection, err := OpenCollectionStore(testStore(filepath.Join(dir, "collection.json")), nil) - if err != nil { - t.Fatal(err) - } - nextCollection := cloneCollection(collection.data) - nextCollection.Costumes = []Costume{{InvenIndex: 88, ID: 65103, UseChar: 77}} - if err := collection.commit(nextCollection); err != nil { - t.Fatal(err) - } - design := &gamedata.CostumePotentialDesign{ - Nodes: map[uint64]map[uint64]gamedata.CostumePotentialNode{65103: { - 1: {ID: 1, Costs: []gamedata.CostumePotentialCost{{Type: 4, Count: 100}}}, - 2: {ID: 2, Prerequisites: []uint64{1}, Costs: []gamedata.CostumePotentialCost{{Type: 8, ID: 114, Count: 2}}}, - 3: {ID: 3, Prerequisites: []uint64{2}, Costs: []gamedata.CostumePotentialCost{{Type: 8, ID: 114, Count: 3}, {Type: 4, Count: 200}}}, - }}, - CostumeUnique: map[uint64]uint64{65103: 651}, - CharacterGrade: map[uint64]uint64{6514: 5}, - CharacterUnique: map[uint64]uint64{6514: 651}, - } - service, err := NewCostumePotentialService(design, collection, characters, inventory, wallet) - if err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 77), 3, 88) - request = wire.AppendVarint(request, 4, 1) - request = wire.AppendBytes(request, 5, ItemWire(Item{Type: 4, Count: 100})) - code, response, handled, err := service.Handle("/CostumeNodeActivation", request) - if err != nil || !handled || code != 261 || len(response) != 0 { - t.Fatalf("single activation code=%d handled=%v response=%x err=%v", code, handled, response, err) - } - oneClick := wire.AppendVarint(wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 77), 3, 88) - oneClick = wire.AppendVarint(wire.AppendVarint(oneClick, 4, 2), 4, 3) - item := materials[0] - oneClick = wire.AppendBytes(oneClick, 5, ItemWire(item)) - oneClick = wire.AppendBytes(oneClick, 5, ItemWire(Item{Type: 4, Count: 200})) - code, _, handled, err = service.Handle("/CostumeNodeActivation", oneClick) - if err != nil || !handled || code != 261 { - t.Fatalf("one-click activation code=%d handled=%v err=%v", code, handled, err) - } - if wallet.Snapshot().Gold != 0 { - t.Fatalf("gold=%d", wallet.Snapshot().Gold) - } - costume, found := collection.CostumeByIndex(88) - if !found || len(costume.PotentialIDs) != 3 || costume.PotentialIDs[0] != 1 || costume.PotentialIDs[2] != 3 { - t.Fatalf("activated costume=%+v found=%v", costume, found) - } - var encoded []uint64 - if err := wire.Walk(CostumeWire(costume), func(field wire.Field) error { - if field.Number == 8 { - v, _ := binary.Uvarint(field.Value) - encoded = append(encoded, v) - } - return nil - }); err != nil { - t.Fatal(err) - } - if len(encoded) != 3 { - t.Fatalf("wire potential IDs=%v", encoded) - } - if _, _, _, err := service.Handle("/CostumeNodeActivation", oneClick); err == nil { - t.Fatal("already active nodes accepted") - } - if wallet.Snapshot().Gold != 0 { - t.Fatal("duplicate request charged wallet") - } -} diff --git a/go/internal/server/player/eat_food.go b/go/internal/server/player/eat_food.go deleted file mode 100644 index 7e94761..0000000 --- a/go/internal/server/player/eat_food.go +++ /dev/null @@ -1,402 +0,0 @@ -package player - -import ( - "crypto/sha256" - "encoding/binary" - "encoding/hex" - "encoding/json" - "errors" - "fmt" - "math" - "strconv" - "sync" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -// FoodService participates in the account request transaction: consuming the -// stack, saving health and recording the replay response commit together. -type FoodService struct { - mu sync.Mutex - store stateio.AtomicEntryStore - design *gamedata.FoodDesign - inventory *Inventory - characters *CharacterStore - session string - currentPack func() (int, error) - battleActive func() bool -} - -type foodReply struct { - Digest string `json:"digest"` - Body []byte `json:"body"` -} - -func OpenFoodService(store stateio.Store, design *gamedata.FoodDesign, inventory *Inventory, characters *CharacterStore) (*FoodService, error) { - entries, ok := store.(stateio.AtomicEntryStore) - if !ok || design == nil || inventory == nil || characters == nil { - return nil, errors.New("player: incomplete food configuration") - } - s := &FoodService{store: entries, design: design, inventory: inventory, characters: characters} - health, err := entries.ListEntries("characters", "current_hp") - if err != nil { - return nil, err - } - for key, raw := range health { - index, err := strconv.ParseUint(key, 10, 64) - var hp uint64 - if err != nil || index == 0 || json.Unmarshal(raw, &hp) != nil { - return nil, errors.New("player: malformed saved current health") - } - if _, err := characters.MaxHealth(index); err != nil { - return nil, err - } - } - replies, err := entries.ListEntries("characters", "food_requests") - if err != nil { - return nil, err - } - for key, raw := range replies { - var reply foodReply - decodedKey, keyErr := hex.DecodeString(key) - if keyErr != nil || len(decodedKey) != sha256.Size || json.Unmarshal(raw, &reply) != nil || len(reply.Body) == 0 { - return nil, errors.New("player: malformed saved food replay") - } - digest, err := hex.DecodeString(reply.Digest) - if err != nil || len(digest) != sha256.Size { - return nil, errors.New("player: malformed saved food replay digest") - } - if err := wire.Walk(reply.Body, func(field wire.Field) error { return nil }); err != nil { - return nil, errors.New("player: malformed saved food replay body") - } - } - return s, nil -} - -func (s *FoodService) BeginSession(id string) { s.mu.Lock(); defer s.mu.Unlock(); s.session = id } - -// AttachContext supplies authoritative pack and active-battle state; the client -// omits PackId in ordinary recovery requests, which means the current pack. -func (s *FoodService) AttachContext(currentPack func() (int, error), battleActive func() bool) error { - if currentPack == nil || battleActive == nil { - return errors.New("player: incomplete food context") - } - s.mu.Lock() - defer s.mu.Unlock() - s.currentPack = currentPack - s.battleActive = battleActive - return nil -} - -func (s *FoodService) Handle(path string, request []byte) (int, []byte, bool, error) { - if path == "/EatFoodAuto" { - return s.eatFoodAuto(request) - } - if path != "/EatFood" { - return 0, nil, false, nil - } - var seq, pack, index uint64 - seenHeaders := map[int]bool{} - parseErr := wire.Walk(request, func(field wire.Field) error { - if field.Number == 4 { - return nil - } - target := map[int]*uint64{1: &seq, 2: &pack, 3: &index}[field.Number] - if target == nil || field.Type != 0 || seenHeaders[field.Number] { - return errors.New("player: invalid EatFood header") - } - seenHeaders[field.Number] = true - *target, _ = binary.Uvarint(field.Value) - return nil - }) - if parseErr != nil || seq == 0 || seq > math.MaxInt32 || pack > math.MaxInt32 || index == 0 || index > math.MaxInt64 { - return 0, nil, true, errors.New("player: invalid EatFood request") - } - items, err := equipmentRequestItems(request, 4, "EatFood") - if err != nil || len(items) == 0 { - return 0, nil, true, errors.New("player: EatFood requires food stacks") - } - s.mu.Lock() - defer s.mu.Unlock() - if s.session == "" || s.currentPack == nil || s.battleActive == nil { - return 0, nil, true, errors.New("player: EatFood session or context unavailable") - } - keyDigest := sha256.Sum256([]byte(s.session + ":EatFood:" + strconv.FormatUint(seq, 10))) - key := hex.EncodeToString(keyDigest[:]) - digest := sha256.Sum256(request) - digestString := hex.EncodeToString(digest[:]) - prior, found, err := s.store.LoadEntry("characters", "food_requests", key) - if err != nil { - return 0, nil, true, err - } - if found { - var reply foodReply - if json.Unmarshal(prior, &reply) != nil || reply.Digest != digestString { - return 0, nil, true, errors.New("player: EatFood sequence reused with different request") - } - return 22, append([]byte(nil), reply.Body...), true, nil - } - currentPack, err := s.currentPack() - if err != nil || currentPack <= 0 || (pack != 0 && pack != uint64(currentPack)) { - return 0, nil, true, errors.New("player: EatFood pack unavailable") - } - if s.battleActive() { - return 0, nil, true, errors.New("player: cannot eat food during battle") - } - character, err := s.recoverCharacter(index, items) - if err != nil { - return 0, nil, true, err - } - if err := s.inventory.CanConsume(items); err != nil { - return 0, nil, true, err - } - body := wire.AppendBytes(nil, 1, CharacterWire(character)) - if err := s.inventory.Consume(items); err != nil { - return 0, nil, true, err - } - healthRaw, _ := json.Marshal(character.HP) - replyRaw, _ := json.Marshal(foodReply{Digest: digestString, Body: body}) - if err := s.store.SaveWithEntries("characters", nil, []stateio.EntryMutation{{Bucket: "current_hp", Key: strconv.FormatUint(index, 10), Payload: healthRaw}, {Bucket: "food_requests", Key: key, Payload: replyRaw}}); err != nil { - return 0, nil, true, err - } - return 22, body, true, nil -} - -// MaxHealth reads the shared stat calculator without replacing it with field HP. -func (s *CharacterStore) MaxHealth(index uint64) (uint64, error) { - var character Character - for _, candidate := range s.RawAll() { - if candidate.InvenIndex == index { - character = candidate - break - } - } - if character.InvenIndex == 0 { - return 0, fmt.Errorf("player: unknown health character %d", index) - } - s.mu.Lock() - calculate := s.maxHealth - s.mu.Unlock() - if calculate != nil { - return calculate(character) - } - if character.HP == 0 { - return 0, errors.New("player: maximum health unavailable") - } - return character.HP, nil -} - -// CurrentHealth resolves persisted current HP independently from maximum HP. -// A missing current_hp entry retains the owned character record's HP, including -// zero. Recomputing equipment or other maximum-health stats must never heal it. -func (s *CharacterStore) CurrentHealth(index uint64) (uint64, error) { - var saved Character - for _, character := range s.RawAll() { - if character.InvenIndex == index { - saved = character - break - } - } - if saved.InvenIndex == 0 { - return 0, fmt.Errorf("player: unknown health character %d", index) - } - return s.savedCurrentHealth(saved) -} - -func (s *CharacterStore) savedCurrentHealth(saved Character) (uint64, error) { - if s.store == nil { - return saved.HP, nil - } - data, found, err := s.store.LoadEntry("characters", "current_hp", strconv.FormatUint(saved.InvenIndex, 10)) - if err != nil { - return 0, err - } - if !found { - return saved.HP, nil - } - var hp uint64 - if json.Unmarshal(data, &hp) != nil { - return 0, errors.New("player: invalid saved current health") - } - return hp, nil -} - -// SetCurrentHealth accepts server-validated battle results and recovery only. -// Food requests never accept health from the caller. -func (s *CharacterStore) SetCurrentHealth(index, hp uint64) error { - maximum, err := s.MaxHealth(index) - if err != nil { - return err - } - if hp > maximum { - return errors.New("player: current health exceeds maximum") - } - raw, _ := json.Marshal(hp) - return s.store.PutEntry("characters", "current_hp", strconv.FormatUint(index, 10), raw) -} - -func (s *CharacterStore) resetCurrentHealth(index uint64) error { - _, err := s.store.DeleteEntry("characters", "current_hp", strconv.FormatUint(index, 10)) - return err -} - -func (s *FoodService) eatFoodAuto(request []byte) (int, []byte, bool, error) { - var seq uint64 - var targets []struct { - index uint64 - items []Item - } - seenSequence := false - err := wire.Walk(request, func(field wire.Field) error { - switch field.Number { - case 1: - if field.Type != 0 || seenSequence { - return errors.New("player: invalid EatFoodAuto sequence") - } - seenSequence = true - seq, _ = binary.Uvarint(field.Value) - case 2: - if field.Type != 2 { - return errors.New("player: invalid EatFoodAuto target") - } - var index uint64 - seenIndex := false - if err := wire.Walk(field.Value, func(inner wire.Field) error { - if inner.Number == 2 { - return nil - } - if inner.Number != 1 || inner.Type != 0 || seenIndex { - return errors.New("player: invalid EatFoodAuto character") - } - seenIndex = true - index, _ = binary.Uvarint(inner.Value) - return nil - }); err != nil { - return err - } - if index == 0 || index > math.MaxInt64 { - return errors.New("player: invalid EatFoodAuto character index") - } - items, err := equipmentRequestItems(field.Value, 2, "EatFoodAuto") - if err != nil { - return err - } - if len(items) == 0 { - return errors.New("player: EatFoodAuto character requires food stacks") - } - targets = append(targets, struct { - index uint64 - items []Item - }{index, items}) - default: - return errors.New("player: unknown EatFoodAuto field") - } - return nil - }) - if err != nil || seq == 0 || seq > math.MaxInt32 || len(targets) == 0 { - return 0, nil, true, errors.New("player: invalid EatFoodAuto request") - } - s.mu.Lock() - defer s.mu.Unlock() - if s.session == "" || s.currentPack == nil || s.battleActive == nil { - return 0, nil, true, errors.New("player: EatFoodAuto context unavailable") - } - keyDigest := sha256.Sum256([]byte(s.session + ":EatFoodAuto:" + strconv.FormatUint(seq, 10))) - key := hex.EncodeToString(keyDigest[:]) - digest := sha256.Sum256(request) - digestString := hex.EncodeToString(digest[:]) - prior, found, err := s.store.LoadEntry("characters", "food_requests", key) - if err != nil { - return 0, nil, true, err - } - if found { - var reply foodReply - if json.Unmarshal(prior, &reply) != nil || reply.Digest != digestString { - return 0, nil, true, errors.New("player: EatFoodAuto sequence reused with different request") - } - return 27, append([]byte(nil), reply.Body...), true, nil - } - pack, err := s.currentPack() - if err != nil || pack <= 0 || s.battleActive() { - return 0, nil, true, errors.New("player: EatFoodAuto unavailable during battle or outside pack") - } - seenCharacters := map[uint64]bool{} - var items []Item - var body []byte - var changes []stateio.EntryMutation - for _, target := range targets { - if seenCharacters[target.index] { - return 0, nil, true, errors.New("player: duplicate EatFoodAuto character") - } - seenCharacters[target.index] = true - character, err := s.recoverCharacter(target.index, target.items) - if err != nil { - return 0, nil, true, err - } - items = append(items, target.items...) - raw, _ := json.Marshal(character.HP) - changes = append(changes, stateio.EntryMutation{Bucket: "current_hp", Key: strconv.FormatUint(target.index, 10), Payload: raw}) - info := wire.AppendVarint(nil, 1, target.index) - info = wire.AppendVarint(info, 2, character.HP) - body = wire.AppendBytes(body, 1, info) - } - // CanConsume accounts for a shared stack requested by several characters. - if err := s.inventory.CanConsume(items); err != nil { - return 0, nil, true, err - } - if err := s.inventory.Consume(items); err != nil { - return 0, nil, true, err - } - replyRaw, _ := json.Marshal(foodReply{Digest: digestString, Body: body}) - changes = append(changes, stateio.EntryMutation{Bucket: "food_requests", Key: key, Payload: replyRaw}) - if err := s.store.SaveWithEntries("characters", nil, changes); err != nil { - return 0, nil, true, err - } - return 27, body, true, nil -} - -func (s *FoodService) recoverCharacter(index uint64, items []Item) (Character, error) { - character, found := s.characters.Find(index) - if !found { - return Character{}, errors.New("player: EatFood character is not owned") - } - maximum, err := s.characters.MaxHealth(index) - if err != nil { - return Character{}, err - } - current, err := s.characters.CurrentHealth(index) - if err != nil { - return Character{}, err - } - var total uint64 - seen := map[uint64]bool{} - for _, item := range items { - if item.Type != 5 || item.InvenIndex == 0 || item.Count == 0 || item.Count > math.MaxInt32 || seen[item.InvenIndex] { - return Character{}, errors.New("player: invalid EatFood inventory stack") - } - seen[item.InvenIndex] = true - food, exists := s.design.Foods[item.ID] - if !exists { - return Character{}, errors.New("player: unknown EatFood dish") - } - value, err := food.Recovery(character.ID, maximum, item.Count) - if err != nil { - return Character{}, err - } - if value > math.MaxUint64-total { - return Character{}, errors.New("player: EatFood recovery overflow") - } - total += value - } - if current > maximum { - current = maximum - } - if total >= maximum-current { - character.HP = maximum - } else { - character.HP = current + total - } - return character, nil -} diff --git a/go/internal/server/player/eat_food_test.go b/go/internal/server/player/eat_food_test.go deleted file mode 100644 index 78ec7a9..0000000 --- a/go/internal/server/player/eat_food_test.go +++ /dev/null @@ -1,517 +0,0 @@ -package player - -import ( - "bytes" - "errors" - "path/filepath" - "testing" - - "bd2server/internal/server/accountstate" - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/wire" -) - -func foodTestService(t *testing.T, store stateio.Store) (*FoodService, *Inventory, *CharacterStore) { - t.Helper() - starter := &Starter{Version: "2.35.10"} - inventory, err := OpenInventory(store, starter) - if err != nil { - t.Fatal(err) - } - characters, err := OpenCharacterStore(store, []Character{{InvenIndex: 77, ID: 350, Level: 1, HP: 7}, {InvenIndex: 78, ID: 360, Level: 1, HP: 7}}, inventory, "", "") - if err != nil { - t.Fatal(err) - } - if err = characters.AttachMaxHealth(func(Character) (uint64, error) { return 100, nil }); err != nil { - t.Fatal(err) - } - if err = characters.EnsurePersisted(); err != nil { - t.Fatal(err) - } - design := &gamedata.FoodDesign{Foods: map[uint64]gamedata.Food{101: {ID: 101, Point: 15}, 102: {ID: 102, Type: 1, Point: 10, FavoritePoint: 25, FavoriteUniqueCharIDs: []uint64{35}}, 103: {ID: 103, Type: 2}, 104: {ID: 104, Point: 100, FoodBuffID: 101}, 105: {ID: 105, Point: 50, RecoveryType: 1}}} - s, err := OpenFoodService(store, design, inventory, characters) - if err != nil { - t.Fatal(err) - } - s.BeginSession("food-session") - if err = s.AttachContext(func() (int, error) { return 21, nil }, func() bool { return false }); err != nil { - t.Fatal(err) - } - return s, inventory, characters -} - -func autoFoodRequest(seq uint64, indices []uint64, items []Item) []byte { - out := wire.AppendVarint(nil, 1, seq) - for i, index := range indices { - target := wire.AppendVarint(nil, 1, index) - target = wire.AppendBytes(target, 2, ItemWire(items[i])) - out = wire.AppendBytes(out, 2, target) - } - return out -} - -func TestEatFoodAutoSharesStacksValidatesWholeRequestAndReplaysSQLite(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - s, inventory, characters := foodTestService(t, repo) - stacks, err := inventory.GrantOnce("auto-food", []gamedata.BattleReward{{Type: 5, ID: 102, Count: 3}}) - if err != nil { - t.Fatal(err) - } - for _, index := range []uint64{77, 78} { - if err = characters.SetCurrentHealth(index, 0); err != nil { - t.Fatal(err) - } - } - stack := stacks[0] - stack.Count = 2 - overspend := autoFoodRequest(1, []uint64{77, 78}, []Item{stack, stack}) - if _, _, _, err = s.Handle("/EatFoodAuto", overspend); err == nil { - t.Fatal("accepted aggregate stack overspend") - } - if hp, _ := characters.CurrentHealth(77); hp != 0 { - t.Fatal("partially healed rejected request") - } - if inventory.All()[0].Count != 3 { - t.Fatal("partially consumed rejected request") - } - stack.Count = 1 - request := autoFoodRequest(2, []uint64{77, 78}, []Item{stack, stack}) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - code, body, _, err := s.Handle("/EatFoodAuto", request) - if err != nil || code != 27 { - t.Fatalf("auto code=%d err=%v", code, err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 25 { - t.Fatalf("favorite hp=%d", hp) - } - if hp, _ := characters.CurrentHealth(78); hp != 10 { - t.Fatalf("ordinary hp=%d", hp) - } - var resultIndices []uint64 - if err = wire.Walk(body, func(field wire.Field) error { - index, _, err := wire.Varint(field.Value, 1) - resultIndices = append(resultIndices, index) - return err - }); err != nil || len(resultIndices) != 2 || resultIndices[0] != 77 || resultIndices[1] != 78 { - t.Fatalf("auto response=%v err=%v", resultIndices, err) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - s, inventory, _ = foodTestService(t, repo) - code, replay, _, err := s.Handle("/EatFoodAuto", request) - if err != nil || code != 27 || !bytes.Equal(replay, body) || inventory.All()[0].Count != 1 { - t.Fatalf("auto replay code=%d err=%v", code, err) - } - // Equal sequences on the two endpoints refer to separate requests. - if _, _, _, err = s.Handle("/EatFood", foodRequest(2, 77, 0, stack)); err != nil { - t.Fatalf("normal/auto sequence collision: %v", err) - } -} - -func TestEatFoodAutoRejectsEmptyTargetWithoutConsumingValidTarget(t *testing.T) { - s, inventory, characters := foodTestService(t, stateio.NewMemory()) - stacks, err := inventory.GrantOnce("auto-empty", []gamedata.BattleReward{{Type: 5, ID: 101, Count: 1}}) - if err != nil { - t.Fatal(err) - } - if err = characters.SetCurrentHealth(77, 0); err != nil { - t.Fatal(err) - } - request := autoFoodRequest(1, []uint64{77}, stacks) - request = wire.AppendBytes(request, 2, wire.AppendVarint(nil, 1, 78)) - if _, _, _, err = s.Handle("/EatFoodAuto", request); err == nil { - t.Fatal("accepted empty recovery target") - } - if inventory.All()[0].Count != 1 { - t.Fatal("empty target consumed valid target food") - } - if hp, _ := characters.CurrentHealth(77); hp != 0 { - t.Fatal("empty target partially healed valid target") - } -} - -func foodRequest(seq, index, pack uint64, items ...Item) []byte { - out := wire.AppendVarint(nil, 1, seq) - if pack != 0 { - out = wire.AppendVarint(out, 2, pack) - } - out = wire.AppendVarint(out, 3, index) - for _, item := range items { - out = wire.AppendBytes(out, 4, ItemWire(item)) - } - return out -} - -func TestEatFoodSQLitePersistsRecoveryAndSequenceReplay(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - s, inventory, characters := foodTestService(t, repo) - stacks, err := inventory.GrantOnce("food", []gamedata.BattleReward{{Type: 5, ID: 101, Count: 4}, {Type: 5, ID: 102, Count: 2}}) - if err != nil { - t.Fatal(err) - } - if err = characters.SetCurrentHealth(77, 20); err != nil { - t.Fatal(err) - } - stacks[0].Count = 2 - stacks[1].Count = 1 - request := foodRequest(10, 77, 0, stacks...) - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - code, body, handled, err := s.Handle("/EatFood", request) - if err != nil || code != 22 || !handled { - t.Fatalf("code=%d handled=%v err=%v", code, handled, err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if hp, err := characters.CurrentHealth(77); err != nil || hp != 75 { - t.Fatalf("hp=%d err=%v", hp, err) - } - if c, _ := characters.Find(77); c.HP != 75 || characters.All()[0].HP != 75 { - t.Fatal("current HP overwritten by maximum") - } - if err = repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - s, inventory, characters = foodTestService(t, repo) - code, replayed, _, err := s.Handle("/EatFood", request) - if err != nil || code != 22 || !bytes.Equal(body, replayed) { - t.Fatalf("replay err=%v", err) - } - if items := inventory.All(); len(items) != 2 || items[0].Count != 2 || items[1].Count != 1 { - t.Fatalf("retry consumed items %+v", items) - } - if hp, _ := characters.CurrentHealth(77); hp != 75 { - t.Fatalf("reopened hp=%d", hp) - } - request = foodRequest(10, 77, 0, stacks[0]) - if _, _, _, err = s.Handle("/EatFood", request); err == nil { - t.Fatal("accepted changed request with reused sequence") - } -} - -func TestEatFoodRejectsInvalidStacksAndContextWithoutMutation(t *testing.T) { - s, inventory, characters := foodTestService(t, stateio.NewMemory()) - stacks, err := inventory.GrantOnce("food", []gamedata.BattleReward{{Type: 5, ID: 101, Count: 2}, {Type: 5, ID: 103, Count: 2}, {Type: 5, ID: 104, Count: 2}, {Type: 5, ID: 999, Count: 2}}) - if err != nil { - t.Fatal(err) - } - if err = characters.SetCurrentHealth(77, 0); err != nil { - t.Fatal(err) - } - wrongID := stacks[0] - wrongID.ID = 102 - wrongType := stacks[0] - wrongType.Type = 8 - over := stacks[0] - over.Count = 3 - missing := stacks[0] - missing.InvenIndex++ - requests := [][]byte{foodRequest(1, 999, 0, stacks[0]), foodRequest(2, 77, 22, stacks[0]), foodRequest(3, 77, 0, wrongID), foodRequest(4, 77, 0, wrongType), foodRequest(5, 77, 0, over), foodRequest(6, 77, 0, missing), foodRequest(7, 77, 0, stacks[1]), foodRequest(8, 77, 0, stacks[2]), foodRequest(9, 77, 0, stacks[3]), foodRequest(10, 77, 0, stacks[0], stacks[0]), foodRequest(11, 77, 0)} - for i, request := range requests { - if _, _, _, err := s.Handle("/EatFood", request); err == nil { - t.Fatalf("invalid request %d accepted", i) - } - } - if err := s.AttachContext(func() (int, error) { return 21, nil }, func() bool { return true }); err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/EatFood", foodRequest(12, 77, 0, stacks[0])); err == nil { - t.Fatal("accepted food during battle") - } - if hp, _ := characters.CurrentHealth(77); hp != 0 { - t.Fatalf("rejected request changed HP=%d", hp) - } - if got := inventory.All(); len(got) != 4 || got[0].Count != 2 { - t.Fatalf("rejected request consumed inventory %+v", got) - } -} - -func TestEatFoodPercentageRecoversDeadCharacterAndClampsToMaximum(t *testing.T) { - s, inventory, characters := foodTestService(t, stateio.NewMemory()) - stacks, err := inventory.GrantOnce("food", []gamedata.BattleReward{{Type: 5, ID: 105, Count: 3}}) - if err != nil { - t.Fatal(err) - } - if err = characters.SetCurrentHealth(77, 0); err != nil { - t.Fatal(err) - } - stack := stacks[0] - stack.Count = 1 - if _, _, _, err = s.Handle("/EatFood", foodRequest(1, 77, 21, stack)); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 50 { - t.Fatalf("percentage hp=%d", hp) - } - stack.Count = 2 - if _, _, _, err = s.Handle("/EatFood", foodRequest(2, 77, 21, stack)); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 100 { - t.Fatalf("clamped hp=%d", hp) - } - if len(inventory.All()) != 0 { - t.Fatal("empty food stack remains") - } - if err = characters.SetCurrentHealth(77, 101); err == nil { - t.Fatal("accepted health above maximum") - } -} - -func TestCurrentHealthGrowthAndImmortalClearPersistedInjury(t *testing.T) { - _, inventory, characters := foodTestService(t, stateio.NewMemory()) - stacks, err := inventory.GrantOnce("growth-food-health", []gamedata.BattleReward{{Type: 8, ID: 1, Count: 1}}) - if err != nil { - t.Fatal(err) - } - characters.grow = func(Character, []gamedata.GrowthMaterial) (uint64, uint64, []gamedata.GrowthMaterial, error) { - return 2, 0, nil, nil - } - if err = characters.SetCurrentHealth(77, 3); err != nil { - t.Fatal(err) - } - request := wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 77) - request = wire.AppendBytes(request, 3, ItemWire(stacks[0])) - if _, _, _, err = characters.Handle("/CharGrowth", request); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 100 { - t.Fatalf("grown current HP=%d", hp) - } - if err = characters.SetCurrentHealth(77, 0); err != nil { - t.Fatal(err) - } - request = wire.AppendVarint(wire.AppendVarint(nil, 1, 2), 2, 77) - target, _, _ := wire.Varint(request, 2) - candidate, _ := characters.Find(target) - if err := characters.AttachImmortalDesign(&gamedata.ImmortalDesign{Characters: map[uint64]uint64{candidate.ID: 42}, FullRestore: map[[2]uint64]bool{{42, candidate.TalentLevel}: true}}); err != nil { - t.Fatal(err) - } - if _, _, _, err = characters.Handle("/CharImmortal", request); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 100 { - t.Fatalf("revived current HP=%d", hp) - } -} - -func TestCurrentHealthRetainsSavedCharacterHPWithoutSeparateEntryAcrossReopen(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - seed := []Character{{InvenIndex: 77, ID: 350, Level: 1, HP: 17}, {InvenIndex: 78, ID: 360, Level: 1, HP: 0}} - open := func(store stateio.Store) *CharacterStore { - inventory, err := OpenInventory(store, &Starter{Version: "2.35.10"}) - if err != nil { - t.Fatal(err) - } - characters, err := OpenCharacterStore(store, seed, inventory, "", "") - if err != nil { - t.Fatal(err) - } - if err = characters.AttachMaxHealth(func(Character) (uint64, error) { return 500, nil }); err != nil { - t.Fatal(err) - } - if err = characters.EnsurePersisted(); err != nil { - t.Fatal(err) - } - return characters - } - check := func(characters *CharacterStore) { - for _, saved := range seed { - hp, err := characters.CurrentHealth(saved.InvenIndex) - if err != nil || hp != saved.HP { - t.Fatalf("current %d=%d want=%d err=%v", saved.InvenIndex, hp, saved.HP, err) - } - character, found := characters.Find(saved.InvenIndex) - if !found || character.HP != saved.HP { - t.Fatalf("Find=%+v found=%v", character, found) - } - } - all := characters.All() - if len(all) != 2 || all[0].HP != 17 || all[1].HP != 0 { - t.Fatalf("All=%+v", all) - } - maximum, err := characters.MaxHealth(77) - if err != nil || maximum != 500 { - t.Fatalf("maximum=%d err=%v", maximum, err) - } - } - characters := open(repo) - check(characters) - rows, err := repo.ListEntries("characters", "current_hp") - if err != nil || len(rows) != 0 { - t.Fatalf("reads created entries=%v err=%v", rows, err) - } - if err = repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - characters = open(repo) - check(characters) - if err = characters.SetCurrentHealth(77, 300); err != nil { - t.Fatal(err) - } - if err = characters.SetCurrentHealth(78, 0); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 300 { - t.Fatalf("explicit health entry lost=%d", hp) - } - // Revival must write the restored value, since deleting the entry alone - // would expose the zero HP in the owned character record again. - request := wire.AppendVarint(wire.AppendVarint(nil, 1, 1), 2, 78) - target, _, _ := wire.Varint(request, 2) - candidate, _ := characters.Find(target) - if err := characters.AttachImmortalDesign(&gamedata.ImmortalDesign{Characters: map[uint64]uint64{candidate.ID: 42}, FullRestore: map[[2]uint64]bool{{42, candidate.TalentLevel}: true}}); err != nil { - t.Fatal(err) - } - if _, _, _, err = characters.Handle("/CharImmortal", request); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(78); hp != 500 { - t.Fatalf("revived health=%d", hp) - } -} - -func TestEatFoodRecoversSavedHPWithoutSeparateCurrentHealthEntry(t *testing.T) { - food, inventory, characters := foodTestService(t, stateio.NewMemory()) - // The seed has current HP 7 and maximum 100, with no current_hp entry. - stacks, err := inventory.GrantOnce("saved-health-food", []gamedata.BattleReward{{Type: 5, ID: 101, Count: 1}}) - if err != nil { - t.Fatal(err) - } - if _, _, _, err = food.Handle("/EatFood", foodRequest(1, 77, 21, stacks[0])); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 22 { - t.Fatalf("recovered saved health=%d want22", hp) - } - if len(inventory.All()) != 0 { - t.Fatal("food was not consumed") - } -} - -type failFoodStore struct { - stateio.AtomicEntryStore - fail bool -} - -func (s *failFoodStore) SaveWithEntries(domain string, core []byte, changes []stateio.EntryMutation) error { - if s.fail && domain == "characters" { - return errors.New("injected food save failure") - } - return s.AtomicEntryStore.SaveWithEntries(domain, core, changes) -} - -func TestEatFoodSQLiteRollbackRestoresInventoryHealthAndReplayLedger(t *testing.T) { - path := filepath.Join(t.TempDir(), "state.db") - repo, err := accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - store := &failFoodStore{AtomicEntryStore: repo} - s, inventory, characters := foodTestService(t, store) - stacks, err := inventory.GrantOnce("food", []gamedata.BattleReward{{Type: 5, ID: 101, Count: 1}}) - if err != nil { - t.Fatal(err) - } - if err = characters.SetCurrentHealth(77, 1); err != nil { - t.Fatal(err) - } - store.fail = true - op, err := repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - request := foodRequest(1, 77, 21, stacks[0]) - if _, _, _, err = s.Handle("/EatFood", request); err == nil { - t.Fatal("injected failure was ignored") - } - if err = op.Rollback(); !errors.Is(err, stateio.ErrStateRecoveryRequired) { - t.Fatalf("expected recovery fencing, got %v", err) - } - if err := repo.Close(); err != nil { - t.Fatal(err) - } - repo, err = accountstate.Open(path) - if err != nil { - t.Fatal(err) - } - defer func() { - if err := repo.Close(); err != nil { - t.Error(err) - } - }() - s, inventory, characters = foodTestService(t, repo) - if inventory.All()[0].Count != 1 { - t.Fatal("rollback lost inventory") - } - if hp, _ := characters.CurrentHealth(77); hp != 1 { - t.Fatalf("rollback HP=%d", hp) - } - ledger, err := repo.ListEntries("characters", "food_requests") - if err != nil || len(ledger) != 0 { - t.Fatalf("rollback ledger=%v err=%v", ledger, err) - } - op, err = repo.BeginOperation() - if err != nil { - t.Fatal(err) - } - if _, _, _, err = s.Handle("/EatFood", request); err != nil { - t.Fatal(err) - } - if err = op.Commit(); err != nil { - t.Fatal(err) - } - if hp, _ := characters.CurrentHealth(77); hp != 16 { - t.Fatalf("recovery after rollback HP=%d", hp) - } -} diff --git a/go/internal/server/player/equipment.go b/go/internal/server/player/equipment.go deleted file mode 100644 index f2eab81..0000000 --- a/go/internal/server/player/equipment.go +++ /dev/null @@ -1,2059 +0,0 @@ -package player - -import ( - "encoding/binary" - "encoding/json" - "errors" - "fmt" - "maps" - "math" - "reflect" - "sort" - "strconv" - "strings" - "sync" - "unicode/utf8" - - "bd2server/internal/server/gamedata" - "bd2server/internal/server/stateio" - "bd2server/internal/server/versionconfig" - "bd2server/internal/server/wire" -) - -// Equipment is one server-owned equipment instance. The immutable definition -// (name, icon, slot and base stats) remains in the client's local GameData. -type Equipment struct { - InvenIndex uint64 `json:"inven_index"` - ID uint64 `json:"id"` - Level uint64 `json:"level"` - UseChar uint64 `json:"use_char,omitempty"` - KeepFlag uint64 `json:"keep_flag,omitempty"` - LockFlag uint64 `json:"lock_flag,omitempty"` - SortID uint64 `json:"sort_id,omitempty"` - Mark string `json:"mark,omitempty"` - MainOption []EquipmentOption `json:"main_option,omitempty"` - SubOption []EquipmentOption `json:"sub_option,omitempty"` - PrivateOption *EquipmentOption `json:"private_option,omitempty"` - Rank []uint64 `json:"rank,omitempty"` - UpgradeAttempts uint64 `json:"upgrade_attempts"` -} - -// EquipmentOption is the exact EquipmentOptionTable composite key used by an -// equipment instance. The client resolves the displayed stat curve locally. -type EquipmentOption struct { - GroupID uint64 `json:"group_id"` - ID uint64 `json:"id"` -} - -type equipmentSnapshot struct { - Version string `json:"version"` - NextIndex uint64 `json:"next_index"` - Equipment []Equipment `json:"-"` - Granted map[string]uint64 `json:"-"` -} - -// EquipmentInventory persists non-stackable equipment independently from -// ItemDBInfo inventory because the wire protocols are different types. -type EquipmentInventory struct { - mu sync.Mutex - store stateio.AtomicEntryStore - owned equipmentSnapshot - persisted equipmentSnapshot - corePresent bool - characters *CharacterStore - slots map[uint64]uint64 - upgrade *gamedata.EquipmentUpgradeDesign - craft *gamedata.EquipmentCraftDesign - smelting *gamedata.EquipmentSmeltingDesign - optionReroll *gamedata.EquipmentOptionRerollDesign - statDesign *gamedata.EquipmentStatDesign - wallet *Wallet - inventory *Inventory - sessionID string - smeltCache map[string]smeltingReply - pendingReroll *equipmentOptionRerollPending - presets map[equipmentPresetKey]equipmentPreset -} - -type smeltingReply struct { - code int - body []byte -} - -type equipmentOptionRerollPending struct { - Equipment Equipment `json:"equipment"` -} - -func (s *EquipmentInventory) AttachUpgrade(design *gamedata.EquipmentUpgradeDesign, wallet *Wallet, inventory *Inventory) error { - if design == nil || wallet == nil || inventory == nil { - return errors.New("player: incomplete equipment upgrade configuration") - } - s.mu.Lock() - defer s.mu.Unlock() - s.upgrade, s.wallet, s.inventory = design, wallet, inventory - return nil -} - -func (s *EquipmentInventory) AttachCraft(design *gamedata.EquipmentCraftDesign) error { - if design == nil { - return errors.New("player: incomplete equipment crafting configuration") - } - s.mu.Lock() - defer s.mu.Unlock() - s.craft = design - return nil -} - -func (s *EquipmentInventory) AttachSmelting(design *gamedata.EquipmentSmeltingDesign, wallet *Wallet, inventory *Inventory) error { - if design == nil || wallet == nil || inventory == nil { - return errors.New("player: incomplete equipment smelting configuration") - } - s.mu.Lock() - defer s.mu.Unlock() - s.smelting, s.wallet, s.inventory = design, wallet, inventory - return nil -} - -func (s *EquipmentInventory) AttachOptionReroll(design *gamedata.EquipmentOptionRerollDesign, wallet *Wallet, inventory *Inventory) error { - if design == nil || wallet == nil || inventory == nil { - return errors.New("player: incomplete equipment option reroll configuration") - } - s.mu.Lock() - defer s.mu.Unlock() - s.optionReroll, s.wallet, s.inventory = design, wallet, inventory - if s.pendingReroll != nil { - if err := s.validatePendingRerollLocked(s.pendingReroll); err != nil { - return err - } - } - return nil -} - -// BeginSession scopes the in-memory request replay cache. A repeated protobuf -// sequence in one login must return the first refinement result without a -// second roll or charge. -func (s *EquipmentInventory) BeginSession(id string) { - s.mu.Lock() - defer s.mu.Unlock() - if id == "" || id == s.sessionID { - return - } - s.sessionID = id - s.smeltCache = make(map[string]smeltingReply) -} - -func (s *EquipmentInventory) AttachSlots(slots map[uint64]uint64) error { - if len(slots) == 0 { - return errors.New("player: empty equipment slot design") - } - s.mu.Lock() - defer s.mu.Unlock() - s.slots = make(map[uint64]uint64, len(slots)) - for id, slot := range slots { - if id == 0 || slot > 4 { - return errors.New("player: invalid equipment slot design") - } - s.slots[id] = slot - } - return nil -} - -func (s *EquipmentInventory) AttachCharacters(characters *CharacterStore) error { - if characters == nil { - return errors.New("player: nil character store") - } - known := make(map[uint64]bool) - for _, character := range characters.RawAll() { - known[character.InvenIndex] = true - } - s.mu.Lock() - defer s.mu.Unlock() - for key := range s.presets { - if !known[key.CharacterIndex] { - return fmt.Errorf("player: equipment preset references unknown character %d", key.CharacterIndex) - } - } - s.characters = characters - return nil -} - -func OpenEquipmentInventory(store stateio.Store) (*EquipmentInventory, error) { - entries, ok := store.(stateio.AtomicEntryStore) - if !ok { - return nil, errors.New("player: nil equipment store") - } - s := &EquipmentInventory{store: entries, smeltCache: make(map[string]smeltingReply), presets: make(map[equipmentPresetKey]equipmentPreset), owned: equipmentSnapshot{ - Version: versionconfig.State(), NextIndex: 910000001, Granted: map[string]uint64{}, - }} - data, err := store.Load("equipment") - if err != nil { - return nil, err - } - if data != nil { - s.corePresent = true - if err := stateio.RequireExactJSONObject(data, "version", "next_index"); err != nil { - return nil, fmt.Errorf("player: incompatible equipment layout: %w", err) - } - var shape map[string]json.RawMessage - if err := json.Unmarshal(data, &shape); err != nil { - return nil, fmt.Errorf("player: decode equipment shape: %w", err) - } - for _, name := range []string{"equipment", "granted"} { - if _, exists := shape[name]; exists { - return nil, fmt.Errorf("player: equipment %s must use entries", name) - } - } - if err := json.Unmarshal(data, &s.owned); err != nil { - return nil, fmt.Errorf("player: decode equipment: %w", err) - } - } else if err := stateio.RequireNoEntries(entries, "equipment", "equipment", "granted", "reroll_pending", "presets"); err != nil { - return nil, fmt.Errorf("player: invalid equipment storage: %w", err) - } - if s.owned.Version != versionconfig.State() || s.owned.NextIndex < 910000001 { - return nil, errors.New("player: invalid saved equipment") - } - s.owned.Granted, err = loadUintEntries(entries, "equipment", "granted") - if err != nil { - return nil, err - } - rawEquipment, err := entries.ListEntries("equipment", "equipment") - if err != nil { - return nil, err - } - for key, value := range rawEquipment { - var entry Equipment - index, parseErr := strconv.ParseUint(key, 10, 64) - var shape map[string]json.RawMessage - if parseErr != nil || json.Unmarshal(value, &shape) != nil || json.Unmarshal(value, &entry) != nil || entry.InvenIndex != index { - return nil, fmt.Errorf("player: invalid equipment entry %q", key) - } - if _, present := shape["upgrade_attempts"]; !present { - return nil, errors.New("player: equipment save requires upgrade_attempts; migrate the development save") - } - s.owned.Equipment = append(s.owned.Equipment, entry) - } - sort.Slice(s.owned.Equipment, func(i, j int) bool { return s.owned.Equipment[i].InvenIndex < s.owned.Equipment[j].InvenIndex }) - for _, entry := range s.owned.Equipment { - if len(entry.Rank) != 3 { - return nil, fmt.Errorf("player: equipment %d requires exactly three rank slots, found %d; repair the development save before starting", entry.InvenIndex, len(entry.Rank)) - } - } - pendingEntries, err := entries.ListEntries("equipment", "reroll_pending") - if err != nil { - return nil, err - } - if len(pendingEntries) > 1 { - return nil, errors.New("player: multiple equipment option reroll candidates") - } - if raw, ok := pendingEntries["current"]; ok { - if err := stateio.RequireExactJSONObject(raw, "equipment"); err != nil { - return nil, fmt.Errorf("player: incompatible equipment option reroll candidate: %w", err) - } - var pending equipmentOptionRerollPending - if err := json.Unmarshal(raw, &pending); err != nil { - return nil, fmt.Errorf("player: decode equipment option reroll candidate: %w", err) - } - s.pendingReroll = &pending - if err := s.validatePendingRerollLocked(s.pendingReroll); err != nil { - return nil, err - } - } else if len(pendingEntries) != 0 { - return nil, errors.New("player: invalid equipment option reroll candidate key") - } - if err := s.loadEquipmentPresets(entries); err != nil { - return nil, err - } - s.persisted = cloneEquipmentSnapshot(s.owned) - return s, nil -} - -func (s *EquipmentInventory) EnsurePersisted() error { - s.mu.Lock() - defer s.mu.Unlock() - data, err := s.store.Load("equipment") - if err != nil { - return err - } - if data != nil { - return nil - } - return s.commitLocked(cloneEquipmentSnapshot(s.owned), "initial account generation") -} - -// GrantOnce returns the same instance on a retry, allowing QuestClear response -// retries without duplicating ownership. -func (s *EquipmentInventory) GrantOnce(identity string, equipmentID uint64) (Equipment, error) { - if identity == "" || equipmentID == 0 { - return Equipment{}, errors.New("player: invalid equipment grant") - } - s.mu.Lock() - defer s.mu.Unlock() - if index := s.owned.Granted[identity]; index != 0 { - for _, current := range s.owned.Equipment { - if current.InvenIndex == index { - return current, nil - } - } - return Equipment{}, errors.New("player: equipment grant index is missing") - } - return s.grantLocked(identity, Equipment{ID: equipmentID}) -} - -// GrantGeneratedOnce saves an independently generated gacha instance. Retry -// calls return the original rolls rather than creating a second copy. -func (s *EquipmentInventory) GrantGeneratedOnce(identity string, entry Equipment) (Equipment, error) { - if identity == "" || entry.ID == 0 || len(entry.Rank) != 3 { - return Equipment{}, errors.New("player: invalid generated equipment") - } - s.mu.Lock() - defer s.mu.Unlock() - return s.grantLocked(identity, entry) -} - -func (s *EquipmentInventory) grantLocked(identity string, entry Equipment) (Equipment, error) { - if index := s.owned.Granted[identity]; index != 0 { - for _, current := range s.owned.Equipment { - if current.InvenIndex == index { - return current, nil - } - } - return Equipment{}, errors.New("player: equipment grant index is missing") - } - if len(entry.Rank) == 0 { - // Rank is a fixed three-slot client field (unlock thresholds +3/+6/+9). - // Even an unenhanced item must have three explicit zero values. - entry.Rank = []uint64{0, 0, 0} - } - if len(entry.Rank) != 3 { - return Equipment{}, errors.New("player: equipment requires exactly three rank slots") - } - entry.InvenIndex = s.owned.NextIndex - next := equipmentSnapshot{Version: s.owned.Version, NextIndex: s.owned.NextIndex + 1, - Equipment: append([]Equipment(nil), s.owned.Equipment...), Granted: make(map[string]uint64, len(s.owned.Granted)+1)} - next.Equipment = append(next.Equipment, entry) - maps.Copy(next.Granted, s.owned.Granted) - next.Granted[identity] = entry.InvenIndex - if err := s.commitLocked(next, "grant"); err != nil { - return Equipment{}, err - } - return entry, nil -} - -func EquipmentWire(entry Equipment) []byte { - base := wire.AppendVarint(nil, 1, entry.ID) - if entry.Level != 0 { - base = wire.AppendVarint(base, 2, entry.Level) - } - for _, option := range entry.MainOption { - base = wire.AppendBytes(base, 3, equipmentOptionWire(option)) - } - for _, option := range entry.SubOption { - base = wire.AppendBytes(base, 4, equipmentOptionWire(option)) - } - // The native ResolvedPrivateOption getter dereferences PrivateOption before - // checking IsValid. A present empty message selects its GameData fallback; - // an omitted field leaves the protobuf object null and crashes the UI. - var private []byte - if entry.PrivateOption != nil { - private = equipmentOptionWire(*entry.PrivateOption) - } - base = wire.AppendBytes(base, 5, private) - for _, rank := range entry.Rank { - base = wire.AppendVarint(base, 6, rank) - } - var out []byte - out = wire.AppendVarint(out, 1, entry.InvenIndex) - if entry.UseChar != 0 { - out = wire.AppendVarint(out, 2, entry.UseChar) - } - if entry.KeepFlag != 0 { - out = wire.AppendVarint(out, 3, entry.KeepFlag) - } - if entry.LockFlag != 0 { - out = wire.AppendVarint(out, 4, entry.LockFlag) - } - out = wire.AppendBytes(out, 5, base) - if entry.SortID != 0 { - out = wire.AppendVarint(out, 7, entry.SortID) - } - if entry.Mark != "" { - out = wire.AppendBytes(out, 8, []byte(entry.Mark)) - } - return out -} - -func equipmentOptionWire(option EquipmentOption) []byte { - out := wire.AppendVarint(nil, 1, option.GroupID) - return wire.AppendVarint(out, 2, option.ID) -} - -func (s *EquipmentInventory) Handle(path string, request []byte) (int, []byte, bool, error) { - if path != "/EquipInfo" && path != "/EquipUse" && path != "/EquipClear" && path != "/EquipChange" && path != "/EquipBatchUse" && path != "/EquipPresetInfo" && path != "/EquipPresetSave" && path != "/EquipPresetNameChange" && path != "/EquipUpgrade" && path != "/EquipSequenceUpgrade" && path != "/EquipSmelting" && path != "/EquipSequenceSmelting" && path != "/EquipOptionReRoll" && path != "/EquipOptionReRollConfirm" && path != "/EquipMainOptChange" && path != "/EquipMarkSet" && path != "/EquipMarkDelete" && path != "/EquipLock" && path != "/EquipMaking" && path != "/EquipBreak" && path != "/EquipMakingToBreakAuto" && path != "/EquipUpgradeToBreakAuto" { - return 0, nil, false, nil - } - if seq, found, err := wire.Varint(request, 1); err != nil || !found || seq == 0 { - return 0, nil, true, fmt.Errorf("player: %s invalid sequence", path) - } - if path == "/EquipUse" { - return s.use(request) - } - if path == "/EquipClear" { - return s.clear(request) - } - if path == "/EquipBatchUse" { - return s.batchUse(request) - } - if path == "/EquipPresetInfo" { - return s.presetInfo() - } - if path == "/EquipPresetSave" { - return s.presetSave(request) - } - if path == "/EquipPresetNameChange" { - return s.presetNameChange(request) - } - if path == "/EquipUpgrade" { - return s.upgradeOnce(request) - } - if path == "/EquipSequenceUpgrade" { - return s.upgradeSequence(request) - } - if path == "/EquipMaking" { - return s.makeEquipment(request) - } - if path == "/EquipBreak" { - return s.breakEquipment(request) - } - if path == "/EquipMakingToBreakAuto" { - return s.makeToBreakAuto(request) - } - if path == "/EquipUpgradeToBreakAuto" { - return s.upgradeToBreakAuto(request) - } - if path == "/EquipSmelting" { - return s.smeltOnce(request) - } - if path == "/EquipSequenceSmelting" { - return s.smeltSequence(request) - } - if path == "/EquipOptionReRoll" { - return s.optionRerollRequest(request) - } - if path == "/EquipOptionReRollConfirm" { - return s.optionRerollConfirm(request) - } - if path == "/EquipMainOptChange" { - return s.mainOptionChange(request) - } - if path == "/EquipChange" { - return s.change(request) - } - if path == "/EquipMarkSet" || path == "/EquipMarkDelete" { - return s.mark(path, request) - } - if path == "/EquipLock" { - return s.lock(request) - } - s.mu.Lock() - defer s.mu.Unlock() - var response []byte - for _, entry := range s.owned.Equipment { - response = wire.AppendBytes(response, 1, EquipmentWire(entry)) - } - if s.pendingReroll != nil { - response = wire.AppendBytes(response, 2, EquipmentWire(s.pendingReroll.Equipment)) - } - return 34, response, true, nil -} - -func (s *EquipmentInventory) optionRerollRequest(request []byte) (int, []byte, bool, error) { - seq, _, _ := wire.Varint(request, 1) - index, found, err := wire.Varint(request, 2) - if err != nil || !found || index == 0 { - return 0, nil, true, errors.New("player: EquipOptionReRoll missing equipment") - } - mainLocks, err := repeatedBoolField(request, 3, "EquipOptionReRoll main lock") - if err != nil { - return 0, nil, true, err - } - subLocks, err := repeatedBoolField(request, 4, "EquipOptionReRoll sub lock") - if err != nil { - return 0, nil, true, err - } - materials, err := equipmentRequestItems(request, 5, "EquipOptionReRoll") - if err != nil { - return 0, nil, true, err - } - rerollType, _, err := wire.Varint(request, 6) - if err != nil || rerollType > 1 { - return 0, nil, true, errors.New("player: EquipOptionReRoll unsupported reroll type") - } - - s.mu.Lock() - defer s.mu.Unlock() - cacheKey := s.smeltingCacheKey("option-reroll", seq) - if reply, ok := s.smeltCache[cacheKey]; ok { - return reply.code, append([]byte(nil), reply.body...), true, nil - } - if s.optionReroll == nil || s.wallet == nil || s.inventory == nil { - return 0, nil, true, errors.New("player: equipment option reroll unavailable") - } - position := s.equipmentPositionLocked(index) - if position < 0 { - return 0, nil, true, fmt.Errorf("player: EquipOptionReRoll unknown equipment %d", index) - } - current := s.owned.Equipment[position] - base := current - if s.pendingReroll != nil { - if s.pendingReroll.Equipment.InvenIndex != index { - return 0, nil, true, errors.New("player: another equipment option reroll is awaiting confirmation") - } - // A retry from the result screen carries only lock masks. The values - // being locked are therefore the last candidate, not the still-official - // equipment options. Keep chaining candidates until confirm/keep clears - // the pending result. - base = s.pendingReroll.Equipment - } - definition, ok := s.optionReroll.Lookup(current.ID) - if !ok { - return 0, nil, true, fmt.Errorf("player: equipment %d has no option reroll design", current.ID) - } - if len(mainLocks) != len(base.MainOption) || len(mainLocks) != len(definition.MainGroups) || - len(subLocks) != len(base.SubOption) || len(subLocks) != len(definition.SubGroups) { - return 0, nil, true, errors.New("player: EquipOptionReRoll lock arrays do not match equipment options") - } - effectiveMainLocks := append([]bool(nil), mainLocks...) - if rerollType == 1 && len(effectiveMainLocks) != 0 { - // The client's has-another-option mode asks the server to leave the - // first main option alone. It is not a paid lock and remains false in - // the request mask. - effectiveMainLocks[0] = true - } - lockedCount, unlockedRerollable := uint64(0), uint64(0) - for i, locked := range mainLocks { - if base.MainOption[i].GroupID != definition.MainGroups[i] || base.MainOption[i].ID == 0 { - return 0, nil, true, fmt.Errorf("player: equipment %d main option %d does not match GameData", index, i) - } - canReroll := len(s.optionReroll.Groups[definition.MainGroups[i]].Choices) >= 2 - if locked && (i == 0 || !canReroll) { - return 0, nil, true, fmt.Errorf("player: EquipOptionReRoll main option %d cannot be locked", i) - } - if locked { - lockedCount++ - } else if canReroll && !(rerollType == 1 && i == 0) { //nolint:staticcheck // QF1001 - unlockedRerollable++ - } - } - for i, locked := range subLocks { - if base.SubOption[i].GroupID != definition.SubGroups[i] || base.SubOption[i].ID == 0 { - return 0, nil, true, fmt.Errorf("player: equipment %d sub option %d does not match GameData", index, i) - } - canReroll := len(s.optionReroll.Groups[definition.SubGroups[i]].Choices) >= 2 - if locked && !canReroll { - return 0, nil, true, fmt.Errorf("player: EquipOptionReRoll sub option %d cannot be locked", i) - } - if locked { - lockedCount++ - } else if canReroll { - unlockedRerollable++ - } - } - if unlockedRerollable == 0 { - return 0, nil, true, errors.New("player: EquipOptionReRoll must leave a rerollable option unlocked") - } - costs, err := s.optionReroll.Cost(current.ID, lockedCount) - if err != nil { - return 0, nil, true, err - } - gold, consumed, err := validateOptionRerollMaterials(costs, materials, s.optionReroll.Conversion) - if err != nil { - return 0, nil, true, err - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - return 0, nil, true, errors.New("player: insufficient gold for equipment option reroll") - } - if len(consumed) != 0 { - if err := s.inventory.CanConsume(consumed); err != nil { - return 0, nil, true, err - } - } - privateLocks := make([]bool, len(definition.PrivateGroups)) - for i := range privateLocks { - privateLocks[i] = true - } - rolled, err := s.optionReroll.RollUnlocked(current.ID, gamedata.EquipmentOptionRerollLocks{Main: effectiveMainLocks, Sub: subLocks, Private: privateLocks}) - if err != nil { - return 0, nil, true, err - } - candidate := cloneEquipment(base) - for i := range candidate.MainOption { - if !effectiveMainLocks[i] { - candidate.MainOption[i] = EquipmentOption{GroupID: rolled.Main[i].GroupID, ID: rolled.Main[i].ID} - } - } - for i := range candidate.SubOption { - if !subLocks[i] { - candidate.SubOption[i] = EquipmentOption{GroupID: rolled.Sub[i].GroupID, ID: rolled.Sub[i].ID} - } - } - pending := &equipmentOptionRerollPending{Equipment: candidate} - if err := s.commitOptionRerollLocked(pending, consumed, gold, "equip-option-reroll:"+cacheKey); err != nil { - return 0, nil, true, err - } - var response []byte - for _, option := range candidate.MainOption { - response = wire.AppendBytes(response, 1, equipmentOptionWire(option)) - } - for _, option := range candidate.SubOption { - response = wire.AppendBytes(response, 2, equipmentOptionWire(option)) - } - s.smeltCache[cacheKey] = smeltingReply{code: 192, body: append([]byte(nil), response...)} - return 192, response, true, nil -} - -func (s *EquipmentInventory) optionRerollConfirm(request []byte) (int, []byte, bool, error) { - seq, _, _ := wire.Varint(request, 1) - index, found, err := wire.Varint(request, 2) - if err != nil || !found || index == 0 { - return 0, nil, true, errors.New("player: EquipOptionReRollConfirm missing equipment") - } - confirm, err := optionalBoolField(request, 3, "EquipOptionReRollConfirm confirm") - if err != nil { - return 0, nil, true, err - } - s.mu.Lock() - cacheKey := s.smeltingCacheKey("option-reroll-confirm", seq) - if reply, ok := s.smeltCache[cacheKey]; ok { - s.mu.Unlock() - return reply.code, append([]byte(nil), reply.body...), true, nil - } - if s.pendingReroll == nil || s.pendingReroll.Equipment.InvenIndex != index { - s.mu.Unlock() - return 0, nil, true, fmt.Errorf("player: equipment %d has no option reroll awaiting confirmation", index) - } - position := s.equipmentPositionLocked(index) - if position < 0 { - s.mu.Unlock() - return 0, nil, true, fmt.Errorf("player: EquipOptionReRollConfirm unknown equipment %d", index) - } - next := cloneEquipmentSnapshot(s.owned) - if confirm { - next.Equipment[position].MainOption = append([]EquipmentOption(nil), s.pendingReroll.Equipment.MainOption...) - next.Equipment[position].SubOption = append([]EquipmentOption(nil), s.pendingReroll.Equipment.SubOption...) - } - if err := s.persistEquipmentState(next, nil, true, "option reroll confirm"); err != nil { - s.mu.Unlock() - return 0, nil, true, err - } - s.owned = next - s.persisted = cloneEquipmentSnapshot(next) - s.pendingReroll = nil - entry := cloneEquipment(next.Equipment[position]) - s.mu.Unlock() - response := wire.AppendBytes(nil, 1, EquipmentWire(entry)) - if character, ok := s.equippedCharacter(entry); ok { - response = wire.AppendBytes(response, 2, CharacterWire(character)) - } - s.mu.Lock() - s.smeltCache[cacheKey] = smeltingReply{code: 193, body: append([]byte(nil), response...)} - s.mu.Unlock() - return 193, response, true, nil -} - -func (s *EquipmentInventory) mainOptionChange(request []byte) (int, []byte, bool, error) { - index, found, err := wire.Varint(request, 2) - if err != nil || !found || index == 0 { - return 0, nil, true, errors.New("player: EquipMainOptChange missing equipment") - } - groupID, groupFound, err := wire.Varint(request, 3) - if err != nil || !groupFound || groupID == 0 { - return 0, nil, true, errors.New("player: EquipMainOptChange missing option group") - } - optionID, optionFound, err := wire.Varint(request, 4) - if err != nil || !optionFound || optionID == 0 { - return 0, nil, true, errors.New("player: EquipMainOptChange missing option") - } - - s.mu.Lock() - locked := true - defer func() { - if locked { - s.mu.Unlock() - } - }() - if s.optionReroll == nil { - return 0, nil, true, errors.New("player: equipment option design unavailable") - } - position := s.equipmentPositionLocked(index) - if position < 0 { - return 0, nil, true, fmt.Errorf("player: EquipMainOptChange unknown equipment %d", index) - } - current := s.owned.Equipment[position] - definition, ok := s.optionReroll.Lookup(current.ID) - if !ok || definition.PrivateUniqueCharID == 0 || len(definition.MainGroups) == 0 || len(current.MainOption) == 0 { - return 0, nil, true, fmt.Errorf("player: equipment %d has no changeable main option", current.ID) - } - group, ok := s.optionReroll.Groups[groupID] - if !ok || groupID != definition.MainGroups[0] || current.MainOption[0].GroupID != groupID || len(group.Choices) < 2 || !optionChoiceExists(group, optionID) { - return 0, nil, true, fmt.Errorf("player: equipment %d main option %d/%d is not allowed", current.ID, groupID, optionID) - } - - next := cloneEquipmentSnapshot(s.owned) - next.Equipment[position].MainOption[0] = EquipmentOption{GroupID: groupID, ID: optionID} - pending := s.pendingReroll - pendingDirty := false - if pending != nil && pending.Equipment.InvenIndex == index { - copy := &equipmentOptionRerollPending{Equipment: cloneEquipment(pending.Equipment)} - if len(copy.Equipment.MainOption) == 0 || copy.Equipment.MainOption[0].GroupID != groupID { - return 0, nil, true, errors.New("player: option reroll candidate has no matching main option") - } - copy.Equipment.MainOption[0] = EquipmentOption{GroupID: groupID, ID: optionID} - pending = copy - pendingDirty = true - } - if err := s.persistEquipmentState(next, pending, pendingDirty, "main option change"); err != nil { - return 0, nil, true, err - } - s.owned = next - s.persisted = cloneEquipmentSnapshot(next) - if pendingDirty { - s.pendingReroll = pending - } - // Character health reads account equipment through pictorial ownership. - // Release the equipment lock before resolving the equipped character. - s.mu.Unlock() - locked = false - var response []byte - if character, ok := s.equippedCharacter(current); ok { - response = wire.AppendBytes(response, 1, CharacterWire(character)) - } - return 537, response, true, nil -} - -func repeatedBoolField(data []byte, number int, name string) ([]bool, error) { - var result []bool - err := wire.Walk(data, func(field wire.Field) error { - if field.Number != number { - return nil - } - switch field.Type { - case 0: - value, count := binary.Uvarint(field.Value) - if count <= 0 || value > 1 { - return fmt.Errorf("player: %s is invalid", name) - } - result = append(result, value == 1) - case 2: - for offset := 0; offset < len(field.Value); { - value, count := binary.Uvarint(field.Value[offset:]) - if count <= 0 || value > 1 { - return fmt.Errorf("player: %s is invalid", name) - } - result = append(result, value == 1) - offset += count - } - default: - return fmt.Errorf("player: %s is invalid", name) - } - return nil - }) - return result, err -} - -func optionalBoolField(data []byte, number int, name string) (bool, error) { - value := false - seen := false - err := wire.Walk(data, func(field wire.Field) error { - if field.Number != number { - return nil - } - if seen || field.Type != 0 { - return fmt.Errorf("player: %s is invalid", name) - } - raw, count := binary.Uvarint(field.Value) - if count <= 0 || raw > 1 { - return fmt.Errorf("player: %s is invalid", name) - } - seen = true - value = raw == 1 - return nil - }) - return value, err -} - -const ( - equipUpgradeSuccess = iota - equipUpgradeFail - equipUpgradeStopMaxLevel - equipUpgradeStopSuccess - equipUpgradeStopNotEnough - equipUpgradeStopGoldLimit - equipUpgradeStopTargetLevel - equipUpgradeStopMaxTryCount -) - -func (s *EquipmentInventory) upgradeOnce(request []byte) (int, []byte, bool, error) { - index, found, err := wire.Varint(request, 2) - if err != nil || !found || index == 0 { - return 0, nil, true, errors.New("player: EquipUpgrade missing equipment") - } - var materials []Item - err = wire.Walk(request, func(field wire.Field) error { - if field.Number != 3 { - return nil - } - if field.Type != 2 { - return errors.New("player: EquipUpgrade invalid material") - } - var item Item - if err := decodeVarints(field.Value, map[int]*uint64{1: &item.InvenIndex, 2: &item.ID, 3: &item.Type, 4: &item.Count, 5: &item.KeepFlag, 6: &item.TimeValue, 8: &item.ExpiryTime, 9: &item.SortID, 10: &item.UseCount}); err != nil { - return err - } - if item.Type == 0 || item.Count == 0 || (item.Type == 4 && (item.ID != 0 || item.InvenIndex != 0)) || (item.Type != 4 && (item.ID == 0 || item.InvenIndex == 0)) { - return errors.New("player: EquipUpgrade invalid material") - } - materials = append(materials, item) - return nil - }) - if err != nil { - return 0, nil, true, err - } - if len(materials) == 0 { - return 0, nil, true, errors.New("player: EquipUpgrade has no material") - } - s.mu.Lock() - entry, success, _, _, err := s.attemptUpgradeLocked(index, materials) - s.mu.Unlock() - if err != nil { - return 0, nil, true, err - } - result := uint64(equipUpgradeFail) - if success { - result = equipUpgradeSuccess - } - response := wire.AppendBytes(nil, 1, EquipmentWire(entry)) - if result != 0 { - response = wire.AppendVarint(response, 2, result) - } - if character, ok := s.equippedCharacter(entry); ok { - response = wire.AppendBytes(response, 3, CharacterWire(character)) - } - return 37, response, true, nil -} - -func (s *EquipmentInventory) upgradeSequence(request []byte) (int, []byte, bool, error) { - index, found, err := wire.Varint(request, 2) - if err != nil || !found || index == 0 { - return 0, nil, true, errors.New("player: EquipSequenceUpgrade missing equipment") - } - count, found, err := wire.Varint(request, 3) - if err != nil || !found || count == 0 || count > 100000 { - return 0, nil, true, errors.New("player: EquipSequenceUpgrade invalid attempt count") - } - goldLimit, _, err := wire.Varint(request, 5) - if err != nil { - return 0, nil, true, errors.New("player: EquipSequenceUpgrade invalid gold limit") - } - target, _, err := wire.Varint(request, 6) - if err != nil { - return 0, nil, true, errors.New("player: EquipSequenceUpgrade invalid target") - } - s.mu.Lock() - if s.upgrade == nil || s.wallet == nil { - s.mu.Unlock() - return 0, nil, true, errors.New("player: equipment upgrade unavailable") - } - position := s.equipmentPositionLocked(index) - if position < 0 { - s.mu.Unlock() - return 0, nil, true, fmt.Errorf("player: EquipSequenceUpgrade unknown equipment %d", index) - } - maximum := s.upgrade.MaxLevel[s.owned.Equipment[position].ID] - if target == 0 || target > maximum { - target = maximum - } - var attempts, usedGold uint64 - result := uint64(equipUpgradeStopMaxTryCount) - var consumed []Item - var lack []Item - for attempts < count { - entry := s.owned.Equipment[position] - if entry.Level >= maximum { - result = equipUpgradeStopMaxLevel - break - } - if entry.Level >= target { - result = equipUpgradeStopTargetLevel - break - } - level, _, designErr := s.upgrade.Level(entry.ID, entry.Level) - if designErr != nil { - s.mu.Unlock() - return 0, nil, true, designErr - } - materials, gold, costErr := s.selectUpgradeCosts(level.Costs) - if costErr != nil { - result = equipUpgradeStopNotEnough - lack = upgradeLackItems(level.Costs) - break - } - if goldLimit != 0 && usedGold+gold > goldLimit { - result = equipUpgradeStopGoldLimit - break - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - result = equipUpgradeStopNotEnough - lack = upgradeLackItems(level.Costs) - break - } - updated, success, spent, actual, attemptErr := s.attemptUpgradeLocked(index, materials) - if attemptErr != nil { - s.mu.Unlock() - return 0, nil, true, attemptErr - } - attempts++ - usedGold += spent - consumed = append(consumed, actual...) - position = s.equipmentPositionLocked(index) - if success && updated.Level >= target { - result = equipUpgradeStopTargetLevel - if updated.Level >= maximum { - result = equipUpgradeStopMaxLevel - } - break - } - } - entry := s.owned.Equipment[position] - s.mu.Unlock() - response := wire.AppendBytes(nil, 1, EquipmentWire(entry)) - if character, ok := s.equippedCharacter(entry); ok { - response = wire.AppendBytes(response, 2, CharacterWire(character)) - } - response = wire.AppendVarint(response, 3, result) - response = wire.AppendVarint(response, 4, attempts) - for _, item := range consumed { - response = wire.AppendBytes(response, 5, ItemWire(item)) - } - for _, item := range lack { - response = wire.AppendBytes(response, 6, ItemWire(item)) - } - if usedGold != 0 { - response = wire.AppendVarint(response, 7, usedGold) - } - return 176, response, true, nil -} - -func (s *EquipmentInventory) smeltOnce(request []byte) (int, []byte, bool, error) { - seq, _, _ := wire.Varint(request, 1) - index, found, err := wire.Varint(request, 2) - if err != nil || !found || index == 0 { - return 0, nil, true, errors.New("player: EquipSmelting missing equipment") - } - materials, err := equipmentRequestItems(request, 3, "EquipSmelting") - if err != nil { - return 0, nil, true, err - } - s.mu.Lock() - locked := true - defer func() { - if locked { - s.mu.Unlock() - } - }() - cacheKey := s.smeltingCacheKey("single", seq) - if reply, ok := s.smeltCache[cacheKey]; ok { - return reply.code, append([]byte(nil), reply.body...), true, nil - } - position, current, err := s.smeltingEquipmentLocked(index) - if err != nil { - return 0, nil, true, err - } - costs, err := s.smelting.Cost(current.ID) - if err != nil { - return 0, nil, true, err - } - gold, _, err := validateSmeltingMaterials(costs, materials) - if err != nil { - return 0, nil, true, err - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - return 0, nil, true, errors.New("player: insufficient gold for equipment smelting") - } - var itemMaterials []Item - var consumedMileageMaterial uint64 - for _, item := range materials { - if item.Type != 4 { - itemMaterials = append(itemMaterials, item) - } - if item.Type == s.smelting.Mileage.UseType && item.ID == s.smelting.Mileage.UseID { - consumedMileageMaterial += item.Count - } - } - if len(itemMaterials) != 0 { - if err := s.inventory.CanConsume(itemMaterials); err != nil { - return 0, nil, true, err - } - } - candidate, err := s.smelting.RollCandidate(current.ID) - if err != nil { - return 0, nil, true, err - } - currentScore, err := s.smelting.Score(current.ID, current.Rank) - if err != nil { - return 0, nil, true, err - } - candidateScore, err := s.smelting.Score(current.ID, candidate) - if err != nil { - return 0, nil, true, err - } - success := candidateScore > currentScore - next := cloneEquipmentSnapshot(s.owned) - if success { - next.Equipment[position].Rank = append([]uint64(nil), candidate...) - } - currency, earned, err := s.commitSmeltingLocked( - next, itemMaterials, gold, consumedMileageMaterial, - "equip-smelting:"+cacheKey, "smelting") - if err != nil { - return 0, nil, true, err - } - current = next.Equipment[position] - response := wire.AppendBytes(nil, 1, EquipmentWire(current)) - s.mu.Unlock() - locked = false - if character, ok := s.equippedCharacter(current); ok { - response = wire.AppendBytes(response, 2, CharacterWire(character)) - } - if !success { - response = wire.AppendVarint(response, 3, equipUpgradeFail) - for _, rank := range candidate { - response = wire.AppendVarint(response, 4, rank) - } - } - response = appendSmeltingMileage(response, 5, 6, currency.EquipMileageExchangeGage, s.smelting.Mileage, earned) - s.mu.Lock() - s.smeltCache[cacheKey] = smeltingReply{code: 105, body: append([]byte(nil), response...)} - s.mu.Unlock() - return 105, response, true, nil -} - -func (s *EquipmentInventory) smeltSequence(request []byte) (int, []byte, bool, error) { - seq, _, _ := wire.Varint(request, 1) - index, found, err := wire.Varint(request, 2) - if err != nil || !found || index == 0 { - return 0, nil, true, errors.New("player: EquipSequenceSmelting missing equipment") - } - count, found, err := wire.Varint(request, 3) - if err != nil || !found || count == 0 { - return 0, nil, true, errors.New("player: EquipSequenceSmelting invalid attempt count") - } - target, _, err := wire.Varint(request, 4) - if err != nil { - return 0, nil, true, errors.New("player: EquipSequenceSmelting invalid target score") - } - s.mu.Lock() - locked := true - defer func() { - if locked { - s.mu.Unlock() - } - }() - cacheKey := s.smeltingCacheKey("sequence", seq) - if reply, ok := s.smeltCache[cacheKey]; ok { - return reply.code, append([]byte(nil), reply.body...), true, nil - } - position, current, err := s.smeltingEquipmentLocked(index) - if err != nil { - return 0, nil, true, err - } - if count > s.smelting.MaxStreak { - return 0, nil, true, fmt.Errorf("player: EquipSequenceSmelting attempt count %d exceeds %d", count, s.smelting.MaxStreak) - } - maximumRanks, err := s.smelting.MaximumRanks(current.ID) - if err != nil { - return 0, nil, true, err - } - maximumScore, err := s.smelting.Score(current.ID, maximumRanks) - if err != nil { - return 0, nil, true, err - } - if target > maximumScore { - return 0, nil, true, fmt.Errorf("player: EquipSequenceSmelting target %d exceeds %d", target, maximumScore) - } - costs, err := s.smelting.Cost(current.ID) - if err != nil { - return 0, nil, true, err - } - currentScore, err := s.smelting.Score(current.ID, current.Rank) - if err != nil { - return 0, nil, true, err - } - // The 2.35.10 client splits a requested sequence into packets of at most - // 1000 attempts. Exhausting this packet is not the user's global max-try - // stop: UPGRADE_SUCCESS tells the client to send the next chunk. Terminal - // stop values are reserved for target/max score and insufficient resources. - result := uint64(equipUpgradeSuccess) - var attempts, successes uint64 - if currentScore >= maximumScore { - result = equipUpgradeStopMaxLevel - } else if target != 0 && currentScore >= target { - result = equipUpgradeStopTargetLevel - } - for attempts < count && result == equipUpgradeSuccess { - if _, _, selectErr := s.selectSmeltingCosts(costs, attempts+1); selectErr != nil { - result = equipUpgradeStopNotEnough - break - } - candidate, rollErr := s.smelting.RollCandidate(current.ID) - if rollErr != nil { - return 0, nil, true, rollErr - } - candidateScore, scoreErr := s.smelting.Score(current.ID, candidate) - if scoreErr != nil { - return 0, nil, true, scoreErr - } - attempts++ - if candidateScore > currentScore { - current.Rank = append([]uint64(nil), candidate...) - currentScore = candidateScore - successes++ - } - if currentScore >= maximumScore { - result = equipUpgradeStopMaxLevel - } else if target != 0 && currentScore >= target { - result = equipUpgradeStopTargetLevel - } - } - var consumed, lack []Item - var gold, mileageMaterial, earned uint64 - currency := s.wallet.Snapshot() - if attempts != 0 { - consumed, gold, err = s.selectSmeltingCosts(costs, attempts) - if err != nil { - return 0, nil, true, err - } - for _, item := range consumed { - if item.Type == s.smelting.Mileage.UseType && item.ID == s.smelting.Mileage.UseID { - mileageMaterial += item.Count - } - } - var itemMaterials []Item - for _, item := range consumed { - if item.Type != 4 { - itemMaterials = append(itemMaterials, item) - } - } - next := cloneEquipmentSnapshot(s.owned) - next.Equipment[position].Rank = append([]uint64(nil), current.Rank...) - currency, earned, err = s.commitSmeltingLocked( - next, itemMaterials, gold, mileageMaterial, - "equip-sequence-smelting:"+cacheKey, "sequence smelting") - if err != nil { - return 0, nil, true, err - } - current = next.Equipment[position] - } - // NotEnough is used only when the next requested attempt could not be - // funded. Exhausting this packet retains Success so the client can continue - // a sequence whose total requested count exceeds the 1000-attempt chunk. - if result == equipUpgradeStopNotEnough { - lack = upgradeLackItems(costs) - } - response := wire.AppendBytes(nil, 1, EquipmentWire(current)) - s.mu.Unlock() - locked = false - if character, ok := s.equippedCharacter(current); ok { - response = wire.AppendBytes(response, 2, CharacterWire(character)) - } - response = wire.AppendVarint(response, 3, result) - response = wire.AppendVarint(response, 4, attempts) - for _, item := range consumed { - response = wire.AppendBytes(response, 5, ItemWire(item)) - } - for _, item := range lack { - response = wire.AppendBytes(response, 6, ItemWire(item)) - } - response = appendSmeltingMileage(response, 7, 8, currency.EquipMileageExchangeGage, s.smelting.Mileage, earned) - if successes != 0 { - response = wire.AppendVarint(response, 9, successes) - } - s.mu.Lock() - s.smeltCache[cacheKey] = smeltingReply{code: 177, body: append([]byte(nil), response...)} - s.mu.Unlock() - return 177, response, true, nil -} - -func (s *EquipmentInventory) smeltingCacheKey(kind string, seq uint64) string { - return kind + ":" + s.sessionID + ":seq:" + strconv.FormatUint(seq, 10) -} - -func (s *EquipmentInventory) smeltingEquipmentLocked(index uint64) (int, Equipment, error) { - if s.smelting == nil || s.wallet == nil || s.inventory == nil { - return -1, Equipment{}, errors.New("player: equipment smelting unavailable") - } - position := s.equipmentPositionLocked(index) - if position < 0 { - return -1, Equipment{}, fmt.Errorf("player: unknown equipment %d", index) - } - entry := s.owned.Equipment[position] - design, ok := s.smelting.Equipment[entry.ID] - if !ok || entry.Level != design.MaxLevel || len(entry.Rank) != 3 { - return -1, Equipment{}, fmt.Errorf("player: equipment %d is not ready for smelting", index) - } - if _, err := s.smelting.Score(entry.ID, entry.Rank); err != nil { - return -1, Equipment{}, fmt.Errorf("player: equipment %d has invalid smelting rank: %w", index, err) - } - return position, entry, nil -} - -func equipmentRequestItems(request []byte, number int, operation string) ([]Item, error) { - var result []Item - err := wire.Walk(request, func(field wire.Field) error { - if field.Number != number { - return nil - } - if field.Type != 2 { - return fmt.Errorf("player: %s invalid material", operation) - } - var item Item - if err := decodeVarints(field.Value, map[int]*uint64{1: &item.InvenIndex, 2: &item.ID, 3: &item.Type, 4: &item.Count, 5: &item.KeepFlag, 6: &item.TimeValue, 8: &item.ExpiryTime, 9: &item.SortID, 10: &item.UseCount}); err != nil { - return err - } - if item.Type == 0 || item.Count == 0 || (item.Type == 4 && (item.ID != 0 || item.InvenIndex != 0)) || (item.Type != 4 && (item.ID == 0 || item.InvenIndex == 0)) { - return fmt.Errorf("player: %s invalid material", operation) - } - result = append(result, item) - return nil - }) - if err != nil { - return nil, err - } - if len(result) == 0 { - return nil, fmt.Errorf("player: %s has no material", operation) - } - return result, nil -} - -func validateSmeltingMaterials(costs []gamedata.PromotionCost, materials []Item) (gold, mileageMaterial uint64, err error) { - want := make(map[[2]uint64]uint64, len(costs)) - for _, cost := range costs { - want[[2]uint64{cost.Type, cost.ID}] += cost.Count - } - got := make(map[[2]uint64]uint64, len(materials)) - for _, item := range materials { - got[[2]uint64{item.Type, item.ID}] += item.Count - if item.Type == 4 { - gold += item.Count - } else { - mileageMaterial += item.Count - } - } - if len(got) != len(want) { - return 0, 0, errors.New("player: equipment smelting material kinds mismatch") - } - for key, count := range want { - if got[key] != count { - return 0, 0, fmt.Errorf("player: equipment smelting material %d/%d=%d want=%d", key[0], key[1], got[key], count) - } - } - return gold, mileageMaterial, nil -} - -func (s *EquipmentInventory) selectSmeltingCosts(costs []gamedata.PromotionCost, attempts uint64) ([]Item, uint64, error) { - var result []Item - var gold uint64 - for _, cost := range costs { - if attempts != 0 && cost.Count > ^uint64(0)/attempts { - return nil, 0, errors.New("player: equipment smelting cost overflow") - } - count := cost.Count * attempts - switch cost.Type { - case 4: - if cost.ID != 0 || gold != 0 || !s.wallet.CanSpendGold(count) { - return nil, 0, errors.New("player: insufficient equipment smelting gold") - } - gold = count - result = append(result, Item{Type: 4, Count: count}) - case 8: - items, err := s.inventory.SelectMutable(cost.Type, cost.ID, count) - if err != nil { - return nil, 0, err - } - result = append(result, items...) - default: - return nil, 0, fmt.Errorf("player: unsupported equipment smelting cost type %d", cost.Type) - } - } - return result, gold, nil -} - -func appendSmeltingMileage(response []byte, gaugeField, rewardField int, gauge uint64, mileage gamedata.EquipmentSmeltingMileage, earned uint64) []byte { - if gauge != 0 { - response = wire.AppendVarint(response, gaugeField, gauge) - } - if earned != 0 { - reward := ItemWire(Item{ID: mileage.RewardID, Type: mileage.RewardType, Count: earned}) - bundle := wire.AppendBytes(nil, 1, reward) - response = wire.AppendBytes(response, rewardField, bundle) - } - return response -} - -// commitSmeltingLocked keeps refinement's three typed snapshots synchronized -// in memory. Caller holds equipment.mu; this method takes the remaining locks -// in wallet -> inventory order, calculates every candidate before writing, -// then publishes all three only after their saves succeed. The surrounding -// request transaction supplies durable all-or-none recovery across the writes. -func (s *EquipmentInventory) commitSmeltingLocked( - nextEquipment equipmentSnapshot, - consumed []Item, - gold, mileageMaterial uint64, - identity, operation string, -) (Currency, uint64, error) { - if identity == "" || mileageMaterial == 0 || s.wallet == nil || s.inventory == nil { - return Currency{}, 0, errors.New("player: invalid transactional equipment smelting") - } - s.wallet.mu.Lock() - defer s.wallet.mu.Unlock() - s.inventory.mu.Lock() - defer s.inventory.mu.Unlock() - - nextWallet := cloneWallet(s.wallet.state) - if nextWallet.Spent[identity] { - return Currency{}, 0, errors.New("player: equipment smelting request was already committed") - } - if nextWallet.Gold < gold { - return Currency{}, 0, errors.New("player: insufficient gold for equipment smelting") - } - nextWallet.Gold -= gold - nextWallet.Spent[identity] = true - threshold, rewardCount := s.smelting.Mileage.UseCount, s.smelting.Mileage.RewardCount - if threshold == 0 || rewardCount == 0 || nextWallet.EquipMileageExchangeGage >= threshold || - math.MaxUint64-nextWallet.EquipMileageExchangeGage < mileageMaterial { - return Currency{}, 0, errors.New("player: invalid equipment smelting gauge") - } - total := nextWallet.EquipMileageExchangeGage + mileageMaterial - exchanges := total / threshold - earned := exchanges * rewardCount - if exchanges != 0 && earned/exchanges != rewardCount || math.MaxUint64-nextWallet.EquipMileage < earned { - return Currency{}, 0, errors.New("player: equipment mileage overflow") - } - nextWallet.EquipMileageExchangeGage = total % threshold - nextWallet.EquipMileage += earned - - nextItems := cloneOwnedSnapshot(s.inventory.owned) - for _, want := range consumed { - if err := consumeOwnedItem(&nextItems, want); err != nil { - return Currency{}, 0, err - } - } - - walletData, err := json.Marshal(nextWallet) - if err != nil { - return Currency{}, 0, err - } - if err := s.store.SaveWithEntries("wallet", walletData, entry("spent", identity, []byte("true"))); err != nil { - return Currency{}, 0, fmt.Errorf("player: persist equipment %s wallet: %w", operation, err) - } - if err := s.inventory.commitOwned(nextItems); err != nil { - return Currency{}, 0, fmt.Errorf("player: persist equipment %s items: %w", operation, err) - } - if err := s.persistEquipment(nextEquipment, operation); err != nil { - return Currency{}, 0, err - } - s.owned = nextEquipment - s.persisted = cloneEquipmentSnapshot(nextEquipment) - s.wallet.state = nextWallet - s.inventory.owned = nextItems - return nextWallet.Currency, earned, nil -} - -func consumeOwnedItem(next *ownedSnapshot, want Item) error { - if next == nil || want.InvenIndex == 0 || want.ID == 0 || want.Type == 0 || want.Count == 0 { - return errors.New("player: invalid item consumption") - } - for i, current := range next.Items { - if current.InvenIndex != want.InvenIndex { - continue - } - if current.ID != want.ID || current.Type != want.Type || current.Count < want.Count { - return fmt.Errorf("player: item %d consumption mismatch", want.InvenIndex) - } - current.Count -= want.Count - if current.Count == 0 { - next.Items = append(next.Items[:i], next.Items[i+1:]...) - } else { - next.Items[i] = current - } - return nil - } - return fmt.Errorf("player: item %d is not mutable-owned", want.InvenIndex) -} - -func validateOptionRerollMaterials(costs []gamedata.PromotionCost, materials []Item, conversion *gamedata.EquipmentOptionRerollConversion) (uint64, []Item, error) { - if len(costs) == 0 || len(materials) == 0 { - return 0, nil, errors.New("player: EquipOptionReRoll has no material") - } - expected := make(map[[2]uint64]uint64, len(costs)) - for _, cost := range costs { - key := [2]uint64{cost.Type, cost.ID} - if cost.Type == 0 || cost.Count == 0 || (cost.Type == 4 && cost.ID != 0) || math.MaxUint64-expected[key] < cost.Count { - return 0, nil, errors.New("player: invalid equipment option reroll cost") - } - expected[key] += cost.Count - } - actual := make(map[[2]uint64]uint64, len(materials)) - consumed := make([]Item, 0, len(materials)) - for _, material := range materials { - key := [2]uint64{material.Type, material.ID} - if math.MaxUint64-actual[key] < material.Count { - return 0, nil, errors.New("player: EquipOptionReRoll material count overflows") - } - actual[key] += material.Count - if material.Type != 4 { - consumed = append(consumed, material) - } - } - for key, want := range expected { - if conversion != nil && key == ([2]uint64{conversion.TargetType, conversion.TargetID}) && - expected[[2]uint64{conversion.SourceType, conversion.SourceID}] == 0 { - targetKey := key - sourceKey := [2]uint64{conversion.SourceType, conversion.SourceID} - targetCount := actual[targetKey] - if targetCount > want || conversion.Ratio == 0 || want-targetCount > math.MaxUint64/conversion.Ratio || actual[sourceKey] != (want-targetCount)*conversion.Ratio { - return 0, nil, fmt.Errorf("player: EquipOptionReRoll converted material %d/%d does not match cost", key[0], key[1]) - } - delete(actual, targetKey) - delete(actual, sourceKey) - continue - } - if actual[key] != want { - return 0, nil, fmt.Errorf("player: EquipOptionReRoll material %d/%d=%d want=%d", key[0], key[1], actual[key], want) - } - delete(actual, key) - } - if len(actual) != 0 { - return 0, nil, errors.New("player: EquipOptionReRoll material kinds mismatch") - } - return expected[[2]uint64{4, 0}], consumed, nil -} - -func (s *EquipmentInventory) commitOptionRerollLocked(pending *equipmentOptionRerollPending, consumed []Item, gold uint64, identity string) error { - if pending == nil || identity == "" || s.wallet == nil || s.inventory == nil { - return errors.New("player: invalid transactional equipment option reroll") - } - s.wallet.mu.Lock() - defer s.wallet.mu.Unlock() - s.inventory.mu.Lock() - defer s.inventory.mu.Unlock() - - nextWallet := cloneWallet(s.wallet.state) - if nextWallet.Spent[identity] { - return errors.New("player: equipment option reroll request was already committed") - } - if nextWallet.Gold < gold { - return errors.New("player: insufficient gold for equipment option reroll") - } - nextWallet.Gold -= gold - nextWallet.Spent[identity] = true - nextItems := cloneOwnedSnapshot(s.inventory.owned) - for _, want := range consumed { - if err := consumeOwnedItem(&nextItems, want); err != nil { - return err - } - } - walletData, err := json.Marshal(nextWallet) - if err != nil { - return err - } - if err := s.store.SaveWithEntries("wallet", walletData, entry("spent", identity, []byte("true"))); err != nil { - return fmt.Errorf("player: persist equipment option reroll wallet: %w", err) - } - if err := s.inventory.commitOwned(nextItems); err != nil { - return fmt.Errorf("player: persist equipment option reroll items: %w", err) - } - if err := s.persistEquipmentState(s.owned, pending, true, "option reroll"); err != nil { - return err - } - s.wallet.state = nextWallet - s.inventory.owned = nextItems - s.pendingReroll = &equipmentOptionRerollPending{Equipment: cloneEquipment(pending.Equipment)} - return nil -} - -func upgradeLackItems(costs []gamedata.PromotionCost) []Item { - items := make([]Item, 0, len(costs)) - for _, cost := range costs { - items = append(items, Item{ID: cost.ID, Type: cost.Type, Count: cost.Count}) - } - return items -} - -func (s *EquipmentInventory) equipmentPositionLocked(index uint64) int { - for i := range s.owned.Equipment { - if s.owned.Equipment[i].InvenIndex == index { - return i - } - } - return -1 -} - -func (s *EquipmentInventory) selectUpgradeCosts(costs []gamedata.PromotionCost) ([]Item, uint64, error) { - var selected []Item - var gold uint64 - for _, cost := range costs { - switch cost.Type { - case 4: - if cost.ID != 0 || cost.Count == 0 || gold != 0 { - return nil, 0, errors.New("player: invalid equipment upgrade gold cost") - } - gold = cost.Count - selected = append(selected, Item{Type: 4, Count: cost.Count}) - case 8: - items, err := s.inventory.SelectMutable(cost.Type, cost.ID, cost.Count) - if err != nil { - return nil, 0, err - } - selected = append(selected, items...) - default: - return nil, 0, fmt.Errorf("player: unsupported equipment upgrade cost type %d", cost.Type) - } - } - return selected, gold, nil -} - -func (s *EquipmentInventory) attemptUpgradeLocked(index uint64, materials []Item) (Equipment, bool, uint64, []Item, error) { - if s.upgrade == nil || s.wallet == nil { - return Equipment{}, false, 0, nil, errors.New("player: equipment upgrade unavailable") - } - position := s.equipmentPositionLocked(index) - if position < 0 { - return Equipment{}, false, 0, nil, fmt.Errorf("player: unknown equipment %d", index) - } - current := s.owned.Equipment[position] - level, _, err := s.upgrade.Level(current.ID, current.Level) - if err != nil { - return Equipment{}, false, 0, nil, err - } - want := make(map[[2]uint64]uint64, len(level.Costs)) - for _, cost := range level.Costs { - want[[2]uint64{cost.Type, cost.ID}] += cost.Count - } - got := make(map[[2]uint64]uint64) - var gold uint64 - var items []Item - for _, material := range materials { - got[[2]uint64{material.Type, material.ID}] += material.Count - if material.Type == 4 { - if material.InvenIndex != 0 || material.ID != 0 || gold != 0 { - return Equipment{}, false, 0, nil, errors.New("player: invalid equipment upgrade currency") - } - gold = material.Count - } else { - items = append(items, material) - } - } - if len(got) != len(want) { - return Equipment{}, false, 0, nil, fmt.Errorf("player: equipment upgrade material kinds mismatch") - } - for key, count := range want { - if got[key] != count { - return Equipment{}, false, 0, nil, fmt.Errorf("player: equipment upgrade material %d/%d=%d want=%d", key[0], key[1], got[key], count) - } - } - if gold != 0 && !s.wallet.CanSpendGold(gold) { - return Equipment{}, false, 0, nil, errors.New("player: insufficient gold for equipment upgrade") - } - if len(items) != 0 { - if s.inventory == nil { - return Equipment{}, false, 0, nil, errors.New("player: equipment upgrade inventory unavailable") - } - if err := s.inventory.CanConsume(items); err != nil { - return Equipment{}, false, 0, nil, err - } - } - // Draw before any cross-store write: RNG failure must never charge the - // player. Ordinary enhancement also unlocks an official grade at the - // +3/+6/+9 pivots; smelting can later change those grades, but is separate. - success, err := s.upgrade.Roll(level.SuccessRatio) - if err != nil { - return Equipment{}, false, 0, nil, fmt.Errorf("player: roll equipment upgrade: %w", err) - } - var rankSlot, rankValue uint64 - if success { - nextLevel := current.Level + 1 - switch nextLevel { - case 3: - rankSlot = 1 - case 6: - rankSlot = 2 - case 9: - rankSlot = 3 - } - if rankSlot != 0 { - if len(current.Rank) != 3 || current.Rank[rankSlot-1] != 0 { - return Equipment{}, false, 0, nil, fmt.Errorf("player: equipment %d invalid rank state at +%d", index, nextLevel) - } - rankValue, err = s.upgrade.RollRank(current.ID, rankSlot) - if err != nil { - return Equipment{}, false, 0, nil, fmt.Errorf("player: roll equipment rank: %w", err) - } - } - } - attempt := current.UpgradeAttempts + 1 - identity := "equip-upgrade:" + strconv.FormatUint(index, 10) + ":" + strconv.FormatUint(attempt, 10) - if gold != 0 { - if _, err := s.wallet.SpendGoldOnce(identity, gold); err != nil { - return Equipment{}, false, 0, nil, fmt.Errorf("player: spend equipment upgrade gold: %w", err) - } - } - if len(items) != 0 { - if err := s.inventory.Consume(items); err != nil { - return Equipment{}, false, 0, nil, fmt.Errorf("player: consume equipment upgrade items: %w", err) - } - } - next := cloneEquipmentSnapshot(s.owned) - next.Equipment[position].UpgradeAttempts = attempt - if success { - next.Equipment[position].Level++ - if rankSlot != 0 { - next.Equipment[position].Rank[rankSlot-1] = rankValue - } - } - if err := s.commitLocked(next, "upgrade"); err != nil { - return Equipment{}, false, 0, nil, err - } - return next.Equipment[position], success, gold, materials, nil -} - -func (s *EquipmentInventory) equippedCharacter(entry Equipment) (Character, bool) { - if entry.UseChar == 0 || s.characters == nil { - return Character{}, false - } - return s.characters.Find(entry.UseChar) -} - -func (s *EquipmentInventory) clear(request []byte) (int, []byte, bool, error) { - equipmentIndex, found, err := wire.Varint(request, 2) - if err != nil || !found || equipmentIndex == 0 { - return 0, nil, true, errors.New("player: EquipClear missing equipment") - } - characterIndex, found, err := wire.Varint(request, 3) - if err != nil || !found || characterIndex == 0 { - return 0, nil, true, errors.New("player: EquipClear missing character") - } - if s.characters == nil { - return 0, nil, true, errors.New("player: EquipClear character store unavailable") - } - if _, found := s.characters.Find(characterIndex); !found { - return 0, nil, true, fmt.Errorf("player: EquipClear unknown character %d", characterIndex) - } - s.mu.Lock() - next := cloneEquipmentSnapshot(s.owned) - position := -1 - for i := range next.Equipment { - if next.Equipment[i].InvenIndex == equipmentIndex { - position = i - break - } - } - if position < 0 { - s.mu.Unlock() - return 0, nil, true, fmt.Errorf("player: EquipClear unknown equipment %d", equipmentIndex) - } - if next.Equipment[position].UseChar != characterIndex { - s.mu.Unlock() - return 0, nil, true, fmt.Errorf("player: EquipClear equipment %d is not used by character %d", equipmentIndex, characterIndex) - } - next.Equipment[position].UseChar = 0 - if err := s.commitLocked(next, "clear"); err != nil { - s.mu.Unlock() - return 0, nil, true, err - } - s.mu.Unlock() - // Re-query after the equipment mutation so the shared stat calculator - // returns HP with the cleared item excluded. - character, found := s.characters.Find(characterIndex) - if !found { - return 0, nil, true, fmt.Errorf("player: EquipClear character %d disappeared", characterIndex) - } - return 36, wire.AppendBytes(nil, 1, CharacterWire(character)), true, nil -} - -func (s *EquipmentInventory) lock(request []byte) (int, []byte, bool, error) { - equipmentIndex, found, err := wire.Varint(request, 2) - if err != nil || !found || equipmentIndex == 0 { - return 0, nil, true, errors.New("player: EquipLock missing equipment") - } - // LockFlag is int32, but proto3 omits its zero value. A missing field 3 is - // therefore the normal unlock request; present values are restricted to 1. - lockFlag, present, err := wire.Varint(request, 3) - if err != nil || (present && lockFlag != 1) { - return 0, nil, true, errors.New("player: EquipLock invalid lock flag") - } - if !present { - lockFlag = 0 - } - s.mu.Lock() - defer s.mu.Unlock() - next := cloneEquipmentSnapshot(s.owned) - position := -1 - for i := range next.Equipment { - if next.Equipment[i].InvenIndex == equipmentIndex { - position = i - break - } - } - if position < 0 { - return 0, nil, true, fmt.Errorf("player: EquipLock unknown equipment %d", equipmentIndex) - } - next.Equipment[position].LockFlag = lockFlag - if err := s.commitLocked(next, "lock"); err != nil { - return 0, nil, true, err - } - return 38, nil, true, nil -} - -func (s *EquipmentInventory) mark(path string, request []byte) (int, []byte, bool, error) { - equipmentIndex, found, err := wire.Varint(request, 2) - if err != nil || !found || equipmentIndex == 0 { - return 0, nil, true, fmt.Errorf("player: %s missing equipment", strings.TrimPrefix(path, "/")) - } - mark := "" - packetCode := 397 - if path == "/EquipMarkSet" { - raw, present, fieldErr := wire.Bytes(request, 3) - if fieldErr != nil || !present || !validEquipmentMark(raw) { - return 0, nil, true, errors.New("player: EquipMarkSet invalid mark") - } - mark = string(raw) - packetCode = 396 - } - s.mu.Lock() - defer s.mu.Unlock() - next := cloneEquipmentSnapshot(s.owned) - position := -1 - for i := range next.Equipment { - if next.Equipment[i].InvenIndex == equipmentIndex { - position = i - break - } - } - if position < 0 { - return 0, nil, true, fmt.Errorf("player: %s unknown equipment %d", strings.TrimPrefix(path, "/"), equipmentIndex) - } - next.Equipment[position].Mark = mark - if err := s.commitLocked(next, strings.TrimPrefix(path, "/")); err != nil { - return 0, nil, true, err - } - return packetCode, nil, true, nil -} - -// EquipmentInfo.MakeStringCustomMarkData emits either "~', dev_mail_grant.PAGE)
- self.assertIn('
', dev_mail_grant.PAGE)
- self.assertIn('max-height:min(40vh,28rem)', dev_mail_grant.PAGE)
- self.assertIn('overflow:auto', dev_mail_grant.PAGE)
- self.assertIn("$('item-picker').open=false", dev_mail_grant.PAGE)
-
- def test_picker_search_includes_random_box_aliases(self):
- self.assertIn("(x.aliases||[]).join(' ')", dev_mail_grant.PAGE)
-
- def test_mail_tool_disables_browser_response_caching(self):
- self.assertIn('self.send_header("Cache-Control", "no-store")', inspect.getsource(dev_mail_grant.Handler.reply))
-
- def test_development_settings_are_strict_and_atomic(self):
- limits = {
- "baseline": {"items": 100, "equipment": 500},
- "enabled_limits": {"items": 500, "equipment": 2000},
- }
- with tempfile.TemporaryDirectory() as temporary:
- path = Path(temporary) / "dev-tools.json"
- store = dev_mail_grant.DevelopmentSettingsStore(path, limits)
- self.assertFalse(store.snapshot()["inventory"]["unlimited"])
- updated = store.set_inventory({"unlimited": True})
- self.assertTrue(updated["inventory"]["unlimited"])
- self.assertEqual(json.loads(path.read_text(encoding="utf-8")), {
- "version": 1, "inventory": {"unlimited": True},
- })
- reopened = dev_mail_grant.DevelopmentSettingsStore(path, limits)
- self.assertTrue(reopened.snapshot()["inventory"]["unlimited"])
- for invalid in ({"unlimited": 1}, {"unlimited": "true"}, {"unlimited": True, "extra": 1}, {}):
- with self.assertRaises(ValueError):
- store.set_inventory(invalid)
- path.write_text('{"version":1,"inventory":{"unlimited":true},"extra":1}', encoding="utf-8")
- with self.assertRaises(ValueError):
- dev_mail_grant.DevelopmentSettingsStore(path, limits)
-
- def test_development_tool_page_places_inventory_after_mail(self):
- self.assertIn("BD2 开发工具 ", dev_mail_grant.PAGE)
- self.assertLess(dev_mail_grant.PAGE.index("开发邮件发放"), dev_mail_grant.PAGE.index("无限背包容量"))
- self.assertIn("重新登录客户端后生效", dev_mail_grant.PAGE)
- self.assertIn("/api/settings/inventory", dev_mail_grant.PAGE)
-
- def test_browser_grant_appends_spool_without_allocating_mail_id(self):
- with tempfile.TemporaryDirectory() as temporary:
- root = Path(temporary)
- output = root / "mail-grants-spool.json"
- store = dev_mail_grant.MailGrantStore(output, [{
- "id": 9, "element_type": 8, "name": "slime",
- }])
- self.assertTrue(output.is_file())
- self.assertEqual(json.loads(output.read_text(encoding="utf-8")), {"version": 1, "grants": []})
- result = store.grant({"item_id": 9, "element_type": 8, "count": 123, "title": "test", "body": "body"})
- written = json.loads(output.read_text(encoding="utf-8"))
- self.assertEqual(written["version"], 1)
- self.assertEqual(len(written["grants"]), 1)
- self.assertEqual(written["grants"][0]["rewards"], [{"type": 8, "id": 9, "count": 123}])
- self.assertNotIn("mail_id", written["grants"][0])
- self.assertEqual(result["grant"], written["grants"][0])
- self.assertFalse(result["restart_required"])
- with self.assertRaises(ValueError):
- store.grant({"item_id": 999, "element_type": 8, "count": 1})
- with self.assertRaises(ValueError):
- store.grant({"item_id": 9, "element_type": 8, "count": dev_mail_grant.MAX_INT32 + 1})
-
- def test_browser_form_blocks_duplicate_submit_until_response(self):
- self.assertIn("if(submitting)return", dev_mail_grant.PAGE)
- self.assertIn("$('grant-submit').disabled=true", dev_mail_grant.PAGE)
- self.assertIn("x.grant.identity", dev_mail_grant.PAGE)
-
-
-class ClientSourceToolTests(unittest.TestCase):
- def test_generated_tree_staging_uses_parent_inheriting_directory_mode(self):
- with tempfile.TemporaryDirectory() as temporary:
- output = Path(temporary) / "dev" / "client-source-readable"
- real_mkdir = inherited_stage.os.mkdir
- calls = []
-
- def recording_mkdir(path, *args, **kwargs):
- calls.append((Path(path), args, kwargs))
- return real_mkdir(path, *args, **kwargs)
-
- with mock.patch.object(inherited_stage.os, "mkdir", side_effect=recording_mkdir):
- stage = inherited_stage.create_inherited_stage(output)
- try:
- self.assertEqual(stage.parent, output.parent)
- stage_call = next(call for call in calls if call[0] == stage)
- self.assertEqual(stage_call[1], ())
- self.assertEqual(stage_call[2], {})
- finally:
- stage.rmdir()
-
- def test_deobfuscates_code_not_comments_or_literals_and_records_collisions(self):
- with tempfile.TemporaryDirectory() as temporary:
- root = Path(temporary)
- source = root / "Assembly-CSharp"
- source.mkdir()
- mapping = root / "translation.obfuscate"
- mapping.write_text(
- "#ReverseOrder\n"
- "α⇨Net.Player/User-Info\n"
- "β⇨Other.User Info\n"
- "γ⇨class\n"
- "δ⇨Meaning.D\n",
- encoding="utf-8",
- )
- (source / "α.cs").write_text(
- "// α β γ δ\n"
- "class α { string v = \"α β γ δ\"; char x = 'α'; α f; β g; γ h; δ i; }\n",
- encoding="utf-8",
- )
- (source / "asset.bin").write_bytes(b"\x00a")
- output = root / "mirror"
- manifest = deobfuscate_client_source.deobfuscate(source, mapping, output)
- text = (output / "User_Info.cs").read_text(encoding="utf-8")
- self.assertIn("// α β γ δ", text)
- self.assertIn('"α β γ δ"', text)
- self.assertIn("char x = 'α'", text)
- self.assertIn("class User_Info", text)
- self.assertIn("User_Info__from_u03B2 g", text)
- self.assertIn("_class h", text)
- self.assertIn("D i", text)
- self.assertEqual((output / "asset.bin").read_bytes(), b"\x00a")
- self.assertEqual(manifest["statistics"]["identifier_replacements"], 5)
- self.assertTrue((output / deobfuscate_client_source.MANIFEST).is_file())
-
- def test_deobfuscator_skips_ambiguous_scoped_symbol(self):
- entries = [
- {"line": 1, "source": "α", "meaning": "One.Value"},
- {"line": 2, "source": "α", "meaning": "Two.Value"},
- ]
- replacements, report, warnings = deobfuscate_client_source.build_replacements(entries)
- self.assertNotIn("α", replacements)
- self.assertEqual(report, [])
- self.assertTrue(any("ambiguous" in warning for warning in warnings))
-
- def test_deobfuscator_refuses_output_inside_source_or_unmanaged_output(self):
- with tempfile.TemporaryDirectory() as temporary:
- root = Path(temporary)
- source = root / "source"
- source.mkdir()
- mapping = root / "map"
- mapping.write_text("a⇨Name\n", encoding="utf-8")
- with self.assertRaises(ValueError):
- deobfuscate_client_source.deobfuscate(source, mapping, source / "out")
- output = root / "output"
- output.mkdir()
- (output / "someone.txt").write_text("keep", encoding="utf-8")
- with self.assertRaises(FileExistsError):
- deobfuscate_client_source.deobfuscate(source, mapping, output)
-
- def test_reconstructs_proto_and_lossless_descriptor_set(self):
- with tempfile.TemporaryDirectory() as temporary:
- root = Path(temporary)
- source = root / "Assembly-CSharp"
- net = source / "Proto" / "Net"
- net.mkdir(parents=True)
- descriptor = descriptor_pb2.FileDescriptorProto(
- name="Request/Login.proto", package="proto.net", syntax="proto3"
- )
- message = descriptor.message_type.add(name="LoginRequest")
- message.field.add(
- name="seq", number=1,
- label=descriptor_pb2.FieldDescriptorProto.LABEL_OPTIONAL,
- type=descriptor_pb2.FieldDescriptorProto.TYPE_INT32,
- )
- encoded = base64.b64encode(descriptor.SerializeToString()).decode("ascii")
- (net / "LoginRequestReflection.cs").write_text(
- "private static FileDescriptor descriptor = "
- "FileDescriptor.FromGeneratedCode(Convert.FromBase64String("
- f"string.Concat(new string[] {{ \"{encoded}\" }})), new FileDescriptor[0], info);",
- encoding="utf-8",
- )
- output = root / "proto-view"
- manifest = extract_client_proto.reconstruct(source, output)
- proto = (output / "Request" / "Login.proto").read_text(encoding="utf-8")
- self.assertIn('syntax = "proto3";', proto)
- self.assertIn("package proto.net;", proto)
- self.assertIn("message LoginRequest", proto)
- self.assertIn("int32 seq = 1;", proto)
- saved = descriptor_pb2.FileDescriptorSet()
- saved.ParseFromString((output / extract_client_proto.DESCRIPTOR_SET).read_bytes())
- self.assertEqual(saved.file[0], descriptor)
- self.assertEqual(manifest["statistics"]["proto_files"], 1)
-
- def test_proto_extractor_rejects_unmanaged_or_nested_outputs(self):
- with tempfile.TemporaryDirectory() as temporary:
- root = Path(temporary)
- source = root / "source"
- (source / "Proto" / "Net").mkdir(parents=True)
- with self.assertRaises(ValueError):
- extract_client_proto.reconstruct(source, source / "view")
- descriptor = descriptor_pb2.FileDescriptorProto(name="X.proto", syntax="proto3")
- encoded = base64.b64encode(descriptor.SerializeToString()).decode("ascii")
- (source / "Proto" / "Net" / "XReflection.cs").write_text(
- "Convert.FromBase64String(string.Concat(new string[] {"
- f"\"{encoded}\"" + "})), new FileDescriptor[0], info);",
- encoding="utf-8",
- )
- output = root / "view"
- output.mkdir()
- with self.assertRaises(FileExistsError):
- extract_client_proto.reconstruct(source, output)
-
-
-if __name__ == "__main__":
- unittest.main()