mirror of
https://github.com/luxfi/graph.git
synced 2026-09-24 00:26:07 +00:00
zeekay
705e38c45e
Red re-verify found the V4 AMM-spoof gate left a parallel bypass.
MEDIUM — V2/V3 AMM handlers were UNGATED: handlePairCreated, handlePoolCreated,
handleSwapV2, handleSwapV3 accepted events from ANY emitter — a parallel path to
seed phantom pairs/pools/swaps and inflate the SAME public Factory/TVL/volume
aggregates the now-gated V4 handlers (0x9999) block. Mirror the V4 gate:
- PairCreated/PoolCreated now require the canonical FactoryV2/FactoryV3 emitter
(the AMM trust root), exactly as V4 requires 0x9999.
- V2/V3 Swaps are emitted by the pair/pool, not the factory, so the gate is
"is the emitter a pool the canonical factory created?" (isKnownPool). The
factory authorizes a pool; only that pool's Swaps are honored. A rogue
factory's PairCreated is rejected, so its pools are never registered, so its
Swaps are rejected — the whole parallel path is closed.
- Pair/pool keys lower-cased at seed so the Swap-side lookup is RPC-casing safe.
- Factory addresses configurable (Config.FactoryV2/V3 + FACTORY_V2/V3 env),
default to canonical Lux mainnet (LuxMainnet.Factory{V2,V3}); main wires env.
INFO — the per-namespace resolver limit lacked the `if limit < 1 { limit = 100 }`
floor that resolvers/dex has, so a negative/zero `first` flowed to ListByType
(`first:-1` = "no limit" in SQLite — an unbounded scan). Brought every pl()/
parseLimit() copy to the dex-canonical body (floor <1->100 + cap 1000); fhe and
platform also gained the previously-missing 1000 cap. All namespaces now identical.
Tests (go test ./...): indexer + resolvers/{dex,exchange,fhe,platform} green.
- TestHandleV2_RejectsNonFactory, TestHandleV3_RejectsNonFactory: rogue
PairCreated/PoolCreated/Swap seeds nothing and does not inflate the factory
aggregate; the canonical-source event IS honored end-to-end.
- TestPL_ClampsFirst / TestParseLimit_ClampsFirst (exchange/fhe/platform):
negative/zero floored to 100, capped at 1000.
90 lines
3.3 KiB
Go
90 lines
3.3 KiB
Go
// Package ai provides resolvers for the A-Chain (AIVM).
|
|
//
|
|
// Indexes: inference proofs, model hashes, compute attestations, jobs, training runs.
|
|
//
|
|
// Entities: InferenceProof, ModelHash, ComputeAttestation, AIJob, TrainingRun
|
|
package ai
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/luxfi/graph/storage"
|
|
)
|
|
|
|
type ResolverFunc = func(context.Context, *storage.Store, map[string]interface{}) (interface{}, error)
|
|
|
|
func Register(resolvers map[string]ResolverFunc) {
|
|
resolvers["inferenceProof"] = resolveInferenceProof
|
|
resolvers["inferenceProofs"] = resolveInferenceProofs
|
|
resolvers["modelHash"] = resolveModelHash
|
|
resolvers["modelHashes"] = resolveModelHashes
|
|
resolvers["computeAttestation"] = resolveComputeAttestation
|
|
resolvers["computeAttestations"] = resolveComputeAttestations
|
|
resolvers["aiJob"] = resolveAIJob
|
|
resolvers["aiJobs"] = resolveAIJobs
|
|
resolvers["trainingRun"] = resolveTrainingRun
|
|
resolvers["trainingRuns"] = resolveTrainingRuns
|
|
resolvers["aiStats"] = resolveAIStats
|
|
}
|
|
|
|
func resolveInferenceProof(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
if id, ok := args["id"]; ok {
|
|
return s.GetByType("InferenceProof", fmt.Sprint(id))
|
|
}
|
|
return nil, fmt.Errorf("inferenceProof requires id")
|
|
}
|
|
func resolveInferenceProofs(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
return s.ListByType("InferenceProof", pl(args))
|
|
}
|
|
func resolveModelHash(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
if id, ok := args["id"]; ok {
|
|
return s.GetByType("ModelHash", fmt.Sprint(id))
|
|
}
|
|
return nil, fmt.Errorf("modelHash requires id")
|
|
}
|
|
func resolveModelHashes(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
return s.ListByType("ModelHash", pl(args))
|
|
}
|
|
func resolveComputeAttestation(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
if id, ok := args["id"]; ok {
|
|
return s.GetByType("ComputeAttestation", fmt.Sprint(id))
|
|
}
|
|
return nil, fmt.Errorf("computeAttestation requires id")
|
|
}
|
|
func resolveComputeAttestations(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
return s.ListByType("ComputeAttestation", pl(args))
|
|
}
|
|
func resolveAIJob(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
if id, ok := args["id"]; ok {
|
|
return s.GetByType("AIJob", fmt.Sprint(id))
|
|
}
|
|
return nil, fmt.Errorf("aiJob requires id")
|
|
}
|
|
func resolveAIJobs(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
return s.ListByType("AIJob", pl(args))
|
|
}
|
|
func resolveTrainingRun(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
if id, ok := args["id"]; ok {
|
|
return s.GetByType("TrainingRun", fmt.Sprint(id))
|
|
}
|
|
return nil, fmt.Errorf("trainingRun requires id")
|
|
}
|
|
func resolveTrainingRuns(_ context.Context, s *storage.Store, args map[string]interface{}) (interface{}, error) {
|
|
return s.ListByType("TrainingRun", pl(args))
|
|
}
|
|
func resolveAIStats(_ context.Context, s *storage.Store, _ map[string]interface{}) (interface{}, error) {
|
|
return s.GetByType("AIStats", "1")
|
|
}
|
|
|
|
func pl(args map[string]interface{}) int {
|
|
limit := 100
|
|
if l, ok := args["first"]; ok {
|
|
fmt.Sscanf(fmt.Sprint(l), "%d", &limit)
|
|
}
|
|
if limit < 1 {
|
|
limit = 100
|
|
}
|
|
return min(limit, 1000)
|
|
}
|