Implement program-type specific header generation and improve struct_ops template handling.

Signed-off-by: Cong Wang <cwang@multikernel.io>

Author: congwang-mk

examples/basic_match.ks

@@ -1,24 +1,7 @@
 // Basic Match Construct Demo for KernelScript
 // Demonstrates packet matching with the new match construct
 
-// XDP context struct (from BTF)
-struct xdp_md {
-  data: u64,
-  data_end: u64,
-  data_meta: u64,
-  ingress_ifindex: u32,
-  rx_queue_index: u32,
-  egress_ifindex: u32,
-}
-
-// XDP action enum (from BTF)
-enum xdp_action {
-  XDP_ABORTED = 0,
-  XDP_DROP = 1,
-  XDP_PASS = 2,
-  XDP_REDIRECT = 3,
-  XDP_TX = 4,
-}
+include "xdp.kh"
 
 // Protocol constants
 enum IpProtocol {

examples/break_continue_unbound.ks

@@ -1,25 +1,8 @@
-// XDP context struct (from BTF)
-struct xdp_md {
-  data: u64,
-  data_end: u64,
-  data_meta: u64,
-  ingress_ifindex: u32,
-  rx_queue_index: u32,
-  egress_ifindex: u32,
-}
-
-// XDP action enum (from BTF)
-enum xdp_action {
-  XDP_ABORTED = 0,
-  XDP_DROP = 1,
-  XDP_PASS = 2,
-  XDP_REDIRECT = 3,
-  XDP_TX = 4,
-}
-
 // Example demonstrating break and continue in truly unbound loops
 // This should force bpf_loop() usage
 
+include "xdp.kh"
+
 var counter_map : hash<u32, u32>(10)
 
 @xdp fn packet_filter(ctx: *xdp_md) -> xdp_action {

examples/dynptr.ks

@@ -1,26 +1,9 @@
-// XDP context struct (from BTF)
-struct xdp_md {
-  data: u64,
-  data_end: u64,
-  data_meta: u64,
-  ingress_ifindex: u32,
-  rx_queue_index: u32,
-  egress_ifindex: u32,
-}
-
-// XDP action enum (from BTF)
-enum xdp_action {
-  XDP_ABORTED = 0,
-  XDP_DROP = 1,
-  XDP_PASS = 2,
-  XDP_REDIRECT = 3,
-  XDP_TX = 4,
-}
-
 // Dynptr showcase - compiler should transparently use dynptr APIs for packet access
 // Example to demonstrate bpf_dynptr_from_mem usage
 // This would be for accessing memory buffers, not packet data
 
+include "xdp.kh"
+
 struct DataBuffer {
     data: u8[32],
     size: u32

examples/error_handling_demo.ks

@@ -1,23 +1,6 @@
-// XDP context struct (from BTF)
-struct xdp_md {
-  data: u64,
-  data_end: u64,
-  data_meta: u64,
-  ingress_ifindex: u32,
-  rx_queue_index: u32,
-  egress_ifindex: u32,
-}
-
-// XDP action enum (from BTF)
-enum xdp_action {
-  XDP_ABORTED = 0,
-  XDP_DROP = 1,
-  XDP_PASS = 2,
-  XDP_REDIRECT = 3,
-  XDP_TX = 4,
-}
-
 // Minimal error handling demo
+include "xdp.kh"
+
 var counters : hash<u32, u64>(1024)
 
 @xdp fn error_demo(ctx: *xdp_md) -> xdp_action {

examples/extern_kfunc_demo.ks

@@ -1,3 +1,5 @@
+include "xdp.kh"
+
 // External kfunc declarations - these would typically be imported from kernel BTF
 extern bpf_ktime_get_ns() -> u64
 extern bpf_trace_printk(fmt: *u8, fmt_size: u32) -> i32

examples/functions.ks

@@ -1,21 +1,4 @@
-// XDP context struct (from BTF)
-struct xdp_md {
-  data: u64,
-  data_end: u64,
-  data_meta: u64,
-  ingress_ifindex: u32,
-  rx_queue_index: u32,
-  egress_ifindex: u32,
-}
-
-// XDP action enum (from BTF)
-enum xdp_action {
-  XDP_ABORTED = 0,
-  XDP_DROP = 1,
-  XDP_PASS = 2,
-  XDP_REDIRECT = 3,
-  XDP_TX = 4,
-}
+include "xdp.kh"
 
 type IpAddress = u32
 

examples/import_demo.ks

@@ -6,6 +6,8 @@ import utils from "./import/simple_utils.ks"
 // Import Python module (uses Python bridge)
 import network_utils from "./import/network_utils.py"
 
+include "xdp.kh"
+
 config network {
     enable_filtering: bool = false,
     status_code: u32 = 0,

examples/include_demo.ks

@@ -4,6 +4,8 @@
 // Include declarations from header files
 include "common_kfuncs.kh"
 include "xdp_kfuncs.kh"
+include "xdp.kh"
+
 
 // XDP program that uses included kfunc declarations
 @xdp

examples/local_global_vars.ks

@@ -1,29 +1,12 @@
-// XDP context struct (from BTF)
-struct xdp_md {
-  data: u64,
-  data_end: u64,
-  data_meta: u64,
-  ingress_ifindex: u32,
-  rx_queue_index: u32,
-  egress_ifindex: u32,
-}
-
-// XDP action enum (from BTF)
-enum xdp_action {
-  XDP_ABORTED = 0,
-  XDP_DROP = 1,
-  XDP_PASS = 2,
-  XDP_REDIRECT = 3,
-  XDP_TX = 4,
-}
-
 // Example demonstrating local vs shared global variables
 //
 // This example shows the difference between:
 // - Regular global variables (shared with userspace via skeleton)
 // - Local global variables (kernel-only, not accessible from userspace)
 
 // Shared global variables - accessible from userspace via skeleton
+include "xdp.kh"
+
 var packet_count: u64 = 0
 var debug_enabled: bool = true
 

examples/map_operations_demo.ks

@@ -4,62 +4,9 @@
 //
 // NOTE: This file uses advanced language features not yet implemented in KernelScript.
 
-// XDP context struct (from BTF)
-struct xdp_md {
-  data: u64,
-  data_end: u64,
-  data_meta: u64,
-  ingress_ifindex: u32,
-  rx_queue_index: u32,
-  egress_ifindex: u32,
-}
-
-// XDP action enum (from BTF)
-enum xdp_action {
-  XDP_ABORTED = 0,
-  XDP_DROP = 1,
-  XDP_PASS = 2,
-  XDP_REDIRECT = 3,
-  XDP_TX = 4,
-}
-
-// TC context struct (from BTF)
-struct __sk_buff {
-  data: u64,
-  data_end: u64,
-  len: u32,
-  ifindex: u32,
-  protocol: u32,
-  mark: u32,
-}
-
-// TC action constants
-enum tc_action {
-  TC_ACT_UNSPEC = 255,
-  TC_ACT_OK = 0,
-  TC_ACT_RECLASSIFY = 1,
-  TC_ACT_SHOT = 2,
-  TC_ACT_PIPE = 3,
-  TC_ACT_STOLEN = 4,
-  TC_ACT_QUEUED = 5,
-  TC_ACT_REPEAT = 6,
-  TC_ACT_REDIRECT = 7,
-}
-
-// Base trace entry struct
-struct trace_entry {
-  entry_type: u16,
-  flags: u8,
-  preempt_count: u8,
-  pid: i32,
-}
-
-// Tracepoint context struct (from BTF) - sys_enter structure
-struct trace_event_raw_sys_enter {
-  ent: trace_entry,
-  id: i64,
-  args: u64[6],
-}
+include "xdp.kh"
+include "tc.kh"
+include "tracepoint.kh"
 
 // This example demonstrates comprehensive map operations with multi-program analysis
 // It shows various access patterns and concurrent access scenarios