87 lines
1.6 KiB
Go
87 lines
1.6 KiB
Go
package phpfpm
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"net"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// Create an fcgi client
|
|
func newFcgiClient(h string, args ...interface{}) (*conn, error) {
|
|
var con net.Conn
|
|
if len(args) != 1 {
|
|
return nil, errors.New("fcgi: not enough params")
|
|
}
|
|
|
|
var err error
|
|
switch args[0].(type) {
|
|
case int:
|
|
addr := h + ":" + strconv.FormatInt(int64(args[0].(int)), 10)
|
|
con, err = net.Dial("tcp", addr)
|
|
case string:
|
|
laddr := net.UnixAddr{Name: args[0].(string), Net: h}
|
|
con, err = net.DialUnix(h, nil, &laddr)
|
|
default:
|
|
err = errors.New("fcgi: we only accept int (port) or string (socket) params.")
|
|
}
|
|
fcgi := &conn{
|
|
rwc: con,
|
|
}
|
|
|
|
return fcgi, err
|
|
}
|
|
|
|
func (client *conn) Request(
|
|
env map[string]string,
|
|
requestData string,
|
|
) (retout []byte, reterr []byte, err error) {
|
|
defer client.rwc.Close()
|
|
var reqId uint16 = 1
|
|
|
|
err = client.writeBeginRequest(reqId, uint16(roleResponder), 0)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = client.writePairs(typeParams, reqId, env)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if len(requestData) > 0 {
|
|
if err = client.writeRecord(typeStdin, reqId, []byte(requestData)); err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
rec := &record{}
|
|
var err1 error
|
|
|
|
// receive until EOF or FCGI_END_REQUEST
|
|
READ_LOOP:
|
|
for {
|
|
err1 = rec.read(client.rwc)
|
|
if err1 != nil && strings.Contains(err1.Error(), "use of closed network connection") {
|
|
if err1 != io.EOF {
|
|
err = err1
|
|
}
|
|
break
|
|
}
|
|
|
|
switch {
|
|
case rec.h.Type == typeStdout:
|
|
retout = append(retout, rec.content()...)
|
|
case rec.h.Type == typeStderr:
|
|
reterr = append(reterr, rec.content()...)
|
|
case rec.h.Type == typeEndRequest:
|
|
fallthrough
|
|
default:
|
|
break READ_LOOP
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|