Files
flock/pkg/cni/rpc.go
T
Donavan Fritz 31fcae2a97
Build flock Image / build (push) Has been cancelled
M2 plumbing: CNI ↔ agent JSON RPC over unix socket
Locks the wire format between /opt/cni/bin/flock and flock-agent. ADD
returns a CNI Result, DEL returns success/error, CHECK returns
success/error. Connection-per-RPC, newline-delimited JSON.

- pkg/cni/rpc.go: shared Op + Request + Response + framed encode/decode.
- pkg/cni/rpc_client.go: net.Dial + EncodeRequest + DecodeResponse;
  rpcSocket overridable for tests.
- pkg/cni/plugin.go: real implementations of CmdAdd/Del/Check that call
  through, mapping agent errors to types.Error.
- pkg/agent/rpc.go: rpcServer with swappable AddHandler/DelHandler/
  CheckHandler (defaults: not-implemented for ADD; idempotent-no-op for
  DEL/CHECK so kubelet teardown of a never-ADDed pod doesn't fail).
- pkg/agent/server.go: replaces the M1 accept-and-close placeholder
  with rpcServer.serve(ctx, listener); listener closes on ctx cancel.

Tests cover: Request/Response JSON roundtrip, end-to-end client →
unix-socket → fake server, agent error → CNI types.Error mapping.

ADD remains "not implemented" until netlink + IPAM wire-up — the agent
returns an error and kubelet will fail pod sandbox creation IF a node
were configured to use this CNI. host001's CNI plane is still 100%
Calico, so this changes nothing observable on the cluster.

Co-Authored-By: Claude Sonnet 4.6 (1M context) <noreply@anthropic.com>
2026-04-24 22:21:33 -05:00

88 lines
2.6 KiB
Go

package cni
import (
"encoding/json"
"fmt"
"io"
current "github.com/containernetworking/cni/pkg/types/100"
)
// Op is the CNI verb the plugin asks the agent to perform.
type Op string
const (
OpAdd Op = "ADD"
OpDel Op = "DEL"
OpCheck Op = "CHECK"
)
// Request is sent over the unix socket from the CNI plugin to flock-agent.
// Field names mirror the kubelet → CNI invocation env vars; the agent uses
// these to look up Pod metadata via the informer cache.
type Request struct {
Op Op `json:"op"`
ContainerID string `json:"container_id"`
Netns string `json:"netns"` // /proc/<pid>/ns/net
IfName string `json:"ifname"` // typically "eth0"
Args string `json:"args"` // raw CNI_ARGS env (K=V;K=V;...)
Path string `json:"path"` // CNI_PATH (plugin search path)
StdinData []byte `json:"stdin_data"` // raw network configuration JSON
}
// Response carries either a typed CNI Result or a single error string.
// We use a string error (not a Go error) because errors traverse a JSON
// boundary; the client converts back to a CNI types.Error.
type Response struct {
Result *current.Result `json:"result,omitempty"`
Error string `json:"error,omitempty"`
}
// EncodeRequest writes req to w as a single JSON object followed by '\n'.
// The newline framing makes the wire-protocol simple to read incrementally
// without a length prefix or full-stream-buffering.
func EncodeRequest(w io.Writer, req Request) error {
b, err := json.Marshal(req)
if err != nil {
return fmt.Errorf("marshal request: %w", err)
}
b = append(b, '\n')
if _, err := w.Write(b); err != nil {
return fmt.Errorf("write request: %w", err)
}
return nil
}
// DecodeRequest reads one newline-delimited JSON request from r.
func DecodeRequest(r io.Reader) (Request, error) {
var req Request
dec := json.NewDecoder(r)
if err := dec.Decode(&req); err != nil {
return Request{}, fmt.Errorf("decode request: %w", err)
}
return req, nil
}
// EncodeResponse writes resp to w as a single JSON object followed by '\n'.
func EncodeResponse(w io.Writer, resp Response) error {
b, err := json.Marshal(resp)
if err != nil {
return fmt.Errorf("marshal response: %w", err)
}
b = append(b, '\n')
if _, err := w.Write(b); err != nil {
return fmt.Errorf("write response: %w", err)
}
return nil
}
// DecodeResponse reads one newline-delimited JSON response from r.
func DecodeResponse(r io.Reader) (Response, error) {
var resp Response
dec := json.NewDecoder(r)
if err := dec.Decode(&resp); err != nil {
return Response{}, fmt.Errorf("decode response: %w", err)
}
return resp, nil
}