Include Shutdown frame for arm64 / cloud-hypervisor

Signed-off-by: Alex Ellis (OpenFaaS Ltd) <[email protected]>

Author: alexellis

main.go

@@ -21,6 +21,7 @@ var (
 const (
 	FrameTypeData       = 0x01
 	FrameTypeWindowSize = 0x02
+	FrameTypeShutdown   = 0x03
 )
 
 func main() {
@@ -229,6 +230,13 @@ func handleConnection(conn net.Conn) {
 						pty.Setsize(ptty, &ws)
 						log.Printf("Resized PTY to %dx%d", width, height)
 					}
+				case FrameTypeShutdown:
+					log.Printf("Received shutdown command, executing reboot")
+					rebootCmd := exec.Command("reboot")
+					if err := rebootCmd.Run(); err != nil {
+						log.Printf("Error executing reboot: %v", err)
+					}
+					return
 				default:
 					log.Printf("Unknown frame type: %d", frameType)
 				}