Files
2021-12-20 17:17:49 +08:00

201 lines
5.1 KiB
C#

using System;
using System.Collections.Generic;
using System.Collections.Specialized;
using System.IO;
using System.Linq;
using System.Net;
using System.Reflection;
using System.Text;
using Getf.Service.Transfer.Client.WinService.Entities;
using Getf.Service.Transfer.Client.WinService.Helpers;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
namespace Getf.Service.Transfer.Client.WinService.HttpHandler.HttpByte
{
public class HttpByteHandler : IDataHandler
{
LogHelper _LogHelper;
public TransInfo HandlerData(TransInfo srcData)
{
var r = new TransInfo();
var action = srcData.Body.Action;
if (String.IsNullOrWhiteSpace(action))
{
r.TransResultInfo = GetResult(-320, "Action(url|method)参数不能为空");
return r;
}
var url = action;
var jsonStr = String.IsNullOrWhiteSpace(srcData.Body.Param) ? "{}" : srcData.Body.Param;
var jObject = JsonConvert.DeserializeObject<JObject>(jsonStr);
var rData = DoRequest(url, jObject, srcData.Data, out string msg, out int htttpStatusCode);
if (rData == null)
{
r.TransResultInfo = new TransResult()
{
Code = htttpStatusCode,
Message = msg
};
return r;
}
r.TransResultInfo = new TransResult()
{
Code = htttpStatusCode,
Message = msg
};
r.Data = rData;
return r;
}
private TransResult GetResult(int code, string message)
{
return new TransResult()
{
Code = code,
Message = message
};
}
public byte[] DoRequest(string url, JObject jObject, byte[] data, out string msg, out int httpStatusCode)
{
try
{
Encoding encoding = new UTF8Encoding(false);
if (jObject["Encoding"] != null)
{
encoding = Encoding.GetEncoding(jObject["Encoding"].Value<string>());
}
var method = "POST";
if (jObject["Method"] != null)
{
method = jObject["Method"].Value<string>();
}
HttpWebRequest httpWebRequest = WebRequest.Create(url) as HttpWebRequest;
var bodyIndex = GetBodyIndex(data);
var headStr = encoding.GetString(data.Take(bodyIndex).ToArray());
var bodyData = data.Skip(bodyIndex)/*.Select(m => (char)m)*/.ToArray();
SetHead(httpWebRequest, method, headStr);
byte[] r;
if (!"GET".Equals(method, StringComparison.OrdinalIgnoreCase))
{
var requestStream = httpWebRequest.GetRequestStream();
requestStream.Write(bodyData, 0, bodyData.Length);
r = GetResponse(httpWebRequest, encoding);
requestStream.Close();
requestStream.Dispose();
}
else
{
r = GetResponse(httpWebRequest, encoding);
}
msg = String.Empty;
httpStatusCode = 200;
return r;
}
catch (WebException e)
{
using (var stream = e?.Response?.GetResponseStream())
{
httpStatusCode = (int?)((System.Net.HttpWebResponse)(e?.Response))?.StatusCode ?? 500;
if (stream != null)
{
using (StreamReader streamReader = new StreamReader(stream))
{
msg = streamReader.ReadToEnd();
}
}
else
{
msg = e.Message;
}
}
}
catch (Exception e)
{
httpStatusCode = 500;
msg = e.Message;
}
return null;
}
private int GetBodyIndex(byte[] data)
{
var index = -1;
for (int i = 0, c = data.Length; i < c; i++)
{
var elem = data[i];
if (elem == '\r' && i + 3 < data.Length && data[i + 1] == '\n' && data[i + 2] == '\r' && data[i + 3] == '\n')
{
index = i + 4;
break;
}
}
return index;
}
private static PropertyInfo InnerCollectionProperty = typeof(WebHeaderCollection).GetProperty("InnerCollection",
System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.NonPublic);
public static void SetHeaderValue(WebHeaderCollection header, string name, string value)
{
var collection = InnerCollectionProperty.GetValue(header, null) as NameValueCollection;
collection[name] = value;
}
private void SetHead(HttpWebRequest httpWebRequest, string method, string headStr)
{
httpWebRequest.Method = method;
var headStrSplitResult = headStr.Split(new String[] { "\r\n" }, StringSplitOptions.RemoveEmptyEntries);
foreach (var elem in headStrSplitResult)
{
var keyValue = elem.Split(':');
var name = keyValue[0];
var value = keyValue[1];
if ("Accept-Encoding".Equals(name, StringComparison.OrdinalIgnoreCase))
{
httpWebRequest.AutomaticDecompression = DecompressionMethods.GZip;
}
SetHeaderValue(httpWebRequest.Headers, name, value);
}
/*if ("post".Equals(method, StringComparison.OrdinalIgnoreCase))
{
if (String.IsNullOrWhiteSpace(httpWebRequest.ContentType))
{
httpWebRequest.ContentType = "application/x-www-form-urlencoded";
}
}*/
}
private byte[] GetResponse(HttpWebRequest httpWebRequest, Encoding encoding)
{
var response = httpWebRequest.GetResponse();
response.Headers.ToByteArray();
using (var stream = response.GetResponseStream())
{
using (MemoryStream ms = new MemoryStream())
{
stream.CopyTo(ms);
var r = new List<byte>();
r.AddRange(response.Headers.ToByteArray());
//r.AddRange(encoding.GetBytes("\r\n"));
r.AddRange(ms.ToArray());
return r.ToArray();
}
}
}
}
}