func (c4 *C4RemoteSession) writeHttpRequest(preq *http.Request) error {
ev := new(event.HTTPRequestEvent)
ev.FromRequest(preq)
ev.SetHash(c4.sess.SessionID)
if strings.Contains(ev.Url, "http://") {
ev.Url = ev.Url[7+len(preq.Host):]
}
//log.Printf("Session[%d]Range Request %s\n", c4.sess.SessionID, ev.Url)
c4.offerRequestEvent(ev)
return nil
}
func (gae *GAEHttpConnection) doRangeFetch(req *http.Request, firstChunkRes *http.Response) {
task := new(rangeFetchTask)
task.FetchLimit = int(gae_cfg.FetchLimitSize)
task.FetchWorkerNum = int(gae_cfg.ConcurrentRangeFetcher)
task.SessionID = gae.sess.SessionID
// task.TaskValidation = func() bool {
// return !util.IsDeadConnection(gae.sess.LocalRawConn)
// }
gae.rangeWorker = task
fetch := func(preq *http.Request) (*http.Response, error) {
ev := new(event.HTTPRequestEvent)
ev.FromRequest(preq)
ev.SetHash(gae.sess.SessionID)
err, xres := gae.requestEvent(gaeHttpClient, gae.sess, ev)
if nil != err {
//try again
err, xres = gae.requestEvent(gaeHttpClient, gae.sess, ev)
}
if nil == err {
httpresev := xres.(*event.HTTPResponseEvent)
httpres := httpresev.ToResponse()
return httpres, err
}
return nil, err
}
pres, err := task.SyncGet(req, firstChunkRes, fetch)
if nil == err {
err = pres.Write(gae.sess.LocalRawConn)
if nil != err {
task.Close()
gae.rangeWorker = nil
}
if nil != pres.Body {
pres.Body.Close()
}
}
if nil != err {
log.Printf("Session[%d]Range task failed for reason:%v\n", gae.sess.SessionID, err)
}
if nil != err || !util.IsResponseKeepAlive(pres) || !util.IsRequestKeepAlive(req) {
gae.sess.LocalRawConn.Close()
gae.sess.State = STATE_SESSION_CLOSE
gae.Close()
}
}
func (session *SessionConnection) process() error {
close_session := func() {
session.LocalRawConn.Close()
if nil != session.RemoteConn {
session.RemoteConn.Close()
}
session.State = STATE_SESSION_CLOSE
}
readRequest := func() (*http.Request, error) {
for {
needCheckRemote := false
if nil != session.RemoteConn {
forward, ok := session.RemoteConn.(*ForwardConnection)
if ok {
if forward.IsClosed() {
return nil, fmt.Errorf("Remote conn closed.")
}
needCheckRemote = true
}
}
if needCheckRemote {
session.LocalRawConn.SetReadDeadline(time.Now().Add(1 * time.Second))
}
req, e := http.ReadRequest(session.LocalBufferConn)
if nil != req {
req.Header.Del("Proxy-Connection")
}
if needCheckRemote {
var zero time.Time
session.LocalRawConn.SetReadDeadline(zero)
}
if nil != e && util.IsTimeoutError(e) {
continue
}
return req, e
}
return nil, nil
}
switch session.State {
case STATE_RECV_HTTP:
req, rerr := readRequest()
if nil == rerr {
var rev event.HTTPRequestEvent
rev.FromRequest(req)
rev.SetHash(session.SessionID)
err := session.processHttpEvent(&rev)
if err != nil {
log.Printf("Session[%d]Failed to read http request:%v\n", session.SessionID, err)
close_session()
return io.EOF
}
}
if nil != rerr {
log.Printf("Session[%d]Browser close connection:%v\n", session.SessionID, rerr)
close_session()
return io.EOF
}
case STATE_RECV_HTTP_CHUNK:
buf := make([]byte, 8192)
n, err := session.LocalBufferConn.Read(buf)
if nil == err {
rev := new(event.HTTPChunkEvent)
rev.Content = buf[0:n]
err = session.processHttpChunkEvent(rev)
}
if nil != err {
close_session()
return io.EOF
}
case STATE_RECV_TCP:
}
return nil
}