func processConn(conn net.Conn, config *sslconn.Config) {
defer conn.Close()
log.Printf("Accepted: %s", conn.RemoteAddr().String())
sslc, err := sslconn.NewConn(conn, conn, config, true)
if err != nil {
log.Printf("New conn error: %s", err.Error())
return
}
defer sslc.Free()
err = sslc.Handshake()
if err != nil {
log.Printf("Handshake error: %s", err.Error())
return
}
reader := bufio.NewReader(sslc)
for {
line, _, err := reader.ReadLine()
if err != nil {
log.Printf("Read error: %s", err.Error())
return
}
if string(line) == "" {
break
}
}
w := bufio.NewWriter(sslc)
body := `<html>
<head>
<title>Welcome!</title>
</head>
<body>
<center><h1>Welcome!</h1></center>
</body>
</html>`
w.WriteString("HTTP/1.1 200 OK\r\n" +
"Content-Type: text/html\r\n" +
fmt.Sprintf("Content-Length: %d\r\n", len(body)) +
"Connection: close\r\n" +
"\r\n")
w.WriteString(body)
w.Flush()
err = sslc.Shutdown()
if err != nil {
log.Printf("Shutdown error: %s", err.Error())
return
}
}
func main() {
flag.Parse()
config := &sslconn.Config{}
if verify {
config.Verify = sslconn.VERIFY_PEER
}
config.CipherList = "ALL:!ADH:!LOW:!EXP:!DES-CBC3-SHA:@STRENGTH"
conn, err := net.Dial("tcp", server)
if err != nil {
log.Fatalf("Dial error: %s", err.Error())
}
sslc, err := sslconn.NewConn(conn, conn, config, false)
reader := bufio.NewReader(sslc)
writer := bufio.NewWriter(sslc)
err = sslc.Handshake()
if err != nil {
log.Fatal("Handshake error: %s", err.Error())
}
writer.WriteString(fmt.Sprintf("GET %s HTTP/1.0\r\n", resource))
writer.WriteString("User-Agent: sslconn-example-client\r\n")
writer.WriteString(fmt.Sprintf("Host: %s\r\n", server))
writer.WriteString("Accept: */*\r\n")
writer.WriteString("\r\n")
writer.Flush()
contentLenRegexp := regexp.MustCompile(`(?i)^Content-Length: (\d+)$`)
contentLen := 0
for {
line, _, err := reader.ReadLine()
if contentLenRegexp.Match(line) {
match := contentLenRegexp.FindSubmatch(line)[1]
contentLen, _ = strconv.Atoi(string(match))
}
if err != nil {
log.Printf("Read error: %s", err.Error())
return
}
if string(line) == "" {
break
}
log.Printf("%s", line)
}
buffer := make([]byte, 32*1024)
for contentLen > 0 {
max := len(buffer)
if max > contentLen {
max = contentLen
}
read, err := reader.Read(buffer[:max])
contentLen -= read
fmt.Print(string(buffer[:read]))
if err != nil {
log.Printf("Read error: %s", err.Error())
}
}
}