mirror of
https://github.com/RoganDawes/P4wnP1_aloa.git
synced 2025-03-17 21:31:56 +01:00
Added missing folders
This commit is contained in:
mame82
parent
3a777c42a1
commit
796a5e5ba8
57
cli_client/cmd_led.go
Normal file
57
cli_client/cmd_led.go
Normal file
@ -0,0 +1,57 @@
|
||||
package cli_client
|
||||
|
||||
import (
|
||||
"github.com/spf13/cobra"
|
||||
|
||||
pb "../proto"
|
||||
"fmt"
|
||||
"log"
|
||||
)
|
||||
|
||||
var blink_count uint32
|
||||
|
||||
// usbCmd represents the usb command
|
||||
var ledCmd = &cobra.Command{
|
||||
Use: "LED",
|
||||
Short: "Set or Get LED state of P4wnP1",
|
||||
}
|
||||
|
||||
var ledGetCmd = &cobra.Command{
|
||||
Use: "get",
|
||||
Short: "Get LED blink count",
|
||||
}
|
||||
|
||||
var ledSetCmd = &cobra.Command{
|
||||
Use: "set",
|
||||
Short: "Set LED blink count",
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
blink := cmd.Flags().Lookup("blink")
|
||||
if blink.Changed {
|
||||
if err := ClientSetLED(StrRemoteHost, StrRemotePort, pb.LEDSettings{BlinkCount: blink_count}); err == nil {
|
||||
fmt.Printf("LED blink count set to %v\n", blink.Value)
|
||||
} else {
|
||||
log.Println(err)
|
||||
}
|
||||
} else {
|
||||
cmd.Usage()
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(ledCmd)
|
||||
ledCmd.AddCommand(ledGetCmd)
|
||||
ledCmd.AddCommand(ledSetCmd)
|
||||
|
||||
// Here you will define your flags and configuration settings.
|
||||
|
||||
// Cobra supports Persistent Flags which will work for this command
|
||||
// and all subcommands, e.g.:
|
||||
// usbCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||
|
||||
// Cobra supports local flags which will only run when this command
|
||||
// is called directly, e.g.:
|
||||
// usbCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||
ledSetCmd.Flags().Uint32Var(&blink_count,"blink", 0,"Set blink count (0: Off, 1..254: blink n times, >254: On)")
|
||||
}
|
||||
45
cli_client/cmd_root.go
Normal file
45
cli_client/cmd_root.go
Normal file
@ -0,0 +1,45 @@
|
||||
package cli_client
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var (
|
||||
StrRemoteHost string
|
||||
StrRemotePort string
|
||||
)
|
||||
|
||||
// rootCmd represents the base command when called without any subcommands
|
||||
var rootCmd = &cobra.Command{
|
||||
Use: "cli_client",
|
||||
Short: "P4wnP1 (remote) CLI configuration",
|
||||
Long: `The cli_client tool could be used to configure P4wnP1
|
||||
from the command line. The tool relies on RPC so it could be used
|
||||
remotely.`,
|
||||
// Uncomment the following line if your bare application
|
||||
// has an action associated with it:
|
||||
// Run: func(cmd *cobra.Command, args []string) { },
|
||||
}
|
||||
|
||||
// Execute adds all child commands to the root command and sets flags appropriately.
|
||||
// This is called by main.main(). It only needs to happen once to the rootCmd.
|
||||
func Execute() {
|
||||
if err := rootCmd.Execute(); err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.PersistentFlags().StringVar(&StrRemoteHost, "host", "localhost", "The host with the listening P4wnP1 RPC server")
|
||||
rootCmd.PersistentFlags().StringVar(&StrRemotePort, "port", "50051", "The port on which the P4wnP1 RPC server is listening")
|
||||
|
||||
/*
|
||||
// Cobra also supports local flags, which will only run
|
||||
// when this action is called directly.
|
||||
rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||
*/
|
||||
}
|
||||
56
cli_client/rpc.go
Normal file
56
cli_client/rpc.go
Normal file
@ -0,0 +1,56 @@
|
||||
package cli_client
|
||||
|
||||
import (
|
||||
"log"
|
||||
"google.golang.org/grpc"
|
||||
|
||||
pb "../proto"
|
||||
"time"
|
||||
"golang.org/x/net/context"
|
||||
)
|
||||
|
||||
func ClientConnectServer(rpcHost string, rpcPort string) (
|
||||
connection *grpc.ClientConn,
|
||||
client pb.P4WNP1Client,
|
||||
ctx context.Context,
|
||||
cancel context.CancelFunc,
|
||||
err error) {
|
||||
// Set up a connection to the server.
|
||||
address := rpcHost + ":" + rpcPort
|
||||
log.Printf("Connecting %s ...", address)
|
||||
connection, err = grpc.Dial(address, grpc.WithInsecure())
|
||||
if err != nil {
|
||||
log.Fatalf("did not connect: %v", err)
|
||||
}
|
||||
//defer conn.Close()
|
||||
|
||||
client = pb.NewP4WNP1Client(connection)
|
||||
|
||||
// Contact the server
|
||||
ctx, cancel = context.WithTimeout(context.Background(), time.Second)
|
||||
//defer cancel()
|
||||
|
||||
err = nil
|
||||
return
|
||||
}
|
||||
|
||||
func ClientSetLED(host string, port string, ls pb.LEDSettings) (err error) {
|
||||
conn, client, ctx, cancel, err := ClientConnectServer(host, port)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
_, err = client.SetLEDSettings(ctx, &ls)
|
||||
if err != nil {
|
||||
log.Printf("Error setting LED blink count %d: %v", ls.BlinkCount, err)
|
||||
}
|
||||
|
||||
ClientDisconnectServer(cancel, conn)
|
||||
return
|
||||
}
|
||||
|
||||
func ClientDisconnectServer(cancel context.CancelFunc, connection *grpc.ClientConn) error {
|
||||
defer connection.Close()
|
||||
defer cancel()
|
||||
return nil
|
||||
}
|
||||
41
cli_client/usb.go
Normal file
41
cli_client/usb.go
Normal file
@ -0,0 +1,41 @@
|
||||
package cli_client
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// usbCmd represents the usb command
|
||||
var usbCmd = &cobra.Command{
|
||||
Use: "usb",
|
||||
Short: "Set or get USB Gadget settings",
|
||||
Long: ``,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
fmt.Println("usb called")
|
||||
},
|
||||
}
|
||||
|
||||
var usbGetCmd = &cobra.Command{
|
||||
Use: "get",
|
||||
Short: "Get USB Gadget settings",
|
||||
Long: ``,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
fmt.Printf("usb get called %v", args)
|
||||
},
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(usbCmd)
|
||||
usbCmd.AddCommand(usbGetCmd)
|
||||
|
||||
// Here you will define your flags and configuration settings.
|
||||
|
||||
// Cobra supports Persistent Flags which will work for this command
|
||||
// and all subcommands, e.g.:
|
||||
// usbCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||
|
||||
// Cobra supports local flags which will only run when this command
|
||||
// is called directly, e.g.:
|
||||
// usbCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||
}
|
||||
73
service/led.go
Normal file
73
service/led.go
Normal file
@ -0,0 +1,73 @@
|
||||
package service
|
||||
|
||||
import(
|
||||
"os"
|
||||
"log"
|
||||
"io/ioutil"
|
||||
"time"
|
||||
"sync/atomic"
|
||||
|
||||
pb "../proto"
|
||||
)
|
||||
|
||||
const (
|
||||
LED_TRIGGER_PATH = "/sys/class/leds/led0/trigger"
|
||||
LED_BRIGHTNESS_PATH = "/sys/class/leds/led0/brightness"
|
||||
LED_TRIGGER_MANUAL = "none"
|
||||
LED_ON = "0"
|
||||
LED_OFF = "1"
|
||||
LED_DELAY_ON = 200 * time.Millisecond
|
||||
LED_DELAY_OFF = 200 * time.Millisecond
|
||||
LED_DELAY_PAUSE = 500 * time.Millisecond
|
||||
)
|
||||
|
||||
var (
|
||||
blink_count uint32 = 0
|
||||
)
|
||||
|
||||
func InitLed(led_on bool) (error) {
|
||||
//set trigger of LED to manual
|
||||
log.Println("Setting LED to manual trigger ...")
|
||||
ioutil.WriteFile(LED_TRIGGER_PATH, []byte(LED_TRIGGER_MANUAL), os.ModePerm)
|
||||
if led_on {
|
||||
log.Println("Setting LED to ON ...")
|
||||
ioutil.WriteFile(LED_BRIGHTNESS_PATH, []byte(LED_ON), os.ModePerm)
|
||||
} else {
|
||||
log.Println("Setting LED to OFF ...")
|
||||
ioutil.WriteFile(LED_BRIGHTNESS_PATH, []byte(LED_OFF), os.ModePerm)
|
||||
}
|
||||
atomic.StoreUint32(&blink_count, 0)
|
||||
|
||||
go led_loop()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func led_loop() {
|
||||
|
||||
for {
|
||||
for i := uint32(0); i < atomic.LoadUint32(&blink_count); i++ {
|
||||
ioutil.WriteFile(LED_BRIGHTNESS_PATH, []byte(LED_ON), os.ModePerm)
|
||||
time.Sleep(LED_DELAY_ON)
|
||||
|
||||
//Don't turn off led if blink_count >= 255 (solid)
|
||||
if 255 > atomic.LoadUint32(&blink_count) {
|
||||
ioutil.WriteFile(LED_BRIGHTNESS_PATH, []byte(LED_OFF), os.ModePerm)
|
||||
time.Sleep(LED_DELAY_OFF)
|
||||
}
|
||||
}
|
||||
time.Sleep(LED_DELAY_PAUSE)
|
||||
}
|
||||
}
|
||||
|
||||
func SetLed(s pb.LEDSettings) (error) {
|
||||
//log.Printf("setLED called with %+v", s)
|
||||
|
||||
atomic.StoreUint32(&blink_count, s.BlinkCount)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func GetLed() (res *pb.LEDSettings, err error) {
|
||||
return &pb.LEDSettings{BlinkCount: atomic.LoadUint32(&blink_count)}, nil
|
||||
}
|
||||
51
service/network.go
Normal file
51
service/network.go
Normal file
@ -0,0 +1,51 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"github.com/docker/libcontainer/netlink"
|
||||
"net"
|
||||
"log"
|
||||
)
|
||||
|
||||
func CreateBridge(name string) error {
|
||||
return netlink.CreateBridge(name, false)
|
||||
}
|
||||
|
||||
func DeleteBridge(name string) error {
|
||||
return netlink.DeleteBridge(name)
|
||||
}
|
||||
|
||||
func CheckInterfaceExistence(name string) (res bool, err error) {
|
||||
_, err = net.InterfaceByName(name)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
return true, err
|
||||
}
|
||||
|
||||
func NetworkLinkUp(name string) (err error) {
|
||||
iface, err := net.InterfaceByName(name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = netlink.NetworkLinkUp(iface)
|
||||
return
|
||||
}
|
||||
|
||||
func AddInterfaceToBridgeIfExistent(bridgeName string, ifName string) (err error) {
|
||||
br, err := net.InterfaceByName(bridgeName)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
iface, err := net.InterfaceByName(ifName)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = netlink.AddToBridge(iface, br)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
log.Printf("Interface %s added to bridge %s", ifName, bridgeName)
|
||||
return nil
|
||||
}
|
||||
141
service/rpc.go
Normal file
141
service/rpc.go
Normal file
@ -0,0 +1,141 @@
|
||||
//package rpcserv
|
||||
package service
|
||||
|
||||
import (
|
||||
"log"
|
||||
pb "../proto"
|
||||
"golang.org/x/net/context"
|
||||
"net"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/reflection"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type server struct {}
|
||||
|
||||
func (s *server) GetDeployedGadgetSetting(ctx context.Context, e *pb.Empty) (gs *pb.GadgetSettings, err error) {
|
||||
gs, err = ParseGadgetState(USB_GADGET_NAME)
|
||||
|
||||
if err == nil {
|
||||
j_usbset, _ := json.Marshal(gs)
|
||||
log.Printf("Gadget settings requested %v", string(j_usbset))
|
||||
} else {
|
||||
log.Printf("Error parsing current gadget config: %v", err)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (s *server) DeployGadgetSetting(context.Context, *pb.Empty) (gs *pb.GadgetSettings, err error) {
|
||||
gs_backup,_ := ParseGadgetState(USB_GADGET_NAME)
|
||||
errg := DeployGadgetSettings(GadgetSettingsState)
|
||||
err = nil
|
||||
if errg != nil {
|
||||
err = errors.New(fmt.Sprintf("Deploying new gadget settings failed, reverted to old ones: %v", errg))
|
||||
DeployGadgetSettings(*gs_backup) //We don't catch the error, as the old settings should have been working
|
||||
}
|
||||
|
||||
gs, _ = ParseGadgetState(USB_GADGET_NAME) //Return settings from deployed gadget
|
||||
return
|
||||
}
|
||||
|
||||
func (s *server) GetGadgetSettings(context.Context, *pb.Empty) (*pb.GadgetSettings, error) {
|
||||
return &GadgetSettingsState, nil
|
||||
}
|
||||
|
||||
func (s *server) SetGadgetSettings(ctx context.Context, gs *pb.GadgetSettings) (res *pb.GadgetSettings, err error) {
|
||||
if err = ValidateGadgetSetting(*gs); err != nil {
|
||||
//We return the validation error and the current (unchanged) GadgetSettingsState
|
||||
res = &GadgetSettingsState
|
||||
return
|
||||
}
|
||||
GadgetSettingsState = *gs
|
||||
res = &GadgetSettingsState
|
||||
return
|
||||
}
|
||||
|
||||
func (s *server) GetLEDSettings(context.Context, *pb.Empty) (res *pb.LEDSettings, err error) {
|
||||
return GetLed()
|
||||
}
|
||||
|
||||
func (s *server) SetLEDSettings(ctx context.Context, ls *pb.LEDSettings) (*pb.Empty, error) {
|
||||
log.Printf("SetLEDSettings %+v", ls)
|
||||
SetLed(*ls)
|
||||
return &pb.Empty{}, nil
|
||||
}
|
||||
|
||||
func (s *server) StopGadget(context.Context, *pb.Empty) (*pb.Empty, error) {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (s *server) StartGadget(context.Context, *pb.Empty) (*pb.Empty, error) {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
/*
|
||||
//Attach handler function implementing the "GetGadgetSettings" interface to server
|
||||
func (s *server) GetGadgetSettings(context.Context, *pb.Empty) (usbset *pb.GadgetSettings, err error) {
|
||||
usbset, err = ParseGadgetState(USB_GADGET_NAME)
|
||||
|
||||
if err == nil {
|
||||
j_usbset, _ := json.Marshal(usbset)
|
||||
log.Printf("Gadget settings requested %v", string(j_usbset))
|
||||
} else {
|
||||
log.Printf("Error parsing current gadget config: %v", err)
|
||||
}
|
||||
|
||||
|
||||
return usbset, err
|
||||
}
|
||||
|
||||
|
||||
|
||||
//Attach handler function implementing the "SetGadgetSettings" interface to server
|
||||
|
||||
|
||||
func (s *server) SetGadgetSettings(context.Context, *pb.GadgetSettings) (*pb.Error, error) {
|
||||
return &pb.Error{Err: 0}, nil
|
||||
}
|
||||
|
||||
//Attach handler function implementing the "StartGadget" interface to server
|
||||
func (s *server) StartGadget(context.Context, *pb.Empty) (*pb.Error, error) {
|
||||
return &pb.Error{Err: 0}, nil
|
||||
}
|
||||
|
||||
//Attach handler function implementing the "StopGadget" interface to server
|
||||
func (s *server) StopGadget(context.Context, *pb.Empty) (*pb.Error, error) {
|
||||
return &pb.Error{Err: 0}, nil
|
||||
}
|
||||
|
||||
func (s *server) GetLEDSettings(context.Context, *pb.Empty) (*pb.LEDSettings, error) {
|
||||
led_settings := &pb.LEDSettings{}
|
||||
return led_settings, nil
|
||||
}
|
||||
|
||||
func (s *server) SetLEDSettings(ctx context.Context, ledSettings *pb.LEDSettings) (rpcerr *pb.Error, err error) {
|
||||
log.Printf("SetLEDSettings %+v", ledSettings)
|
||||
setLed(*ledSettings)
|
||||
return &pb.Error{Err: 0}, nil
|
||||
}
|
||||
*/
|
||||
|
||||
func StartRpcServer(host string, port string) {
|
||||
listen_address := host + ":" + port
|
||||
//Open TCP listener
|
||||
log.Printf("P4wnP1 RPC server listening on " + listen_address)
|
||||
lis, err := net.Listen("tcp", listen_address)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to listen: %v", err)
|
||||
}
|
||||
|
||||
//Create gRPC Server
|
||||
s := grpc.NewServer()
|
||||
pb.RegisterP4WNP1Server(s, &server{})
|
||||
// Register reflection service on gRPC server.
|
||||
reflection.Register(s)
|
||||
if err := s.Serve(lis); err != nil {
|
||||
log.Fatalf("Failed to serve: %v", err)
|
||||
}
|
||||
}
|
||||
615
service/usb.go
Normal file
615
service/usb.go
Normal file
@ -0,0 +1,615 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strings"
|
||||
|
||||
pb "../proto"
|
||||
"time"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
const (
|
||||
USB_GADGET_NAME = "mame82_gadget"
|
||||
USB_GADGET_DIR_BASE = "/sys/kernel/config/usb_gadget"
|
||||
USB_GADGET_DIR = USB_GADGET_DIR_BASE + "/" + USB_GADGET_NAME
|
||||
|
||||
USB_ETHERNET_BRIDGE_NAME = "usbeth"
|
||||
|
||||
USB_bcdDevice = "0x0100" //Version 1.00
|
||||
USB_bcdUSB = "0x0200" //mode: USB 2.0
|
||||
|
||||
// composite class / subclass / proto (needs single configuration)
|
||||
USB_bDeviceClass = "0xEF"
|
||||
USB_bDeviceSubClass = "0x02"
|
||||
USB_bDeviceProtocol = "0x01"
|
||||
|
||||
USB_CONFIGURATION_MaxPower = "250"
|
||||
USB_CONFIGURATION_bmAttributes = "0x80" //should be 0x03 for USB_OTG_SRP | USB_OTG_HNP
|
||||
|
||||
/*
|
||||
//RNDIS function constants
|
||||
USB_FUNCTION_RNDIS_DEFAULT_host_addr = "42:63:65:12:34:56"
|
||||
USB_FUNCTION_RNDIS_DEFAULT_dev_addr = "42:63:65:56:34:12"
|
||||
*/
|
||||
//OS descriptors for RNDIS composite function on Windows
|
||||
USB_FUNCTION_RNDIS_os_desc_use = "1"
|
||||
USB_FUNCTION_RNDIS_os_desc_b_vendor_code = "0xbc"
|
||||
USB_FUNCTION_RNDIS_os_desc_qw_sign = "MSFT100"
|
||||
USB_FUNCTION_RNDIS_os_desc_interface_compatible_id = "RNDIS"
|
||||
USB_FUNCTION_RNDIS_os_desc_interface_sub_compatible_id = "5162001"
|
||||
|
||||
/*
|
||||
//CDC ECM function constants
|
||||
USB_FUNCTION_CDC_ECM_DEFAULT_host_addr = "42:63:66:12:34:56"
|
||||
USB_FUNCTION_CDC_ECM_DEFAULT_dev_addr = "42:63:66:56:34:12"
|
||||
*/
|
||||
|
||||
//HID function, keyboard constants
|
||||
USB_FUNCTION_HID_KEYBOARD_protocol = "1"
|
||||
USB_FUNCTION_HID_KEYBOARD_subclass = "1"
|
||||
USB_FUNCTION_HID_KEYBOARD_report_length = "8"
|
||||
USB_FUNCTION_HID_KEYBOARD_report_desc = "\x05\x01\t\x06\xa1\x01\x05\x07\x19\xe0)\xe7\x15\x00%\x01u\x01\x95\x08\x81\x02\x95\x01u\x08\x81\x03\x95\x05u\x01\x05\x08\x19\x01)\x05\x91\x02\x95\x01u\x03\x91\x03\x95\x06u\x08\x15\x00%e\x05\x07\x19\x00)e\x81\x00\xc0"
|
||||
USB_FUNCTION_HID_KEYBOARD_name = "hid.keyboard"
|
||||
|
||||
//HID function, mouse constants
|
||||
USB_FUNCTION_HID_MOUSE_protocol = "2"
|
||||
USB_FUNCTION_HID_MOUSE_subclass = "1"
|
||||
USB_FUNCTION_HID_MOUSE_report_length = "6"
|
||||
USB_FUNCTION_HID_MOUSE_report_desc = "\x05\x01\t\x02\xa1\x01\t\x01\xa1\x00\x85\x01\x05\t\x19\x01)\x03\x15\x00%\x01\x95\x03u\x01\x81\x02\x95\x01u\x05\x81\x03\x05\x01\t0\t1\x15\x81%\x7fu\x08\x95\x02\x81\x06\x95\x02u\x08\x81\x01\xc0\xc0\x05\x01\t\x02\xa1\x01\t\x01\xa1\x00\x85\x02\x05\t\x19\x01)\x03\x15\x00%\x01\x95\x03u\x01\x81\x02\x95\x01u\x05\x81\x01\x05\x01\t0\t1\x15\x00&\xff\x7f\x95\x02u\x10\x81\x02\xc0\xc0"
|
||||
USB_FUNCTION_HID_MOUSE_name = "hid.mouse"
|
||||
|
||||
//HID function, custom vendor device constants
|
||||
USB_FUNCTION_HID_RAW_protocol = "1"
|
||||
USB_FUNCTION_HID_RAW_subclass = "1"
|
||||
USB_FUNCTION_HID_RAW_report_length = "64"
|
||||
USB_FUNCTION_HID_RAW_report_desc = "\x06\x00\xff\t\x01\xa1\x01\t\x01\x15\x00&\xff\x00u\x08\x95@\x81\x02\t\x02\x15\x00&\xff\x00u\x08\x95@\x91\x02\xc0"
|
||||
USB_FUNCTION_HID_RAW_name = "hid.raw"
|
||||
|
||||
)
|
||||
|
||||
var (
|
||||
DefaultGadgetSettings pb.GadgetSettings = GetDefaultGadgetSettings()
|
||||
GadgetSettingsState pb.GadgetSettings = GetDefaultGadgetSettings()
|
||||
)
|
||||
|
||||
func ValidateGadgetSetting(gs pb.GadgetSettings) error {
|
||||
/* ToDo: validations
|
||||
- check host_addr/dev_addr of RNDIS + CDC ECM to be valid MAC adresses via regex
|
||||
- check host_addr/dev_addr of RNDIS + CDC ECM for duplicates
|
||||
- check EP consumption to be not more than 7 (ECM 2 EP, RNDIS 2 EP, HID Mouse 1 EP, HID Keyboard 1 EP, HID Raw 1 EP, Serial 2 EP ??, UMS ??)
|
||||
- check serial, product, Manufacturer to not be empty
|
||||
- check Pid, Vid with regex (Note: we don't check if Vid+Pid have been used for another composite function setup, yet)
|
||||
*/
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func addUSBEthernetBridge() {
|
||||
//Create the bridge
|
||||
CreateBridge(USB_ETHERNET_BRIDGE_NAME)
|
||||
|
||||
//add the interfaces
|
||||
if err := AddInterfaceToBridgeIfExistent(USB_ETHERNET_BRIDGE_NAME, "usb0"); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
if err := AddInterfaceToBridgeIfExistent(USB_ETHERNET_BRIDGE_NAME, "usb1"); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
//enable the bridge
|
||||
NetworkLinkUp(USB_ETHERNET_BRIDGE_NAME)
|
||||
}
|
||||
|
||||
func deleteUSBEthernetBridge() {
|
||||
//we ignore error results
|
||||
DeleteBridge(USB_ETHERNET_BRIDGE_NAME)
|
||||
}
|
||||
|
||||
/*
|
||||
Polls for presence of "usb0" / "usb1" till one of both is active or timeout is reached
|
||||
*/
|
||||
|
||||
func pollForUSBEthernet(timeout time.Duration) error {
|
||||
for startTime := time.Now(); time.Since(startTime) < timeout; {
|
||||
if present, _ := CheckInterfaceExistence("usb0"); present {
|
||||
return nil
|
||||
}
|
||||
if present, _ := CheckInterfaceExistence("usb1"); present {
|
||||
return nil
|
||||
}
|
||||
|
||||
//Take a breath
|
||||
time.Sleep(100*time.Millisecond)
|
||||
fmt.Print(".")
|
||||
}
|
||||
return errors.New(fmt.Sprintf("Timeout %v reached before usb0 or usb1 cam up"))
|
||||
}
|
||||
|
||||
func InitDefaultGadgetSettings() error {
|
||||
return DeployGadgetSettings(DefaultGadgetSettings)
|
||||
}
|
||||
|
||||
func GetDefaultGadgetSettings() (res pb.GadgetSettings) {
|
||||
res = pb.GadgetSettings{
|
||||
Enabled: false,
|
||||
Vid: "0x1d6b",
|
||||
Pid: "0x1337",
|
||||
Manufacturer: "MaMe82",
|
||||
Product: "P4wnP1 by MaMe82",
|
||||
Serial: "deadbeef1337",
|
||||
Use_CDC_ECM: false,
|
||||
Use_RNDIS: false,
|
||||
Use_HID_KEYBOARD: false,
|
||||
Use_HID_MOUSE: false,
|
||||
Use_HID_RAW: false,
|
||||
Use_UMS: false,
|
||||
Use_SERIAL: false,
|
||||
RndisSettings: &pb.GadgetSettingsEthernet{
|
||||
HostAddr: "42:63:65:12:34:56",
|
||||
DevAddr: "42:63:65:56:34:12",
|
||||
},
|
||||
CdcEcmSettings: &pb.GadgetSettingsEthernet{
|
||||
HostAddr: "42:63:66:12:34:56",
|
||||
DevAddr: "42:63:66:56:34:12",
|
||||
},
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
//depends on `bash`, `grep` and `lsmod` binary
|
||||
func CheckLibComposite() error {
|
||||
log.Printf("Checking for libcomposite...")
|
||||
out, err := exec.Command("lsmod").Output()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
if strings.Contains(string(out), "libcomposite") {
|
||||
log.Printf("... libcomposite loaded")
|
||||
return nil
|
||||
}
|
||||
|
||||
//if here, libcomposite isn't loaded ... try to load
|
||||
log.Printf("Libcomposite not loaded, trying to fix ...")
|
||||
err = exec.Command("modprobe", "libcomposite").Run()
|
||||
if err == nil {
|
||||
log.Printf("... libcomposite loaded")
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func getUDCName() (string, error) {
|
||||
files, err := ioutil.ReadDir("/sys/class/udc")
|
||||
if err != nil {
|
||||
return "", errors.New("Couldn't find working UDC driver")
|
||||
}
|
||||
if len(files) < 1 {
|
||||
return "", errors.New("Couldn't find working UDC driver")
|
||||
}
|
||||
return files[0].Name(), nil
|
||||
|
||||
}
|
||||
|
||||
func ParseGadgetState(gadgetName string) (result *pb.GadgetSettings, err error) {
|
||||
err = nil
|
||||
result = &pb.GadgetSettings{}
|
||||
result.CdcEcmSettings = &pb.GadgetSettingsEthernet{}
|
||||
result.RndisSettings = &pb.GadgetSettingsEthernet{}
|
||||
|
||||
//gadget_root := "./test"
|
||||
gadget_dir := USB_GADGET_DIR_BASE + "/" + gadgetName
|
||||
|
||||
//check if root exists, return error otherwise
|
||||
if _, err = os.Stat(gadget_dir); os.IsNotExist(err) {
|
||||
err = errors.New(fmt.Sprintf("Gadget %s doesn't exist", gadgetName))
|
||||
result = nil
|
||||
return
|
||||
}
|
||||
|
||||
//ToDo: check if enabled (UDC in functionfs is set to content of /sys/class/udc)
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/idVendor"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading Vid", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.Vid = strings.TrimSuffix(string(res), "\n")
|
||||
}
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/idProduct"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading Pid", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.Pid = strings.TrimSuffix(string(res), "\n")
|
||||
}
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/strings/0x409/serialnumber"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading Serial", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.Serial = strings.TrimSuffix(string(res), "\n")
|
||||
}
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/strings/0x409/manufacturer"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading Manufacturer", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.Manufacturer = strings.TrimSuffix(string(res), "\n")
|
||||
}
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/strings/0x409/product"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading Product", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.Product = strings.TrimSuffix(string(res), "\n")
|
||||
}
|
||||
|
||||
//Check enabled functions in configuration
|
||||
|
||||
//USB RNDIS
|
||||
if _, err1 := os.Stat(gadget_dir+"/configs/c.1/rndis.usb0"); !os.IsNotExist(err1) {
|
||||
result.Use_RNDIS = true
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/functions/rndis.usb0/host_addr"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading RNDIS host_addr", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.RndisSettings.HostAddr = strings.TrimSuffix(string(res), "\000\n")
|
||||
}
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/functions/rndis.usb0/dev_addr"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading RNDIS dev_addr", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.RndisSettings.DevAddr = strings.TrimSuffix(string(res), "\000\n")
|
||||
}
|
||||
}
|
||||
|
||||
//USB CDC ECM
|
||||
if _, err1 := os.Stat(gadget_dir+"/configs/c.1/ecm.usb1"); !os.IsNotExist(err1) {
|
||||
result.Use_CDC_ECM = true
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/functions/ecm.usb1/host_addr"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading CDC ECM host_addr", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.CdcEcmSettings.HostAddr = strings.TrimSuffix(string(res), "\000\n")
|
||||
}
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/functions/ecm.usb1/dev_addr"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading CDC ECM dev_addr", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
result.CdcEcmSettings.DevAddr = strings.TrimSuffix(string(res), "\000\n")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
//USB serial
|
||||
if _, err1 := os.Stat(gadget_dir+"/configs/c.1/acm.GS0"); !os.IsNotExist(err1) {
|
||||
result.Use_SERIAL = true
|
||||
}
|
||||
|
||||
//USB HID Keyboard
|
||||
if _, err1 := os.Stat(gadget_dir+"/configs/c.1/"+USB_FUNCTION_HID_KEYBOARD_name); !os.IsNotExist(err1) {
|
||||
result.Use_HID_KEYBOARD = true
|
||||
}
|
||||
|
||||
//USB HID Mouse
|
||||
if _, err1 := os.Stat(gadget_dir+"/configs/c.1/"+USB_FUNCTION_HID_MOUSE_name); !os.IsNotExist(err1) {
|
||||
result.Use_HID_KEYBOARD = true
|
||||
}
|
||||
|
||||
//USB HID RAW
|
||||
if _, err1 := os.Stat(gadget_dir+"/configs/c.1/"+USB_FUNCTION_HID_RAW_name); !os.IsNotExist(err1) {
|
||||
result.Use_HID_RAW = true
|
||||
}
|
||||
|
||||
//check if UDC is set (Gadget enabled)
|
||||
udc_name, _ := getUDCName()
|
||||
|
||||
if res, err := ioutil.ReadFile(gadget_dir + "/UDC"); err != nil {
|
||||
err1 := errors.New(fmt.Sprintf("Gadget %s error reading UDC", gadgetName))
|
||||
return nil, err1
|
||||
} else {
|
||||
udc_name_set := strings.TrimSuffix(string(res), "\n")
|
||||
if udc_name == udc_name_set {
|
||||
result.Enabled = true
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
|
||||
}
|
||||
|
||||
func DeployGadgetSettings(settings pb.GadgetSettings) error {
|
||||
var usesUSBEthernet bool
|
||||
|
||||
//gadget_root := "./test"
|
||||
gadget_root := USB_GADGET_DIR_BASE
|
||||
|
||||
//check if root exists, return error otherwise
|
||||
if _, err := os.Stat(gadget_root); os.IsNotExist(err) {
|
||||
return errors.New("Configfs path for gadget doesn't exist")
|
||||
}
|
||||
|
||||
//ToDo: check if UDC is present and usable
|
||||
|
||||
//create gadget folder
|
||||
os.Mkdir(USB_GADGET_DIR, os.ModePerm)
|
||||
log.Printf("Creating composite gadget '%s'", USB_GADGET_NAME)
|
||||
|
||||
//set vendor ID, product ID
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/idVendor", []byte(settings.Vid), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/idProduct", []byte(settings.Pid), os.ModePerm)
|
||||
|
||||
//set USB mode to 2.0 and device version to 1.0
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/bcdUSB", []byte(USB_bcdUSB), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/bcdDevice", []byte(USB_bcdDevice), os.ModePerm)
|
||||
|
||||
//composite class / subclass / proto (needs single configuration)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/bDeviceClass", []byte(USB_bDeviceClass), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/bDeviceSubClass", []byte(USB_bDeviceSubClass), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/bDeviceProtocol", []byte(USB_bDeviceProtocol), os.ModePerm)
|
||||
|
||||
// set device descriptions
|
||||
os.Mkdir(USB_GADGET_DIR+"/strings/0x409", os.ModePerm) // English language strings
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/strings/0x409/serialnumber", []byte(settings.Serial), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/strings/0x409/manufacturer", []byte(settings.Manufacturer), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/strings/0x409/product", []byte(settings.Product), os.ModePerm)
|
||||
|
||||
// create configuration instance (only one, as multiple configs aren't valid for Windows composite devices)
|
||||
os.MkdirAll(USB_GADGET_DIR+"/configs/c.1/strings/0x409", os.ModePerm) // English language strings
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/configs/c.1/strings/0x409/configuration", []byte("Config 1: Composite"), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/configs/c.1/MaxPower", []byte(USB_CONFIGURATION_MaxPower), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/configs/c.1/bmAttributes", []byte(USB_CONFIGURATION_bmAttributes), os.ModePerm)
|
||||
|
||||
// RNDIS has to be the first interface on Composite device for Windows (first function initialized)
|
||||
if settings.Use_RNDIS {
|
||||
log.Printf("... creating USB RNDIS function")
|
||||
usesUSBEthernet = true
|
||||
os.Mkdir(USB_GADGET_DIR+"/functions/rndis.usb0", os.ModePerm) //create RNDIS function
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/functions/rndis.usb0/host_addr", []byte(settings.RndisSettings.HostAddr), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/functions/rndis.usb0/dev_addr", []byte(settings.RndisSettings.DevAddr), os.ModePerm)
|
||||
|
||||
/*
|
||||
add OS specific device descriptors to force Windows to load RNDIS drivers
|
||||
=============================================================================
|
||||
Witout this additional descriptors, most Windows system detect the RNDIS interface as "Serial COM port"
|
||||
To prevent this, the Microsoft specific OS descriptors are added in here
|
||||
!! Important:
|
||||
If the device already has been connected to the Windows System without providing the
|
||||
OS descriptor, Windows never asks again for them and thus never installs the RNDIS driver
|
||||
This behavior is driven by creation of an registry hive, the first time a device without
|
||||
OS descriptors is attached. The key is build like this:
|
||||
|
||||
HKLM\SYSTEM\CurrentControlSet\Control\usbflags\[USB_VID+USB_PID+bcdRelease\osvc
|
||||
|
||||
To allow Windows to read the OS descriptors again, the according registry hive has to be
|
||||
deleted manually or USB descriptor values have to be cahnged (f.e. USB_PID).
|
||||
*/
|
||||
|
||||
//set OS descriptors for Windows
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/os_desc/use", []byte(USB_FUNCTION_RNDIS_os_desc_use), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/os_desc/b_vendor_code", []byte(USB_FUNCTION_RNDIS_os_desc_b_vendor_code), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/os_desc/qw_sign", []byte(USB_FUNCTION_RNDIS_os_desc_qw_sign), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/functions/rndis.usb0/os_desc/interface.rndis/compatible_id", []byte(USB_FUNCTION_RNDIS_os_desc_interface_compatible_id), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/functions/rndis.usb0/os_desc/interface.rndis/sub_compatible_id", []byte(USB_FUNCTION_RNDIS_os_desc_interface_sub_compatible_id), os.ModePerm)
|
||||
|
||||
//activate function by symlinking to config 1
|
||||
err := os.Symlink(USB_GADGET_DIR+"/functions/rndis.usb0", USB_GADGET_DIR+"/configs/c.1/rndis.usb0")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
// add config 1 to OS descriptors
|
||||
err = os.Symlink(USB_GADGET_DIR+"/configs/c.1", USB_GADGET_DIR+"/os_desc/c.1")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
if settings.Use_CDC_ECM {
|
||||
log.Printf("... creating USB CDC ECM function")
|
||||
usesUSBEthernet = true
|
||||
os.Mkdir(USB_GADGET_DIR+"/functions/ecm.usb1", os.ModePerm) //create CDC ECM function
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/functions/ecm.usb1/host_addr", []byte(settings.CdcEcmSettings.HostAddr), os.ModePerm)
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/functions/ecm.usb1/dev_addr", []byte(settings.CdcEcmSettings.DevAddr), os.ModePerm)
|
||||
|
||||
//activate function by symlinking to config 1
|
||||
err := os.Symlink(USB_GADGET_DIR+"/functions/ecm.usb1", USB_GADGET_DIR+"/configs/c.1/ecm.usb1")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
if settings.Use_SERIAL {
|
||||
log.Printf("... creating USB serial function")
|
||||
os.Mkdir(USB_GADGET_DIR+"/functions/acm.GS0", os.ModePerm) //create ACM function
|
||||
|
||||
//activate function by symlinking to config 1
|
||||
err := os.Symlink(USB_GADGET_DIR+"/functions/acm.GS0", USB_GADGET_DIR+"/configs/c.1/acm.GS0")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
if settings.Use_HID_KEYBOARD {
|
||||
log.Printf("... creating USB HID Keyboard function")
|
||||
funcdir := USB_GADGET_DIR + "/functions/" + USB_FUNCTION_HID_KEYBOARD_name
|
||||
os.Mkdir(funcdir, os.ModePerm) //create HID function for keyboard
|
||||
|
||||
ioutil.WriteFile(funcdir+"/protocol", []byte(USB_FUNCTION_HID_KEYBOARD_protocol), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/subclass", []byte(USB_FUNCTION_HID_KEYBOARD_subclass), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/report_length", []byte(USB_FUNCTION_HID_KEYBOARD_report_length), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/report_desc", []byte(USB_FUNCTION_HID_KEYBOARD_report_desc), os.ModePerm)
|
||||
|
||||
err := os.Symlink(funcdir, USB_GADGET_DIR+"/configs/c.1/"+USB_FUNCTION_HID_KEYBOARD_name)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
if settings.Use_HID_MOUSE {
|
||||
log.Printf("... creating USB HID Mouse function")
|
||||
funcdir := USB_GADGET_DIR + "/functions/" + USB_FUNCTION_HID_MOUSE_name
|
||||
os.Mkdir(funcdir, os.ModePerm) //create HID function for mouse
|
||||
|
||||
ioutil.WriteFile(funcdir+"/protocol", []byte(USB_FUNCTION_HID_MOUSE_protocol), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/subclass", []byte(USB_FUNCTION_HID_MOUSE_subclass), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/report_length", []byte(USB_FUNCTION_HID_MOUSE_report_length), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/report_desc", []byte(USB_FUNCTION_HID_MOUSE_report_desc), os.ModePerm)
|
||||
|
||||
err := os.Symlink(funcdir, USB_GADGET_DIR+"/configs/c.1/"+USB_FUNCTION_HID_MOUSE_name)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
if settings.Use_HID_RAW {
|
||||
log.Printf("... creating USB HID Generic device function")
|
||||
funcdir := USB_GADGET_DIR + "/functions/" + USB_FUNCTION_HID_RAW_name
|
||||
os.Mkdir(funcdir, os.ModePerm) //create HID function for mouse
|
||||
|
||||
ioutil.WriteFile(funcdir+"/protocol", []byte(USB_FUNCTION_HID_RAW_protocol), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/subclass", []byte(USB_FUNCTION_HID_RAW_subclass), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/report_length", []byte(USB_FUNCTION_HID_RAW_report_length), os.ModePerm)
|
||||
ioutil.WriteFile(funcdir+"/report_desc", []byte(USB_FUNCTION_HID_RAW_report_desc), os.ModePerm)
|
||||
|
||||
err := os.Symlink(funcdir, USB_GADGET_DIR+"/configs/c.1/"+USB_FUNCTION_HID_RAW_name)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
//get UDC driver name and bind to gadget
|
||||
if settings.Enabled {
|
||||
udc_name, err := getUDCName()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
ioutil.WriteFile(USB_GADGET_DIR+"/UDC", []byte(udc_name), os.ModePerm)
|
||||
}
|
||||
|
||||
|
||||
deleteUSBEthernetBridge() //delete former used bridge, if there's any
|
||||
//In case USB ethernet is uesd (RNDIS or CDC ECM), we add a bridge interface
|
||||
if usesUSBEthernet {
|
||||
//wait till "usb0" or "usb1" comes up
|
||||
err := pollForUSBEthernet(10*time.Second)
|
||||
if err == nil {
|
||||
//add USBEthernet bridge including the usb interfaces
|
||||
log.Printf("... creating network bridge for USB ethernet devices")
|
||||
addUSBEthernetBridge()
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
log.Printf("... done")
|
||||
return nil
|
||||
}
|
||||
|
||||
func DestroyAllGadgets() error {
|
||||
//gadget_root := "./test"
|
||||
gadget_root := USB_GADGET_DIR_BASE
|
||||
|
||||
//check if root exists, return error otherwise
|
||||
if _, err := os.Stat(gadget_root); os.IsNotExist(err) {
|
||||
return errors.New("Configfs path for gadget doesn't exist")
|
||||
}
|
||||
|
||||
gadget_dirs, err := ioutil.ReadDir(gadget_root)
|
||||
if err != nil {
|
||||
return errors.New("No gadgets")
|
||||
}
|
||||
|
||||
for _, gadget_dir_obj := range gadget_dirs {
|
||||
gadget_name := gadget_dir_obj.Name()
|
||||
log.Println("Found gadget: " + gadget_name)
|
||||
err = DestroyGadget(gadget_name)
|
||||
if err != nil {
|
||||
log.Println(err) //don't return, continue with next
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func DestroyGadget(Gadget_name string) error {
|
||||
//gadget_root := "./test"
|
||||
gadget_dir := USB_GADGET_DIR_BASE + "/" + Gadget_name
|
||||
|
||||
//check if root exists, return error otherwise
|
||||
if _, err := os.Stat(USB_GADGET_DIR_BASE); os.IsNotExist(err) {
|
||||
return errors.New("Gadget " + Gadget_name + " doesn't exist")
|
||||
}
|
||||
log.Println("Deconstructing gadget " + Gadget_name + "...")
|
||||
|
||||
//Assure gadget gets unbound from UDC
|
||||
ioutil.WriteFile(gadget_dir+"/UDC", []byte("\x00"), os.ModePerm)
|
||||
|
||||
//Iterate over configurations
|
||||
config_dirs, _ := ioutil.ReadDir(gadget_dir + "/configs")
|
||||
for _, conf_dir_obj := range config_dirs {
|
||||
conf_name := conf_dir_obj.Name()
|
||||
conf_dir := gadget_dir + "/configs/" + conf_name
|
||||
log.Println("Found config: " + conf_name)
|
||||
|
||||
//find linked functions
|
||||
conf_content, _ := ioutil.ReadDir(conf_dir)
|
||||
for _, function := range conf_content {
|
||||
//Remove link from function to config
|
||||
if function.Mode()&os.ModeSymlink > 0 {
|
||||
log.Println("\tRemoving function " + function.Name() + " from config " + conf_name)
|
||||
os.Remove(conf_dir + "/" + function.Name())
|
||||
}
|
||||
}
|
||||
|
||||
//find string directories in config
|
||||
strings_content, _ := ioutil.ReadDir(conf_dir + "/strings")
|
||||
for _, str := range strings_content {
|
||||
string_dir := str.Name()
|
||||
//Remove string from config
|
||||
log.Println("\tRemoving string dir '" + string_dir + "' from configuration")
|
||||
os.Remove(conf_dir + "/strings/" + string_dir)
|
||||
}
|
||||
|
||||
//Check if there's an OS descriptor refering this config
|
||||
if _, err := os.Stat(gadget_dir + "/os_desc/" + conf_name); !os.IsNotExist(err) {
|
||||
log.Println("\tDeleting link to '" + conf_name + "' from gadgets OS descriptor")
|
||||
os.Remove(gadget_dir + "/os_desc/" + conf_name)
|
||||
}
|
||||
|
||||
// remove config folder, finally
|
||||
log.Println("\tDeleting configuration '" + conf_name + "'")
|
||||
os.Remove(conf_dir)
|
||||
}
|
||||
|
||||
// remove functions
|
||||
log.Println("Removing functions from '" + Gadget_name + "'")
|
||||
os.RemoveAll(gadget_dir + "/functions/")
|
||||
|
||||
//find string directories in gadget
|
||||
strings_content, _ := ioutil.ReadDir(gadget_dir + "/strings")
|
||||
for _, str := range strings_content {
|
||||
string_dir := str.Name()
|
||||
//Remove string from config
|
||||
log.Println("Removing string dir '" + string_dir + "' from " + Gadget_name)
|
||||
os.Remove(gadget_dir + "/strings/" + string_dir)
|
||||
}
|
||||
|
||||
//And now remove the gadget itself
|
||||
log.Println("Removing gadget " + Gadget_name)
|
||||
os.Remove(gadget_dir)
|
||||
|
||||
return nil
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user