feat: sync multilayer td integration

internal/compat/android/layer_rpc.go

@@ -0,0 +1,38 @@
+package android
+
+import (
+	"errors"
+
+	"github.com/iamxvbaba/td/bin"
+	"github.com/iamxvbaba/td/tg"
+)
+
+var ErrPrivateLayerRPCInvalid = errors.New("android private layer RPC is invalid")
+
+// AdaptPrivateLayerRPC invokes the provenance-locked static gotdgen overlay
+// from the generated unknown-method view. Nested values decode with the exact
+// connection profile and the canonical request is re-profiled by gotd core.
+func AdaptPrivateLayerRPC(view tg.LayerRPCUnknownMethodView) (tg.LayerOutboundCall, bool, error) {
+	outbound, handled, err := view.AdaptClientRPCOverlay(tg.LayerClientRPCOverlayDrkloAndroid)
+	if err == nil && !handled {
+		outbound, handled, err = view.AdaptClientRPCOverlay(tg.LayerClientRPCOverlayDrkloAndroidTheme)
+	}
+	if err != nil {
+		return tg.LayerOutboundCall{}, handled, errors.Join(ErrPrivateLayerRPCInvalid, err)
+	}
+	return outbound, handled, nil
+}
+
+// UpgradePrivateLayerRPC is retained only for Router.Dispatch's legacy test
+// seam. Production admission uses AdaptPrivateLayerRPC above so its decode
+// shares the outer generated request budget.
+func UpgradePrivateLayerRPC(profile tg.LayerProfile, in *bin.Buffer, limits tg.LayerDecodeLimits) (*bin.Buffer, bool, error) {
+	upgraded, handled, err := tg.AdaptClientRPCOverlayWithLimits(profile, tg.LayerClientRPCOverlayDrkloAndroid, in, limits)
+	if err == nil && !handled {
+		upgraded, handled, err = tg.AdaptClientRPCOverlayWithLimits(profile, tg.LayerClientRPCOverlayDrkloAndroidTheme, in, limits)
+	}
+	if err != nil {
+		return nil, handled, errors.Join(ErrPrivateLayerRPCInvalid, err)
+	}
+	return upgraded, handled, nil
+}

internal/compat/android/layer_rpc_test.go

@@ -0,0 +1,54 @@
+package android
+
+import (
+	"errors"
+	"testing"
+
+	"github.com/iamxvbaba/td/bin"
+	"github.com/iamxvbaba/td/tg"
+)
+
+func TestUpgradePrivateLayerRPCOnlyAcceptsAuditedAndroidConstructors(t *testing.T) {
+	// DrKLO messages.forwardMessages private CRC has a body identical to the
+	// canonical request for the flags=0 empty-vector case.
+	private := bin.Buffer{}
+	private.PutID(0x41d41ade)
+	private.PutInt(0)
+	private.PutID(0x7f3b18ea) // inputPeerEmpty
+	private.PutVectorHeader(0)
+	private.PutVectorHeader(0)
+	private.PutID(0x7f3b18ea) // inputPeerEmpty
+
+	in := &bin.Buffer{Buf: private.Copy()}
+	upgraded, ok, err := UpgradePrivateLayerRPC(tg.LayerProfileCanonical, in, tg.LayerDecodeLimits{})
+	if err != nil || !ok {
+		t.Fatalf("upgrade private method = ok:%v err:%v", ok, err)
+	}
+	if in.Len() != 0 {
+		t.Fatalf("successful private method left %d bytes", in.Len())
+	}
+	if id, peekErr := upgraded.PeekID(); peekErr != nil || id != 0x13704a7c {
+		t.Fatalf("canonical id = %#x err=%v", id, peekErr)
+	}
+
+	official := bin.Buffer{}
+	official.PutID(0xb921bd04) // arbitrary non-private/official constructor
+	if value, handled, err := UpgradePrivateLayerRPC(tg.LayerProfileCanonical, &official, tg.LayerDecodeLimits{}); value != nil || handled || err != nil {
+		t.Fatalf("non-private method = value:%v handled:%v err:%v", value, handled, err)
+	}
+}
+
+func TestGeneratedPrivateLayerRPCOverlayHasAllAuditedMethods(t *testing.T) {
+	if got, want := tg.LayerClientRPCOverlayMethodCount(tg.LayerClientRPCOverlayDrkloAndroid), 15; got != want {
+		t.Fatalf("generated DrKLO method count = %d, want %d", got, want)
+	}
+}
+
+func TestUpgradePrivateLayerRPCRejectsMalformedBody(t *testing.T) {
+	malformed := bin.Buffer{}
+	malformed.PutID(0x41d41ade)
+	_, ok, err := UpgradePrivateLayerRPC(tg.LayerProfileCanonical, &malformed, tg.LayerDecodeLimits{})
+	if !ok || !errors.Is(err, ErrPrivateLayerRPCInvalid) {
+		t.Fatalf("malformed private method = ok:%v err:%v", ok, err)
+	}
+}