func (p *DirectProxy) Serve(session *proxy.ProxySession, ev event.Event) error {
if nil == session.Remote || session.Remote.C.Closed() {
switch ev.(type) {
case *event.TCPOpenEvent:
case *event.HTTPRequestEvent:
case *event.UDPEvent:
default:
session.Close()
return fmt.Errorf("Can NOT create direct channel by event:%T", ev)
}
c, err := newDirectChannel(ev, &p.conf)
if nil != err {
session.Close()
return err
}
session.Remote = &proxy.RemoteChannel{
Addr: c.addr,
DirectIO: true,
}
session.Remote.C = c
if c.httpsProxyConn {
session.Hijacked = true
return nil
}
}
if nil == session.Remote {
return fmt.Errorf("No remote connected.")
}
switch ev.(type) {
case *event.UDPEvent:
session.Remote.WriteRaw(ev.(*event.UDPEvent).Content)
case *event.ConnCloseEvent:
session.Remote.Close()
case *event.TCPOpenEvent:
case *event.ConnTestEvent:
//do nothing
case *event.TCPChunkEvent:
session.Remote.WriteRaw(ev.(*event.TCPChunkEvent).Content)
case *event.HTTPRequestEvent:
req := ev.(*event.HTTPRequestEvent)
content := req.HTTPEncode()
_, err := session.Remote.WriteRaw(content)
if nil != err {
closeEv := &event.ConnCloseEvent{}
closeEv.SetId(ev.GetId())
proxy.HandleEvent(closeEv)
return err
}
return nil
default:
log.Printf("Invalid event type:%T to process", ev)
}
return nil
}
