using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.IO; using Web.Core; using FLocal.Common.dataobjects; using Web.Core.DB; using Web.Core.DB.conditions; using System.Net; namespace FLocal.Common { public static class UploadManager { private class _AlreadyUploadedException : FLocalException { public _AlreadyUploadedException() : base("Already uploaded") { } } public class AlreadyUploadedException : FLocalException { public readonly int uploadId; public AlreadyUploadedException(int uploadId) : base("Already uploaded " + uploadId) { this.uploadId = uploadId; } } private const int MAX_UPLOAD_FILESIZE = 1024*1024; private static HashSet allowedExtensions = new HashSet() { "jpg", "jpe", "jpeg", "jfif", "jif", "svg", "gif", "png", }; public static Upload UploadFile(Stream fileStream, string filename, DateTime uploadDate, User uploader, int? id) { string extension = filename.Split('.').Last().ToLower(); if(!allowedExtensions.Contains(extension)) throw new FLocalException("Unsupported extension '" + extension + "'"); if(fileStream.Length > MAX_UPLOAD_FILESIZE) throw new FLocalException("File is too big"); byte[] data = new byte[fileStream.Length]; int readBytes = fileStream.Read(data, 0, (int)fileStream.Length); if(readBytes != fileStream.Length) { throw new FLocalException("File is incomplete (read " + readBytes + " of " + fileStream.Length + ")"); } string file_md5 = Util.md5(fileStream); AbstractCondition condition = new ComparisonCondition( Upload.TableSpec.instance.getColumnSpec(Upload.TableSpec.FIELD_HASH), ComparisonType.EQUAL, file_md5 ); int? uploadId = null; try { if(Config.instance.mainConnection.GetCountByConditions(Upload.TableSpec.instance, condition) > 0) { throw new _AlreadyUploadedException(); } Config.Transactional(transaction => { Config.instance.mainConnection.lockTable(transaction, Upload.TableSpec.instance); /*if(Config.instance.mainConnection.GetCountByConditions(Upload.TableSpec.instance, condition, new JoinSpec[0]) > 0) { throw new _AlreadyUploadedException(); }*/ //TODO: ??? HttpWebRequest request = (HttpWebRequest)WebRequest.Create(Config.instance.UploaderUrl + "Upload/?extension=" + extension + "&signature=" + Util.md5(file_md5 + " " + Config.instance.SaltUploader)); request.Method = "POST"; request.ContentLength = data.Length; Stream requestStream = request.GetRequestStream(); requestStream.Write(data, 0, data.Length); requestStream.Close(); HttpWebResponse response; try { response = (HttpWebResponse)request.GetResponse(); } catch(WebException e) { response = (HttpWebResponse)e.Response; } using(StreamReader reader = new StreamReader(response.GetResponseStream())) { string result = reader.ReadToEnd(); if(result != "OK") { throw new CriticalException("Cannot upload file to upload service: " + result); } } Dictionary row = new Dictionary(); if(id.HasValue) row[Upload.TableSpec.FIELD_ID] = id.ToString(); row[Upload.TableSpec.FIELD_HASH] = file_md5; row[Upload.TableSpec.FIELD_EXTENSION] = extension; row[Upload.TableSpec.FIELD_UPLOADDATE] = uploadDate.ToUTCString(); row[Upload.TableSpec.FIELD_USERID] = uploader.id.ToString(); row[Upload.TableSpec.FIELD_SIZE] = data.Length.ToString(); row[Upload.TableSpec.FIELD_FILENAME] = filename; uploadId = int.Parse(Config.instance.mainConnection.insert(transaction, Upload.TableSpec.instance, row)); }); } catch(_AlreadyUploadedException) { throw new AlreadyUploadedException( int.Parse( Config.instance.mainConnection.LoadIdsByConditions( Upload.TableSpec.instance, condition, Diapasone.unlimited ).Single() ) ); } return Upload.LoadById(uploadId.Value); } public static Upload SafeUploadFile(Stream fileStream, string filename, User uploader) { Upload upload; try { upload = UploadManager.UploadFile(fileStream, filename, DateTime.Now, uploader, null); } catch(UploadManager.AlreadyUploadedException e) { upload = Upload.LoadById(e.uploadId); } return upload; } public static void WriteUpload(Upload upload, Stream output) { HttpWebRequest request = (HttpWebRequest)WebRequest.Create(Config.instance.UploaderUrl + "Data/" + upload.hash + "." + upload.extension); request.Method = "GET"; HttpWebResponse response = null; try { try { response = (HttpWebResponse)request.GetResponse(); } catch(WebException e) { response = (HttpWebResponse)e.Response; if(response == null) { throw; } } using(Stream responseStream = response.GetResponseStream()) { responseStream.WriteTo(output); } } finally { if(response != null) { response.Close(); } } } } }