freqy/protomolecule
3
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

command line flag parsing

Browse Source
This commit is contained in:
kayos 2021-05-13 11:38:07 -07:00
parent 6461cbb62e
commit 563f5bbaf9
2 changed files with 15 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
main.go
View File

@ -1,7 +1,7 @@
package main package main
import ( import (
//"flag" "flag"
"os" "os"
"protomolecule/src/dust" "protomolecule/src/dust"
"protomolecule/src/eros" "protomolecule/src/eros"
@ -56,25 +56,12 @@ func init() {
// suppress debug messages unless -d is called // suppress debug messages unless -d is called
zerolog.SetGlobalLevel(zerolog.InfoLevel) zerolog.SetGlobalLevel(zerolog.InfoLevel)
for _, arg := range os.Args {
if arg == "-d" {
zerolog.SetGlobalLevel(zerolog.DebugLevel)
}
}
log.Debug().Msg("Logging initialized") log.Debug().Msg("Logging initialized")
log.Debug().Msg("Initializing database engine") log.Debug().Msg("Initializing database engine")
// initialize database engine // initialize database engine
eros.Awaken() eros.Awaken()
for _, flag := range os.Args {
if flag == "-A" {
projVars.AttackMode = true
}
}
} }
func main() { func main() {
@ -82,8 +69,6 @@ func main() {
var scanID int var scanID int
var scan *scanStuff.Scan var scan *scanStuff.Scan
//values := flag.Args()
scanID = ScanMgr.NewScan() scanID = ScanMgr.NewScan()
scan = ScanMgr.Scans[scanID] scan = ScanMgr.Scans[scanID]

14
src/vars/projVars.go
View File

@ -32,7 +32,19 @@ var ScanAdapter = bluetooth.DefaultAdapter
var AttackAdapter = bluetooth.DefaultAdapter var AttackAdapter = bluetooth.DefaultAdapter
//Device to be targeted --not fully implemented //Device to be targeted --not fully implemented
var Target = flag.String("t", "00:00:00:00:00:00", "Target device to attack") //var Target = flag.String("t", "00:00:00:00:00:00", "Target device to attack")
var dFlag = flag.Bool("d", false, "global debug logging and pretty printing")
var aFlag = flag.Bool("a", false, "attack mode")
flag.Parse()
if *aFlag {
projVars.AttackMode = true
}
if *dFlag {
zerolog.SetGlobalLevel(zerolog.DebugLevel)
}
//var attacker bluetooth.Addresser //var attacker bluetooth.Addresser
//var connected bool //var connected bool