import "fmt"
import "net"
import "bufio"
+import "strings"
+
+var cmdHandler = map[string]func([]string,string,*bufio.ReadWriter ) {
+ "test":handleCmdTest,
+}
+
+
+func readLineSafe(rw *bufio.ReadWriter) (string, error) {
+ wasPrefix:=false
+ var line string
+ for isPrefix:=true;isPrefix; {
+ var lineBuf []byte
+ var err error
+ lineBuf,isPrefix,err = rw.ReadLine()
+ if err != nil {
+ return "",err
+ }
+ if isPrefix {
+ wasPrefix=true
+ } else {
+ line=string(lineBuf)
+ }
+ }
+ if wasPrefix {
+ fmt.Println("line too long")
+ fmt.Fprintf(rw,"line too long\n")
+ rw.Flush()
+ return "",nil
+ }
+ return line,nil
+}
+
+func connToReadWriter(c net.Conn) (*bufio.ReadWriter) {
+ client_r := bufio.NewReaderSize(c,14)
+ client_w := bufio.NewWriterSize(c,14)
+ return bufio.NewReadWriter(client_r,client_w)
+}
func handleConnection(c net.Conn) () {
+ client:=connToReadWriter(c)
fmt.Println("new connection")
- var client *bufio.ReadWriter;
- {
- client_r := bufio.NewReaderSize(c,14)
- client_w := bufio.NewWriterSize(c,14)
- client = bufio.NewReadWriter(client_r,client_w)
- }
for {
- wasPrefix:=false
- var line string
- for isPrefix:=true;isPrefix; {
- var lineBuf []byte
- var err error
- lineBuf,isPrefix,err = client.ReadLine()
- if err != nil {
- fmt.Printf("Readline Error: %s\n",err.Error())
- c.Close()
- return
- }
- if isPrefix {
- wasPrefix=true
- } else {
- line=string(lineBuf)
- }
- }
- if wasPrefix {
- fmt.Println("line too long")
- fmt.Fprintf(client,"line too long\n")
- client.Flush()
- continue
+ line,err:=readLineSafe(client)
+ if err != nil {
+ if err.Error() != "EOF" {
+ fmt.Printf("Error: readLineSafe returned %v\n",err.Error())
+ } else {
+ fmt.Printf("Connection closed by remote host\n");
}
- fmt.Printf("Received: %v\n", string(line))
-//
-// data := buf[0:nr]
-// fmt.Printf("Received: %v", string(data))
-// _, err = c.Write(data)
-// if err != nil {
-// fmt.Println("Write: " + err.Error())
-// return
-// }
+ c.Close()
+ return
+ }
+ if line == "" {
+ continue
+ }
+ fmt.Printf("Received: %v\n", line)
+ tokens:=strings.Fields(line)
+ remainStr:=strings.Join(tokens[1:]," ")
+ handleCmd(tokens,remainStr,client)
}
}
+func handleCmd(tokens []string, remainStr string,client * bufio.ReadWriter) {
+ cmd:=tokens[0]
+ func_ptr,present := cmdHandler[cmd]
+ if present {
+ func_ptr(tokens, remainStr,client)
+ } else {
+ fmt.Printf("Error: unknown Cmd: %v\n", cmd)
+ }
+}
+
+func handleCmdTest(tokens []string, remainStr string, client * bufio.ReadWriter) {
+ //cmd:=tokens[0]
+ fmt.Printf("Test: %v\n", remainStr)
+}
func main() {
ln, err := net.Listen("unix", "/tmp/test.sock")