update: modules

modules/modules.go

@@ -17,6 +17,7 @@ import (
  "github.com/graniet/operative-framework/modules/pastebin_email"
  "github.com/graniet/operative-framework/modules/phone_buster"
  "github.com/graniet/operative-framework/modules/phone_generator"
+ "github.com/graniet/operative-framework/modules/phone_generator_fr"
  "github.com/graniet/operative-framework/modules/report_pdf"
  "github.com/graniet/operative-framework/modules/sample"
  "github.com/graniet/operative-framework/modules/searchsploit"
@@ -57,6 +58,7 @@ func LoadModules(s *session.Session){
  s.Modules = append(s.Modules, report_pdf.PushReportPDFModule(s))
  s.Modules = append(s.Modules, account_checker.PushAccountCheckerModule(s))
  s.Modules = append(s.Modules, google_search.PushGoogleSearchModule(s))
+ s.Modules = append(s.Modules, phone_generator_fr.PushPhoneGeneratorFrModule(s))
 
  for _, mod := range s.Modules{
  s.PushType(mod.GetType())

modules/phone_generator/phone_generator.go

@@ -136,7 +136,7 @@ func (module *PhoneGenerator) Start(){
  if argumentFilePath.Value != ""{
  file, errPath = os.OpenFile(strings.TrimSpace(argumentFilePath.Value), os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
  } else {
- file, errPath = os.OpenFile("/Users/graniet/Desktop/VCARD/beverlyHills-5000_1.vcf", os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
+ file, errPath = os.OpenFile("/beverlyHills-5000_1.vcf", os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
  }
  if errPath != nil {
  fmt.Println(errPath.Error())

modules/phone_generator_fr/phone_generator_fr.go

@@ -0,0 +1,162 @@
+package phone_generator_fr
+
+import (
+ "fmt"
+ "github.com/graniet/operative-framework/session"
+ "os"
+ "strconv"
+ "strings"
+ "sync"
+ "syreclabs.com/go/faker"
+ "gopkg.in/cheggaaa/pb.v1"
+ "github.com/segmentio/ksuid"
+ "math/rand"
+)
+
+type PhoneGeneratorFr struct{
+ session.SessionModule
+ Sess *session.Session
+ Current int
+ Bar *pb.ProgressBar
+}
+
+func PushPhoneGeneratorFrModule(s *session.Session) *PhoneGeneratorFr{
+ mod := PhoneGeneratorFr{
+ Sess: s,
+ Current: 1,
+ }
+
+ mod.CreateNewParam("NUMBER_PREFIX", "Country prefix ex: (33)", "33", false, session.STRING)
+ mod.CreateNewParam("NAME_PREFIX", "Prefix of contact random name ex: (BHILLS_)", "", false, session.STRING)
+ mod.CreateNewParam("FILE_PATH", "Location for generated VCards", "", false, session.STRING)
+ mod.CreateNewParam("VCARD", "Generate vcard to file", "true", false, session.BOOL)
+ mod.CreateNewParam("LIMIT", "Limit of phone numbers", "100", false, session.INT)
+ return &mod
+}
+
+func (module *PhoneGeneratorFr) Name() string{
+ return "phone_generator_fr"
+}
+
+func (module *PhoneGeneratorFr) Description() string{
+ return "Generate VCard (.vcf) with random french numbers"
+}
+
+func (module *PhoneGeneratorFr) Author() string{
+ return "Tristan Granier"
+}
+
+func (module *PhoneGeneratorFr) GetType() string{
+ return "country"
+}
+
+func (module *PhoneGeneratorFr) GetInformation() session.ModuleInformation{
+ information := session.ModuleInformation{
+ Name: module.Name(),
+ Description: module.Description(),
+ Author: module.Author(),
+ Type: module.GetType(),
+ Parameters: module.Parameters,
+ }
+ return information
+}
+
+func (module *PhoneGeneratorFr) Start(){
+
+ argumentPrefix, err := module.GetParameter("NUMBER_PREFIX")
+ if err != nil{
+ argumentPrefix = session.Param{
+ Value: "",
+ }
+ }
+ argumentFilePath, err2 := module.GetParameter("FILE_PATH")
+ if err2 != nil{
+ argumentFilePath = session.Param{
+ Value: "",
+ }
+ }
+ argumentNamePrefix, err3 := module.GetParameter("NAME_PREFIX")
+ if err3 != nil{
+ argumentNamePrefix = session.Param{
+ Value: "",
+ }
+ }
+
+ argumentVCard, err4 := module.GetParameter("VCARD")
+ if err4 != nil{
+ argumentVCard = session.Param{
+ Value: "",
+ }
+ }
+
+ argumentLimit, err5 := module.GetParameter("LIMIT")
+ if err5 != nil{
+ fmt.Println(err5.Error())
+ return
+ }
+ if argumentLimit.Value == ""{
+ argumentLimit.Value = "100"
+ }
+
+
+ module.Bar = pb.New(module.Sess.StringToInteger(argumentLimit.Value))
+
+ pool, err := pb.StartPool(module.Bar)
+ if err != nil {
+ panic(err)
+ }
+ wg := new(sync.WaitGroup)
+ for{
+ if module.Current < module.Sess.StringToInteger(argumentLimit.Value) {
+ wg.Add(1)
+ go func(module *PhoneGeneratorFr, bar *pb.ProgressBar) {
+ phone := faker.PhoneNumber().CellPhone()
+ if strings.Contains(phone, "(") && strings.Contains(phone, ")") {
+ newPhone := strings.Split(phone, ")")[1]
+ if argumentPrefix.Value != "" {
+ randomNumber := rand.Intn(9)
+ newPhone = "+"+strings.TrimSpace(argumentPrefix.Value) + " 6" + strings.TrimSpace(strings.Replace(newPhone, "-", "", -1)) + strconv.Itoa(randomNumber)
+ } else{
+ newPhone = "+1 (213)" + newPhone
+ }
+ module.Results = append(module.Results, newPhone)
+ module.Current = module.Current + 1
+ bar.Increment()
+ }
+ }(module, module.Bar)
+ wg.Done()
+ } else{
+ break
+ }
+ }
+ wg.Wait()
+ _ = pool.Stop()
+
+ var file *os.File
+ var errPath error
+
+ if argumentFilePath.Value != ""{
+ file, errPath = os.OpenFile(strings.TrimSpace(argumentFilePath.Value), os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
+ } else {
+ file, errPath = os.OpenFile("/beverlyHills-5000_1.vcf", os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
+ }
+ if errPath != nil {
+ fmt.Println(errPath.Error())
+ return
+ }
+ defer file.Close()
+ for _, number := range module.Results{
+ var uuid string
+ if argumentNamePrefix.Value == "" {
+ uuid = "BHills_GO_" + ksuid.New().String()
+ } else{
+ uuid = strings.TrimSpace(argumentNamePrefix.Value) + "_" + ksuid.New().String()
+ }
+ if argumentVCard.Value == "true" {
+ _, _ = file.WriteString("BEGIN:VCARD\nVERSION:3.0\nN:" + uuid + ";;;\nFN:" + uuid + "\nTEL;type=HOME:" + number + "\nEND:VCARD\n")
+ } else{
+ _, _ = file.WriteString("\"" + number + "\",\n")
+ }
+ }
+ module.Sess.Stream.Success("VCards successfully generated to '" + argumentFilePath.Value + "'")
+}

modules/whatsapp/whatsapp_extractor.go

@@ -37,6 +37,7 @@ func PushWhatsappExtractorModule(s *session.Session) *WhatsappExtractor{
  mod := WhatsappExtractor{
  Sess: s,
  }
+ mod.CreateNewParam("FILE_PATH", "File to extract contacts", "contacts.json", true, session.STRING)
  return &mod
 }
 
@@ -83,6 +84,13 @@ func readSession() (whatsapp.Session, error) {
 }
 
 func (module *WhatsappExtractor) Start(){
+
+ file_ouput, err := module.GetParameter("FILE_PATH")
+ if err != nil{
+ module.Sess.Stream.Error(err.Error())
+ return
+
+ }
  //create new WhatsApp connection
  wac, err := whatsapp.NewConn(5 * time.Second)
  if err != nil {
@@ -152,7 +160,7 @@ func (module *WhatsappExtractor) Start(){
  module.Sess.Stream.Render(t)
  module.Sess.Stream.Success("Total contacts : " + strconv.Itoa(len(contacts)))
  rankingsJson, _ := json.Marshal(module.Contacts)
- err = ioutil.WriteFile("contacts.json", rankingsJson, 0644)
+ err = ioutil.WriteFile(file_ouput.Value, rankingsJson, 0644)
  }
 
 }