diff options
Diffstat (limited to 'Sockets/HttpClientSocket.cpp')
| -rw-r--r-- | Sockets/HttpClientSocket.cpp | 277 |
1 files changed, 277 insertions, 0 deletions
diff --git a/Sockets/HttpClientSocket.cpp b/Sockets/HttpClientSocket.cpp new file mode 100644 index 0000000..c6f43e4 --- /dev/null +++ b/Sockets/HttpClientSocket.cpp @@ -0,0 +1,277 @@ +/** + ** \file HttpClientSocket.cpp + ** \date 2006-04-20 + ** \author grymse@alhem.net +**/ +/* +Copyright (C) 2007 Anders Hedstrom + +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 2 +of the License, or (at your option) any later version. + +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, write to the Free Software +Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. +*/ +#ifdef _MSC_VER +#pragma warning(disable:4786) +#endif +#include "HttpClientSocket.h" +#include "StdLog.h" +#include "ISocketHandler.h" +#include "Utility.h" + + +#ifdef SOCKETS_NAMESPACE +namespace SOCKETS_NAMESPACE { +#endif + + +HttpClientSocket::HttpClientSocket(ISocketHandler& h) +:HTTPSocket(h) +,m_data_ptr(NULL) +,m_data_size(0) +,m_content_length(0) +,m_data_ptr_set(false) +,m_fil(NULL) +,m_content_ptr(0) +,m_b_complete(false) +,m_b_close_when_complete(false) +{ +} + + +HttpClientSocket::HttpClientSocket(ISocketHandler& h,const std::string& url_in) +:HTTPSocket(h) +,m_data_ptr(NULL) +,m_data_size(0) +,m_content_length(0) +,m_data_ptr_set(false) +,m_fil(NULL) +,m_content_ptr(0) +,m_b_complete(false) +,m_b_close_when_complete(false) +{ + std::string url; + url_this(url_in, m_protocol, m_host, m_port, url, m_url_filename); + SetUrl( url ); +} + + +HttpClientSocket::~HttpClientSocket() +{ + if (m_data_ptr && !m_data_ptr_set) + { + delete[] m_data_ptr; + } + if (m_fil) + { + fclose(m_fil); + } +} + + +void HttpClientSocket::OnFirst() +{ + if (!IsResponse()) + { + Handler().LogError(this, "OnFirst", 0, "Response expected but not received - aborting", LOG_LEVEL_FATAL); + SetCloseAndDelete(); + } + m_content = GetHttpVersion() + " " + GetStatus() + " " + GetStatusText() + "\r\n"; +} + + +void HttpClientSocket::OnHeader(const std::string& key,const std::string& value) +{ + m_content += key + ": " + value + "\r\n"; + if (!strcasecmp(key.c_str(), "content-length")) + { + m_content_length = atoi(value.c_str()); + } + else + if (!strcasecmp(key.c_str(), "content-type")) + { + m_content_type = value; + } +} + + +void HttpClientSocket::OnHeaderComplete() +{ + if (m_filename.size()) + { + m_fil = fopen(m_filename.c_str(), "wb"); + } + else + if (!m_data_ptr && m_content_length) + { + m_data_ptr = new unsigned char[m_content_length]; + m_data_size = m_content_length; + } +} + + +void HttpClientSocket::OnData(const char *buf,size_t len) +{ + if (m_fil) + { + fwrite(buf, 1, len, m_fil); + } + else + if (m_data_ptr) + { + if (m_content_ptr + len > m_data_size) + { + Handler().LogError(this, "OnData", -1, "content buffer overflow", LOG_LEVEL_ERROR); + } + else + { + memcpy(m_data_ptr + m_content_ptr, buf, len); + } + } + m_content_ptr += len; + if (m_content_ptr == m_content_length && m_content_length) + { + if (m_fil) + { + fclose(m_fil); + m_fil = NULL; + } + m_b_complete = true; + OnContent(); + if (m_b_close_when_complete) + { + SetCloseAndDelete(); + } + } +} + + +void HttpClientSocket::OnDelete() +{ + if (!m_b_complete) + { + if (m_fil) + { + fclose(m_fil); + m_fil = NULL; + } + m_b_complete = true; + OnContent(); + } +} + + +void HttpClientSocket::SetFilename(const std::string& x) +{ + m_filename = x; +} + + +void HttpClientSocket::SetDataPtr(unsigned char *buf,size_t len) +{ + m_data_ptr = buf; + m_data_size = len; + m_data_ptr_set = true; +} + + +const std::string& HttpClientSocket::GetContent() +{ + return m_content; +} + + +size_t HttpClientSocket::GetContentLength() +{ + return m_content_length; +} + + +size_t HttpClientSocket::GetContentPtr() +{ + return m_content_ptr; +} + + +size_t HttpClientSocket::GetPos() +{ + return m_content_ptr; +} + + +bool HttpClientSocket::Complete() +{ + return m_b_complete; +} + + +const unsigned char *HttpClientSocket::GetDataPtr() const +{ + return m_data_ptr; +} + + +void HttpClientSocket::OnContent() +{ +} + + +void HttpClientSocket::SetCloseOnComplete(bool x) +{ + m_b_close_when_complete = x; +} + + +const std::string& HttpClientSocket::GetUrlProtocol() +{ + return m_protocol; +} + + +const std::string& HttpClientSocket::GetUrlHost() +{ + return m_host; +} + + +port_t HttpClientSocket::GetUrlPort() +{ + return m_port; +} + + +const std::string& HttpClientSocket::GetUrlFilename() +{ + return m_url_filename; +} + + +const std::string& HttpClientSocket::GetContentType() +{ + return m_content_type; +} + + +void HttpClientSocket::Url(const std::string& url_in,std::string& host,port_t& port) +{ + std::string url; + url_this(url_in, m_protocol, m_host, m_port, url, m_url_filename); + SetUrl(url); + host = GetUrlHost(); + port = GetUrlPort(); +} + + +#ifdef SOCKETS_NAMESPACE +} // namespace SOCKETS_NAMESPACE { +#endif + |