An alternative to UBB.threads
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
FLocal/IISUploadHandler/UploadHandler.cs

79 lines
2.8 KiB

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Web;
using System.IO;
using FLocal.Core;
namespace FLocal.IISUploadHandler {
class UploadHandler : IHttpHandler {
public bool IsReusable {
get {
return true;
}
}
public void ProcessRequest(HttpContext httpcontext) {
if(httpcontext.Request.Path.ToLower() == "/upload/") {
this.ProcessUpload(httpcontext);
} else if(httpcontext.Request.Path.ToLower().StartsWith("/data/")) {
this.ProcessRetrieve(httpcontext);
} else {
throw new HttpException(403, "wrong url");
}
}
private static string getFilePath(string md5, string extension) {
foreach(char chr in (md5 + extension)) {
if(!Char.IsLetterOrDigit(chr)) throw new HttpException(403, "wrong md5 or extension");
}
return Config.instance.storageDir + md5.PHPSubstring(0, 2) + Path.DirectorySeparatorChar + md5.PHPSubstring(2, 2) + Path.DirectorySeparatorChar + md5.PHPSubstring(4) + "." + extension;
}
private static void CreateDirectoryIfNotExists(DirectoryInfo directoryInfo) {
if(!directoryInfo.Exists) {
CreateDirectoryIfNotExists(directoryInfo.Parent);
directoryInfo.Create();
}
}
private void ProcessUpload(HttpContext context) {
byte[] data = new byte[context.Request.InputStream.Length];
if(context.Request.InputStream.Read(data, 0, (int)context.Request.InputStream.Length) != context.Request.InputStream.Length) {
throw new FLocalException("File is not uploaded correctly");
}
string file_md5 = Util.md5(context.Request.InputStream);
string md5 = Util.md5(file_md5 + " " + Config.instance.salt);
if(md5 != context.Request.QueryString["signature"]) {
throw new HttpException(403, "signature mismatch");
}
string filePath = getFilePath(file_md5, context.Request.QueryString["extension"]);
CreateDirectoryIfNotExists((new FileInfo(filePath)).Directory);
using(FileStream stream = new FileStream(filePath, FileMode.CreateNew)) {
stream.Write(data, 0, data.Length);
}
context.Response.Write("OK");
}
private void ProcessRetrieve(HttpContext context) {
string[] requestParts = context.Request.Path.Split(new char[] { '/' }, StringSplitOptions.RemoveEmptyEntries);
if(requestParts.Length != 2) throw new HttpException(403, "wrong url");
string[] fileParts = requestParts[1].Split(new char[] { '.' }, StringSplitOptions.RemoveEmptyEntries);
if(fileParts.Length != 2) throw new HttpException(403, "wrong url");
string mime = Util.getMimeByExtension(fileParts[1]);
if(mime == null) throw new FLocalException("unknown extension '" + fileParts[1] + "'");
context.Response.ContentType = mime;
context.Response.TransmitFile(getFilePath(fileParts[0], fileParts[1]));
}
}
}