示例#1
0
文件: main.go 项目: marksee/trudy
func connectionDispatcher(listener listener.TrudyListener, name string, show bool) {
	defer listener.Close()
	for {
		fd, conn, err := listener.Accept()
		if err != nil {
			continue
		}
		var p pipe.TrudyPipe
		if name == "TLS" {
			p = new(pipe.TLSPipe)
			err = p.New(connectionCount, fd, conn)
		} else {
			p = new(pipe.TCPPipe)
			err = p.New(connectionCount, fd, conn)
		}
		if err != nil {
			log.Println("[ERR] Error creating new pipe.")
			continue
		}
		if show {
			log.Printf("[INFO] ( %v ) %v Connection accepted!\n", connectionCount, name)
		}
		go clientHandler(p, show)
		go serverHandler(p)
		connectionCount++
	}
}
示例#2
0
文件: main.go 项目: marksee/trudy
func serverHandler(pipe pipe.TrudyPipe) {
	buffer := make([]byte, 65535)

	for {
		bytesRead, err := pipe.ReadDestination(buffer)
		if err != nil {
			break
		}
		data := module.Data{FromClient: false,
			Bytes:    buffer[:bytesRead],
			DestAddr: pipe.SourceInfo(),
			SrcAddr:  pipe.DestinationInfo()}

		if data.Drop() {
			continue
		}

		if data.DoMangle() {
			data.Mangle()
			bytesRead = len(data.Bytes)
		}

		if data.DoIntercept() {
			if websocketConn == nil {
				log.Printf("[ERR] Websocket Connection has not been setup yet! Cannot intercept.")
				continue
			}
			websocketMutex.Lock()
			bs := fmt.Sprintf("% x", data.Bytes)
			if err := websocketConn.WriteMessage(websocket.TextMessage, []byte(bs)); err != nil {
				log.Printf("[ERR] Failed to write to websocket: %v\n", err)
				websocketMutex.Unlock()
				continue
			}
			_, moddedBytes, err := websocketConn.ReadMessage()
			websocketMutex.Unlock()
			if err != nil {
				log.Printf("[ERR] Failed to read from websocket: %v\n", err)
				continue
			}
			str := string(moddedBytes)
			str = strings.Replace(str, " ", "", -1)
			moddedBytes, err = hex.DecodeString(str)
			if err != nil {
				log.Printf("[ERR] Failed to decode hexedited data.")
				continue
			}
			data.Bytes = moddedBytes
			bytesRead = len(moddedBytes)
		}

		if data.DoPrint() {
			log.Printf("%v -> %v\n%v\n", data.DestAddr.String(), data.SrcAddr.String(), data.PrettyPrint())
		}
		_, err = pipe.WriteSource(data.Bytes[:bytesRead])
		if err != nil {
			break
		}
	}
}