Add network namespace parameter to run snapshot loaded VMs in a custom network namespace

Signed-off-by: Amory Hoste <[email protected]>

Author: amohoste

proto/firecracker.pb.go

@@ -118,6 +118,7 @@ message LoadSnapshotRequest {
     string SnapshotFilePath = 2;
     string MemFilePath = 3;
     bool EnableUserPF = 4;
+    string NetworkNamespace = 5;
 }
 
 message LoadResponse {

proto/firecracker.proto

@@ -14,6 +14,7 @@
 package main
 
 import (
+	"github.com/containernetworking/plugins/pkg/ns"
 	"net"
 	"time"
 
@@ -118,14 +119,34 @@ func networkConfigFromProto(nwIface *proto.FirecrackerNetworkInterface, vmID str
 	return result, nil
 }
 
-// netNSFromProto returns the network namespace set, if any in the protobuf
+// netNSFromProto returns the name and handle of the network namespace set, if any in the protobuf
 // message.
-func netNSFromProto(request *proto.CreateVMRequest) string {
+func netNSFromProto(request *proto.CreateVMRequest) (string, ns.NetNS) {
 	if request != nil {
-		return request.NetworkNamespace
+		netNS, err := ns.GetNS(request.NetworkNamespace)
+		if err != nil {
+			return "", nil
+		}
+
+		return request.NetworkNamespace, netNS
+	}
+
+	return "", nil
+}
+
+// netNSFromProto returns the name and handle of the network namespace set, if any in the protobuf
+// message.
+func netNSFromSnapRequest(request *proto.LoadSnapshotRequest) (string, ns.NetNS) {
+	if request != nil {
+		netNS, err := ns.GetNS(request.NetworkNamespace)
+		if err != nil {
+			return "", nil
+		}
+
+		return request.NetworkNamespace, netNS
 	}
 
-	return ""
+	return "", nil
 }
 
 // rateLimiterFromProto creates a firecracker RateLimiter object from the