From 3bee1bcc0f92e01a48d6e8ead47475d68c4ba82d Mon Sep 17 00:00:00 2001
From: pdliyan <pdliyan@didiglobal.com>
Date: Sat, 23 Mar 2024 19:34:54 +0800
Subject: [PATCH] Reverse of sk_msg.

---
 {{project-name}}/src/main.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/{{project-name}}/src/main.rs b/{{project-name}}/src/main.rs
index 07910a9..174a0ff 100644
--- a/{{project-name}}/src/main.rs
+++ b/{{project-name}}/src/main.rs
@@ -122,11 +122,11 @@ async fn main() -> Result<(), anyhow::Error> {
     program.load()?;
     program.attach(cgroup)?;
     {%- when "sk_msg" -%}
-    let prog: &mut SkMsg = bpf.program_mut("{{crate_name}}").unwrap().try_into()?;
-    prog.load()?;
-
     let sock_map: SockHash::<_, SockKey> = bpf.map("{{sock_map}}").unwrap().try_into()?;
     let map_fd = sock_map.fd();
+
+    let prog: &mut SkMsg = bpf.program_mut("{{crate_name}}").unwrap().try_into()?;
+    prog.load()?;
     prog.attach(map_fd)?;
     // insert sockets to the map using sock_map.insert here, or from a sock_ops program
     {%- when "xdp" -%}