├── README.md
├── ai
    ├── root.go
    └── type.go
├── common
    ├── config.go
    ├── flag.go
    ├── parse.go
    ├── request.go
    ├── requests
    │   └── config.go
    └── utils
    │   ├── ExportExcel.go
    │   ├── fileutil.go
    │   └── gologger
    │       └── gologger.go
├── go.mod
├── go.sum
├── img
    ├── Pasted image 20220401194624.png
    ├── Pasted image 20220401195020.png
    ├── Pasted image 20220401195229.png
    └── Pasted image 20220401195416.png
├── main.go
└── runner
    └── runner.go


/README.md:
--------------------------------------------------------------------------------
 1 | ## 更新说明 
 2 | 
 3 | ------------20230402------------
 4 | 
 5 | 1. 爱企查接口更新，参考ENScan_GO
 6 | 
 7 | ------------20220615------------
 8 | 
 9 | 1. 适配linux系统，解决linux系统下不能输出问题
10 | 
11 | # 参数说明
12 | 对ENScanPublic.exe进行二开(只保留爱企查接口),能对开业的分支公司、投资占比50%以上的公司进行递归查询!并对Domain和Title进行聚合整理!
13 | 
14 | ***author  :i11us0ry***
15 | ***version :0.0.1***
16 | 
17 | Usage of AScan.exe:
18 | 
19 | > -b    bool    是否递归查询开业状态的分支公司
20 | > 
21 | > -f     string  包含公司名称的文件，公司名按行存储
22 | > 
23 | > -n    string  公司名称,最好是爱企查上公司对应的`完整名称`
24 | > 
25 | > -s    bool     是否递归查询对外投资50%以上的开业公司
26 |   
27 | # 使用说明
28 |   首次使用先运行AScan.exe，会自动生成配置文件和保存目录。生成完配置文件后手动添加cookie，程序提示cookie过期或提示需要更新cookie时也需要更新cookie
29 |   
30 |   ![image](https://github.com/i11us0ry/AScan/blob/main/img/Pasted%20image%2020220401194624.png)
31 |   
32 |   运行时根据需求输入参数即可
33 |   
34 |   ![image](https://github.com/i11us0ry/AScan/blob/main/img/Pasted%20image%2020220401195416.png)
35 | 
36 | # 保存结果
37 |   AScan会将查询结果分为两部分。
38 |   
39 |   第一部分以查询公司、分支公司及对外投资公司为单位的xlsx表格。表格内容包括基本信息、网站备案、分支机构、微信公众号、对外投资、软件著作权信息(APP)，方便红队和攻防中做备用计划
40 |   
41 |   ![image](https://github.com/i11us0ry/AScan/blob/main/img/Pasted%20image%2020220401195020.png)
42 | 
43 |   第二部分为查询公司、分支公司、备案信息及对外投资50%以上公司的域名信息整合，方便快速扫描或做下一步信息收集
44 |     
45 |   ![image](https://github.com/i11us0ry/AScan/blob/main/img/Pasted%20image%2020220401195229.png)
46 | 


--------------------------------------------------------------------------------
/ai/root.go:
--------------------------------------------------------------------------------
  1 | package ai
  2 | 
  3 | import (
  4 | 	"AScan/common"
  5 | 	"AScan/common/utils"
  6 | 	"AScan/common/utils/gologger"
  7 | 	"github.com/olekukonko/tablewriter"
  8 | 	"github.com/tidwall/gjson"
  9 | 	"github.com/tidwall/sjson"
 10 | 	"github.com/xuri/excelize/v2"
 11 | 	"os"
 12 | 	"runtime"
 13 | 	"strconv"
 14 | 	"strings"
 15 | 	"time"
 16 | )
 17 | 
 18 | var (
 19 | 	Info common.Web
 20 | )
 21 | 
 22 | func Init(info common.Web){
 23 | 	Info = info
 24 | }
 25 | 
 26 | // pageParseJson 提取页面中的JSON字段
 27 | func pageParseJson(content string) gjson.Result {
 28 | 	tag1 := "window.pageData ="
 29 | 	tag2 := "window.isSpider ="
 30 | 	//tag2 := "/* eslint-enable */</script><script data-app"
 31 | 	idx1 := strings.Index(content, tag1)
 32 | 	idx2 := strings.Index(content, tag2)
 33 | 	if idx2 > idx1 {
 34 | 		str := content[idx1+len(tag1) : idx2]
 35 | 		str = strings.Replace(str, "\n", "", -1)
 36 | 		str = strings.Replace(str, " ", "", -1)
 37 | 		str = str[:len(str)-1]
 38 | 		return gjson.Get(string(str), "result")
 39 | 	}
 40 | 	return gjson.Result{}
 41 | }
 42 | 
 43 | // 第一步 GetEnInfoByPid 根据PID获取公司信息
 44 | func GetEnInfoByPid(options common.Options,sn *common.Sun) {
 45 | 	pid := ""
 46 | 	enop := common.Options{}
 47 | 	if options.CompanyID == "" {
 48 | 		_, enop = SearchName(options)
 49 | 	} else {
 50 | 		enop = options
 51 | 	}
 52 | 	pid = enop.CompanyID
 53 | 	if pid == "" {
 54 | 		gologger.Errorf("没有获取到PID\n")
 55 | 		return
 56 | 	}
 57 | 	gologger.Infof("查询PID %s\n", pid)
 58 | 
 59 | 	//获取公司信息
 60 | 	res := getCompanyInfoById(pid, true, enop,sn)
 61 | 	outPutExcelByEnInfo(res, enop, sn)
 62 | 
 63 | }
 64 | 
 65 | func outPutExcelByEnInfo(enInfo EnInfo, options common.Options,sn *common.Sun) {
 66 | 	f := excelize.NewFile()
 67 | 	//Base info
 68 | 	baseHeaders := []string{"信息", "值"}
 69 | 	baseData := [][]interface{}{
 70 | 		{"PID", enInfo.Pid},
 71 | 		{"企业名称", enInfo.EntName},
 72 | 		{"法人代表", enInfo.legalPerson},
 73 | 		{"开业状态", enInfo.openStatus},
 74 | 		{"电话", enInfo.telephone},
 75 | 		{"邮箱", enInfo.email},
 76 | 		{"网址",enInfo.webSite},
 77 | 	}
 78 | 	if enInfo.webSite!="-" && enInfo.webSite!=""{
 79 | 		Info.Check(enInfo.webSite,enInfo.EntName)
 80 | 	}
 81 | 	f, _ = utils.ExportExcel("基本信息", baseHeaders, baseData, f)
 82 | 
 83 | 	for k, s := range enInfo.ensMap {
 84 | 		if s.total > 0 && s.api != "" {
 85 | 			//gologger.Infof("正在导出%s\n", s.name)
 86 | 			headers := s.keyWord
 87 | 			var data [][]interface{}
 88 | 			for _, y := range enInfo.infos[k] {
 89 | 				results := gjson.GetMany(y.Raw, s.field...)
 90 | 				var str []interface{}
 91 | 				for _, s1 := range results {
 92 | 					str = append(str, s1.String())
 93 | 					// 新增 聚合备案和基本信息中的website
 94 | 					if s.name == "网站备案"{
 95 | 						if results[0].String()!="-" && results[0].String()!=""{
 96 | 							Info.Check(results[0].String(),results[1].String())
 97 | 						}
 98 | 					}
 99 | 				}
100 | 				data = append(data, str)
101 | 			}
102 | 			f, _ = utils.ExportExcel(s.name, headers, data, f)
103 | 		}
104 | 	}
105 | 
106 | 	f.DeleteSheet("Sheet1")
107 | 	// Save spreadsheet by the given path.
108 | 	sysType := runtime.GOOS
109 | 	savaPath := ""
110 | 	if sysType == "windows"{
111 | 		savaPath = options.Output + "\\" +
112 | 			time.Now().Format("2006-01-02") +
113 | 			enInfo.EntName + strconv.FormatInt(time.Now().Unix(), 10) + ".xlsx"
114 | 	} else {
115 | 		savaPath = options.Output + "/" +
116 | 			time.Now().Format("2006-01-02") +
117 | 			enInfo.EntName + strconv.FormatInt(time.Now().Unix(), 10) + ".xlsx"
118 | 	}
119 | 
120 | 	// 将所有子公司名字收集
121 | 	sn.Name = append(sn.Name, enInfo.EntName)
122 | 	if err := f.SaveAs(savaPath); err != nil {
123 | 		gologger.Fatalf("导出失败：%s", err)
124 | 	}
125 | 	//gologger.Infof("导出成功路径： %s\n", savaPath)
126 | 
127 | }
128 | 
129 | // getCompanyInfoById 获取公司基本信息
130 | // pid 公司id
131 | // isSearch 是否递归搜索信息【分支机构、对外投资信息】
132 | // options options
133 | func getCompanyInfoById(pid string, isSearch bool, options common.Options,sn *common.Sun) EnInfo {
134 | 	var enInfo EnInfo
135 | 	enInfo.infos = make(map[string][]gjson.Result)
136 | 	urls := "https://aiqicha.baidu.com/company_detail_" + pid
137 | 	content := common.GetReq(urls, options)
138 | 	res := pageParseJson(string(content))
139 | 	//获取企业基本信息情况
140 | 	enInfo.Pid = res.Get("pid").String()
141 | 	enInfo.EntName = res.Get("entName").String()
142 | 	enInfo.legalPerson = res.Get("legalPerson").String()
143 | 	enInfo.openStatus = res.Get("openStatus").String()
144 | 	enInfo.telephone = res.Get("telephone").String()
145 | 	enInfo.email = res.Get("email").String()
146 | 	enInfo.webSite = res.Get("website").String()
147 | 	//gologger.Infof("企业基本信息\n")
148 | 	data := [][]string{
149 | 		{"PID", enInfo.Pid},
150 | 		{"企业名称", enInfo.EntName},
151 | 		{"法人代表", enInfo.legalPerson},
152 | 		{"开业状态", enInfo.openStatus},
153 | 		{"电话", enInfo.telephone},
154 | 		{"邮箱", enInfo.email},
155 | 		{"网址",enInfo.webSite},
156 | 	}
157 | 	table := tablewriter.NewWriter(os.Stdout)
158 | 	table.SetAlignment(tablewriter.ALIGN_CENTER)
159 | 	table.AppendBulk(data)
160 | 	table.Render()
161 | 
162 | 	//判断企业状态，不然就可以跳过了
163 | 	if enInfo.openStatus == "注销" || enInfo.openStatus == "吊销" {
164 | 		return enInfo
165 | 	}
166 | 
167 | 	//获取企业信息
168 | 	enInfoUrl := "https://aiqicha.baidu.com/compdata/navigationListAjax?pid=" + pid
169 | 	enInfoRes := common.GetReq(enInfoUrl, options)
170 | 	ensInfoMap := make(map[string]*EnsGo)
171 | 	if gjson.Get(string(enInfoRes), "status").String() == "0" {
172 | 		data := gjson.Get(string(enInfoRes), "data").Array()
173 | 		for _, s := range data {
174 | 			for _, t := range s.Get("children").Array() {
175 | 				ensInfoMap[t.Get("id").String()] = &EnsGo{
176 | 					t.Get("name").String(),
177 | 					t.Get("total").Int(),
178 | 					t.Get("avaliable").Int(),
179 | 					"",
180 | 					[]string{},
181 | 					[]string{},
182 | 				}
183 | 			}
184 | 		}
185 | 	}
186 | 
187 | 	//赋值API数据
188 | 	ensInfoMap["webRecord"].api = "detail/icpinfoAjax"
189 | 	ensInfoMap["webRecord"].field = []string{"domain", "siteName", "homeSite", "icpNo"}
190 | 	ensInfoMap["webRecord"].keyWord = []string{"域名", "站点名称", "首页", "ICP备案号"}
191 | 
192 | 	//ensInfoMap["appinfo"].api = "c/appinfoAjax"
193 | 	//ensInfoMap["appinfo"].field = []string{"name", "classify", "logoWord", "logoBrief", "entName"}
194 | 	//ensInfoMap["appinfo"].keyWord = []string{"APP名称", "分类", "LOGO文字", "描述", "所属公司"}
195 | 
196 | 	//ensInfoMap["microblog"].api = "c/microblogAjax"
197 | 	//ensInfoMap["microblog"].field = []string{"nickname", "weiboLink", "logo"}
198 | 	//ensInfoMap["microblog"].keyWord = []string{"微博昵称", "链接", "LOGO"}
199 | 
200 | 	ensInfoMap["wechatoa"].api = "c/wechatoaAjax"
201 | 	ensInfoMap["wechatoa"].field = []string{"wechatName", "wechatId", "wechatIntruduction", "wechatLogo", "qrcode", "entName"}
202 | 	ensInfoMap["wechatoa"].keyWord = []string{"名称", "ID", "描述", "LOGO", "二维码", "归属公司"}
203 | 
204 | 	//ensInfoMap["enterprisejob"].api = "c/enterprisejobAjax"
205 | 	//ensInfoMap["enterprisejob"].field = []string{"jobTitle", "location", "education", "publishDate", "desc"}
206 | 	//ensInfoMap["enterprisejob"].keyWord = []string{"职位名称", "工作地点", "学历要求", "发布日期", "招聘描述"}
207 | 
208 | 	ensInfoMap["copyright"].api = "detail/copyrightAjax"
209 | 	ensInfoMap["copyright"].field = []string{"softwareName", "shortName", "softwareType", "typeCode", "regDate"}
210 | 	ensInfoMap["copyright"].keyWord = []string{"软件名称", "软件简介", "分类", "行业", "日期"}
211 | 
212 | 	//ensInfoMap["supplier"].api = "c/supplierAjax"
213 | 	//ensInfoMap["supplier"].field = []string{"supplier", "source", "principalNameClient", "cooperationDate"}
214 | 	//ensInfoMap["supplier"].keyWord = []string{"供应商名称", "来源", "所属公司", "日期"}
215 | 
216 | 	ensInfoMap["invest"].api = "detail/investajax" //对外投资
217 | 	ensInfoMap["invest"].field = []string{"entName", "openStatus", "regRate", "data"}
218 | 	ensInfoMap["invest"].keyWord = []string{"公司名称", "状态", "投资比例", "数据信息"}
219 | 
220 | 	ensInfoMap["branch"].api = "detail/branchajax" //分支机构
221 | 	ensInfoMap["branch"].field = []string{"entName", "openStatus", "data"}
222 | 	ensInfoMap["branch"].keyWord = []string{"公司名称", "状态", "数据信息"}
223 | 
224 | 	enInfo.ensMap = ensInfoMap
225 | 
226 | 	//获取数据
227 | 	for k, s := range ensInfoMap {
228 | 		if s.total > 0 && s.api != "" {
229 | 			gologger.Infof("正在查询 %s\n", s.name)
230 | 			t := getInfoList(res.Get("pid").String(), s.api, options)
231 | 
232 | 			//判断下网站备案，然后提取出来，留个坑看看有没有更好的解决方案
233 | 			if k == "webRecord" {
234 | 				var tmp []gjson.Result
235 | 				for _, y := range t {
236 | 					for _, o := range y.Get("domain").Array() {
237 | 						value, _ := sjson.Set(y.Raw, "domain", o.String())
238 | 						tmp = append(tmp, gjson.Parse(value))
239 | 					}
240 | 				}
241 | 				t = tmp
242 | 			}
243 | 			//保存数据
244 | 			enInfo.infos[k] = t
245 | 
246 | 			//命令输出展示
247 | 			table := tablewriter.NewWriter(os.Stdout)
248 | 			table.SetHeader(ensInfoMap[k].keyWord)
249 | 			for _, y := range t {
250 | 				results := gjson.GetMany(y.Raw, ensInfoMap[k].field...)
251 | 				var str []string
252 | 				for _, s := range results {
253 | 					str = append(str, s.String())
254 | 				}
255 | 				table.Append(str)
256 | 			}
257 | 			table.Render()
258 | 
259 | 		}
260 | 	}
261 | 
262 | 	// 查询对外投资详细信息
263 | 	// 对外投资>0 && 是否递归 && 参数投资信息大于0
264 | 	if ensInfoMap["invest"].total > 0 && isSearch && options.Invest {
265 | 		enInfo.investInfos = make(map[string]EnInfo)
266 | 		for _, t := range enInfo.infos["invest"] {
267 | 			openStatus := t.Get("openStatus").String()
268 | 			if openStatus == "注销" || openStatus == "吊销" {
269 | 				continue
270 | 			}
271 | 			investNum := 0.00
272 | 			if t.Get("regRate").String() == "-" {
273 | 				investNum = -1
274 | 			} else {
275 | 				str := strings.Replace(t.Get("regRate").String(), "%", "", -1)
276 | 				investNum, _ = strconv.ParseFloat(str, 2)
277 | 			}
278 | 			// 50%以上控股递归查询
279 | 			if investNum >= 50 {
280 | 				gologger.Infof("企业名称：%s 投资占比：%v\n", t.Get("entName"), investNum)
281 | 				options.CompanyID = t.Get("pid").String()
282 | 				options.KeyWord = ""
283 | 				GetEnInfoByPid(options,sn)
284 | 				//n := getCompanyInfoById(t.Get("pid").String(), true, options)
285 | 				//enInfo.investInfos[t.Get("pid").String()] = n
286 | 			}
287 | 		}
288 | 	}
289 | 
290 | 	// 查询分支机构公司详细信息
291 | 	// 分支机构大于0 && 是否递归模式 && 参数是否开启查询
292 | 	if ensInfoMap["branch"].total > 0 && isSearch && options.IsGetBranch {
293 | 		enInfo.branchInfos = make(map[string]EnInfo)
294 | 		for _, t := range enInfo.infos["branch"] {
295 | 			if t.Get("openStatus").String() == "开业"{
296 | 				gologger.Infof("分支名称：%s 状态：%s\n", t.Get("entName"), t.Get("openStatus"))
297 | 				options.CompanyID = t.Get("pid").String()
298 | 				options.KeyWord = ""
299 | 				GetEnInfoByPid(options,sn)
300 | 				//n := getCompanyInfoById(t.Get("pid").String(), false, options)
301 | 				//enInfo.branchInfos[t.Get("pid").String()] = n
302 | 			}
303 | 		}
304 | 	}
305 | 	return enInfo
306 | }
307 | 
308 | // getInfoList 获取信息列表
309 | func getInfoList(pid string, types string, options common.Options) []gjson.Result {
310 | 	urls := "https://aiqicha.baidu.com/" + types + "?pid=" + pid
311 | 	content := common.GetReq(urls, options)
312 | 	var listData []gjson.Result
313 | 	if gjson.Get(string(content), "status").String() == "0" {
314 | 		data := gjson.Get(string(content), "data")
315 | 		//判断一个获取的特殊值
316 | 		if types == "relations/relationalMapAjax" {
317 | 			data = gjson.Get(string(content), "data.investRecordData")
318 | 		}
319 | 		//判断是否多页，遍历获取所有数据
320 | 		pageCount := data.Get("pageCount").Int()
321 | 		if pageCount > 1 {
322 | 			for i := 1; int(pageCount) >= i; i++ {
323 | 				gologger.Infof("当前：%s,%d\n", types, i)
324 | 				reqUrls := urls + "&p=" + strconv.Itoa(i)
325 | 				content = common.GetReq(reqUrls, options)
326 | 				listData = append(listData, gjson.Get(string(content), "data.list").Array()...)
327 | 			}
328 | 		} else {
329 | 			listData = data.Get("list").Array()
330 | 		}
331 | 	}
332 | 	return listData
333 | 
334 | }
335 | 
336 | // SearchName 根据企业名称搜索信息
337 | func SearchName(options common.Options) ([]gjson.Result,common.Options) {
338 | 	//fmt.Println(options.KeyWord)
339 | 	name := options.KeyWord
340 | 	urls := "https://aiqicha.baidu.com/s?q=" + name + "&t=0"
341 | 	content := common.GetReq(urls, options)
342 | 	enList1 := pageParseJson(string(content))
343 | 	enList := enList1.Get("resultList").Array()
344 | 
345 | 	if len(enList) == 0 {
346 | 		gologger.Errorf("没有查询到关键词 “%s”\n", name)
347 | 		return enList,options
348 | 	} else {
349 | 		gologger.Infof("关键词：“%s” 查询到 %d 个结果，默认选择第一个 \n", name, len(enList))
350 | 	}
351 | 	options.CompanyID = enList[0].Get("pid").String()
352 | 
353 | 	table := tablewriter.NewWriter(os.Stdout)
354 | 	table.SetHeader([]string{"PID", "企业名称", "法人代表"})
355 | 	for _, v := range enList {
356 | 		table.Append([]string{v.Get("pid").String(), v.Get("titleName").String(), v.Get("titleLegal").String()})
357 | 	}
358 | 	table.Render()
359 | 	return enList,options
360 | }
361 | 


--------------------------------------------------------------------------------
/ai/type.go:
--------------------------------------------------------------------------------
 1 | package ai
 2 | 
 3 | import (
 4 | 	"github.com/tidwall/gjson"
 5 | )
 6 | 
 7 | type EnBen struct {
 8 | 	Pid           string `json:"pid"`
 9 | 	EntName       string `json:"entName"`
10 | 	EntType       string `json:"entType"`
11 | 	ValidityFrom  string `json:"validityFrom"`
12 | 	Domicile      string `json:"domicile"`
13 | 	EntLogo       string `json:"entLogo"`
14 | 	OpenStatus    string `json:"openStatus"`
15 | 	LegalPerson   string `json:"legalPerson"`
16 | 	LogoWord      string `json:"logoWord"`
17 | 	TitleName     string `json:"titleName"`
18 | 	TitleLegal    string `json:"titleLegal"`
19 | 	TitleDomicile string `json:"titleDomicile"`
20 | 	RegCap        string `json:"regCap"`
21 | 	Scope         string `json:"scope"`
22 | 	RegNo         string `json:"regNo"`
23 | 	PersonTitle   string `json:"personTitle"`
24 | 	PersonID      string `json:"personId"`
25 | }
26 | 
27 | type EnsGo struct {
28 | 	name      string
29 | 	total     int64
30 | 	available int64
31 | 	api       string
32 | 	field     []string
33 | 	keyWord   []string
34 | }
35 | 
36 | type EnInfo struct {
37 | 	Pid         string `json:"pid"`
38 | 	EntName     string `json:"entName"`
39 | 	legalPerson string
40 | 	openStatus  string
41 | 	email       string
42 | 	telephone   string
43 | 	branchNum   int64
44 | 	investNum   int64
45 | 	webSite 	string
46 | 	//info
47 | 	infos  map[string][]gjson.Result
48 | 	ensMap map[string]*EnsGo
49 | 	//other
50 | 	investInfos map[string]EnInfo
51 | 	branchInfos map[string]EnInfo
52 | }
53 | 


--------------------------------------------------------------------------------
/common/config.go:
--------------------------------------------------------------------------------
  1 | package common
  2 | 
  3 | import (
  4 | 	"AScan/common/utils/gologger"
  5 | 	"fmt"
  6 | 	"gopkg.in/yaml.v3"
  7 | 	"io"
  8 | 	"io/ioutil"
  9 | 	"os"
 10 | 	"path/filepath"
 11 | 	"runtime"
 12 | )
 13 | 
 14 | type Options struct {
 15 | 	KeyWord     string // Keyword of Search
 16 | 	CompanyID   string // Company ID
 17 | 	InputFile   string // Scan Input File
 18 | 	Output      string
 19 | 	CookieInfo  string
 20 | 	ScanType    string
 21 | 	IsGetBranch bool
 22 | 	Invest      bool
 23 | 	GetFlags    string
 24 | 	Version     bool
 25 | }
 26 | 
 27 | type Sun struct {
 28 | 	Name 		[]string
 29 | }
 30 | 
 31 | type Conf struct {
 32 | 	Version 	string 			`yaml:"version"`
 33 | 	Output 		string 			`yaml:"output"`
 34 | 	Cookies 	string 			`yaml:"cookies"`
 35 | }
 36 | 
 37 | type WebInfo struct {
 38 | 	FileName string
 39 | 	Info  []Info
 40 | }
 41 | 
 42 | type Info struct{
 43 | 	Domain	string
 44 | 	Title 	string
 45 | }
 46 | 
 47 | type Web interface {
 48 | 	Check(w, t string)
 49 | 	SetFileName(fn string)
 50 | }
 51 | 
 52 | func CheckConf() *Conf{
 53 | 	sysType := runtime.GOOS
 54 | 	config := ""
 55 | 	if sysType == "windows"{
 56 | 		config = GetConfigDir() + "\\conf.yml"
 57 | 	} else {
 58 | 		config = GetConfigDir() + "/conf.yml"
 59 | 	}
 60 | 	_, exist := os.Stat(config)
 61 | 	// 文件不存在
 62 | 	if os.IsNotExist(exist) {
 63 | 		return writeConf(config)
 64 | 		gologger.Printf("已自动生成配置文件 conf.yml")
 65 | 	}
 66 | 	return readConf()
 67 | }
 68 | 
 69 | func writeConf(fileName string) *Conf{
 70 | 	conf := &Conf{}
 71 | 	conf.Version = Version
 72 | 	sysType := runtime.GOOS
 73 | 	if sysType == "windows"{
 74 | 		conf.Output = GetConfigDir() + "\\result"
 75 | 	} else {
 76 | 		conf.Output = GetConfigDir() + "/result"
 77 | 	}
 78 | 	conf.Cookies = "cookie"
 79 | 	file, _ := os.OpenFile(fileName, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)
 80 | 	defer file.Close()
 81 | 	enc := yaml.NewEncoder(file)
 82 | 	err := enc.Encode(conf)
 83 | 	if err != nil {
 84 | 		gologger.Labelf("%v" ,err)
 85 | 		os.Exit(1)
 86 | 	}
 87 | 	return conf
 88 | }
 89 | 
 90 | func readConf() *Conf{
 91 | 	conf := &Conf{}
 92 | 	sysType := runtime.GOOS
 93 | 	var yamlFile []byte
 94 | 	if sysType == "windows"{
 95 | 		yamlFile, _ = ioutil.ReadFile(GetConfigDir() + "\\conf.yml")
 96 | 	} else {
 97 | 		yamlFile, _ = ioutil.ReadFile(GetConfigDir() + "/conf.yml")
 98 | 	}
 99 | 	err := yaml.Unmarshal(yamlFile, conf)
100 | 	if err != nil {
101 | 		fmt.Println()
102 | 		gologger.Labelf("conf.yml read err!\n")
103 | 		os.Exit(1)
104 | 	}
105 | 	return conf
106 | }
107 | 
108 | func GetConfigDir() string{
109 | 	dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
110 | 	if err != nil {
111 | 		gologger.Labelf("%v\n",err)
112 | 		os.Exit(1)
113 | 	}
114 | 	//strings.Replace(dir, "\\", "/", -1)
115 | 	return fmt.Sprintf("%v",dir)
116 | }
117 | 
118 | func (info *WebInfo)SetFileName (fn string){
119 | 	info.FileName = fn
120 | }
121 | 
122 | func (info *WebInfo)Check (d, t string){
123 | 	flag := false
124 | 	for _, v := range(info.Info){
125 | 		if v.Domain == d {
126 | 			flag = true
127 | 			break
128 | 		}
129 | 	}
130 | 	if !flag {
131 | 		info.Info = append(info.Info,Info{d,t})
132 | 		writeFile(info.FileName,d,t)
133 | 	}
134 | }
135 | 
136 | // 写入domain_title
137 | func writeFile(fileName string, d, t string) {
138 | 	f, err := os.OpenFile(fileName, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0777) //打开文件
139 | 	defer f.Close()
140 | 	if err != nil {
141 | 		gologger.Labelf("%v 打开失败!",fileName)
142 | 		return
143 | 	}
144 | 	// 将文件写进去
145 | 	if _, err = io.WriteString(f, fmt.Sprintf("Domain:%-60v Title:%v\n",d,t)); err != nil {
146 | 		gologger.Labelf("%v 写入失败! %v",fileName,err)
147 | 		return
148 | 	}
149 | }
150 | 
151 | func WriteSun(fileName string,sn []string){
152 | 	f, err := os.OpenFile(fileName, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0777) //打开文件
153 | 	defer f.Close()
154 | 	if err != nil {
155 | 		gologger.Labelf("%v 打开失败! %v",fileName,err)
156 | 		return
157 | 	}
158 | 	// 将文件写进去
159 | 	if _, err = io.WriteString(f, fmt.Sprintf("\n%s\n","子公司：")); err != nil {
160 | 		gologger.Labelf("%v 写入失败! %v",fileName,err)
161 | 		return
162 | 	}
163 | 	for _,v := range(sn){
164 | 		if _, err = io.WriteString(f, fmt.Sprintf("%s\n",v)); err != nil {
165 | 			gologger.Labelf("%v 写入失败! %v",fileName,err)
166 | 			return
167 | 		}
168 | 	}
169 | }


--------------------------------------------------------------------------------
/common/flag.go:
--------------------------------------------------------------------------------
 1 | package common
 2 | 
 3 | import (
 4 | 	"flag"
 5 | 	"fmt"
 6 | )
 7 | 
 8 | const banner = `对ENScanPublic.exe进行二开,能对开业的分支公司、投资占比50%以上的公司进行递归查询!并对Domain和Title进行聚合整理!`
 9 | const author = `i11us0ry`
10 | const Version = `0.0.1`
11 | 
12 | func Banner() {
13 | 	fmt.Println(fmt.Sprintf("\n%s\nauthor	:%s\nversion	:%s\n", banner, author, Version))
14 | }
15 | 
16 | func Flag(Info *Options) {
17 | 	Banner()
18 | 	flag.StringVar(&Info.KeyWord, "n", "", "公司名称,最好是爱企查上公司对应的完整名称")
19 | 	flag.StringVar(&Info.InputFile, "f", "", "包含公司名称的文件，公司名按行存储")
20 | 	flag.BoolVar(&Info.IsGetBranch, "b", false, "是否递归查询开业状态的分支公司")
21 | 	flag.BoolVar(&Info.Invest, "s", false, "是否递归查询对外投资50%以上的开业公司")
22 | 	flag.Parse()
23 | }
24 | 


--------------------------------------------------------------------------------
/common/parse.go:
--------------------------------------------------------------------------------
 1 | package common
 2 | 
 3 | import (
 4 | 	"flag"
 5 | 	"AScan/common/utils/gologger"
 6 | 	"os"
 7 | )
 8 | 
 9 | func Parse(options *Options) {
10 | 	if options.Version {
11 | 		gologger.Infof("Current Version: %s\n", Version)
12 | 		os.Exit(0)
13 | 	}
14 | 	if options.KeyWord == "" && options.CompanyID == "" && options.InputFile == "" {
15 | 		flag.PrintDefaults()
16 | 		os.Exit(0)
17 | 	}
18 | 
19 | }
20 | 


--------------------------------------------------------------------------------
/common/request.go:
--------------------------------------------------------------------------------
 1 | package common
 2 | 
 3 | import (
 4 | 	"AScan/common/requests"
 5 | 	"AScan/common/utils/gologger"
 6 | 	"io/ioutil"
 7 | 	"net/http"
 8 | 	"time"
 9 | )
10 | 
11 | func GetReq(url string, options Options) []byte {
12 | 	var transport = requests.DefaultTransport()
13 | 	var client = &http.Client{
14 | 		Transport: transport,
15 | 		CheckRedirect: func(req *http.Request, via []*http.Request) error {
16 | 			return http.ErrUseLastResponse /* 不进入重定向 */
17 | 		},
18 | 	}
19 | 	req, _ := http.NewRequest("GET", url, nil)
20 | 	req.Header = http.Header{
21 | 		"User-Agent": {"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.80 Safari/537.36 Edg/98.0.1108.43"},
22 | 		"Accept":     {"text/html, application/xhtml+xml, image/jxr, */*"},
23 | 		"Cookie":     {options.CookieInfo},
24 | 		//"Accept-Encoding": {"gzip, deflate"},
25 | 		"Referer": {"https://www.baidu.com"},
26 | 	}
27 | 	//fmt.Println(options.CookieInfo)
28 | 	resp, err := client.Do(req)
29 | 	if err != nil {
30 | 		gologger.Errorf("请求发生错误，5秒后重试\n%s\n", err)
31 | 		time.Sleep(5 * time.Second)
32 | 		return GetReq(url, options)
33 | 	}
34 | 	if resp.StatusCode == 403 {
35 | 		gologger.Fatalf("ip被禁止访问网站，请更换ip\n")
36 | 	} else if resp.StatusCode == 401 {
37 | 		gologger.Fatalf("Cookie有问题或过期，请重新获取\n")
38 | 	} else if resp.StatusCode == 302 {
39 | 		gologger.Fatalf("需要更新Cookie\n")
40 | 	}
41 | 
42 | 	body, _ := ioutil.ReadAll(resp.Body)
43 | 	_ = resp.Body.Close()
44 | 	//page, _ := htmlquery.Parse(strings.NewReader(string(body)))
45 | 	return body
46 | }
47 | 


--------------------------------------------------------------------------------
/common/requests/config.go:
--------------------------------------------------------------------------------
 1 | package requests
 2 | 
 3 | import (
 4 | 	"crypto/tls"
 5 | 	"golang.org/x/net/html"
 6 | 	"net"
 7 | 	"net/http"
 8 | 	"time"
 9 | )
10 | 
11 | type Request struct {
12 | 	Url    string
13 | 	Cookie string
14 | }
15 | 
16 | type Response struct {
17 | 	Body []byte
18 | 	Page *html.Node
19 | }
20 | 
21 | func DefaultTransport() *http.Transport {
22 | 	transport := &http.Transport{
23 | 		Proxy: http.ProxyFromEnvironment,
24 | 		DialContext: (&net.Dialer{
25 | 			Timeout:   60 * time.Second,
26 | 			KeepAlive: 60 * time.Second,
27 | 		}).DialContext,
28 | 		MaxIdleConnsPerHost: -1,
29 | 		TLSClientConfig: &tls.Config{
30 | 			InsecureSkipVerify: true,
31 | 		},
32 | 		DisableKeepAlives: true,
33 | 	}
34 | 	return transport
35 | }
36 | 


--------------------------------------------------------------------------------
/common/utils/ExportExcel.go:
--------------------------------------------------------------------------------
 1 | package utils
 2 | 
 3 | /**
 4 |  * @Description
 5 |  * @Author ggr
 6 |  * @Date 2021/7/8 21:06
 7 |  * @Form https://cloud.tencent.com/developer/article/1846410
 8 |  **/
 9 | 
10 | import (
11 | 	"github.com/xuri/excelize/v2"
12 | 	"strconv"
13 | )
14 | 
15 | // maxCharCount 最多26个字符A-Z
16 | const maxCharCount = 26
17 | 
18 | // ExportExcel 导出Excel文件
19 | // sheetName 工作表名称, 注意这里不要取sheet1这种名字,否则导致文件打开时发生部分错误。
20 | // headers 列名切片， 表头
21 | // rows 数据切片，是一个二维数组
22 | // f 为了创建多个单元格，得把标签往外放
23 | func ExportExcel(sheetName string, headers []string, rows [][]interface{}, f *excelize.File) (*excelize.File, error) {
24 | 	sheetIndex := f.NewSheet(sheetName)
25 | 	maxColumnRowNameLen := 1 + len(strconv.Itoa(len(rows)))
26 | 	columnCount := len(headers)
27 | 	if columnCount > maxCharCount {
28 | 		maxColumnRowNameLen++
29 | 	} else if columnCount > maxCharCount*maxCharCount {
30 | 		maxColumnRowNameLen += 2
31 | 	}
32 | 	columnNames := make([][]byte, 0, columnCount)
33 | 	for i, header := range headers {
34 | 		columnName := getColumnName(i, maxColumnRowNameLen)
35 | 		columnNames = append(columnNames, columnName)
36 | 		// 初始化excel表头，这里的index从1开始要注意
37 | 		curColumnName := getColumnRowName(columnName, 1)
38 | 		err := f.SetCellValue(sheetName, curColumnName, header)
39 | 		if err != nil {
40 | 			return nil, err
41 | 		}
42 | 	}
43 | 	for rowIndex, row := range rows {
44 | 		for columnIndex, columnName := range columnNames {
45 | 			// 从第二行开始
46 | 			err := f.SetCellValue(sheetName, getColumnRowName(columnName, rowIndex+2), row[columnIndex])
47 | 			if err != nil {
48 | 				return nil, err
49 | 			}
50 | 		}
51 | 	}
52 | 	f.SetActiveSheet(sheetIndex)
53 | 	return f, nil
54 | }
55 | 
56 | // getColumnName 生成列名
57 | // Excel的列名规则是从A-Z往后排;超过Z以后用两个字母表示，比如AA,AB,AC;两个字母不够以后用三个字母表示，比如AAA,AAB,AAC
58 | // 这里做数字到列名的映射：0 -> A, 1 -> B, 2 -> C
59 | // maxColumnRowNameLen 表示名称框的最大长度，假设数据是10行，1000列，则最后一个名称框是J1000(如果有表头，则是J1001),是4位
60 | // 这里根据 maxColumnRowNameLen 生成切片，后面生成名称框的时候可以复用这个切片，而无需扩容
61 | func getColumnName(column, maxColumnRowNameLen int) []byte {
62 | 	const A = 'A'
63 | 	if column < maxCharCount {
64 | 		// 第一次就分配好切片的容量
65 | 		slice := make([]byte, 0, maxColumnRowNameLen)
66 | 		return append(slice, byte(A+column))
67 | 	} else {
68 | 		// 递归生成类似AA,AB,AAA,AAB这种形式的列名
69 | 		return append(getColumnName(column/maxCharCount-1, maxColumnRowNameLen), byte(A+column%maxCharCount))
70 | 	}
71 | }
72 | 
73 | // getColumnRowName 生成名称框
74 | // Excel的名称框是用A1,A2,B1,B2来表示的，这里需要传入前一步生成的列名切片，然后直接加上行索引来生成名称框，就无需每次分配内存
75 | func getColumnRowName(columnName []byte, rowIndex int) (columnRowName string) {
76 | 	l := len(columnName)
77 | 	columnName = strconv.AppendInt(columnName, int64(rowIndex), 10)
78 | 	columnRowName = string(columnName)
79 | 	// 将列名恢复回去
80 | 	columnName = columnName[:l]
81 | 	return
82 | }
83 | 


--------------------------------------------------------------------------------
/common/utils/fileutil.go:
--------------------------------------------------------------------------------
 1 | package utils
 2 | 
 3 | import (
 4 | 	"AScan/common/utils/gologger"
 5 | 	"bufio"
 6 | 	"io"
 7 | 	"os"
 8 | )
 9 | 
10 | // FileExists checks if a file exists and is not a directory
11 | func FileExists(filename string) bool {
12 | 	info, err := os.Stat(filename)
13 | 	if os.IsNotExist(err) {
14 | 		return false
15 | 	}
16 | 	return !info.IsDir()
17 | }
18 | 
19 | // FolderExists checks if a folder exists
20 | func FolderExists(folderpath string) bool {
21 | 	_, err := os.Stat(folderpath)
22 | 	return !os.IsNotExist(err)
23 | }
24 | 
25 | // HasStdin determines if the user has piped input
26 | func HasStdin() bool {
27 | 	stat, err := os.Stdin.Stat()
28 | 	if err != nil {
29 | 		return false
30 | 	}
31 | 
32 | 	mode := stat.Mode()
33 | 
34 | 	isPipedFromChrDev := (mode & os.ModeCharDevice) == 0
35 | 	isPipedFromFIFO := (mode & os.ModeNamedPipe) != 0
36 | 
37 | 	return isPipedFromChrDev || isPipedFromFIFO
38 | }
39 | 
40 | func ReadImf(input io.Reader) []string {
41 | 	var imf []string
42 | 	scanner := bufio.NewScanner(input)
43 | 	for scanner.Scan() {
44 | 		imf = append(imf, scanner.Text())
45 | 	}
46 | 	return imf
47 | }
48 | 
49 | // 文件读取，将读取内容按行保存到数组返回
50 | func FileReadByline(fileName string) []string{
51 | 	datas := []string{}
52 | 	fileData := FileRead(fileName)
53 | 	for fileData.Scan(){
54 | 		datas = append(datas, fileData.Text())
55 | 	}
56 | 	return datas
57 | }
58 | 
59 | //文件读取，返回文件流
60 | func FileRead(fileName string) *bufio.Scanner{
61 | 	f, err := os.OpenFile(fileName, os.O_RDONLY, 0)
62 | 	if err != nil {
63 | 		gologger.Fatalf("file:%v read err!",fileName)
64 | 		os.Exit(0)
65 | 	}
66 | 	datas := bufio.NewScanner(f)
67 | 	return datas
68 | }


--------------------------------------------------------------------------------
/common/utils/gologger/gologger.go:
--------------------------------------------------------------------------------
  1 | package gologger
  2 | 
  3 | import (
  4 | 	"fmt"
  5 | 	"github.com/logrusorgru/aurora"
  6 | 	"github.com/mattn/go-colorable"
  7 | 	"os"
  8 | 	"strings"
  9 | 	"sync"
 10 | )
 11 | 
 12 | // Level defines all the available levels we can log at
 13 | type Level int
 14 | 
 15 | // Available logging levels
 16 | const (
 17 | 	Null Level = iota
 18 | 	Fatal
 19 | 	Silent
 20 | 	Label
 21 | 	Misc
 22 | 	Error
 23 | 	Warning
 24 | 	Info
 25 | 	Debug
 26 | 	Verbose
 27 | )
 28 | 
 29 | var (
 30 | 	// UseColors can be used to control coloring of the output
 31 | 	UseColors = true
 32 | 	// MaxLevel is the maximum level to log at. By default, logging
 33 | 	// is done at Info level. Using verbose will display all the errors too,
 34 | 	// Using silent will display only the most relevant information.
 35 | 	MaxLevel = Info
 36 | 
 37 | 	labels = map[Level]string{
 38 | 		Warning: "Warning",
 39 | 		Error:   "Error",
 40 | 		Label:   "WRN",
 41 | 		Fatal:   "Fatal",
 42 | 		Debug:   "DEBUG",
 43 | 		Info:    "INFO",
 44 | 	}
 45 | 
 46 | 	mutex  = &sync.Mutex{}
 47 | 	output = colorable.NewColorableStdout()
 48 | )
 49 | 
 50 | var stringBuilderPool = &sync.Pool{New: func() interface{} {
 51 | 	return new(strings.Builder)
 52 | }}
 53 | 
 54 | // wrap wraps a given label for a message to a logg-able representation.
 55 | // It checks if colors are specified and what level we are logging at.
 56 | func wrap(label string, level Level) string {
 57 | 	// Check if we are not using colors, if not, return
 58 | 	if !UseColors {
 59 | 		return label
 60 | 	}
 61 | 
 62 | 	switch level {
 63 | 	case Silent:
 64 | 		return label
 65 | 	case Info, Verbose:
 66 | 		return aurora.Blue(label).String()
 67 | 	case Fatal:
 68 | 		return aurora.Bold(aurora.Red(label)).String()
 69 | 	case Error:
 70 | 		return aurora.Red(label).String()
 71 | 	case Debug:
 72 | 		return aurora.Magenta(label).String()
 73 | 	case Warning, Label:
 74 | 		return aurora.Yellow(label).String()
 75 | 	default:
 76 | 		return label
 77 | 	}
 78 | }
 79 | 
 80 | // getLabel generates a label for a given message, depending on the level
 81 | // and the label passed.
 82 | func getLabel(level Level, label string, sb *strings.Builder) {
 83 | 	switch level {
 84 | 	case Silent, Misc:
 85 | 		return
 86 | 	case Error, Fatal, Info, Warning, Debug, Label:
 87 | 		sb.WriteString("[")
 88 | 		sb.WriteString(wrap(labels[level], level))
 89 | 		sb.WriteString("]")
 90 | 		sb.WriteString(" ")
 91 | 		return
 92 | 	case Verbose:
 93 | 		sb.WriteString("[")
 94 | 		sb.WriteString(wrap(label, level))
 95 | 		sb.WriteString("]")
 96 | 		sb.WriteString(" ")
 97 | 		return
 98 | 	default:
 99 | 		return
100 | 	}
101 | }
102 | 
103 | // log logs the actual message to the screen
104 | func log(level Level, label string, format string, args ...interface{}) {
105 | 	// Don't log if the level is null
106 | 	if level == Null {
107 | 		return
108 | 	}
109 | 
110 | 	if level <= MaxLevel {
111 | 		// Build the log message using the string builder pool
112 | 		sb := stringBuilderPool.Get().(*strings.Builder)
113 | 
114 | 		// Get the label and append it to string builder
115 | 		getLabel(level, label, sb)
116 | 
117 | 		message := fmt.Sprintf(format, args...)
118 | 		sb.WriteString(message)
119 | 
120 | 		//if strings.HasSuffix(message, "\n") == false {
121 | 		//	sb.WriteString("\n")
122 | 		//}
123 | 
124 | 		mutex.Lock()
125 | 		switch level {
126 | 		case Silent:
127 | 			_, _ = fmt.Fprint(os.Stdout, sb.String())
128 | 		default:
129 | 			_, _ = fmt.Fprint(output, sb.String())
130 | 		}
131 | 		mutex.Unlock()
132 | 
133 | 		sb.Reset()
134 | 		stringBuilderPool.Put(sb)
135 | 	}
136 | }
137 | 
138 | // Infof writes a info message on the screen with the default label
139 | func Infof(format string, args ...interface{}) {
140 | 	log(Info, "", format, args...)
141 | }
142 | 
143 | // Warningf writes a warning message on the screen with the default label
144 | func Warningf(format string, args ...interface{}) {
145 | 	log(Warning, "", format, args...)
146 | }
147 | 
148 | // Errorf writes an error message on the screen with the default label
149 | func Errorf(format string, args ...interface{}) {
150 | 	log(Error, "", format, args...)
151 | }
152 | 
153 | // Debugf writes an error message on the screen with the default label
154 | func Debugf(format string, args ...interface{}) {
155 | 	log(Debug, "", format, args...)
156 | }
157 | 
158 | // Verbosef writes a verbose message on the screen with a tabel
159 | func Verbosef(format string, label string, args ...interface{}) {
160 | 	log(Verbose, label, format, args...)
161 | }
162 | 
163 | // Silentf writes a message on the stdout with no label
164 | func Silentf(format string, args ...interface{}) {
165 | 	log(Silent, "", format, args...)
166 | }
167 | 
168 | // Fatalf exits the program if we encounter a fatal error
169 | func Fatalf(format string, args ...interface{}) {
170 | 	log(Fatal, "", format, args...)
171 | }
172 | 
173 | // Printf prints a string on screen without any extra stuff
174 | func Printf(format string, args ...interface{}) {
175 | 	log(Misc, "", format, args...)
176 | }
177 | 
178 | // Labelf prints a string on screen with a label interface
179 | func Labelf(format string, args ...interface{}) {
180 | 	log(Label, "", format, args...)
181 | }
182 | 


--------------------------------------------------------------------------------
/go.mod:
--------------------------------------------------------------------------------
 1 | module AScan
 2 | 
 3 | go 1.16
 4 | 
 5 | require (
 6 | 	github.com/logrusorgru/aurora v2.0.3+incompatible
 7 | 	github.com/mattn/go-colorable v0.1.12
 8 | 	github.com/olekukonko/tablewriter v0.0.5
 9 | 	github.com/tidwall/gjson v1.14.0
10 | 	github.com/tidwall/sjson v1.2.4
11 | 	github.com/xuri/excelize/v2 v2.5.0
12 | 	golang.org/x/net v0.0.0-20220325170049-de3da57026de
13 | 	gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c
14 | )
15 | 


--------------------------------------------------------------------------------
/go.sum:
--------------------------------------------------------------------------------
 1 | github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 2 | github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 3 | github.com/logrusorgru/aurora v2.0.3+incompatible h1:tOpm7WcpBTn4fjmVfgpQq0EfczGlG91VSDkswnjF5A8=
 4 | github.com/logrusorgru/aurora v2.0.3+incompatible/go.mod h1:7rIyQOR62GCctdiQpZ/zOJlFyk6y+94wXzv6RNZgaR4=
 5 | github.com/mattn/go-colorable v0.1.12 h1:jF+Du6AlPIjs2BiUiQlKOX0rt3SujHxPnksPKZbaA40=
 6 | github.com/mattn/go-colorable v0.1.12/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4=
 7 | github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y=
 8 | github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=
 9 | github.com/mattn/go-runewidth v0.0.9 h1:Lm995f3rfxdpd6TSmuVCHVb/QhupuXlYr8sCI/QdE+0=
10 | github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=
11 | github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 h1:RWengNIwukTxcDr9M+97sNutRR1RKhG96O6jWumTTnw=
12 | github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826/go.mod h1:TaXosZuwdSHYgviHp1DAtfrULt5eUgsSMsZf+YrPgl8=
13 | github.com/olekukonko/tablewriter v0.0.5 h1:P2Ga83D34wi1o9J6Wh1mRuqd4mF/x/lgBS7N7AbDhec=
14 | github.com/olekukonko/tablewriter v0.0.5/go.mod h1:hPp6KlRPjbx+hW8ykQs1w3UBbZlj6HuIJcUGPhkA7kY=
15 | github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
16 | github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
17 | github.com/richardlehane/mscfb v1.0.3 h1:rD8TBkYWkObWO0oLDFCbwMeZ4KoalxQy+QgniCj3nKI=
18 | github.com/richardlehane/mscfb v1.0.3/go.mod h1:YzVpcZg9czvAuhk9T+a3avCpcFPMUWm7gK3DypaEsUk=
19 | github.com/richardlehane/msoleps v1.0.1 h1:RfrALnSNXzmXLbGct/P2b4xkFz4e8Gmj/0Vj9M9xC1o=
20 | github.com/richardlehane/msoleps v1.0.1/go.mod h1:BWev5JBpU9Ko2WAgmZEuiz4/u3ZYTKbjLycmwiWUfWg=
21 | github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
22 | github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0=
23 | github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
24 | github.com/tidwall/gjson v1.12.1/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
25 | github.com/tidwall/gjson v1.14.0 h1:6aeJ0bzojgWLa82gDQHcx3S0Lr/O51I9bJ5nv6JFx5w=
26 | github.com/tidwall/gjson v1.14.0/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
27 | github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
28 | github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
29 | github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs=
30 | github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
31 | github.com/tidwall/sjson v1.2.4 h1:cuiLzLnaMeBhRmEv00Lpk3tkYrcxpmbU81tAY4Dw0tc=
32 | github.com/tidwall/sjson v1.2.4/go.mod h1:098SZ494YoMWPmMO6ct4dcFnqxwj9r/gF0Etp19pSNM=
33 | github.com/xuri/efp v0.0.0-20210322160811-ab561f5b45e3 h1:EpI0bqf/eX9SdZDwlMmahKM+CDBgNbsXMhsN28XrM8o=
34 | github.com/xuri/efp v0.0.0-20210322160811-ab561f5b45e3/go.mod h1:ybY/Jr0T0GTCnYjKqmdwxyxn2BQf2RcQIIvex5QldPI=
35 | github.com/xuri/excelize/v2 v2.5.0 h1:nDDVfX0qaDuGjAvb+5zTd0Bxxoqa1Ffv9B4kiE23PTM=
36 | github.com/xuri/excelize/v2 v2.5.0/go.mod h1:rSu0C3papjzxQA3sdK8cU544TebhrPUoTOaGPIh0Q1A=
37 | golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97 h1:/UOmuWzQfxxo9UtlXMwuQU8CMgg1eZXqTRwkSQJWKOI=
38 | golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
39 | golang.org/x/image v0.0.0-20210220032944-ac19c3e999fb h1:fqpd0EBDzlHRCjiphRR5Zo/RSWWQlWv34418dnEixWk=
40 | golang.org/x/image v0.0.0-20210220032944-ac19c3e999fb/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0=
41 | golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
42 | golang.org/x/net v0.0.0-20210726213435-c6fcb2dbf985/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
43 | golang.org/x/net v0.0.0-20220325170049-de3da57026de h1:pZB1TWnKi+o4bENlbzAgLrEbY4RMYmUIRobMcSmfeYc=
44 | golang.org/x/net v0.0.0-20220325170049-de3da57026de/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=
45 | golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
46 | golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
47 | golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
48 | golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
49 | golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
50 | golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e h1:fLOSk5Q00efkSvAm+4xcoXD+RRmLmmulPn5I3Y9F2EM=
51 | golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
52 | golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
53 | golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
54 | golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
55 | golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
56 | golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
57 | golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk=
58 | golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
59 | golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
60 | gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
61 | gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
62 | gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
63 | gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
64 | 


--------------------------------------------------------------------------------
/img/Pasted image 20220401194624.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/i11us0ry/AScan/d5d574da075c3efdd42012192459481cae9cdb98/img/Pasted image 20220401194624.png


--------------------------------------------------------------------------------
/img/Pasted image 20220401195020.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/i11us0ry/AScan/d5d574da075c3efdd42012192459481cae9cdb98/img/Pasted image 20220401195020.png


--------------------------------------------------------------------------------
/img/Pasted image 20220401195229.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/i11us0ry/AScan/d5d574da075c3efdd42012192459481cae9cdb98/img/Pasted image 20220401195229.png


--------------------------------------------------------------------------------
/img/Pasted image 20220401195416.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/i11us0ry/AScan/d5d574da075c3efdd42012192459481cae9cdb98/img/Pasted image 20220401195416.png


--------------------------------------------------------------------------------
/main.go:
--------------------------------------------------------------------------------
 1 | package main
 2 | 
 3 | import (
 4 | 	"AScan/common"
 5 | 	"AScan/common/utils"
 6 | 	"AScan/runner"
 7 | 	"os"
 8 | )
 9 | 
10 | func main() {
11 | 	var op common.Options
12 | 	conf := common.CheckConf()
13 | 	if !utils.FolderExists(conf.Output){
14 | 		os.Mkdir(conf.Output, 0777)
15 | 	}
16 | 	common.Flag(&op)
17 | 	common.Parse(&op)
18 | 	op.CookieInfo = conf.Cookies
19 | 	op.Output = conf.Output
20 | 	runner.RunEnumeration(op)
21 | }


--------------------------------------------------------------------------------
/runner/runner.go:
--------------------------------------------------------------------------------
 1 | package runner
 2 | 
 3 | import (
 4 | 	"AScan/ai"
 5 | 	"AScan/common"
 6 | 	"AScan/common/utils"
 7 | 	"AScan/common/utils/gologger"
 8 | 	"os"
 9 | 	"runtime"
10 | 	"strconv"
11 | 	"time"
12 | )
13 | 
14 | func RunEnumeration(options common.Options) {
15 | 	var info common.Web
16 | 	var sn common.Sun
17 | 	info = new(common.WebInfo)
18 | 	sysType := runtime.GOOS
19 | 	fn := ""
20 | 	if sysType == "windows"{
21 | 		fn = common.GetConfigDir()+"\\result\\website_"+strconv.FormatInt(time.Now().Unix(), 10) + ".txt"
22 | 	} else {
23 | 		fn = common.GetConfigDir()+"/result/website_"+strconv.FormatInt(time.Now().Unix(), 10) + ".txt"
24 | 		f, err := os.OpenFile(fn, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0777) //打开文件
25 | 		if err!=nil{
26 | 			gologger.Errorf("file create err!")
27 | 		}
28 | 		f.Close()
29 | 	}
30 | 	info.SetFileName(fn)
31 | 	ai.Init(info)
32 | 	if options.InputFile!=""{
33 | 		ns := utils.FileReadByline(options.InputFile)
34 | 		for _, n := range(ns){
35 | 			options.KeyWord = n
36 | 			ai.GetEnInfoByPid(options,&sn)
37 | 		}
38 | 	} else {
39 | 		ai.GetEnInfoByPid(options,&sn)
40 | 	}
41 | 	// 输出所有子公司名称
42 | 	if len(sn.Name)!=0{
43 | 		common.WriteSun(fn,sn.Name)
44 | 	}
45 | }


--------------------------------------------------------------------------------