changeset 208:de640de989b8

HttpFileSender
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 28 Nov 2012 17:00:46 +0100
parents 7f74209ea0f8
children 9960642f0f45
files CMakeLists.txt Core/FileStorage.h Core/HttpServer/FilesystemHttpSender.cpp Core/HttpServer/FilesystemHttpSender.h Core/HttpServer/HttpFileSender.cpp Core/HttpServer/HttpFileSender.h OrthancServer/OrthancRestApi.cpp UnitTests/ServerIndex.cpp
diffstat 8 files changed, 314 insertions(+), 137 deletions(-) [+]
line wrap: on
line diff
--- a/CMakeLists.txt	Wed Nov 28 16:23:11 2012 +0100
+++ b/CMakeLists.txt	Wed Nov 28 17:00:46 2012 +0100
@@ -109,6 +109,8 @@
   Core/HttpServer/HttpHandler.cpp
   Core/HttpServer/HttpOutput.cpp
   Core/HttpServer/MongooseServer.cpp
+  Core/HttpServer/HttpFileSender.cpp
+  Core/HttpServer/FilesystemHttpSender.cpp
   Core/MultiThreading/BagOfRunnablesBySteps.cpp
   Core/PngWriter.cpp
   Core/SQLite/Connection.cpp
--- a/Core/FileStorage.h	Wed Nov 28 16:23:11 2012 +0100
+++ b/Core/FileStorage.h	Wed Nov 28 17:00:46 2012 +0100
@@ -41,7 +41,9 @@
 {
   class FileStorage : public boost::noncopyable
   {
+    // TODO REMOVE THIS
     friend class HttpOutput;
+    friend class FilesystemHttpSender;
 
   private:
     std::auto_ptr<BufferCompressor> compressor_;
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Core/HttpServer/FilesystemHttpSender.cpp	Wed Nov 28 17:00:46 2012 +0100
@@ -0,0 +1,91 @@
+/**
+ * Orthanc - A Lightweight, RESTful DICOM Store
+ * Copyright (C) 2012 Medical Physics Department, CHU of Liege,
+ * Belgium
+ *
+ * This program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * In addition, as a special exception, the copyright holders of this
+ * program give permission to link the code of its release with the
+ * OpenSSL project's "OpenSSL" library (or with modified versions of it
+ * that use the same license as the "OpenSSL" library), and distribute
+ * the linked executables. You must obey the GNU General Public License
+ * in all respects for all of the code used other than "OpenSSL". If you
+ * modify file(s) with this exception, you may extend this exception to
+ * your version of the file(s), but you are not obligated to do so. If
+ * you do not wish to do so, delete this exception statement from your
+ * version. If you delete this exception statement from all source files
+ * in the program, then also delete it here.
+ * 
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ **/
+
+#include "FilesystemHttpSender.h"
+
+#include "../Toolbox.h"
+
+#include <stdio.h>
+
+namespace Orthanc
+{
+  void FilesystemHttpSender::Setup()
+  {
+    boost::filesystem::path p(path_);
+    SetFilename(p.filename().string());
+    SetContentType(Toolbox::AutodetectMimeType(p.filename().string()));
+  }
+
+  uint64_t FilesystemHttpSender::GetFileSize()
+  {
+    return Toolbox::GetFileSize(path_);
+  }
+
+  bool FilesystemHttpSender::SendData(HttpOutput& output)
+  {
+    FILE* fp = fopen(path_.c_str(), "rb");
+    if (!fp)
+    {
+      return false;
+    }
+
+    std::vector<uint8_t> buffer(1024 * 1024);  // Chunks of 1MB
+
+    for (;;)
+    {
+      size_t nbytes = fread(&buffer[0], 1, buffer.size(), fp);
+      if (nbytes == 0)
+      {
+        break;
+      }
+      else
+      {
+        output.Send(&buffer[0], nbytes);
+      }
+    }
+
+    fclose(fp);
+
+    return true;
+  }
+
+  FilesystemHttpSender::FilesystemHttpSender(const char* path) : path_(path)
+  {
+    Setup();
+  }
+
+  FilesystemHttpSender::FilesystemHttpSender(const FileStorage& storage,
+                                             const std::string& uuid)
+  {
+    path_ = storage.GetPath(uuid).string();
+    Setup();
+  }
+}
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Core/HttpServer/FilesystemHttpSender.h	Wed Nov 28 17:00:46 2012 +0100
@@ -0,0 +1,57 @@
+/**
+ * Orthanc - A Lightweight, RESTful DICOM Store
+ * Copyright (C) 2012 Medical Physics Department, CHU of Liege,
+ * Belgium
+ *
+ * This program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * In addition, as a special exception, the copyright holders of this
+ * program give permission to link the code of its release with the
+ * OpenSSL project's "OpenSSL" library (or with modified versions of it
+ * that use the same license as the "OpenSSL" library), and distribute
+ * the linked executables. You must obey the GNU General Public License
+ * in all respects for all of the code used other than "OpenSSL". If you
+ * modify file(s) with this exception, you may extend this exception to
+ * your version of the file(s), but you are not obligated to do so. If
+ * you do not wish to do so, delete this exception statement from your
+ * version. If you delete this exception statement from all source files
+ * in the program, then also delete it here.
+ * 
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ **/
+
+#pragma once
+
+#include "HttpFileSender.h"
+#include "../FileStorage.h"
+
+namespace Orthanc
+{
+  class FilesystemHttpSender : public HttpFileSender
+  {
+  private:
+    std::string path_;
+
+    void Setup();
+
+  protected:
+    virtual uint64_t GetFileSize();
+
+    virtual bool SendData(HttpOutput& output);
+
+  public:
+    FilesystemHttpSender(const char* path);
+
+    FilesystemHttpSender(const FileStorage& storage,
+                         const std::string& uuid);
+  };
+}
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Core/HttpServer/HttpFileSender.cpp	Wed Nov 28 17:00:46 2012 +0100
@@ -0,0 +1,66 @@
+/**
+ * Orthanc - A Lightweight, RESTful DICOM Store
+ * Copyright (C) 2012 Medical Physics Department, CHU of Liege,
+ * Belgium
+ *
+ * This program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * In addition, as a special exception, the copyright holders of this
+ * program give permission to link the code of its release with the
+ * OpenSSL project's "OpenSSL" library (or with modified versions of it
+ * that use the same license as the "OpenSSL" library), and distribute
+ * the linked executables. You must obey the GNU General Public License
+ * in all respects for all of the code used other than "OpenSSL". If you
+ * modify file(s) with this exception, you may extend this exception to
+ * your version of the file(s), but you are not obligated to do so. If
+ * you do not wish to do so, delete this exception statement from your
+ * version. If you delete this exception statement from all source files
+ * in the program, then also delete it here.
+ * 
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ **/
+
+
+#include "HttpFileSender.h"
+
+#include <boost/lexical_cast.hpp>
+
+namespace Orthanc
+{
+  void HttpFileSender::SendHeader(HttpOutput& output)
+  {
+    std::string header;
+    header += "Content-Length: " + boost::lexical_cast<std::string>(GetFileSize()) + "\r\n";
+
+    if (contentType_.size() > 0)
+    {
+      header += "Content-Type: " + contentType_ + "\r\n";
+    }
+
+    if (filename_.size() > 0)
+    {
+      header += "Content-Disposition: attachment; filename=\"" + filename_ + "\"\r\n";
+    }
+  
+    output.SendCustomOkHeader(header);
+  }
+
+  void HttpFileSender::Send(HttpOutput& output)
+  {
+    SendHeader(output);
+
+    if (!SendData(output))
+    {
+      output.SendHeader(Orthanc_HttpStatus_500_InternalServerError);
+    }
+  }
+}
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Core/HttpServer/HttpFileSender.h	Wed Nov 28 17:00:46 2012 +0100
@@ -0,0 +1,89 @@
+/**
+ * Orthanc - A Lightweight, RESTful DICOM Store
+ * Copyright (C) 2012 Medical Physics Department, CHU of Liege,
+ * Belgium
+ *
+ * This program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * In addition, as a special exception, the copyright holders of this
+ * program give permission to link the code of its release with the
+ * OpenSSL project's "OpenSSL" library (or with modified versions of it
+ * that use the same license as the "OpenSSL" library), and distribute
+ * the linked executables. You must obey the GNU General Public License
+ * in all respects for all of the code used other than "OpenSSL". If you
+ * modify file(s) with this exception, you may extend this exception to
+ * your version of the file(s), but you are not obligated to do so. If
+ * you do not wish to do so, delete this exception statement from your
+ * version. If you delete this exception statement from all source files
+ * in the program, then also delete it here.
+ * 
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ **/
+
+
+#pragma once
+
+#include "HttpOutput.h"
+
+namespace Orthanc
+{
+  class HttpFileSender
+  {
+  private:
+    std::string contentType_;
+    std::string filename_;
+
+    void SendHeader(HttpOutput& output);
+
+  protected:
+    virtual uint64_t GetFileSize() = 0;
+
+    virtual bool SendData(HttpOutput& output) = 0;
+
+  public:
+    virtual ~HttpFileSender()
+    {
+    }
+
+    void ResetContentType()
+    {
+      contentType_.clear();
+    }
+
+    void SetContentType(const std::string& contentType)
+    {
+      contentType_ = contentType;
+    }
+
+    const std::string& GetContentType() const
+    {
+      return contentType_;
+    }
+
+    void ResetFilename()
+    {
+      contentType_.clear();
+    }
+
+    void SetFilename(const std::string& filename)
+    {
+      filename_ = filename;
+    }
+
+    const std::string& GetFilename() const
+    {
+      return filename_;
+    }
+
+    void Send(HttpOutput& output);
+  };
+}
--- a/OrthancServer/OrthancRestApi.cpp	Wed Nov 28 16:23:11 2012 +0100
+++ b/OrthancServer/OrthancRestApi.cpp	Wed Nov 28 17:00:46 2012 +0100
@@ -35,6 +35,7 @@
 #include "OrthancInitialization.h"
 #include "FromDcmtkBridge.h"
 #include "../Core/Uuid.h"
+#include "../Core/HttpServer/FilesystemHttpSender.h"
 
 #include <dcmtk/dcmdata/dcistrmb.h>
 #include <dcmtk/dcmdata/dcfilefo.h>
--- a/UnitTests/ServerIndex.cpp	Wed Nov 28 16:23:11 2012 +0100
+++ b/UnitTests/ServerIndex.cpp	Wed Nov 28 17:00:46 2012 +0100
@@ -5,7 +5,6 @@
 #include <ctype.h>
 #include <glog/logging.h>
 
-
 using namespace Orthanc;
 
 namespace
@@ -261,10 +260,16 @@
 
 
 
+
+#include "../Core/HttpServer/FilesystemHttpSender.h"
+
 #include "../Core/Toolbox.h"
 #include "../Core/HttpServer/HttpOutput.h"
 #include "../Core/HttpServer/HttpHandler.h"
 
+#include "../Core/HttpServer/HttpFileSender.h"
+
+
 namespace Orthanc
 {
   class RestApiPath
@@ -377,131 +382,6 @@
   };
 
 
-  class HttpFileSender
-  {
-  private:
-    std::string contentType_;
-    std::string filename_;
-
-    void SendHeader(HttpOutput& output)
-    {
-      std::string header;
-      header += "Content-Length: " + boost::lexical_cast<std::string>(GetFileSize()) + "\r\n";
-
-      if (contentType_.size() > 0)
-      {
-        header += "Content-Type: " + contentType_ + "\r\n";
-      }
-
-      if (filename_.size() > 0)
-      {
-        header += "Content-Disposition: attachment; filename=\"" + filename_ + "\"\r\n";
-      }
-  
-      output.SendCustomOkHeader(header);
-    }
-
-  protected:
-    virtual uint64_t GetFileSize() = 0;
-
-    virtual bool SendData(HttpOutput& output) = 0;
-
-  public:
-    virtual ~HttpFileSender()
-    {
-    }
-
-    void ResetContentType()
-    {
-      contentType_.clear();
-    }
-
-    void SetContentType(const std::string& contentType)
-    {
-      contentType_ = contentType;
-    }
-
-    const std::string& GetContentType() const
-    {
-      return contentType_;
-    }
-
-    void ResetFilename()
-    {
-      contentType_.clear();
-    }
-
-    void SetFilename(const std::string& filename)
-    {
-      filename_ = filename;
-    }
-
-    const std::string& GetFilename() const
-    {
-      return filename_;
-    }
-
-    void Send(HttpOutput& output)
-    {
-      SendHeader(output);
-
-      if (!SendData(output))
-      {
-        output.SendHeader(Orthanc_HttpStatus_500_InternalServerError);
-      }
-    }
-  };
-
-
-  class FilesystemHttpSender : public HttpFileSender
-  {
-  private:
-    std::string path_;
-
-  protected:
-    virtual uint64_t GetFileSize()
-    {
-      return Toolbox::GetFileSize(path_);
-    }
-
-    virtual bool SendData(HttpOutput& output)
-    {
-      FILE* fp = fopen(path_.c_str(), "rb");
-      if (!fp)
-      {
-        return false;
-      }
-
-      std::vector<uint8_t> buffer(1024 * 1024);  // Chunks of 1MB
-
-      for (;;)
-      {
-        size_t nbytes = fread(&buffer[0], 1, buffer.size(), fp);
-        if (nbytes == 0)
-        {
-          break;
-        }
-        else
-        {
-          output.Send(&buffer[0], nbytes);
-        }
-      }
-
-      fclose(fp);
-
-      return true;
-    }
-
-  public:
-    FilesystemHttpSender(const char* path) : path_(path)
-    {
-      boost::filesystem::path p(path);
-      SetFilename(p.filename().string());
-      SetContentType(Toolbox::AutodetectMimeType(p.filename().string()));
-    }
-  };
-
-
   class RestApiOutput
   {
   private:
@@ -994,14 +874,3 @@
   /*LOG(WARNING) << "REST has started";
     Toolbox::ServerBarrier();*/
 }
-
-
-/**
-
-   output.AnswerBufferWithContentType(s, "application/json");
-   output.AnswerFile(storage_, fileUuid, contentType, filename.c_str());
-   output.Redirect("app/explorer.html");
-   output.SendHeader(Orthanc_HttpStatus_415_UnsupportedMediaType);
-   output.SendMethodNotAllowedError("GET");
-
-**/