CodeMirror/ShareX
1
0
Fork 0
mirror of https://github.com/ShareX/ShareX.git synced 2024-06-30 20:20:34 +12:00
Code Issues Packages Projects Releases Wiki Activity

Don't restrict custom uploaders with http method

Browse source
This commit is contained in:
Jaex 2018-11-11 11:51:44 +03:00
parent f1b46519d4
commit 4169dfe64a
6 changed files with 21 additions and 47 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ShareX.UploadersLib/BaseUploaders/Uploader.cs
View file

@ -170,7 +170,7 @@ public virtual void StopUpload()
}
protected string SendRequestMultiPart(string url, Dictionary<string, string> args, NameValueCollection headers = null, CookieCollection cookies = null,
ResponseType responseType = ResponseType.Text)
ResponseType responseType = ResponseType.Text, HttpMethod method = HttpMethod.POST)
{
string boundary = UploadHelpers.CreateBoundary();
string contentType = UploadHelpers.ContentTypeMultipartFormData + "; boundary=" + boundary;
@ -180,7 +180,7 @@ public virtual void StopUpload()
{
stream.Write(data, 0, data.Length);
using (HttpWebResponse webResponse = GetResponse(HttpMethod.POST, url, stream, contentType, null, headers, cookies))
using (HttpWebResponse webResponse = GetResponse(method, url, stream, contentType, null, headers, cookies))
{
string response = UploadHelpers.ResponseToString(webResponse, responseType);

7
ShareX.UploadersLib/FileUploaders/CustomFileUploader.cs
View file

@ -80,15 +80,10 @@ public CustomFileUploader(CustomUploaderItem customUploaderItem)
public override UploadResult Upload(Stream stream, string fileName)
{
if (customUploader.RequestType != CustomUploaderRequestMethod.POST)
{
throw new Exception("'Request type' must be 'POST' when using custom file uploader.");
}
CustomUploaderArgumentInput input = new CustomUploaderArgumentInput(fileName, "");
UploadResult result = SendRequestFile(customUploader.GetRequestURL(), stream, fileName, customUploader.GetFileFormName(),
customUploader.GetArguments(input), customUploader.GetHeaders(input), responseType: customUploader.ResponseType);
customUploader.GetArguments(input), customUploader.GetHeaders(input), null, customUploader.ResponseType, customUploader.GetHttpMethod());
if (result.IsSuccess)
{

7
ShareX.UploadersLib/ImageUploaders/CustomImageUploader.cs
View file

@ -77,15 +77,10 @@ public CustomImageUploader(CustomUploaderItem customUploaderItem)
public override UploadResult Upload(Stream stream, string fileName)
{
if (customUploader.RequestType != CustomUploaderRequestMethod.POST)
{
throw new Exception("'Request type' must be 'POST' when using custom image uploader.");
}
CustomUploaderArgumentInput input = new CustomUploaderArgumentInput(fileName, "");
UploadResult result = SendRequestFile(customUploader.GetRequestURL(), stream, fileName, customUploader.GetFileFormName(),
customUploader.GetArguments(input), customUploader.GetHeaders(input), responseType: customUploader.ResponseType);
customUploader.GetArguments(input), customUploader.GetHeaders(input), null, customUploader.ResponseType, customUploader.GetHttpMethod());
if (result.IsSuccess)
{

14
ShareX.UploadersLib/SharingServices/CustomURLSharingService.cs
View file

@ -74,28 +74,22 @@ public CustomURLSharer(CustomUploaderItem customUploaderItem)
public override UploadResult ShareURL(string url)
{
if (customUploader.RequestType == CustomUploaderRequestMethod.POST && !string.IsNullOrEmpty(customUploader.FileFormName))
throw new Exception("'File form name' cannot be used with custom URL sharing service.");
if ((customUploader.Arguments == null || !customUploader.Arguments.Any(x => x.Value.Contains("$input$"))) &&
(customUploader.Headers == null || !customUploader.Headers.Any(x => x.Value.Contains("$input$"))))
throw new Exception("Atleast one '$input$' required for argument or header value.");
UploadResult result = new UploadResult { URL = url, IsURLExpected = false };
CustomUploaderArgumentInput input = new CustomUploaderArgumentInput("", url);
Dictionary<string, string> args = customUploader.GetArguments(input);
if (customUploader.RequestType == CustomUploaderRequestMethod.POST)
{
result.Response = SendRequestMultiPart(customUploader.GetRequestURL(), args, customUploader.GetHeaders(input),
responseType: customUploader.ResponseType);
result.Response = SendRequestMultiPart(customUploader.GetRequestURL(), customUploader.GetArguments(input), customUploader.GetHeaders(input), null,
customUploader.ResponseType, customUploader.GetHttpMethod());
}
else
{
result.Response = SendRequest(customUploader.GetHttpMethod(), customUploader.GetRequestURL(), args, customUploader.GetHeaders(input),
responseType: customUploader.ResponseType);
result.Response = SendRequest(customUploader.GetHttpMethod(), customUploader.GetRequestURL(), customUploader.GetArguments(input),
customUploader.GetHeaders(input), null, customUploader.ResponseType);
}
return result;

22
ShareX.UploadersLib/TextUploaders/CustomTextUploader.cs
View file

@ -80,39 +80,35 @@ public CustomTextUploader(CustomUploaderItem customUploaderItem)
public override UploadResult UploadText(string text, string fileName)
{
UploadResult result = new UploadResult();
string requestURL = customUploader.GetRequestURL();
if ((customUploader.RequestType != CustomUploaderRequestMethod.POST || string.IsNullOrEmpty(customUploader.FileFormName)) &&
((customUploader.Arguments == null || !customUploader.Arguments.Any(x => x.Value.Contains("$input$"))) &&
(customUploader.Headers == null || !customUploader.Headers.Any(x => x.Value.Contains("$input$")))))
throw new Exception("Atleast one '$input$' required for argument or header value.");
UploadResult result = new UploadResult();
CustomUploaderArgumentInput input = new CustomUploaderArgumentInput(fileName, text);
Dictionary<string, string> args = customUploader.GetArguments(input);
if (customUploader.RequestType == CustomUploaderRequestMethod.POST)
{
if (string.IsNullOrEmpty(customUploader.FileFormName))
{
result.Response = SendRequestMultiPart(requestURL, args, customUploader.GetHeaders(input), responseType: customUploader.ResponseType);
result.Response = SendRequestMultiPart(customUploader.GetRequestURL(), customUploader.GetArguments(input),
customUploader.GetHeaders(input), null, customUploader.ResponseType, customUploader.GetHttpMethod());
}
else
{
byte[] byteArray = Encoding.UTF8.GetBytes(text);
using (MemoryStream stream = new MemoryStream(byteArray))
byte[] bytes = Encoding.UTF8.GetBytes(text);
using (MemoryStream stream = new MemoryStream(bytes))
{
result = SendRequestFile(requestURL, stream, fileName, customUploader.GetFileFormName(), args, customUploader.GetHeaders(input),
responseType: customUploader.ResponseType);
result = SendRequestFile(customUploader.GetRequestURL(), stream, fileName, customUploader.GetFileFormName(),
customUploader.GetArguments(input), customUploader.GetHeaders(input), null, customUploader.ResponseType, customUploader.GetHttpMethod());
}
}
}
else
{
result.Response = SendRequest(customUploader.GetHttpMethod(), requestURL, args, customUploader.GetHeaders(input),
responseType: customUploader.ResponseType);
result.Response = SendRequest(customUploader.GetHttpMethod(), customUploader.GetRequestURL(), customUploader.GetArguments(input),
customUploader.GetHeaders(input), null, customUploader.ResponseType);
}
try

14
ShareX.UploadersLib/URLShorteners/CustomURLShortener.cs
View file

@ -75,28 +75,22 @@ public CustomURLShortener(CustomUploaderItem customUploaderItem)
public override UploadResult ShortenURL(string url)
{
if (customUploader.RequestType == CustomUploaderRequestMethod.POST && !string.IsNullOrEmpty(customUploader.FileFormName))
throw new Exception("'File form name' cannot be used with custom URL shortener.");
if ((customUploader.Arguments == null || !customUploader.Arguments.Any(x => x.Value.Contains("$input$"))) &&
(customUploader.Headers == null || !customUploader.Headers.Any(x => x.Value.Contains("$input$"))))
throw new Exception("Atleast one '$input$' required for argument or header value.");
UploadResult result = new UploadResult { URL = url };
CustomUploaderArgumentInput input = new CustomUploaderArgumentInput("", url);
Dictionary<string, string> args = customUploader.GetArguments(input);
if (customUploader.RequestType == CustomUploaderRequestMethod.POST)
{
result.Response = SendRequestMultiPart(customUploader.GetRequestURL(), args, customUploader.GetHeaders(input),
responseType: customUploader.ResponseType);
result.Response = SendRequestMultiPart(customUploader.GetRequestURL(), customUploader.GetArguments(input), customUploader.GetHeaders(input), null,
customUploader.ResponseType, customUploader.GetHttpMethod());
}
else
{
result.Response = SendRequest(customUploader.GetHttpMethod(), customUploader.GetRequestURL(), args, customUploader.GetHeaders(input),
responseType: customUploader.ResponseType);
result.Response = SendRequest(customUploader.GetHttpMethod(), customUploader.GetRequestURL(), customUploader.GetArguments(input),
customUploader.GetHeaders(input), null, customUploader.ResponseType);
}
try