go 전송 파일 수신

12788 단어

 
package main

import (
    "fmt"
    "os"
    "net"
    "io"
)

//      
func SendFile(path string,conn net.Conn){
    //          
    f,err  := os.Open(path)
    if err != nil{
        fmt.Println("os.Open err=",err)
        return
    }

    defer f.Close()

    buf := make([]byte,1024*4)
    for{
        n,err := f.Read(buf)
        if err !=nil{
            if err == io.EOF{
                fmt.Println("      ")
            }else{
                fmt.Println(" f.Read err=",err)
            }
            return
        }
        //    
        conn.Write(buf[:n])
    }


}

func main()  {
    //      
    fmt.Println("          ")
    var path string
    fmt.Scan(&path)

    //      info.Name()
    info,err := os.Stat(path)

    if err != nil{
        fmt.Println("os.Stat err= ",err)
        return
    }

    //          
    conn,err := net.Dial("tcp","127.0.0.1:8000")
    if err !=nil{
        fmt.Println("net.Dial err=",err)
        return
    }

    //         
    _,err = conn.Write([]byte(info.Name()))
    if err != nil{
        fmt.Println("conn.Write err =",err)
        return
    }

    var n int
    buf := make([]byte,1024)
    n,err = conn.Read(buf)
    if err !=nil{
        fmt.Println("conn.Read err=",err)
        return
    }

    if "ok"== string(buf[:n]){
        //      
        SendFile(path,conn)
    }





}
package main

import (
    "net"
    "fmt"
    "os"
    "io"
)
//       
func RecvFile(filename string,conn net.Conn)  {
   //    
   f,err :=  os.Create(filename)
   if err != nil{
       fmt.Println("os.Create err=",err)
       return
   }

   buf := make([]byte,1024)
   for{
        n,err := conn.Read(buf) //             
        if err != nil{
            if err == io.EOF{
                fmt.Println("      ")
        }else{
            fmt.Println("conn.Read err=",err)

        }
        return
     }
     if n == 0{
         fmt.Println("n==0       ")
         break
     }
     f.Write(buf[:n])
   }
}

func main()  {
    //  
    listernner,err := net.Listen("tcp","127.0.0.1:8000")
    if err != nil{
        fmt.Println("net.Listen err =",err)
        return
    }
    defer  listernner.Close()
    //        
    conn,err := listernner.Accept()
    if err != nil{
        fmt.Println("listenner.Accept err=",err)
        return
    }
    defer  conn.Close()

    //  
    buf := make([]byte,1024)

     n,err := conn.Read(buf)
     if err != nil{
         fmt.Println("conn.Read err =",err)
         return
     }

     filename := string(buf[:n])
     //  ok
     conn.Write([]byte("ok"))

     //      
     RecvFile(filename,conn)




}

좋은 웹페이지 즐겨찾기