mirror of https://github.com/schollz/croc.git
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"strconv"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
func runClient(connectionType string, codePhrase string) {
|
|
var wg sync.WaitGroup
|
|
wg.Add(numberConnections)
|
|
for id := 0; id < numberConnections; id++ {
|
|
go func(id int) {
|
|
defer wg.Done()
|
|
port := strconv.Itoa(27001 + id)
|
|
connection, err := net.Dial("tcp", "localhost:"+port)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer connection.Close()
|
|
|
|
message := receiveMessage(connection)
|
|
fmt.Println(message)
|
|
sendMessage(connectionType+"."+codePhrase, connection)
|
|
if connectionType == "s" {
|
|
message = receiveMessage(connection)
|
|
fmt.Println(message)
|
|
// TODO: Write data from file
|
|
// Send file name
|
|
sendMessage("filename", connection)
|
|
// Send file size
|
|
time.Sleep(3 * time.Second)
|
|
sendMessage("filesize", connection)
|
|
} else {
|
|
// TODO: Pull data and write to file
|
|
fileName := receiveMessage(connection)
|
|
fileSize := receiveMessage(connection)
|
|
fmt.Println(fileName, fileSize)
|
|
}
|
|
|
|
}(id)
|
|
}
|
|
wg.Wait()
|
|
}
|