1
0
mirror of https://github.com/golang/go synced 2024-11-22 21:10:03 -07:00
go/test/tcpserver.go
Russ Cox efc86a74e4 change meaning of $GOMAXPROCS to number of cpus to use,
not number of threads.  can still starve all the other threads,
but only by looping, not by waiting in a system call.

fix darwin syscall.Syscall6 bug.

fix chanclient bug.

delete $GOMAXPROCS from network tests.

add stripped down printf, sys.printhex to runtime.

R=r
DELTA=355  (217 added, 36 deleted, 102 changed)
OCL=20017
CL=20019
2008-11-25 16:48:10 -08:00

98 lines
1.9 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 && ./$A.out
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
}