Ejemplo n.º 1
0
Archivo: tcp.go Proyecto: jiangzx/ztcp
// 程序的主入口
func main() {
	flag.Parse()
	to := new(TcpObj)
	to.Type = *contentType
	to.Header = make(map[string]string)

	// 分析 Header
	switch {
	// 从文件读取
	case !z.IsBlank(*headerf):
		str, _ := z.Utf8f(z.Ph(*headerf))
		z.JsonFromString(str, &to.Header)
	// 来自参数
	case !z.IsBlank(*header):
		z.JsonFromString(*header, &to.Header)
	}

	// 分析 cookie
	switch {
	// 从文件读取
	case !z.IsBlank(*cookief):
		to.Cookie, _ = z.Utf8f(*cookief)
	// 来自参数
	default:
		to.Cookie = *cookie
	}

	// 请求体的内容,考虑 -body 或者 -f
	to.Body = *body
	if len(*file) > 0 {
		f, err := os.Open(z.Ph(*file))
		if nil != err {
			panic(err)
		}
		to.File = f
	}

	// 分析输出方式
	switch *out {
	case "all":
		to.OutputRequest = true
		to.OutputResponse = true
	case "req":
		to.OutputRequest = true
	case "resp":
		to.OutputResponse = true
	}

	// 执行
	switch {
	// 发送 HTTP 请求
	case !z.IsBlank(*httpUrl):
		to.Target = *httpUrl
		to.DoHttp()
	// 默认,打印参数说明
	default:
		flag.PrintDefaults()
	}

}
Ejemplo n.º 2
0
// 生成 HTTP 请求对象,主要是填充头部信息
func (to *TcpObj) createRequestHeader() *http.Request {
	var req *http.Request
	if len(to.Body) > 0 || nil != to.File {
		req, _ = http.NewRequest("POST", "http://"+to.Target, nil)
	} else {
		req, _ = http.NewRequest("GET", "http://"+to.Target, nil)
	}
	// 计算 Host
	pos := strings.Index(to.Target, "/")
	var host string
	if pos > 0 {
		host = string(to.Target[:pos])
	} else {
		host = to.Target
	}
	pos = strings.Index(host, ":")
	if pos > 0 {
		host = string(host[:pos])
	}
	// 填充其他信息
	req.Header.Set("Accept", "text/plain,text/html,application/json,application/xml;q=0.9,*/*;q=0.8")
	req.Header.Set("Accept-Encoding", "gzip,deflate,sdch")
	req.Header.Set("Accept-Language", "zh-CN,zh;q=0.8")
	req.Header.Set("Cache-Control", "max-age=0")
	req.Header.Set("Connection", "keep-alive")
	req.Header.Set("Host", host)
	req.Header.Set("User-Agent", user_agents["chrome"])

	// 填充自定义头部信息
	if len(to.Header) > 0 {
		for key, val := range to.Header {
			req.Header.Set(key, val)
		}
	}

	// 填充 Cookie
	if !z.IsBlank(to.Cookie) {
		req.Header.Set("Cookie", to.Cookie)
	}

	return req
}
Ejemplo n.º 3
0
// 读取解析规则, 传入一个实现了io.Reader的对象即可
func (qb *QWBuilder) LoadRules(reader io.Reader) *QWBuilder {
	bufreader := bufio.NewReader(reader)
	var line string
	var lnum int
	var err error
	var rules []*QWordRule = make([]*QWordRule, 0)
	for true {
		line, err = bufreader.ReadString('\n')
		if err == io.EOF {
			err = nil
			break
		} else if err != nil {
			break
		}
		lnum++
		z.DebugPrintf("rule %2d : %s", lnum, line)
		// 忽略空行与注释行
		line = strings.TrimSpace(line)
		if z.IsBlank(line) || strings.HasPrefix(line, "#") {
			continue
		}
		// 解析行信息
		if strings.HasPrefix(line, "$") {
			qwRule := new(QWordRule)
			// 获得key
			var fpos int = strings.Index(line, ":")
			if fpos < 0 {
				err = errors.New(fmt.Sprintf("invalid rule line %d : %s", lnum, line))
				break
			}
			qwRule.Key = strings.TrimSpace(line[1:fpos])

			// 获得regex
			var expr string
			if line[fpos+1] == ':' {
				// 简要模式
				expr = "^(" + strings.TrimSpace(line[fpos+2:]) + ")(.*)$"
			} else {
				// 普通模式
				expr = strings.TrimSpace(line[fpos+1:])
			}
			regex, regerr := regexp.Compile(expr)
			if regerr != nil {
				err = regerr
				break
			}
			qwRule.Regex = regex

			// 读取下一行,必须存在哟
			line, err = bufreader.ReadString('\n')
			if err == io.EOF {
				err = errors.New(fmt.Sprintf("rule line(%d) miss next line", lnum))
				break
			} else if err != nil {
				break
			}
			lnum++
			z.DebugPrintf("rule %2d : %s", lnum, line)
			line = strings.TrimSpace(line)
			spos := strings.Index(line, "=")
			if z.IsBlank(line) || !strings.HasPrefix(line, "$") || spos < 0 {
				err = errors.New(fmt.Sprintf("invalid rule line %d : %s", lnum, line))
				break
			}
			// 获得seg
			qwRule.Seg = z.Trim(line[:spos])
			// 获得type
			qcType, qcterr := QCType(strings.TrimSpace(line[spos+1:]))
			if qcterr != nil {
				err = qcterr
				break
			}
			qwRule.Type = qcType
			// 加入到rules
			rules = append(rules, qwRule)
		}
	}
	// 是否有错误
	if err != nil {
		panic(err)
	}
	qb.Rules = rules
	return qb
}