mirror of
https://github.com/danderson/netboot.git
synced 2025-08-09 08:07:11 +02:00
5b23ef31b7
This just runs `tcpdump` as a subprocess, with the right arguments to capture interesting parts of the boot process (DHCP, PXE, TFTP).
71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
// Copyright 2016 Google Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
debugCmd = &cobra.Command{
|
|
Use: "debug",
|
|
Short: "Internal debugging commands",
|
|
Hidden: true,
|
|
}
|
|
dumpIpxeCmd = &cobra.Command{
|
|
Use: "dump-ipxe",
|
|
Short: "Dump the builtin ipxe binaries to disk",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
for fwtype, bs := range Ipxe {
|
|
path := fmt.Sprintf("builtin-ipxe-%d", fwtype)
|
|
if err := ioutil.WriteFile(path, bs, 0644); err != nil {
|
|
fmt.Printf("Error writing %s: %s\n", path, err)
|
|
} else {
|
|
fmt.Println("Wrote", path)
|
|
}
|
|
}
|
|
},
|
|
}
|
|
tcpdumpCmd = &cobra.Command{
|
|
Use: "tcpdump interface pcap-file",
|
|
Short: "Run tcpdump to capture some parts of a boot exchange, for debugging",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if len(args) != 2 {
|
|
fatalf("Need exactly 2 argument, the interface to capture on and the pcap file to write")
|
|
}
|
|
filter := "port 67 or port 68 or port 69 or port 4011"
|
|
args = []string{"-w", args[1], "-e", "-i", args[0]}
|
|
c := exec.Command("tcpdump", append(args, strings.Split(filter, " ")...)...)
|
|
c.Stdout = os.Stdout
|
|
c.Stderr = os.Stderr
|
|
if err := c.Run(); err != nil {
|
|
fatalf("Running tcpdump: %s", err)
|
|
}
|
|
},
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
debugCmd.AddCommand(dumpIpxeCmd)
|
|
debugCmd.AddCommand(tcpdumpCmd)
|
|
rootCmd.AddCommand(debugCmd)
|
|
}
|