initial import

Author: wg

.gitignore

@@ -0,0 +1,3 @@
+bin/
+pkg/
+libkflow.*

demo.c

@@ -0,0 +1,43 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include "kflow.h"
+
+int main(int argc, char **argv) {
+    int r;
+    kflowConfig cfg = {
+        .URL = "http://chdev:20012/chf",
+        .API = {
+            .email = "[email protected]",
+            .token = "81b7262feceecc94eef3ddafbc2c152f",
+            .URL   = "http://chdev:8080/api/v5",
+        },
+        .device_id = 1001,
+        .verbose   = 1,
+    };
+
+    if ((r = kflowInit(&cfg)) != 0) {
+        printf("error initializing libkflow: %d", r);
+        exit(1);
+    };
+
+    kflow flow = {
+        .deviceId    = cfg.device_id,
+        .ipv4SrcAddr = 167772161,
+        .ipv4DstAddr = 167772162,
+        .srcAs       = 1234,
+        .inPkts      = 20,
+        .inBytes     = 40,
+    };
+
+    if ((r = kflowSend(&flow)) != 0) {
+        printf("error sending flow: %d", r);
+        exit(1);
+    }
+
+    if ((r = kflowStop(10*1000)) != 0) {
+        printf("error stopping libkflow: %d", r);
+        exit(1);
+    }
+
+    return 0;
+}

src/github.com/kentik/libkflow/api/client.go

@@ -0,0 +1,73 @@
+package api
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"time"
+)
+
+type Client struct {
+	Header http.Header
+	*http.Client
+}
+
+func NewClient(email, token string, timeout time.Duration) *Client {
+	client := &http.Client{
+		Timeout: timeout,
+	}
+
+	header := http.Header{
+		"X-CH-Auth-Email":     {email},
+		"X-CH-Auth-API-Token": {token},
+	}
+
+	return &Client{
+		Header: header,
+		Client: client,
+	}
+}
+
+func (c *Client) GetDevice(url string, did int) (*Device, error) {
+	r, err := c.do("GET", fmt.Sprintf("%s/device/%d", url, did), nil)
+	if err != nil {
+		return nil, err
+	}
+	defer r.Body.Close()
+
+	if r.StatusCode != 200 {
+		return nil, fmt.Errorf("api: HTTP status code %d", r.StatusCode)
+	}
+
+	dr := &DeviceResponse{}
+	if err := json.NewDecoder(r.Body).Decode(dr); err != nil {
+		return nil, err
+	}
+
+	return &dr.Device, nil
+}
+
+func (c *Client) SendFlow(url string, buf *bytes.Buffer) error {
+	r, err := c.do("POST", url, buf)
+	if err != nil {
+		return err
+	}
+	defer r.Body.Close()
+
+	if r.StatusCode != 200 {
+		return fmt.Errorf("api: HTTP status code %d", r.StatusCode)
+	}
+
+	return nil
+}
+
+func (c *Client) do(method, url string, body io.Reader) (*http.Response, error) {
+	r, err := http.NewRequest(method, url, body)
+	if err != nil {
+		return nil, err
+	}
+	r.Header = c.Header
+	return c.Client.Do(r)
+}

src/github.com/kentik/libkflow/api/types.go

@@ -0,0 +1,37 @@
+package api
+
+import (
+	"fmt"
+	"regexp"
+	"strconv"
+)
+
+type DeviceResponse struct {
+	Device Device `json:"device"`
+}
+
+type Device struct {
+	ID        string        `json:"id"`
+	Name      string        `json:"device_name"`
+	CompanyID string        `json:"company_id"`
+	Custom    CustomColumns `json:"custom_columns"`
+}
+
+type CustomColumns map[string]uint64
+
+func (d *Device) ClientID() string {
+	return fmt.Sprintf("%s:%s:%s", d.CompanyID, d.Name, d.ID)
+}
+
+func (c *CustomColumns) UnmarshalJSON(data []byte) error {
+	m := map[string]uint64{}
+	for _, match := range split.FindAllSubmatchIndex(data, -1) {
+		key := string(data[match[2]:match[3]])
+		val := string(data[match[4]:match[5]])
+		m[key], _ = strconv.ParseUint(val, 10, 64)
+	}
+	*c = m
+	return nil
+}
+
+var split = regexp.MustCompile(`(\w+)=(\d+),?`)