From 3ddacbb9c9062a599a61ee7b3c8cc17b1a54e0bf Mon Sep 17 00:00:00 2001
From: Darien Raymond <admin@v2ray.com>
Date: Fri, 23 Dec 2016 16:54:15 +0100
Subject: [PATCH] prototype for payload inspector

---
 app/dispatcher/config.pb.go    | 50 ++++++++++++++++++++++++++++++----
 app/dispatcher/config.proto    | 10 +++++--
 app/dispatcher/impl/default.go | 15 ++++++++++
 app/dispatcher/impl/http.go    | 10 +++++++
 4 files changed, 77 insertions(+), 8 deletions(-)
 create mode 100644 app/dispatcher/impl/http.go

diff --git a/app/dispatcher/config.pb.go b/app/dispatcher/config.pb.go
index 7d370a54f..4a7a2fab1 100644
--- a/app/dispatcher/config.pb.go
+++ b/app/dispatcher/config.pb.go
@@ -28,7 +28,32 @@ var _ = math.Inf
 // proto package needs to be updated.
 const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package
 
+type Config_FixDestination int32
+
+const (
+	Config_Auto     Config_FixDestination = 0
+	Config_Enabled  Config_FixDestination = 1
+	Config_Disabled Config_FixDestination = 2
+)
+
+var Config_FixDestination_name = map[int32]string{
+	0: "Auto",
+	1: "Enabled",
+	2: "Disabled",
+}
+var Config_FixDestination_value = map[string]int32{
+	"Auto":     0,
+	"Enabled":  1,
+	"Disabled": 2,
+}
+
+func (x Config_FixDestination) String() string {
+	return proto.EnumName(Config_FixDestination_name, int32(x))
+}
+func (Config_FixDestination) EnumDescriptor() ([]byte, []int) { return fileDescriptor0, []int{0, 0} }
+
 type Config struct {
+	FixDestination Config_FixDestination `protobuf:"varint,1,opt,name=fix_destination,json=fixDestination,enum=v2ray.core.app.dispatcher.Config_FixDestination" json:"fix_destination,omitempty"`
 }
 
 func (m *Config) Reset()                    { *m = Config{} }
