Search Site ~Sockets~  C++ Sockets   Tutorial  Classes  License   Java Sockets  ~Examples~  SMTPProxy   PortProxy   ChatServer   Small MUD   ClientServer   HttpGet   DChat   WebServer   UDP   ICMP  ~Contact~ message board  email author

HttpClientSocket.cpp Go to the documentation of this file. /* Copyright (C) 2007-2011 Anders Hedstrom This library is made available under the terms of the GNU GPL, with the additional exemption that compiling, linking, and/or using OpenSSL is allowed. If you would like to use this library in a closed-source application, a separate license agreement is available. For information about the closed-source license agreement for the C++ sockets library, please visit http://www.alhem.net/Sockets/license.html and/or email license@alhem.net. 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 "File.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(ISocketHandler& h,const std::string& host, port_t port, 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; std::string tmp = "http://" + host + ":" + Utility::l2string(port) + url_in; url_this(tmp, 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) { m_fil -> fclose(); } } 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 = new File; if (!m_fil -> fopen(m_filename, "wb")) { delete m_fil; m_fil = NULL; } } 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) { m_fil -> fwrite(buf, 1, len); } 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) { m_fil -> fclose(); delete 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) { m_fil -> fclose(); delete 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

Page, code, and content Copyright (C) 2007 by Anders Hedström

---