simgear/io/HTTPFileRequest.cxx

   1 // HTTP request writing response to a file.
   2 //
   3 // Copyright (C) 2013  Thomas Geymayer <tomgey@gmail.com>
   4 //
   5 // This library is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU Library General Public
   7 // License as published by the Free Software Foundation; either
   8 // version 2 of the License, or (at your option) any later version.
   9 //
  10 // This library is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 // Library General Public License for more details.
  14 //
  15 // You should have received a copy of the GNU Library General Public
  16 // License along with this library; if not, write to the Free Software
  17 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301, USA
  18
  19 #include "HTTPFileRequest.hxx"
  20 #include <simgear/debug/logstream.hxx>
  21
  22 namespace simgear
  23 {
  24 namespace HTTP
  25 {
  26
  27   //----------------------------------------------------------------------------
  28   FileRequest::FileRequest(const std::string& url, const std::string& path):
  29     Request(url, "GET"),
  30     _filename(path)
  31   {
  32
  33   }
  34
  35   //----------------------------------------------------------------------------
  36   void FileRequest::responseHeadersComplete()
  37   {
  38     Request::responseHeadersComplete();
  39
  40     if( !_filename.empty() )
  41       // TODO validate path? (would require to expose fgValidatePath somehow to
  42       //      simgear)
  43       _file.open(_filename.c_str(), std::ios::binary | std::ios::trunc);
  44
  45     if( !_file )
  46     {
  47       SG_LOG
  48       (
  49         SG_IO,
  50         SG_WARN,
  51         "HTTP::FileRequest: failed to open file '" << _filename << "'"
  52       );
  53
  54       abort("Failed to open file.");
  55     }
  56   }
  57
  58   //----------------------------------------------------------------------------
  59   void FileRequest::gotBodyData(const char* s, int n)
  60   {
  61     if( !_file )
  62     {
  63       SG_LOG
  64       (
  65         SG_IO,
  66         SG_WARN,
  67         "HTTP::FileRequest: error writing to '" << _filename << "'"
  68       );
  69       return;
  70     }
  71
  72     _file.write(s, n);
  73   }
  74
  75   //----------------------------------------------------------------------------
  76   void FileRequest::onAlways()
  77   {
  78     _file.close();
  79   }
  80
  81 } // namespace HTTP
  82 } // namespace simgear