@@ -36,21 +61,34 @@ func (m *Config) String() string            { return proto.CompactTextString(m)
 func (*Config) ProtoMessage()               {}
 func (*Config) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
 
+func (m *Config) GetFixDestination() Config_FixDestination {
+	if m != nil {
+		return m.FixDestination
+	}
+	return Config_Auto
+}
+
 func init() {
 	proto.RegisterType((*Config)(nil), "v2ray.core.app.dispatcher.Config")
+	proto.RegisterEnum("v2ray.core.app.dispatcher.Config_FixDestination", Config_FixDestination_name, Config_FixDestination_value)
 }
 
 func init() { proto.RegisterFile("v2ray.com/core/app/dispatcher/config.proto", fileDescriptor0) }
 
 var fileDescriptor0 = []byte{
-	// 134 bytes of a gzipped FileDescriptorProto
+	// 218 bytes of a gzipped FileDescriptorProto
 	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xe2, 0xd2, 0x2a, 0x33, 0x2a, 0x4a,
 	0xac, 0xd4, 0x4b, 0xce, 0xcf, 0xd5, 0x4f, 0xce, 0x2f, 0x4a, 0xd5, 0x4f, 0x2c, 0x28, 0xd0, 0x4f,
 	0xc9, 0x2c, 0x2e, 0x48, 0x2c, 0x49, 0xce, 0x48, 0x2d, 0xd2, 0x4f, 0xce, 0xcf, 0x4b, 0xcb, 0x4c,
 	0xd7, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x92, 0x84, 0xa9, 0x2d, 0x4a, 0xd5, 0x4b, 0x2c, 0x28,
-	0xd0, 0x43, 0xa8, 0x53, 0xe2, 0xe0, 0x62, 0x73, 0x06, 0x2b, 0x75, 0x0a, 0xe1, 0x92, 0x4d, 0xce,
-	0xcf, 0xd5, 0xc3, 0xa9, 0xd4, 0x89, 0x1b, 0xa2, 0x30, 0x00, 0x64, 0x64, 0x14, 0x17, 0x42, 0x62,
-	0x15, 0x93, 0x64, 0x98, 0x51, 0x50, 0x62, 0xa5, 0x9e, 0x33, 0x48, 0x93, 0x63, 0x41, 0x81, 0x9e,
-	0x0b, 0x5c, 0x2e, 0x89, 0x0d, 0xec, 0x02, 0x63, 0x40, 0x00, 0x00, 0x00, 0xff, 0xff, 0x60, 0xf8,
-	0x2f, 0x3b, 0xaf, 0x00, 0x00, 0x00,
+	0xd0, 0x43, 0xa8, 0x53, 0x9a, 0xc5, 0xc8, 0xc5, 0xe6, 0x0c, 0x56, 0x2b, 0x14, 0xc9, 0xc5, 0x9f,
+	0x96, 0x59, 0x11, 0x9f, 0x92, 0x5a, 0x5c, 0x92, 0x99, 0x97, 0x58, 0x92, 0x99, 0x9f, 0x27, 0xc1,
+	0xa8, 0xc0, 0xa8, 0xc1, 0x67, 0x64, 0xa0, 0x87, 0x53, 0xbf, 0x1e, 0x44, 0xaf, 0x9e, 0x5b, 0x66,
+	0x85, 0x0b, 0x42, 0x5f, 0x10, 0x5f, 0x1a, 0x0a, 0x5f, 0xc9, 0x94, 0x8b, 0x0f, 0x55, 0x85, 0x10,
+	0x07, 0x17, 0x8b, 0x63, 0x69, 0x49, 0xbe, 0x00, 0x83, 0x10, 0x37, 0x17, 0xbb, 0x6b, 0x5e, 0x62,
+	0x52, 0x4e, 0x6a, 0x8a, 0x00, 0xa3, 0x10, 0x0f, 0x17, 0x87, 0x4b, 0x66, 0x31, 0x84, 0xc7, 0xe4,
+	0x14, 0xc2, 0x25, 0x9b, 0x9c, 0x9f, 0x8b, 0xdb, 0x76, 0x27, 0x6e, 0x88, 0xf5, 0x01, 0x20, 0x5f,
+	0x46, 0x71, 0x21, 0x24, 0x56, 0x31, 0x49, 0x86, 0x19, 0x05, 0x25, 0x56, 0xea, 0x39, 0x83, 0x34,
+	0x39, 0x16, 0x14, 0xe8, 0xb9, 0xc0, 0xe5, 0x92, 0xd8, 0xc0, 0x81, 0x62, 0x0c, 0x08, 0x00, 0x00,
+	0xff, 0xff, 0x63, 0xfd, 0xa7, 0xdb, 0x42, 0x01, 0x00, 0x00,
 }
diff --git a/app/dispatcher/config.proto b/app/dispatcher/config.proto
index 7074b9207..d1b39609f 100644
--- a/app/dispatcher/config.proto
+++ b/app/dispatcher/config.proto
@@ -7,5 +7,11 @@ option java_package = "com.v2ray.core.app.dispatcher";
 option java_outer_classname = "ConfigProto";
 
 message Config {
-  
-}
\ No newline at end of file
+  enum FixDestination {
+    Auto = 0;
+    Enabled = 1;
+    Disabled = 2;
+  }
+
+  FixDestination fix_destination = 1;
+}
diff --git a/app/dispatcher/impl/default.go b/app/dispatcher/impl/default.go
index e0b9f843a..e2ec58f85 100644
--- a/app/dispatcher/impl/default.go
+++ b/app/dispatcher/impl/default.go
@@ -85,6 +85,21 @@ func (v *DefaultDispatcher) FilterPacketAndDispatch(destination v2net.Destinatio
 	dispatcher.Dispatch(destination, payload, link)
 }
 
+func (v *DefaultDispatcher) FixDestination(payload *buf.Buffer, dest v2net.Destination) v2net.Destination {
+	if dest.Address.Family().IsDomain() || dest.Network != v2net.Network_TCP {
+		return dest
+	}
+
+	switch dest.Port {
+	case 80:
+		addr := DetectHTTPHost(payload)
+		if addr != nil {
+			dest.Address = addr
+		}
+	}
+	return dest
+}
+
 type DefaultDispatcherFactory struct{}
 
 func (v DefaultDispatcherFactory) Create(space app.Space, config interface{}) (app.Application, error) {
diff --git a/app/dispatcher/impl/http.go b/app/dispatcher/impl/http.go
new file mode 100644
index 000000000..2358e38c7
--- /dev/null
+++ b/app/dispatcher/impl/http.go
@@ -0,0 +1,10 @@
+package impl
+
+import (
+	"v2ray.com/core/common/buf"
+	v2net "v2ray.com/core/common/net"
+)
+
+func DetectHTTPHost(payload *buf.Buffer) v2net.Address {
+	return nil
+}