-
Notifications
You must be signed in to change notification settings - Fork 43
/
mitm.go
460 lines (416 loc) · 15.1 KB
/
mitm.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
package yaklib
import (
"context"
"fmt"
"net/http"
"github.com/yaklang/yaklang/common/crep"
"github.com/yaklang/yaklang/common/log"
"github.com/yaklang/yaklang/common/utils"
"github.com/yaklang/yaklang/common/utils/lowhttp"
"github.com/yaklang/yaklang/common/utils/lowhttp/httpctx"
)
var MitmExports = map[string]interface{}{
"Start": startMitm,
"Bridge": startBridge,
"maxContentLength": mitmMaxContentLength,
"isTransparent": mitmConfigIsTransparent,
"context": mitmConfigContext,
"host": mitmConfigHost,
"callback": mitmConfigCallback,
"hijackHTTPRequest": mitmConfigHijackHTTPRequest,
"hijackHTTPResponse": mitmConfigHijackHTTPResponse,
"hijackHTTPResponseEx": mitmConfigHijackHTTPResponseEx,
"wscallback": mitmConfigWSCallback,
"wsforcetext": mitmConfigWSForceTextFrame,
"rootCA": mitmConfigCertAndKey,
"useDefaultCA": mitmConfigUseDefault,
}
// Start 启动一个 MITM (中间人)代理服务器,它的第一个参数是端口,接下来可以接收零个到多个选项函数,用于影响中间人代理服务器的行为
// 如果没有指定 CA 证书和私钥,那么将使用内置的证书和私钥
// Example:
// ```
// mitm.Start(8080, mitm.host("127.0.0.1"), mitm.callback(func(isHttps, urlStr, req, rsp) { http.dump(req); http.dump(rsp) })) // 启动一个中间人代理服务器,并将请求和响应打印到标准输出
// ```
func startMitm(
port int,
opts ...MitmConfigOpt,
) error {
return startBridge(port, "", opts...)
}
type mitmConfig struct {
ctx context.Context
host string
callback func(isHttps bool, urlStr string, r *http.Request, rsp *http.Response)
wsForceTextFrame bool
wscallback func(data []byte, isRequest bool) interface{}
mitmCert, mitmPkey []byte
useDefaultMitmCert bool
maxContentLength int
// 是否开启透明劫持
isTransparent bool
hijackRequest func(isHttps bool, urlStr string, req []byte, forward func([]byte), reject func())
hijackWebsocketDataFrame func(isHttps bool, urlStr string, req []byte, forward func([]byte), reject func())
hijackResponse func(isHttps bool, urlStr string, rsp []byte, forward func([]byte), reject func())
hijackResponseEx func(isHttps bool, urlStr string, req, rsp []byte, forward func([]byte), reject func())
}
type MitmConfigOpt func(config *mitmConfig)
// isTransparent 是一个选项函数,用于指定中间人代理服务器是否开启透明劫持模式,默认为false
// 在开启透明模式下,所有流量都会被默认转发,所有的回调函数都会被忽略
// Example:
// ```
// mitm.Start(8080, mitm.isTransparent(true))
// ```
func mitmConfigIsTransparent(b bool) MitmConfigOpt {
return func(config *mitmConfig) {
config.isTransparent = b
}
}
var MitmConfigContext = mitmConfigContext
// context 是一个选项函数,用于指定中间人代理服务器的上下文
// Example:
// ```
// mitm.Start(8080, mitm.context(context.Background()))
// ```
func mitmConfigContext(ctx context.Context) MitmConfigOpt {
return func(config *mitmConfig) {
config.ctx = ctx
}
}
// useDefaultCA 是一个选项函数,用于指定中间人代理服务器是否使用内置的证书和私钥,默认为true
// 默认的证书与私钥路径:~/yakit-projects/yak-mitm-ca.crt 和 ~/yakit-projects/yak-mitm-ca.key
// Example:
// ```
// mitm.Start(8080, mitm.useDefaultCA(true))
// ```
func mitmConfigUseDefault(t bool) MitmConfigOpt {
return func(config *mitmConfig) {
config.useDefaultMitmCert = t
}
}
// host 是一个选项函数,用于指定中间人代理服务器的监听地址,默认为空,即监听所有网卡
// Example:
// ```
// mitm.Start(8080, mitm.host("127.0.0.1"))
// ```
func mitmConfigHost(host string) MitmConfigOpt {
return func(config *mitmConfig) {
config.host = host
}
}
// callback 是一个选项函数,用于指定中间人代理服务器的回调函数,当接收到请求和响应后,会调用该回调函数
// Example:
// ```
// mitm.Start(8080, mitm.callback(func(isHttps, urlStr, req, rsp) { http.dump(req); http.dump(rsp) }))
// ```
func mitmConfigCallback(f func(bool, string, *http.Request, *http.Response)) MitmConfigOpt {
return func(config *mitmConfig) {
config.callback = f
}
}
// hijackHTTPRequest 是一个选项函数,用于指定中间人代理服务器的请求劫持函数,当接收到请求后,会调用该回调函数
// 通过调用该回调函数的第四个参数,可以修改请求内容,通过调用该回调函数的第五个参数,可以丢弃请求
// Example:
// ```
// mitm.Start(8080, mitm.hijackHTTPRequest(func(isHttps, urlStr, req, modified, dropped) {
// // 添加一个额外的请求头
// req = poc.ReplaceHTTPPacketHeader(req, "AAA", "BBB")
// modified(req)
// }
// ))
// ```
func mitmConfigHijackHTTPRequest(h func(isHttps bool, u string, req []byte, modified func([]byte), dropped func())) MitmConfigOpt {
return func(config *mitmConfig) {
config.hijackRequest = h
}
}
// hijackHTTPResponse 是一个选项函数,用于指定中间人代理服务器的响应劫持函数,当接收到响应后,会调用该回调函数
// 通过调用该回调函数的第四个参数,可以修改响应内容,通过调用该回调函数的第五个参数,可以丢弃响应
// Example:
// ```
// mitm.Start(8080, mitm.hijackHTTPResponse(func(isHttps, urlStr, rsp, modified, dropped) {
// // 修改响应体为hijacked
// rsp = poc.ReplaceBody(rsp, b"hijacked", false)
// modified(rsp)
// }
// ))
// ```
func mitmConfigHijackHTTPResponse(h func(isHttps bool, u string, rsp []byte, modified func([]byte), dropped func())) MitmConfigOpt {
return func(config *mitmConfig) {
config.hijackResponse = h
}
}
// hijackHTTPResponseEx 是一个选项函数,用于指定中间人代理服务器的响应劫持函数,当接收到响应后,会调用该回调函数
// 通过调用该回调函数的第五个参数,可以修改响应内容,通过调用该回调函数的第六个参数,可以丢弃响应
// 它与 hijackHTTPResponse 的区别在于,它可以获取到原始请求报文
// Example:
// ```
// mitm.Start(8080, mitm.hijackHTTPResponseEx(func(isHttps, urlStr, req, rsp, modified, dropped) {
// // 修改响应体为hijacked
// rsp = poc.ReplaceBody(rsp, b"hijacked", false)
// modified(rsp)
// }
// ))
// ```
func mitmConfigHijackHTTPResponseEx(h func(bool, string, []byte, []byte, func([]byte), func())) MitmConfigOpt {
return func(config *mitmConfig) {
config.hijackResponseEx = h
}
}
// wsforcetext 是一个选项函数,用于强制指定中间人代理服务器的 websocket 劫持的数据帧转换为文本帧,默认为false
// ! 已弃用
// Example:
// ```
// mitm.Start(8080, mitm.wsforcetext(true))
// ```
func mitmConfigWSForceTextFrame(b bool) MitmConfigOpt {
return func(config *mitmConfig) {
config.wsForceTextFrame = b
}
}
// wscallback 是一个选项函数,用于指定中间人代理服务器的 websocket 劫持函数,当接收到 websocket 请求或响应后,会调用该回调函数
// 该回调函数的第一个参数是请求或响应的内容
// 第二个参数是一个布尔值,用于指示该内容是请求还是响应,true 表示请求,false 表示响应
// 通过该回调函数的返回值,可以修改请求或响应的内容
// Example:
// ```
// mitm.Start(8080, mitm.wscallback(func(data, isRequest) { println(data); return data }))
// ```
func mitmConfigWSCallback(f func([]byte, bool) interface{}) MitmConfigOpt {
return func(config *mitmConfig) {
config.wscallback = f
}
}
// rootCA 是一个选项函数,用于指定中间人代理服务器的根证书和私钥
// Example:
// ```
// mitm.Start(8080, mitm.rootCA(cert, key))
// ```
func mitmConfigCertAndKey(cert, key []byte) MitmConfigOpt {
return func(config *mitmConfig) {
config.mitmCert = cert
config.mitmPkey = key
}
}
// maxContentLength 是一个选项函数,用于指定中间人代理服务器的最大的请求和响应内容长度,默认为10MB
// Example:
// ```
// mitm.Start(8080, mitm.maxContentLength(100 * 1000 * 1000))
// ```
func mitmMaxContentLength(i int) MitmConfigOpt {
return func(config *mitmConfig) {
config.maxContentLength = i
}
}
// Bridge 启动一个 MITM (中间人)代理服务器,它的第一个参数是端口,第二个参数是下游代理服务器地址,接下来可以接收零个到多个选项函数,用于影响中间人代理服务器的行为
// Bridge 与 Start 类似,但略有不同,Bridge可以指定下游代理服务器地址,同时默认会在接收到请求和响应时打印到标准输出
// 如果没有指定 CA 证书和私钥,那么将使用内置的证书和私钥
// Example:
// ```
// mitm.Bridge(8080, "", mitm.host("127.0.0.1"), mitm.callback(func(isHttps, urlStr, req, rsp) { http.dump(req); http.dump(rsp) })) // 启动一个中间人代理服务器,并将请求和响应打印到标准输出
// ```
func startBridge(
port interface{},
downstreamProxy string,
opts ...MitmConfigOpt,
) error {
config := &mitmConfig{
ctx: context.Background(),
host: "",
callback: nil,
mitmCert: nil,
mitmPkey: nil,
useDefaultMitmCert: true,
maxContentLength: 10 * 1000 * 1000,
}
for _, opt := range opts {
opt(config)
}
if config.host == "" {
config.host = "127.0.0.1"
}
if config.mitmPkey == nil || config.mitmCert == nil {
if !config.useDefaultMitmCert {
return utils.Errorf("empty root CA, please use tls to generate or use mitm.useDefaultCA(true) to allow buildin ca.")
}
log.Infof("mitm proxy use the default cert and key")
}
if config.isTransparent && downstreamProxy != "" {
log.Errorf("mitm.Bridge cannot be 'isTransparent'")
}
if config.ctx == nil {
config.ctx = context.Background()
}
server, err := crep.NewMITMServer(
crep.MITM_SetWebsocketHijackMode(true),
crep.MITM_SetForceTextFrame(config.wsForceTextFrame),
crep.MITM_SetWebsocketRequestHijackRaw(func(req []byte, r *http.Request, rspIns *http.Response, t int64) []byte {
var i interface{}
defer func() {
if err := recover(); err != nil {
log.Error(err)
}
}()
if config.wscallback != nil {
i = config.wscallback(req, true)
req = utils.InterfaceToBytes(i)
}
return req
}),
crep.MITM_SetWebsocketResponseHijackRaw(func(rsp []byte, r *http.Request, rspIns *http.Response, t int64) []byte {
var i interface{}
defer func() {
if err := recover(); err != nil {
log.Error(err)
}
}()
if config.wscallback != nil {
i = config.wscallback(rsp, false)
rsp = utils.InterfaceToBytes(i)
}
return rsp
}),
crep.MITM_SetHijackedMaxContentLength(config.maxContentLength),
crep.MITM_SetTransparentHijackMode(config.isTransparent),
crep.MITM_SetTransparentMirrorHTTP(func(isHttps bool, r *http.Request, rsp *http.Response) {
defer func() {
if err := recover(); err != nil {
log.Error(err)
}
}()
urlIns, err := lowhttp.ExtractURLFromHTTPRequest(r, isHttps)
if urlIns == nil {
log.Errorf("parse to url instance failed...")
return
}
if config.callback != nil {
config.callback(isHttps, urlIns.String(), r, rsp)
return
}
println("RECV request:", urlIns.String())
println("REQUEST: ")
raw, err := utils.HttpDumpWithBody(r, false)
if err != nil {
println("Parse Request Failed: %s")
}
println(string(raw))
println("RESPONSE: ")
raw, err = utils.HttpDumpWithBody(rsp, false)
if err != nil {
println("Parse Response Failed: %s")
}
println(string(raw))
println("-----------------------------")
}),
crep.MITM_SetHTTPResponseMirror(func(isHttps bool, u string, r *http.Request, rsp *http.Response, remoteAddr string) {
defer func() {
if err := recover(); err != nil {
log.Error(err)
}
}()
if config.callback != nil {
config.callback(isHttps, u, r, rsp)
return
}
urlIns, _ := lowhttp.ExtractURLFromHTTPRequest(r, isHttps)
if urlIns == nil {
log.Errorf("parse to url instance failed...")
return
}
fmt.Println("RECV request:", urlIns.String())
fmt.Println("REQUEST: ")
raw, err := utils.HttpDumpWithBody(r, false)
if err != nil {
log.Errorf("Parse Request Failed: %v", err)
}
fmt.Println(string(raw))
fmt.Println("RESPONSE: ")
raw, err = utils.HttpDumpWithBody(rsp, false)
if err != nil {
log.Errorf("Parse Request Failed: %v", err)
}
fmt.Println(string(raw))
fmt.Println("-----------------------------")
}),
crep.MITM_SetDownstreamProxy(downstreamProxy),
crep.MITM_SetCaCertAndPrivKey(config.mitmCert, config.mitmPkey),
crep.MITM_SetHTTPRequestHijackRaw(func(isHttps bool, reqIns *http.Request, req []byte) []byte {
if config.hijackRequest == nil {
return req
}
if reqIns.Method == "CONNECT" {
return req
}
req = lowhttp.FixHTTPRequest(req)
urlStrIns, _ := lowhttp.ExtractURLFromHTTPRequestRaw(req, isHttps)
after := req
isDropped := utils.NewBool(false)
config.hijackRequest(isHttps, urlStrIns.String(), req, func(bytes []byte) {
after = lowhttp.FixHTTPRequest(bytes)
httpctx.SetRequestModified(reqIns, "user")
httpctx.SetHijackedRequestBytes(reqIns, after)
}, func() {
isDropped.Set()
})
if isDropped.IsSet() {
httpctx.SetContextValueInfoFromRequest(reqIns, httpctx.REQUEST_CONTEXT_KEY_IsDropped, true)
return nil
}
return after
}),
crep.MITM_SetHTTPResponseHijackRaw(func(isHttps bool, req *http.Request, rspInstance *http.Response, rsp []byte, remoteAddr string) []byte {
if config.hijackResponse == nil && config.hijackResponseEx == nil {
return rsp
}
if req.Method == "CONNECT" {
return rsp
}
fixedResp, _, _ := lowhttp.FixHTTPResponse(rsp)
if fixedResp == nil {
fixedResp = rsp
}
urlStrIns, err := lowhttp.ExtractURLFromHTTPRequest(req, isHttps)
if err != nil {
log.Errorf("extract url from httpRequest failed: %s", err)
}
after := fixedResp
isDropped := utils.NewBool(false)
if config.hijackResponse != nil {
config.hijackResponse(isHttps, urlStrIns.String(), fixedResp, func(bytes []byte) {
after = bytes
}, func() {
isDropped.Set()
})
}
if config.hijackResponseEx != nil {
reqRaw := httpctx.GetRequestBytes(req)
if reqRaw != nil {
reqRaw, _ = utils.HttpDumpWithBody(req, true)
if reqRaw != nil && !httpctx.GetContextBoolInfoFromRequest(req, httpctx.REQUEST_CONTEXT_KEY_RequestIsStrippedGzip) {
reqRaw = lowhttp.DeletePacketEncoding(reqRaw)
}
}
config.hijackResponseEx(isHttps, urlStrIns.String(), reqRaw, fixedResp, func(bytes []byte) {
after = lowhttp.FixHTTPRequest(bytes)
httpctx.SetResponseModified(req, "user")
httpctx.SetHijackedResponseBytes(req, after)
}, func() {
isDropped.Set()
})
}
if isDropped.IsSet() {
httpctx.SetContextValueInfoFromRequest(req, httpctx.RESPONSE_CONTEXT_KEY_IsDropped, true)
return nil
}
return after
}),
)
if err != nil {
return utils.Errorf("create mitm server failed: %s", err)
}
err = server.Serve(config.ctx, utils.HostPort(config.host, port))
if err != nil {
log.Errorf("server mitm failed: %s", err)
return err
}
return nil
}