proxi 0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
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: