package main
import (
"bufio"
"code.google.com/p/mahonia"
"fmt"
"io"
"net"
"os"
)
func main() {
fmt.Println("create a server or client?")
reader := bufio.NewReader(os.Stdin)
input, _, _ := reader.ReadLine()
if string(input) == "server" {
Server()
}
if string(input) == "client" {
Client()
} else {
fmt.Println(Show("err arguments,entering again!.\r\n alternaltive argument is server or client"))
os.Exit(0)
}
}
func Show(s string) string {
enc := mahonia.NewEncoder("gbk") //中文转码有错误的函数。
return enc.ConvertString(s)
}
func Server() {
exit := make(chan bool)
ip := net.ParseIP("127.0.0.1")
addr := net.TCPAddr{ip, 8888, ""}
go func() {
listener, err := net.ListenTCP("tcp", &addr) //TCPListener listen
if err != nil {
fmt.Println("Initialize error", err.Error())
exit <- true
return
}
fmt.Println("Server listening...")
tcpcon, err := listener.AcceptTCP() //TCPConn client
if err != nil {
fmt.Println(err.Error())
//continue
}
fmt.Println("Client connect")
data := make([]byte, 1024)
if err != nil {
fmt.Println("tcpcon.Read(data)" + err.Error())
}
//recv file name
wc, err := tcpcon.Read(data)
fo, err := os.Create("F:\\uploads\\" + string(data[0:wc]))
if err != nil {
fmt.Println("os.Create" + err.Error())
}
fmt.Println("the file's name is:", string(data[0:wc]))
//recb file size
wc, err = tcpcon.Read(data)
fmt.Println("the file's size is:", string(data[0:wc]))
defer fo.Close()
for {
c, err := tcpcon.Read(data) //???为何调用conn类的Read
if err != nil {
fmt.Println("tcpcon.Read(data)" + err.Error())
}
if string(data[0:c]) == "filerecvend" {
fmt.Println("string(data[0:c]) == filerecvend is true")
tcpcon.Write([]byte("file recv finished!\r\n"))
tcpcon.Close()
break
}
//write to the file
_, err = fo.Write(data[0:c])
if err != nil {
fmt.Println("write err" + err.Error())
}
}
}()
<-exit
fmt.Println(Show("server close!"))
}
func Client() {
fmt.Println("send ur file to the destination", "input ur filename:")
reader := bufio.NewReader(os.Stdin)
input, _, _ := reader.ReadLine()
fmt.Println(string(input))
fi, err := os.Open(string(input))
if err != nil {
panic(err)
}
defer fi.Close()
fiinfo, err := fi.Stat()
fmt.Println("the size of file is ", fiinfo.Size(), "bytes") //fiinfo.Size() return int64 type
conn, err := net.Dial("tcp", "127.0.0.1:8888")
if err != nil {
fmt.Println(Show("connect server fail!"), Show(err.Error()))
return
}
defer conn.Close()
//send filename
_, err = conn.Write(input)
if err != nil {
fmt.Println("conn.Write", err.Error())
}
//send file size
_, err = conn.Write([]byte(string(fiinfo.Size())))
if err != nil {
fmt.Println("conn.Write", err.Error())
}
buff := make([]byte, 1024)
for {
n, err := fi.Read(buff)
if err != nil && err != io.EOF {
panic(err)
}
if n == 0 {
conn.Write([]byte("filerecvend"))
fmt.Println("filerecvend")
break
}
_, err = conn.Write(buff)
if err != nil {
fmt.Println(err.Error())
}
}
}