forked from X-rus/xNet
-
Notifications
You must be signed in to change notification settings - Fork 1
/
ProxyHandler.cs
88 lines (71 loc) · 2.87 KB
/
ProxyHandler.cs
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
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Net.Http;
namespace Extreme.Net
{
public class ProxyHandler : HttpClientHandler
{
private ProxyClient proxyClient;
public ProxyHandler(ProxyClient proxyClient)
{
this.proxyClient = proxyClient;
}
protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
{
HttpResponseMessage response = null;
HttpResponse httpResponse;
var httpRequest = request.ToHttpRequest();
httpRequest.Proxy = proxyClient;
cancellationToken.ThrowIfCancellationRequested();
if (request.Method == System.Net.Http.HttpMethod.Get)
{
httpResponse = await GetAsync(httpRequest, request);
}
else if(request.Method == System.Net.Http.HttpMethod.Post)
{
httpResponse = await PostAsync(httpRequest, request);
}
else
{
httpResponse = await RawAsync(httpRequest, request);
}
cancellationToken.ThrowIfCancellationRequested();
response = httpResponse.ToHttpResponseMessage();
response.RequestMessage = request;
var stream = httpResponse.ToMemoryStream();
if (stream != null)
{
response.Content = new ProgressStreamContent(stream, CancellationToken.None);
}
return response;
}
private async Task<HttpResponse> GetAsync(HttpRequest request, HttpRequestMessage message)
{
return await request.GetAsync(message.RequestUri.ToString());
}
private async Task<HttpResponse> PostAsync(HttpRequest request, HttpRequestMessage message)
{
var content = await message.Content.ReadAsByteArrayAsync();
return await request.PostAsync(message.RequestUri.ToString(), content);
}
private async Task<HttpResponse> RawAsync(HttpRequest request, HttpRequestMessage message)
{
var method = ConvertMethod(message.Method);
return await request.RawAsync(method, message.RequestUri.ToString());
}
private HttpMethod ConvertMethod(System.Net.Http.HttpMethod netHttpMethod)
{
if (netHttpMethod == System.Net.Http.HttpMethod.Head)
return HttpMethod.HEAD;
if (netHttpMethod == System.Net.Http.HttpMethod.Delete)
return HttpMethod.DELETE;
if (netHttpMethod == System.Net.Http.HttpMethod.Put)
return HttpMethod.PUT;
throw new HttpException($"Method {netHttpMethod} not supported");
}
}
}