mirror of
https://github.com/golang/go
synced 2024-09-19 16:01:33 +00:00
100 lines
2 KiB
Go
100 lines
2 KiB
Go
// Copyright 2009 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// $G $F.go && $L $F.$A && GOMAXPROCS=3 ./$A.out
|
|
// # TODO(rsc): GOMAXPROCS will go away eventually.
|
|
// # 3 is one for Echo, one for Serve, one for Connect.
|
|
|
|
package main
|
|
import (
|
|
"os";
|
|
"io";
|
|
"net";
|
|
"syscall"
|
|
)
|
|
|
|
func StringToBuf(s string) *[]byte {
|
|
l := len(s);
|
|
b := new([]byte, l);
|
|
for i := 0; i < l; i++ {
|
|
b[i] = s[i];
|
|
}
|
|
return b;
|
|
}
|
|
|
|
func Echo(fd io.ReadWrite, done *chan<- int) {
|
|
var buf [1024]byte;
|
|
|
|
for {
|
|
n, err := fd.Read(&buf);
|
|
if err != nil || n == 0 {
|
|
break;
|
|
}
|
|
fd.Write((&buf)[0:n])
|
|
}
|
|
done <- 1
|
|
}
|
|
|
|
func Serve(network, addr string, listening, done *chan<- int) {
|
|
l, err := net.Listen(network, addr);
|
|
if err != nil {
|
|
panic("listen: "+err.String());
|
|
}
|
|
listening <- 1;
|
|
|
|
for {
|
|
fd, addr, err := l.Accept();
|
|
if err != nil {
|
|
break;
|
|
}
|
|
echodone := new(chan int)
|
|
go Echo(fd, echodone);
|
|
<-echodone; // make sure Echo stops
|
|
l.Close();
|
|
}
|
|
done <- 1
|
|
}
|
|
|
|
func Connect(network, addr string) {
|
|
fd, err := net.Dial(network, "", addr);
|
|
if err != nil {
|
|
panic("connect: "+err.String());
|
|
}
|
|
|
|
b := StringToBuf("hello, world\n");
|
|
var b1 [100]byte;
|
|
|
|
n, errno := fd.Write(b);
|
|
if n != len(b) {
|
|
panic("syscall.write in connect");
|
|
}
|
|
|
|
n, errno = fd.Read(&b1);
|
|
if n != len(b) {
|
|
panic("syscall.read in connect");
|
|
}
|
|
|
|
// os.Stdout.Write((&b1)[0:n]);
|
|
fd.Close();
|
|
}
|
|
|
|
func Test(network, listenaddr, dialaddr string) {
|
|
// print("Test ", network, " ", listenaddr, " ", dialaddr, "\n");
|
|
listening := new(chan int);
|
|
done := new(chan int);
|
|
go Serve(network, listenaddr, listening, done);
|
|
<-listening; // wait for server to start
|
|
Connect(network, dialaddr);
|
|
<-done; // make sure server stopped
|
|
}
|
|
|
|
func main() {
|
|
Test("tcp", "0.0.0.0:9999", "127.0.0.1:9999");
|
|
Test("tcp", "[::]:9999", "[::ffff:127.0.0.1]:9999");
|
|
Test("tcp", "[::]:9999", "127.0.0.1:9999");
|
|
Test("tcp", "0.0.0.0:9999", "[::ffff:127.0.0.1]:9999");
|
|
sys.exit(0); // supposed to happen on return, doesn't
|
|
}
|
|
|