Shikong/HFish
1
0
Fork 0
mirror of https://gitee.com/lauix/HFish synced 2025-05-12 04:48:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加Memcache UDP蜜罐支持

Browse Source
This commit is contained in:
Srar 2019-08-14 12:37:54 +08:00
parent 219cab2d93
commit 6b441c520c
1 changed files with 53 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
core/protocol/memcache/memcache.go
View File

@ -404,7 +404,7 @@ func tcpServer(address string, exitChan chan int) {
exitChan <- 1 exitChan <- 1
} }
log.Println("[Memcache] Listning on " + address) log.Println("[Memcache TCP] Listning on " + address)
defer l.Close() defer l.Close()
@ -420,7 +420,7 @@ func tcpServer(address string, exitChan chan int) {
go func() { go func() {
skip := false skip := false
reader := bufio.NewReader(conn) reader := bufio.NewReader(conn)
log.Printf("[Memcache %d] Accepted a client socket from %s\n", trackID, conn.RemoteAddr().String()) log.Printf("[Memcache TCP %d] Accepted a client socket from %s\n", trackID, conn.RemoteAddr().String())
for { for {
str, err := reader.ReadString('\n') str, err := reader.ReadString('\n')
if skip { if skip {
@ -428,13 +428,13 @@ func tcpServer(address string, exitChan chan int) {
continue continue
} }
if err != nil { if err != nil {
log.Printf("[Memcache %d] Closed a client socket.", trackID) log.Printf("[Memcache TCP %d] Closed a client socket.\n", trackID)
conn.Close() conn.Close()
break break
} }
str = strings.TrimSpace(str) str = strings.TrimSpace(str)
log.Printf("[Memcache %d] Client request: %s.", trackID, str) log.Printf("[Memcache TCP %d] Client request: %s.\n", trackID, str)
args := strings.Split(str, " ") args := strings.Split(str, " ")
function, exist := commands[args[0]] function, exist := commands[args[0]]
if !exist { if !exist {
@ -470,29 +470,57 @@ func tcpServer(address string, exitChan chan int) {
} }
// func udpServer(address string, exitChan chan int) { func udpServer(address string, exitChan chan int) {
// udpAddr, err := net.ResolveUDPAddr("udp", address) udpAddr, err := net.ResolveUDPAddr("udp", address)
// if err != nil { if err != nil {
// fmt.Println(err.Error()) fmt.Println(err.Error())
// exitChan <- 1 exitChan <- 1
// } }
// l, err := net.ListenUDP("udp", udpAddr) l, err := net.ListenUDP("udp", udpAddr)
// if err != nil { if err != nil {
// fmt.Println(err.Error()) fmt.Println(err.Error())
// exitChan <- 1 exitChan <- 1
// } }
// go func() { log.Println("[Memcache UDP] Listning on " + address)
// buf := make([]byte, 1500)
// for {
// buf := make([]byte, 1500)
// plen, addr, _ := l.ReadFromUDP(buf)
// } go func() {
// } buf := make([]byte, 1500)
// }() for {
// } plen, addr, _ := l.ReadFromUDP(buf)
/* UDP协议需要8个字节的头 */
if plen < 8 {
continue
}
requestStr := string(buf[8:plen])
for _, request := range strings.Split(requestStr, "\n") {
request = strings.TrimSpace(request)
if request == "" {
continue
}
log.Printf("[Memcache UDP] Client request: [%s] from: %s\n", request, addr.String())
args := strings.Split(request, " ")
function, exist := commands[args[0]]
if !exist {
continue
}
args = args[1:]
response, requiredBytes := function(args)
if requiredBytes != 0 {
continue
}
if len(response) > 1300 {
response = response[:1300]
}
l.WriteTo(append([]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00}, response...), addr)
}
}
}()
}
func Start(addr string) { func Start(addr string) {
// 创建一个程序结束码的通道 // 创建一个程序结束码的通道
@ -500,7 +528,7 @@ func Start(addr string) {
// 将服务器并发运行 // 将服务器并发运行
go tcpServer(addr, exitChan) go tcpServer(addr, exitChan)
// go udpServer(addr, exitChan) go udpServer(addr, exitChan)
// 通道阻塞，等待接受返回值 // 通道阻塞，等待接受返回值
code := <-exitChan code := <-exitChan