#include <uapi/linux/bpf.h>
#include <uapi/linux/if_ether.h>
#include <uapi/linux/if_packet.h>
#include <uapi/linux/ip.h>
#include "bpf_helpers.h"

struct bpf_flow_keys {
	long src;
};

struct bpf_map_def SEC("maps") my_map = {
	.type = BPF_MAP_TYPE_ARRAY,
	.key_size = sizeof(u32),
	.value_size = sizeof(long),
	.max_entries = 256,
};

SEC("socket1")
int bpf_prog1(struct __sk_buff *skb)
{
	int index = load_byte(skb, ETH_HLEN + offsetof(struct iphdr, protocol));
	long *value;

#if 1// WORK 
	struct bpf_flow_keys flow_mask_array[2];
	flow_mask_array[0].src = 0xf00;
	flow_mask_array[1].src = 4444;

#else //NOT WORK!
   struct bpf_flow_keys flow_mask_array[] = { 
        {.src = 5555},
        {.src = 4444},
    };
#endif

	if (skb->pkt_type != PACKET_OUTGOING)
		return 0;

    bpf_map_update_elem(&my_map, &index, &(flow_mask_array[1].src), BPF_ANY);

	return 0;
}
char _license[] SEC("license") = "GPL";