generated from cybozu-go/neco-template
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move proxy-related helper functions to a dedicated file
Signed-off-by: Daichi Sakaue <[email protected]>
- Loading branch information
Showing
3 changed files
with
124 additions
and
110 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,123 @@ | ||
| package app | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
|
|
||
| "github.com/cilium/cilium/pkg/client" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "k8s.io/client-go/dynamic" | ||
| "k8s.io/client-go/kubernetes" | ||
| ) | ||
|
|
||
| func getProxyEndpoint(ctx context.Context, c *kubernetes.Clientset, namespace, name string) (string, error) { | ||
| targetPod, err := c.CoreV1().Pods(namespace).Get(ctx, name, metav1.GetOptions{}) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| targetNode := targetPod.Spec.NodeName | ||
|
|
||
| pods, err := c.CoreV1().Pods("kube-system").List(ctx, metav1.ListOptions{ | ||
| FieldSelector: "spec.nodeName=" + targetNode, | ||
| LabelSelector: rootOptions.proxySelector, | ||
| }) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| if num := len(pods.Items); num != 1 { | ||
| err := fmt.Errorf("failed to find cilium-agent-proxy. found %d pods", num) | ||
| return "", err | ||
| } | ||
|
|
||
| podIP := pods.Items[0].Status.PodIP | ||
| return fmt.Sprintf("http://%s:%d", podIP, rootOptions.proxyPort), nil | ||
| } | ||
|
|
||
| func createCiliumClient(ctx context.Context, clientset *kubernetes.Clientset, namespace, name string) (*client.Client, error) { | ||
| endpoint, err := getProxyEndpoint(ctx, clientset, namespace, name) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| if cached, ok := cachedCiliumClients[endpoint]; ok { | ||
| return cached, nil | ||
| } | ||
|
|
||
| ciliumClient, err := client.NewClient(endpoint) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| cachedCiliumClients[endpoint] = ciliumClient | ||
|
|
||
| return ciliumClient, err | ||
| } | ||
|
|
||
| type policyEntryKey struct { | ||
| Identity int `json:"Identity"` | ||
| Direction int `json:"TrafficDirection"` | ||
| Protocol int `json:"Nexthdr"` | ||
| BigPort int `json:"DestPortNetwork"` // big endian | ||
| } | ||
|
|
||
| func (p policyEntryKey) Port() int { | ||
| return ((p.BigPort & 0xFF) << 8) + ((p.BigPort & 0xFF00) >> 8) | ||
| } | ||
|
|
||
| // For the meanings of the flags, see: | ||
| // https://github.com/cilium/cilium/blob/v1.16.3/bpf/lib/common.h#L394 | ||
| type policyEntry struct { | ||
| Flags int `json:"Flags"` | ||
| Packets int `json:"Packets"` | ||
| Bytes int `json:"Bytes"` | ||
| Key policyEntryKey `json:"Key"` | ||
| } | ||
|
|
||
| func (p policyEntry) IsDenyRule() bool { | ||
| return (p.Flags & 1) > 0 | ||
| } | ||
|
|
||
| func (p policyEntry) IsEgressRule() bool { | ||
| return p.Key.Direction > 0 | ||
| } | ||
|
|
||
| func (p policyEntry) IsWildcardProtocol() bool { | ||
| return (p.Flags & 2) > 0 | ||
| } | ||
|
|
||
| func (p policyEntry) IsWildcardPort() bool { | ||
| return (p.Flags & 4) > 0 | ||
| } | ||
|
|
||
| func queryPolicyMap(ctx context.Context, clientset *kubernetes.Clientset, dynamicClient *dynamic.DynamicClient, namespace, name string) ([]policyEntry, error) { | ||
| endpointID, err := getPodEndpointID(ctx, dynamicClient, namespace, name) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to get pod endpoint ID: %w", err) | ||
| } | ||
|
|
||
| url, err := getProxyEndpoint(ctx, clientset, namespace, name) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to get proxy endpoint: %w", err) | ||
| } | ||
|
|
||
| url = fmt.Sprintf("%s/policy/%d", url, endpointID) | ||
| resp, err := http.Get(url) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to request policy: %w", err) | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| data, err := io.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to read response: %w", err) | ||
| } | ||
|
|
||
| policies := make([]policyEntry, 0) | ||
| if err = json.Unmarshal(data, &policies); err != nil { | ||
| return nil, fmt.Errorf("failed to unmarshal response: %w", err) | ||
| } | ||
|
|
||
| return policies, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters