proxi 0.1

Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml ADDED
@@ -0,0 +1,7 @@
1
+ ---
2
+ SHA1:
3
+ metadata.gz: 125abf0d6cecec0a987df416390f7d903f896ea2
4
+ data.tar.gz: 524c9c84a6494d9838b448fa1723d0e7b838ebae
5
+ SHA512:
6
+ metadata.gz: dd8bf72dfb3dabd7088fbb5fc73d8cb02d6be8013659fda142e9f9ea439a5f4b876b434d22676c0b7616630bba6887768342720b602c0ed7b452647b4bb0acbd
7
+ data.tar.gz: 4b10253015c9ab46428ae0494fa5a2d4188e5f5087e8f40e8d45355fdf26709d705c78f082b749055948f6f0c8dc31a2ae6816f934e0f75f1076fa274fc46f75
data/.gitignore ADDED
@@ -0,0 +1 @@
1
+ scratch.rb
data/Gemfile ADDED
@@ -0,0 +1,3 @@
1
+ source 'https://rubygems.org'
2
+
3
+ gemspec
data/Gemfile.lock ADDED
@@ -0,0 +1,31 @@
1
+ PATH
2
+ remote: .
3
+ specs:
4
+ proxi (0.1)
5
+ wisper (> 0)
6
+
7
+ GEM
8
+ remote: https://rubygems.org/
9
+ specs:
10
+ diff-lcs (1.2.5)
11
+ rspec (3.3.0)
12
+ rspec-core (~> 3.3.0)
13
+ rspec-expectations (~> 3.3.0)
14
+ rspec-mocks (~> 3.3.0)
15
+ rspec-core (3.3.2)
16
+ rspec-support (~> 3.3.0)
17
+ rspec-expectations (3.3.1)
18
+ diff-lcs (>= 1.2.0, < 2.0)
19
+ rspec-support (~> 3.3.0)
20
+ rspec-mocks (3.3.2)
21
+ diff-lcs (>= 1.2.0, < 2.0)
22
+ rspec-support (~> 3.3.0)
23
+ rspec-support (3.3.0)
24
+ wisper (1.6.1)
25
+
26
+ PLATFORMS
27
+ ruby
28
+
29
+ DEPENDENCIES
30
+ proxi!
31
+ rspec (~> 3.0)
data/LICENSE ADDED
@@ -0,0 +1,373 @@
1
+ Mozilla Public License Version 2.0
2
+ ==================================
3
+
4
+ 1. Definitions
5
+ --------------
6
+
7
+ 1.1. "Contributor"
8
+ means each individual or legal entity that creates, contributes to
9
+ the creation of, or owns Covered Software.
10
+
11
+ 1.2. "Contributor Version"
12
+ means the combination of the Contributions of others (if any) used
13
+ by a Contributor and that particular Contributor's Contribution.
14
+
15
+ 1.3. "Contribution"
16
+ means Covered Software of a particular Contributor.
17
+
18
+ 1.4. "Covered Software"
19
+ means Source Code Form to which the initial Contributor has attached
20
+ the notice in Exhibit A, the Executable Form of such Source Code
21
+ Form, and Modifications of such Source Code Form, in each case
22
+ including portions thereof.
23
+
24
+ 1.5. "Incompatible With Secondary Licenses"
25
+ means
26
+
27
+ (a) that the initial Contributor has attached the notice described
28
+ in Exhibit B to the Covered Software; or
29
+
30
+ (b) that the Covered Software was made available under the terms of
31
+ version 1.1 or earlier of the License, but not also under the
32
+ terms of a Secondary License.
33
+
34
+ 1.6. "Executable Form"
35
+ means any form of the work other than Source Code Form.
36
+
37
+ 1.7. "Larger Work"
38
+ means a work that combines Covered Software with other material, in
39
+ a separate file or files, that is not Covered Software.
40
+
41
+ 1.8. "License"
42
+ means this document.
43
+
44
+ 1.9. "Licensable"
45
+ means having the right to grant, to the maximum extent possible,
46
+ whether at the time of the initial grant or subsequently, any and
47
+ all of the rights conveyed by this License.
48
+
49
+ 1.10. "Modifications"
50
+ means any of the following:
51
+
52
+ (a) any file in Source Code Form that results from an addition to,
53
+ deletion from, or modification of the contents of Covered
54
+ Software; or
55
+
56
+ (b) any new file in Source Code Form that contains any Covered
57
+ Software.
58
+
59
+ 1.11. "Patent Claims" of a Contributor
60
+ means any patent claim(s), including without limitation, method,
61
+ process, and apparatus claims, in any patent Licensable by such
62
+ Contributor that would be infringed, but for the grant of the
63
+ License, by the making, using, selling, offering for sale, having
64
+ made, import, or transfer of either its Contributions or its
65
+ Contributor Version.
66
+
67
+ 1.12. "Secondary License"
68
+ means either the GNU General Public License, Version 2.0, the GNU
69
+ Lesser General Public License, Version 2.1, the GNU Affero General
70
+ Public License, Version 3.0, or any later versions of those
71
+ licenses.
72
+
73
+ 1.13. "Source Code Form"
74
+ means the form of the work preferred for making modifications.
75
+
76
+ 1.14. "You" (or "Your")
77
+ means an individual or a legal entity exercising rights under this
78
+ License. For legal entities, "You" includes any entity that
79
+ controls, is controlled by, or is under common control with You. For
80
+ purposes of this definition, "control" means (a) the power, direct
81
+ or indirect, to cause the direction or management of such entity,
82
+ whether by contract or otherwise, or (b) ownership of more than
83
+ fifty percent (50%) of the outstanding shares or beneficial
84
+ ownership of such entity.
85
+
86
+ 2. License Grants and Conditions
87
+ --------------------------------
88
+
89
+ 2.1. Grants
90
+
91
+ Each Contributor hereby grants You a world-wide, royalty-free,
92
+ non-exclusive license:
93
+
94
+ (a) under intellectual property rights (other than patent or trademark)
95
+ Licensable by such Contributor to use, reproduce, make available,
96
+ modify, display, perform, distribute, and otherwise exploit its
97
+ Contributions, either on an unmodified basis, with Modifications, or
98
+ as part of a Larger Work; and
99
+
100
+ (b) under Patent Claims of such Contributor to make, use, sell, offer
101
+ for sale, have made, import, and otherwise transfer either its
102
+ Contributions or its Contributor Version.
103
+
104
+ 2.2. Effective Date
105
+
106
+ The licenses granted in Section 2.1 with respect to any Contribution
107
+ become effective for each Contribution on the date the Contributor first
108
+ distributes such Contribution.
109
+
110
+ 2.3. Limitations on Grant Scope
111
+
112
+ The licenses granted in this Section 2 are the only rights granted under
113
+ this License. No additional rights or licenses will be implied from the
114
+ distribution or licensing of Covered Software under this License.
115
+ Notwithstanding Section 2.1(b) above, no patent license is granted by a
116
+ Contributor:
117
+
118
+ (a) for any code that a Contributor has removed from Covered Software;
119
+ or
120
+
121
+ (b) for infringements caused by: (i) Your and any other third party's
122
+ modifications of Covered Software, or (ii) the combination of its
123
+ Contributions with other software (except as part of its Contributor
124
+ Version); or
125
+
126
+ (c) under Patent Claims infringed by Covered Software in the absence of
127
+ its Contributions.
128
+
129
+ This License does not grant any rights in the trademarks, service marks,
130
+ or logos of any Contributor (except as may be necessary to comply with
131
+ the notice requirements in Section 3.4).
132
+
133
+ 2.4. Subsequent Licenses
134
+
135
+ No Contributor makes additional grants as a result of Your choice to
136
+ distribute the Covered Software under a subsequent version of this
137
+ License (see Section 10.2) or under the terms of a Secondary License (if
138
+ permitted under the terms of Section 3.3).
139
+
140
+ 2.5. Representation
141
+
142
+ Each Contributor represents that the Contributor believes its
143
+ Contributions are its original creation(s) or it has sufficient rights
144
+ to grant the rights to its Contributions conveyed by this License.
145
+
146
+ 2.6. Fair Use
147
+
148
+ This License is not intended to limit any rights You have under
149
+ applicable copyright doctrines of fair use, fair dealing, or other
150
+ equivalents.
151
+
152
+ 2.7. Conditions
153
+
154
+ Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted
155
+ in Section 2.1.
156
+
157
+ 3. Responsibilities
158
+ -------------------
159
+
160
+ 3.1. Distribution of Source Form
161
+
162
+ All distribution of Covered Software in Source Code Form, including any
163
+ Modifications that You create or to which You contribute, must be under
164
+ the terms of this License. You must inform recipients that the Source
165
+ Code Form of the Covered Software is governed by the terms of this
166
+ License, and how they can obtain a copy of this License. You may not
167
+ attempt to alter or restrict the recipients' rights in the Source Code
168
+ Form.
169
+
170
+ 3.2. Distribution of Executable Form
171
+
172
+ If You distribute Covered Software in Executable Form then:
173
+
174
+ (a) such Covered Software must also be made available in Source Code
175
+ Form, as described in Section 3.1, and You must inform recipients of
176
+ the Executable Form how they can obtain a copy of such Source Code
177
+ Form by reasonable means in a timely manner, at a charge no more
178
+ than the cost of distribution to the recipient; and
179
+
180
+ (b) You may distribute such Executable Form under the terms of this
181
+ License, or sublicense it under different terms, provided that the
182
+ license for the Executable Form does not attempt to limit or alter
183
+ the recipients' rights in the Source Code Form under this License.
184
+
185
+ 3.3. Distribution of a Larger Work
186
+
187
+ You may create and distribute a Larger Work under terms of Your choice,
188
+ provided that You also comply with the requirements of this License for
189
+ the Covered Software. If the Larger Work is a combination of Covered
190
+ Software with a work governed by one or more Secondary Licenses, and the
191
+ Covered Software is not Incompatible With Secondary Licenses, this
192
+ License permits You to additionally distribute such Covered Software
193
+ under the terms of such Secondary License(s), so that the recipient of
194
+ the Larger Work may, at their option, further distribute the Covered
195
+ Software under the terms of either this License or such Secondary
196
+ License(s).
197
+
198
+ 3.4. Notices
199
+
200
+ You may not remove or alter the substance of any license notices
201
+ (including copyright notices, patent notices, disclaimers of warranty,
202
+ or limitations of liability) contained within the Source Code Form of
203
+ the Covered Software, except that You may alter any license notices to
204
+ the extent required to remedy known factual inaccuracies.
205
+
206
+ 3.5. Application of Additional Terms
207
+
208
+ You may choose to offer, and to charge a fee for, warranty, support,
209
+ indemnity or liability obligations to one or more recipients of Covered
210
+ Software. However, You may do so only on Your own behalf, and not on
211
+ behalf of any Contributor. You must make it absolutely clear that any
212
+ such warranty, support, indemnity, or liability obligation is offered by
213
+ You alone, and You hereby agree to indemnify every Contributor for any
214
+ liability incurred by such Contributor as a result of warranty, support,
215
+ indemnity or liability terms You offer. You may include additional
216
+ disclaimers of warranty and limitations of liability specific to any
217
+ jurisdiction.
218
+
219
+ 4. Inability to Comply Due to Statute or Regulation
220
+ ---------------------------------------------------
221
+
222
+ If it is impossible for You to comply with any of the terms of this
223
+ License with respect to some or all of the Covered Software due to
224
+ statute, judicial order, or regulation then You must: (a) comply with
225
+ the terms of this License to the maximum extent possible; and (b)
226
+ describe the limitations and the code they affect. Such description must
227
+ be placed in a text file included with all distributions of the Covered
228
+ Software under this License. Except to the extent prohibited by statute
229
+ or regulation, such description must be sufficiently detailed for a
230
+ recipient of ordinary skill to be able to understand it.
231
+
232
+ 5. Termination
233
+ --------------
234
+
235
+ 5.1. The rights granted under this License will terminate automatically
236
+ if You fail to comply with any of its terms. However, if You become
237
+ compliant, then the rights granted under this License from a particular
238
+ Contributor are reinstated (a) provisionally, unless and until such
239
+ Contributor explicitly and finally terminates Your grants, and (b) on an
240
+ ongoing basis, if such Contributor fails to notify You of the
241
+ non-compliance by some reasonable means prior to 60 days after You have
242
+ come back into compliance. Moreover, Your grants from a particular
243
+ Contributor are reinstated on an ongoing basis if such Contributor
244
+ notifies You of the non-compliance by some reasonable means, this is the
245
+ first time You have received notice of non-compliance with this License
246
+ from such Contributor, and You become compliant prior to 30 days after
247
+ Your receipt of the notice.
248
+
249
+ 5.2. If You initiate litigation against any entity by asserting a patent
250
+ infringement claim (excluding declaratory judgment actions,
251
+ counter-claims, and cross-claims) alleging that a Contributor Version
252
+ directly or indirectly infringes any patent, then the rights granted to
253
+ You by any and all Contributors for the Covered Software under Section
254
+ 2.1 of this License shall terminate.
255
+
256
+ 5.3. In the event of termination under Sections 5.1 or 5.2 above, all
257
+ end user license agreements (excluding distributors and resellers) which
258
+ have been validly granted by You or Your distributors under this License
259
+ prior to termination shall survive termination.
260
+
261
+ ************************************************************************
262
+ * *
263
+ * 6. Disclaimer of Warranty *
264
+ * ------------------------- *
265
+ * *
266
+ * Covered Software is provided under this License on an "as is" *
267
+ * basis, without warranty of any kind, either expressed, implied, or *
268
+ * statutory, including, without limitation, warranties that the *
269
+ * Covered Software is free of defects, merchantable, fit for a *
270
+ * particular purpose or non-infringing. The entire risk as to the *
271
+ * quality and performance of the Covered Software is with You. *
272
+ * Should any Covered Software prove defective in any respect, You *
273
+ * (not any Contributor) assume the cost of any necessary servicing, *
274
+ * repair, or correction. This disclaimer of warranty constitutes an *
275
+ * essential part of this License. No use of any Covered Software is *
276
+ * authorized under this License except under this disclaimer. *
277
+ * *
278
+ ************************************************************************
279
+
280
+ ************************************************************************
281
+ * *
282
+ * 7. Limitation of Liability *
283
+ * -------------------------- *
284
+ * *
285
+ * Under no circumstances and under no legal theory, whether tort *
286
+ * (including negligence), contract, or otherwise, shall any *
287
+ * Contributor, or anyone who distributes Covered Software as *
288
+ * permitted above, be liable to You for any direct, indirect, *
289
+ * special, incidental, or consequential damages of any character *
290
+ * including, without limitation, damages for lost profits, loss of *
291
+ * goodwill, work stoppage, computer failure or malfunction, or any *
292
+ * and all other commercial damages or losses, even if such party *
293
+ * shall have been informed of the possibility of such damages. This *
294
+ * limitation of liability shall not apply to liability for death or *
295
+ * personal injury resulting from such party's negligence to the *
296
+ * extent applicable law prohibits such limitation. Some *
297
+ * jurisdictions do not allow the exclusion or limitation of *
298
+ * incidental or consequential damages, so this exclusion and *
299
+ * limitation may not apply to You. *
300
+ * *
301
+ ************************************************************************
302
+
303
+ 8. Litigation
304
+ -------------
305
+
306
+ Any litigation relating to this License may be brought only in the
307
+ courts of a jurisdiction where the defendant maintains its principal
308
+ place of business and such litigation shall be governed by laws of that
309
+ jurisdiction, without reference to its conflict-of-law provisions.
310
+ Nothing in this Section shall prevent a party's ability to bring
311
+ cross-claims or counter-claims.
312
+
313
+ 9. Miscellaneous
314
+ ----------------
315
+
316
+ This License represents the complete agreement concerning the subject
317
+ matter hereof. If any provision of this License is held to be
318
+ unenforceable, such provision shall be reformed only to the extent
319
+ necessary to make it enforceable. Any law or regulation which provides
320
+ that the language of a contract shall be construed against the drafter
321
+ shall not be used to construe this License against a Contributor.
322
+
323
+ 10. Versions of the License
324
+ ---------------------------
325
+
326
+ 10.1. New Versions
327
+
328
+ Mozilla Foundation is the license steward. Except as provided in Section
329
+ 10.3, no one other than the license steward has the right to modify or
330
+ publish new versions of this License. Each version will be given a
331
+ distinguishing version number.
332
+
333
+ 10.2. Effect of New Versions
334
+
335
+ You may distribute the Covered Software under the terms of the version
336
+ of the License under which You originally received the Covered Software,
337
+ or under the terms of any subsequent version published by the license
338
+ steward.
339
+
340
+ 10.3. Modified Versions
341
+
342
+ If you create software not governed by this License, and you want to
343
+ create a new license for such software, you may create and use a
344
+ modified version of this License if you rename the license and remove
345
+ any references to the name of the license steward (except to note that
346
+ such modified license differs from this License).
347
+
348
+ 10.4. Distributing Source Code Form that is Incompatible With Secondary
349
+ Licenses
350
+
351
+ If You choose to distribute Source Code Form that is Incompatible With
352
+ Secondary Licenses under the terms of this version of the License, the
353
+ notice described in Exhibit B of this License must be attached.
354
+
355
+ Exhibit A - Source Code Form License Notice
356
+ -------------------------------------------
357
+
358
+ This Source Code Form is subject to the terms of the Mozilla Public
359
+ License, v. 2.0. If a copy of the MPL was not distributed with this
360
+ file, You can obtain one at http://mozilla.org/MPL/2.0/.
361
+
362
+ If it is not possible or desirable to put the notice in a particular
363
+ file, then You may include the notice in a location (such as a LICENSE
364
+ file in a relevant directory) where a recipient would be likely to look
365
+ for such a notice.
366
+
367
+ You may add additional accurate notices of copyright ownership.
368
+
369
+ Exhibit B - "Incompatible With Secondary Licenses" Notice
370
+ ---------------------------------------------------------
371
+
372
+ This Source Code Form is "Incompatible With Secondary Licenses", as
373
+ defined by the Mozilla Public License, v. 2.0.
data/README.md ADDED
@@ -0,0 +1,14 @@
1
+ # Proxi
2
+
3
+ Local TCP/HTTP proxy
4
+
5
+ This is intended as a development tool, for when you want to see exactly what
6
+ goes over the wire, want to log or save specific requests, responses, simulate
7
+ timeouts or slow connections, etc.
8
+
9
+
10
+ ## License
11
+
12
+ © Arne Brasseur 2015
13
+
14
+ Mozilla Public License Version 2.0. See LICENSE file.
data/lib/proxi.rb ADDED
@@ -0,0 +1,13 @@
1
+ require 'socket'
2
+ require 'thread'
3
+ require 'optparse'
4
+ require 'tmpdir'
5
+ require 'openssl'
6
+
7
+ require 'wisper'
8
+
9
+ module Proxi
10
+ end
11
+
12
+ require_relative 'proxi/server'
13
+ require_relative 'proxi/connection'
@@ -0,0 +1,135 @@
1
+ module Proxi
2
+ class Connection
3
+ include Wisper::Publisher
4
+
5
+ attr_reader :in_socket, :thread, :remote_host, :remote_port
6
+
7
+ def initialize(in_socket, remote_host, remote_port)
8
+ @in_socket = in_socket
9
+ @remote_host, @remote_port = remote_host, remote_port
10
+ @state = :new
11
+ end
12
+
13
+ def call
14
+ broadcast(:start_connection, self)
15
+ @thread = Thread.new { proxy_loop }
16
+ end
17
+
18
+ def alive?
19
+ thread.alive?
20
+ end
21
+
22
+ def join_thread
23
+ thread.join
24
+ end
25
+
26
+ private
27
+
28
+ def out_socket
29
+ @out_socket ||= TCPSocket.new(remote_host, remote_port)
30
+ end
31
+
32
+ def proxy_loop
33
+ begin
34
+ loop do
35
+ begin
36
+ ready_sockets.each do |socket|
37
+ handle_socket(socket)
38
+ end
39
+ rescue EOFError
40
+ break
41
+ end
42
+ end
43
+ rescue Object => e
44
+ broadcast(:main_loop_error, self, e)
45
+ raise
46
+ ensure
47
+ in_socket.close rescue StandardError
48
+ out_socket.close rescue StandardError
49
+ broadcast(:end_connection, self)
50
+ end
51
+ end
52
+
53
+ def ready_sockets
54
+ IO.select([in_socket, out_socket]).first
55
+ end
56
+
57
+ def handle_socket(socket)
58
+ data = socket.readpartial(4096)
59
+
60
+ if socket == in_socket
61
+ broadcast(:data_in, self, data)
62
+ out_socket.write data
63
+ out_socket.flush
64
+ else
65
+ broadcast(:data_out, self, data)
66
+ in_socket.write data
67
+ in_socket.flush
68
+ end
69
+ end
70
+ end
71
+
72
+ module SSLConnection
73
+ def connect
74
+ @out_socket = OpenSSL::SSL::SSLSocket.new(super)
75
+ @out_socket.connect
76
+ end
77
+ end
78
+
79
+ class HTTPConnection < Connection
80
+ def initialize(in_socket, host_to_ip)
81
+ @in_socket, @host_to_ip = in_socket, host_to_ip
82
+ @new = true
83
+ end
84
+
85
+ def handle_socket(socket)
86
+ data = socket.readpartial(4096)
87
+
88
+ if socket == in_socket
89
+ broadcast(:data_in, self, data)
90
+
91
+ if @new
92
+ @first_packet = data
93
+ @new = false
94
+ end
95
+
96
+ out_socket.write data
97
+ out_socket.flush
98
+ else
99
+ broadcast(:data_out, self, data)
100
+ in_socket.write data
101
+ in_socket.flush
102
+ end
103
+ end
104
+
105
+ def ready_sockets
106
+ if @new
107
+ sockets = [in_socket]
108
+ else
109
+ sockets = [in_socket, out_socket]
110
+ end
111
+ IO.select(sockets).first
112
+ end
113
+
114
+ def out_socket
115
+ host, port = @host_to_ip.fetch(headers["host"]).split(':')
116
+ port ||= 80
117
+ @out_socket ||= TCPSocket.new(host, port.to_i)
118
+ end
119
+
120
+ def headers
121
+ Hash[
122
+ @first_packet
123
+ .sub(/\r\n\r\n.*/m, '')
124
+ .each_line
125
+ .drop(1) # GET / HTTP/1.1
126
+ .map do |line|
127
+ k,v = line.split(':', 2)
128
+ [k.downcase, v.strip].tap do |header|
129
+ broadcast(:header, self, *header)
130
+ end
131
+ end
132
+ ]
133
+ end
134
+ end
135
+ end
@@ -0,0 +1,45 @@
1
+ module Proxi
2
+ class Server
3
+ include Wisper::Publisher
4
+
5
+ MAX_CONNECTIONS = 5
6
+
7
+ def initialize(listen_port, connection_factory)
8
+ @connection_factory = connection_factory
9
+
10
+ @server = TCPServer.new(nil, listen_port)
11
+
12
+ @connections = []
13
+ end
14
+
15
+ def call
16
+ loop do
17
+ in_socket = @server.accept
18
+ connection = @connection_factory.call(in_socket)
19
+
20
+ broadcast(:new_connection, connection)
21
+
22
+ @connections.push(connection)
23
+
24
+ connection.call
25
+
26
+ reap_connections
27
+ while @connections.size >= MAX_CONNECTIONS
28
+ sleep 1
29
+ reap_connections
30
+ end
31
+ end
32
+ end
33
+
34
+ def reap_connections
35
+ @connections = @connections.select do |t|
36
+ if t.alive?
37
+ true
38
+ else
39
+ t.join_thread
40
+ false
41
+ end
42
+ end
43
+ end
44
+ end
45
+ end
data/proxi.gemspec ADDED
@@ -0,0 +1,19 @@
1
+ Gem::Specification.new do |gem|
2
+ gem.name = 'proxi'
3
+ gem.version = '0.1'
4
+ gem.authors = [ 'Arne Brasseur' ]
5
+ gem.email = [ 'arne@arnebrasseur.net' ]
6
+ gem.description = 'TCP and HTTP proxy scripts'
7
+ gem.summary = gem.description
8
+ gem.homepage = 'https://github.com/plexus/proxi'
9
+ gem.license = 'MPL'
10
+
11
+ gem.require_paths = %w[lib]
12
+ gem.files = `git ls-files`.split $/
13
+ gem.test_files = gem.files.grep(/^spec/)
14
+ gem.extra_rdoc_files = %w[README.md]
15
+
16
+ gem.add_runtime_dependency 'wisper', '> 0'
17
+
18
+ gem.add_development_dependency 'rspec', '~> 3.0'
19
+ end
data/proxy.rb ADDED
@@ -0,0 +1,178 @@
1
+ #!/usr/bin/ruby
2
+
3
+ require 'socket'
4
+ require 'thread'
5
+ require 'optparse'
6
+ require 'tmpdir'
7
+
8
+ class ProxyServer
9
+ attr_reader :switches
10
+ attr_reader :listen_port, :remote_host, :remote_port
11
+ attr_reader :mutex
12
+ attr_accessor :threads
13
+
14
+ def initialize(listen_port, remote_host, remote_port, switches = [])
15
+ @listen_port, @remote_host, @remote_port = listen_port, remote_host, remote_port
16
+ @server = TCPServer.new(nil, @listen_port)
17
+ @switches = switches
18
+
19
+ @threads = []
20
+ @max_threads = 5
21
+ @count = 0
22
+ @mutex = Mutex.new
23
+
24
+ @count += 1 while File.exist? log_name(@count, :in)
25
+ end
26
+
27
+ def start
28
+ loop do
29
+ # begin
30
+ threads << Thread.new(@server.accept) do |in_socket|
31
+ handle_new_thread(in_socket)
32
+ end
33
+ # rescue Interrupt => e
34
+ # STDERR << e.message
35
+ # end
36
+
37
+ threads.select {|t| not t.alive? }.each do |dead|
38
+ dead[:logfd].values {|f| f.close} if dead[:logfd]
39
+ end
40
+
41
+ self.threads = threads.select { |t| t.alive? ? true : (t.join; false) }
42
+ while threads.size >= @max_threads
43
+ sleep 1
44
+ self.threads = threads.select { |t| t.alive? ? true : (t.join; false) }
45
+ end
46
+ end
47
+ end
48
+
49
+ def handle_new_thread(in_socket)
50
+ @mutex.synchronize { Thread.current[:count] = @count+=1 }
51
+ begin
52
+
53
+ begin
54
+ out_socket = TCPSocket.new(remote_host, remote_port)
55
+ if switches[:s]
56
+ puts "OPENSSL"
57
+ require 'openssl'
58
+ out_socket = OpenSSL::SSL::SSLSocket.new( out_socket )
59
+ out_socket.connect
60
+ end
61
+ rescue Errno::ECONNREFUSED
62
+ in_socket.close
63
+ raise
64
+ end
65
+
66
+ proxy_data(in_socket, out_socket)
67
+
68
+ rescue StandardError => e
69
+ puts "Thread #{Thread.current} got exception #{e.inspect}"
70
+ end
71
+
72
+ in_socket.close rescue StandardError
73
+ out_socket.close rescue StandardError
74
+ end
75
+
76
+ def proxy_data(in_socket, out_socket)
77
+ loop do
78
+ (ready_sockets, dummy, dummy) = IO.select([in_socket, out_socket])
79
+ begin
80
+ if switches[:wait]
81
+ sleep switches[:wait]
82
+ end
83
+ ready_sockets.each do |socket|
84
+ data = socket.readpartial(4096)
85
+ if socket == in_socket
86
+ if switches[:vhost]
87
+ vhost = remote_port.to_i == 80 ? switches[:vhost] : "#{switches[:vhost]}:#{remote_port}"
88
+ data.gsub!(/^Host: [^\r]+/, "Host: #{vhost}")
89
+ end
90
+ log(:in, data)
91
+ out_socket.write data
92
+ out_socket.flush
93
+ else
94
+ log(:out, data)
95
+ in_socket.write data
96
+ in_socket.flush
97
+ end
98
+ end
99
+ rescue EOFError
100
+ break
101
+ end
102
+ end
103
+ end
104
+
105
+ def name
106
+ switches[:name] || 'proxy'
107
+ end
108
+
109
+ def dirname
110
+ switches[:dir] || Dir.tmpdir
111
+ end
112
+
113
+ def log_name(num, suffix)
114
+ '%s/%s.%d.%d.%s' % [dirname, name, listen_port, num, suffix]
115
+ end
116
+
117
+ def log(in_out, str)
118
+ if switches[:m] || switches[:mm]
119
+ if in_out == :in
120
+ printf "Thread # %d : input %d bytes\n" % [Thread.current[:count], str.size]
121
+ Thread.current[:starttime] ||= Time.now
122
+ else
123
+ if Thread.current[:gotdata]
124
+ if switches[:mm]
125
+ printf "Thread # %d : continued %.2fs , %d bytes\n" % [Thread.current[:count], Time.now - Thread.current[:starttime], str.size]
126
+ end
127
+ else
128
+ Thread.current[:gotdata] = true
129
+ printf "Thread # %d : reply started %.2fs , %d bytes\n" % [Thread.current[:count], Time.now - Thread.current[:starttime], str.size]
130
+ end
131
+ end
132
+ end
133
+ if switches[:l]
134
+ unless Thread.current[:logfd]
135
+ num = Thread.current[:count]
136
+ Thread.current[:logfd] = {
137
+ :in => File.open(log_name(num, :in), 'w'),
138
+ :out => File.open(log_name(num, :out), 'w')
139
+ }
140
+ end
141
+ Thread.current[:logfd][in_out] << str
142
+ Thread.current[:logfd][in_out].flush
143
+ end
144
+ end
145
+ end
146
+
147
+
148
+
149
+ switches = {}
150
+
151
+ optparse = OptionParser.new do |opts|
152
+ opts.banner = "#{File.basename($0)} [options] <src port> <host> <dst port>"
153
+ opts.on('-w', '--wait SECONDS', 'wait after accepting the connection') { |sec| switches[:wait] = Integer( sec ) }
154
+ opts.on('-l', '--log' , 'log to /tmp') { switches[:l] = true }
155
+ opts.on('-v', '--vhost HOST' , 'rewrite HTTP "Host:" header, useful when the server uses virtual hosts') { |host| switches[:vhost] = host }
156
+ opts.on('-n', '--name NAME' , 'specify an alternative prefix for the log files (default "proxy")') { |name| switches[:name] = name }
157
+ opts.on('-m', '--timing' , 'print time and size information to stdout') { switches[:m] = true }
158
+ opts.on('-x', '--xtiming' , 'give more elaborate timing information on stdout') { switches[:mm] = true }
159
+ opts.on('-s', '--ssl' , 'connect to HTTPS, i.e. do SSL socket termination"') { switches[:s] = true }
160
+ opts.on('-d', '--dir DIR' , 'Directory to place the logs, defaults to the temp dir') { |dir| switches[:dir] = dir }
161
+ opts.on('-o', '--hostmap ip:vhost,ip:vhost', 'provide a mapper from ip to hostname') { |m| switches[:hostmap] = Hash[m.split(',').map{|o| o.split(':').reverse}] }
162
+ opts.on('-h', '--help' , 'Print this help message') { puts opts ; exit}
163
+ end
164
+
165
+ optparse.parse! ARGV
166
+
167
+ p ARGV
168
+
169
+ unless ARGV.count == 3
170
+ puts optparse
171
+ exit
172
+ end
173
+
174
+ p switches
175
+
176
+ listen_port, remote_host, remote_port = *ARGV
177
+
178
+ ProxyServer.new(listen_port, remote_host, remote_port, switches).start
metadata ADDED
@@ -0,0 +1,84 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: proxi
3
+ version: !ruby/object:Gem::Version
4
+ version: '0.1'
5
+ platform: ruby
6
+ authors:
7
+ - Arne Brasseur
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2015-10-24 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: wisper
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - ">"
18
+ - !ruby/object:Gem::Version
19
+ version: '0'
20
+ type: :runtime
21
+ prerelease: false
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - ">"
25
+ - !ruby/object:Gem::Version
26
+ version: '0'
27
+ - !ruby/object:Gem::Dependency
28
+ name: rspec
29
+ requirement: !ruby/object:Gem::Requirement
30
+ requirements:
31
+ - - "~>"
32
+ - !ruby/object:Gem::Version
33
+ version: '3.0'
34
+ type: :development
35
+ prerelease: false
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - "~>"
39
+ - !ruby/object:Gem::Version
40
+ version: '3.0'
41
+ description: TCP and HTTP proxy scripts
42
+ email:
43
+ - arne@arnebrasseur.net
44
+ executables: []
45
+ extensions: []
46
+ extra_rdoc_files:
47
+ - README.md
48
+ files:
49
+ - ".gitignore"
50
+ - Gemfile
51
+ - Gemfile.lock
52
+ - LICENSE
53
+ - README.md
54
+ - lib/proxi.rb
55
+ - lib/proxi/connection.rb
56
+ - lib/proxi/server.rb
57
+ - proxi.gemspec
58
+ - proxy.rb
59
+ homepage: https://github.com/plexus/proxi
60
+ licenses:
61
+ - MPL
62
+ metadata: {}
63
+ post_install_message:
64
+ rdoc_options: []
65
+ require_paths:
66
+ - lib
67
+ required_ruby_version: !ruby/object:Gem::Requirement
68
+ requirements:
69
+ - - ">="
70
+ - !ruby/object:Gem::Version
71
+ version: '0'
72
+ required_rubygems_version: !ruby/object:Gem::Requirement
73
+ requirements:
74
+ - - ">="
75
+ - !ruby/object:Gem::Version
76
+ version: '0'
77
+ requirements: []
78
+ rubyforge_project:
79
+ rubygems_version: 2.2.3
80
+ signing_key:
81
+ specification_version: 4
82
+ summary: TCP and HTTP proxy scripts
83
+ test_files: []
84
+ has_rdoc: