Filter: TLV

1 files changed, 79 insertions, 2 deletions

diff --git a/filter/config.Y b/filter/config.Y
index 5cd52e40..9f3faba0 100644
--- a/filter/config.Y
+++ b/filter/config.Y
@@ -112,6 +112,69 @@ f_new_pair_set(int fa, int ta, int fb, int tb)
 #define LC_ALL 0xFFFFFFFF
 
 static struct f_tree *
+f_new_sub_tlv_item(u32 type, u32 v1)
+{
+  struct f_tree *t = f_new_tree();
+  t->right = t;
+  t->from.type = t->to.type = T_SUB_TLV;
+  struct tlv v;
+  v.type = type;
+  switch (type) {
+  case TLV_COLOR:
+       v.u.color = v1;
+       break;
+  case TLV_UDP_DEST_PORT:
+       v.u.udp_dest_port = v1;
+       break;
+  default:
+       // FIXME error
+       break;
+  }
+  t->from.val.tlv = v;
+  t->to.val.tlv = v;
+  return t;
+}
+
+static struct f_tree *
+f_new_sub_tlv_tunnel_ep(u32 asn, ip_addr ip)
+{
+  struct f_tree *t = f_new_tree();
+  t->right = t;
+  t->from.type = t->to.type = T_SUB_TLV;
+  struct tlv v;
+  v.type = TLV_TUNNEL_ENDPOINT;
+  v.u.tunnel_endpoint.asn = asn;
+  v.u.tunnel_endpoint.ip = ip;
+  t->from.val.tlv = v;
+  t->to.val.tlv = v;
+  return t;
+}
+
+typedef char wg_key_b64_string[45];
+int wg_key_from_base64(u8 key[32], const wg_key_b64_string base64);
+
+static struct f_tree *
+f_new_sub_tlv_encap(u32 type, const char *v1)
+{
+  struct f_tree *t = f_new_tree();
+  t->right = t;
+  t->from.type = t->to.type = T_SUB_TLV;
+  struct tlv v;
+  v.type = TLV_ENCAPSULATION;
+  v.u.tunnel_encap.type = type;
+  // FIXME base64 decode v1
+  int len = 32;
+  v.u.tunnel_encap.data = cfg_alloc(len);
+  v.u.tunnel_encap.length = len;
+  // FIXME
+  wg_key_from_base64(v.u.tunnel_encap.data, v1);
+//  memcpy(v.u.tunnel_encap.data, v1, len);
+  t->from.val.tlv = v;
+  t->to.val.tlv = v;
+  return t;
+}
+
+static struct f_tree *
 f_new_ec_item(u32 kind, u32 ipv4_used, u32 key, u32 vf, u32 vt)
 {
   u64 fm, to;
@@ -178,6 +241,9 @@ f_generate_empty(struct f_dynamic_attr dyn)
     case EAF_TYPE_LC_SET:
       empty = f_const_empty_lclist;
       break;
+    case EAF_TYPE_TUNNEL_ENCAP:
+      empty = f_const_empty_set;
+      break;
     default:
       cf_error("Can't empty that attribute");
   }
@@ -288,7 +354,8 @@ CF_KEYWORDS(FUNCTION, PRINT, PRINTN, UNSET, RETURN,
 	PREPEND, FIRST, LAST, LAST_NONAGGREGATED, MATCH,
 	EMPTY,
 	FILTER, WHERE, EVAL, ATTRIBUTE,
-	BT_ASSERT, BT_TEST_SUITE, BT_CHECK_ASSIGN, BT_TEST_SAME, FORMAT)
+	BT_ASSERT, BT_TEST_SUITE, BT_CHECK_ASSIGN, BT_TEST_SAME,
+	TLV, TUNNEL_ENCAP, TUNNEL_ENDPOINT, UDP_DEST_PORT, COLOR, FORMAT)
 
 %nonassoc THEN
 %nonassoc ELSE
@@ -304,7 +371,7 @@ CF_KEYWORDS(FUNCTION, PRINT, PRINTN, UNSET, RETURN,
 %type <ecs> ec_kind
 %type <fret> break_command 
 %type <i32> cnum
-%type <e> pair_item ec_item lc_item set_item switch_item set_items switch_items switch_body
+%type <e> pair_item ec_item lc_item set_item switch_item set_items switch_items switch_body sub_tlv_item
 %type <trie> fprefix_set
 %type <v> set_atom switch_atom fipa
 %type <px> fprefix
@@ -378,6 +445,7 @@ type:
  | CLIST { $$ = T_CLIST; }
  | ECLIST { $$ = T_ECLIST; }
  | LCLIST { $$ = T_LCLIST; }
+ | TLV { $$ = T_TLV; }
  | type SET {
 	switch ($1) {
 	  case T_INT:
@@ -387,6 +455,7 @@ type:
 	  case T_LC:
 	  case T_RD:
 	  case T_IP:
+	  case T_TLV:
 	       $$ = T_SET;
 	       break;
 
@@ -584,10 +653,17 @@ lc_item:
    { $$ = f_new_lc_item($2, $10, $4, $12, $6, $14); }
 ;
 
+sub_tlv_item:
+   '(' TUNNEL_ENCAP ',' cnum ',' text ')'  { $$ = f_new_sub_tlv_encap($4, $6); }
+ | '(' UDP_DEST_PORT ',' cnum ')' { $$ = f_new_sub_tlv_item(TLV_UDP_DEST_PORT, $4); }
+ | '(' COLOR ',' cnum ')' { $$ = f_new_sub_tlv_item(TLV_COLOR, $4); }
+ | '(' TUNNEL_ENDPOINT ',' NUM ',' ipa ')' { $$ = f_new_sub_tlv_tunnel_ep($4, $6); }
+
 set_item:
    pair_item
  | ec_item
  | lc_item
+ | sub_tlv_item
  | set_atom { $$ = f_new_item($1, $1); }
  | set_atom DDOT set_atom { $$ = f_new_item($1, $3); }
  ;
@@ -596,6 +672,7 @@ switch_item:
    pair_item
  | ec_item
  | lc_item
+ | sub_tlv_item
  | switch_atom { $$ = f_new_item($1, $1); }
  | switch_atom DDOT switch_atom { $$ = f_new_item($1, $3); }
  ;