-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathserver.coffee
213 lines (174 loc) · 7.13 KB
/
server.coffee
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
Fs = require 'fs'
Url = require 'url'
Http = require 'http'
Crypto = require 'crypto'
QueryString = require 'querystring'
port = parseInt process.env.PORT || 8081
version = "1.0.5-adn"
excluded = process.env.CAMO_HOST_EXCLUSIONS || '*.example.org'
shared_key = process.env.CAMO_KEY || '0x24FEEDFACEDEADBEEFCAFE'
max_redirects = process.env.CAMO_MAX_REDIRECTS || 4
camo_hostname = process.env.CAMO_HOSTNAME || "unknown"
logging_enabled = process.env.CAMO_LOGGING_ENABLED || "disabled"
log = (msg) ->
unless logging_enabled == "disabled"
console.log("--------------------------------------------")
console.log(msg)
console.log("--------------------------------------------")
EXCLUDED_HOSTS = new RegExp(excluded.replace(".", "\\.").replace("*", "\\.*"))
RESTRICTED_IPS = /^((10\.)|(127\.)|(169\.254)|(192\.168)|(172\.((1[6-9])|(2[0-9])|(3[0-1]))))/
total_connections = 0
current_connections = 0
started_at = new Date
four_oh_four = (resp, msg) ->
log msg
resp.writeHead 404
finish resp, "Not Found"
finish = (resp, str) ->
current_connections -= 1
current_connections = 0 if current_connections < 1
resp.connection && resp.end str
process_url = (url, transferred_headers, resp, remaining_redirects) ->
if url.host? && !url.host.match(RESTRICTED_IPS)
if url.host.match(EXCLUDED_HOSTS)
return four_oh_four(resp, "Hitting excluded hostnames")
transferred_headers.host = url.host
log transferred_headers
srcReq = Http.request
hostname : url.hostname
port : url.port || 80
path : url.path
headers : transferred_headers
srcReq.on 'error', (error) ->
four_oh_four(resp, "Client Request error #{error.stack}")
srcReq.on 'response', (srcResp) ->
is_finished = true
log srcResp.headers
content_length = srcResp.headers['content-length']
if content_length > 5242880
four_oh_four(resp, "Content-Length exceeded")
else
newHeaders =
'Content-Type' : srcResp.headers['content-type']
'Date' : srcResp.headers['date'] || new Date
'Cache-control' : srcResp.headers['cache-control'] || 'public, max-age=31536000'
'Camo-Host' : camo_hostname
'X-Content-Type-Options' : 'nosniff'
# Handle chunked responses properly
if content_length?
newHeaders['content-length'] = content_length
if srcResp.headers['transfer-encoding']
newHeaders['transfer-encoding'] = srcResp.headers['transfer-encoding']
if srcResp.headers['content-encoding']
newHeaders['content-encoding'] = srcResp.headers['content-encoding']
if srcResp.headers['last-modified-date']
newHeaders['last-modified-date'] = srcResp.headers['last-modified-date']
if srcResp.headers['etag']
newHeaders['etag'] = srcResp.headers['etag']
srcResp.on 'end', ->
if is_finished
finish resp
srcResp.on 'error', ->
if is_finished
finish resp
switch srcResp.statusCode
when 200
if newHeaders['content-type'] && newHeaders['content-type'].slice(0, 5) != 'image'
four_oh_four(resp, "Non-Image content-type returned")
log newHeaders
resp.writeHead srcResp.statusCode, newHeaders
srcResp.on 'data', (chunk) ->
resp.write chunk
when 301, 302, 303, 307
if remaining_redirects <= 0
four_oh_four(resp, "Exceeded max depth")
else
newUrl = Url.parse srcResp.headers['location']
send_parsed = false
unless newUrl.host? and newUrl.hostname?
newUrl.host = newUrl.hostname = url.hostname
newUrl.protocol = url.protocol
send_parsed = true
if newUrl.protocol is 'https:'
if send_parsed
newHeaders['location'] = Url.format newUrl
else
newHeaders['location'] = srcResp.headers['location']
resp.writeHead srcResp.statusCode, newHeaders
else
is_finished = false
process_url newUrl, transferred_headers, resp, remaining_redirects - 1
when 304
resp.writeHead srcResp.statusCode, newHeaders
else
four_oh_four(resp, "Responded with " + srcResp.statusCode + ":" + srcResp.headers)
srcReq.on 'error', ->
finish resp
srcReq.end()
else
four_oh_four(resp, "No host found " + url.host)
# decode a string of two char hex digits
hexdec = (str) ->
if str and str.length > 0 and str.length % 2 == 0 and not str.match(/[^0-9a-f]/)
buf = new Buffer(str.length / 2)
for i in [0...str.length] by 2
buf[i/2] = parseInt(str[i..i+1], 16)
buf.toString()
server = Http.createServer (req, resp) ->
if req.method != 'GET' || req.url == '/'
resp.writeHead 200
resp.end 'hwhat'
else if req.url == '/favicon.ico'
resp.writeHead 200
resp.end 'ok'
else if req.url == '/status'
resp.writeHead 200
resp.end "ok #{current_connections}/#{total_connections} since #{started_at.toString()}"
else
total_connections += 1
current_connections += 1
url = Url.parse req.url
user_agent = process.env.CAMO_HEADER_VIA or= "Camo Asset Proxy #{version}"
transferred_headers =
'Via' : user_agent
'User-Agent' : user_agent
'Accept' : 'image/*;q=0.9,*/*;q=0.8'
'x-forwarded-for' : req.headers['x-forwarded-for']
'x-content-type-options' : 'nosniff'
if req.headers['if-modified-since']?
transferred_headers['If-Modified-Since'] = req.headers['if-modified-since']
if req.headers['if-none-match']?
transferred_headers['If-None-Match'] = req.headers['if-none-match']
delete(req.headers.cookie)
[query_digest, encoded_url] = url.pathname.replace(/^\//, '').split("/", 2)
if encoded_url = hexdec(encoded_url)
url_type = 'path'
dest_url = encoded_url
else
url_type = 'query'
dest_url = QueryString.parse(url.query).url
log({
type: url_type
url: req.url
headers: req.headers
dest: dest_url
digest: query_digest
})
if req.headers['via'] && req.headers['via'].indexOf(user_agent) != -1
return four_oh_four(resp, "Requesting from self")
if url.pathname? && dest_url
hmac = Crypto.createHmac("sha1", shared_key)
hmac.update(dest_url)
hmac_digest = hmac.digest('hex')
if hmac_digest == query_digest
url = Url.parse dest_url
process_url url, transferred_headers, resp, max_redirects
else
four_oh_four(resp, "checksum mismatch #{hmac_digest}:#{query_digest}")
else
four_oh_four(resp, "No pathname provided on the server")
console.log "SSL-Proxy running on #{port} with pid:#{process.pid}."
console.log "Using the secret key #{shared_key}"
Fs.open "tmp/camo.pid", "w", 0o600, (err, fd) ->
Fs.writeSync fd, process.pid
server.listen port