diff options
author | Anas Nashif <anas.nashif@intel.com> | 2012-10-30 12:57:26 -0700 |
---|---|---|
committer | Anas Nashif <anas.nashif@intel.com> | 2012-10-30 12:57:26 -0700 |
commit | 1a78a62555be32868418fe52f8e330c9d0f95d5a (patch) | |
tree | d3765a80e7d3b9640ec2e930743630cd6b9fce2b /doc/html/boost_asio/example/windows/transmit_file.cpp | |
download | boost-1a78a62555be32868418fe52f8e330c9d0f95d5a.tar.gz boost-1a78a62555be32868418fe52f8e330c9d0f95d5a.tar.bz2 boost-1a78a62555be32868418fe52f8e330c9d0f95d5a.zip |
Imported Upstream version 1.49.0upstream/1.49.0
Diffstat (limited to 'doc/html/boost_asio/example/windows/transmit_file.cpp')
-rwxr-xr-x | doc/html/boost_asio/example/windows/transmit_file.cpp | 171 |
1 files changed, 171 insertions, 0 deletions
diff --git a/doc/html/boost_asio/example/windows/transmit_file.cpp b/doc/html/boost_asio/example/windows/transmit_file.cpp new file mode 100755 index 0000000000..c1e00da32e --- /dev/null +++ b/doc/html/boost_asio/example/windows/transmit_file.cpp @@ -0,0 +1,171 @@ +// +// transmit_file.cpp +// ~~~~~~~~~~~~~~~~~ +// +// Copyright (c) 2003-2012 Christopher M. Kohlhoff (chris at kohlhoff dot com) +// +// Distributed under the Boost Software License, Version 1.0. (See accompanying +// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) +// + +#include <ctime> +#include <iostream> +#include <string> +#include <boost/bind.hpp> +#include <boost/shared_ptr.hpp> +#include <boost/enable_shared_from_this.hpp> +#include <boost/asio.hpp> + +#if defined(BOOST_ASIO_HAS_WINDOWS_OVERLAPPED_PTR) + +using boost::asio::ip::tcp; +using boost::asio::windows::overlapped_ptr; +using boost::asio::windows::random_access_handle; + +// A wrapper for the TransmitFile overlapped I/O operation. +template <typename Handler> +void transmit_file(tcp::socket& socket, + random_access_handle& file, Handler handler) +{ + // Construct an OVERLAPPED-derived object to contain the handler. + overlapped_ptr overlapped(socket.get_io_service(), handler); + + // Initiate the TransmitFile operation. + BOOL ok = ::TransmitFile(socket.native_handle(), + file.native_handle(), 0, 0, overlapped.get(), 0, 0); + DWORD last_error = ::GetLastError(); + + // Check if the operation completed immediately. + if (!ok && last_error != ERROR_IO_PENDING) + { + // The operation completed immediately, so a completion notification needs + // to be posted. When complete() is called, ownership of the OVERLAPPED- + // derived object passes to the io_service. + boost::system::error_code ec(last_error, + boost::asio::error::get_system_category()); + overlapped.complete(ec, 0); + } + else + { + // The operation was successfully initiated, so ownership of the + // OVERLAPPED-derived object has passed to the io_service. + overlapped.release(); + } +} + +class connection + : public boost::enable_shared_from_this<connection> +{ +public: + typedef boost::shared_ptr<connection> pointer; + + static pointer create(boost::asio::io_service& io_service, + const std::string& filename) + { + return pointer(new connection(io_service, filename)); + } + + tcp::socket& socket() + { + return socket_; + } + + void start() + { + boost::system::error_code ec; + file_.assign(::CreateFile(filename_.c_str(), GENERIC_READ, 0, 0, + OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED, 0), ec); + if (file_.is_open()) + { + transmit_file(socket_, file_, + boost::bind(&connection::handle_write, shared_from_this(), + boost::asio::placeholders::error, + boost::asio::placeholders::bytes_transferred)); + } + } + +private: + connection(boost::asio::io_service& io_service, const std::string& filename) + : socket_(io_service), + filename_(filename), + file_(io_service) + { + } + + void handle_write(const boost::system::error_code& /*error*/, + size_t /*bytes_transferred*/) + { + boost::system::error_code ignored_ec; + socket_.shutdown(tcp::socket::shutdown_both, ignored_ec); + } + + tcp::socket socket_; + std::string filename_; + random_access_handle file_; +}; + +class server +{ +public: + server(boost::asio::io_service& io_service, + unsigned short port, const std::string& filename) + : acceptor_(io_service, tcp::endpoint(tcp::v4(), port)), + filename_(filename) + { + start_accept(); + } + +private: + void start_accept() + { + connection::pointer new_connection = + connection::create(acceptor_.get_io_service(), filename_); + + acceptor_.async_accept(new_connection->socket(), + boost::bind(&server::handle_accept, this, new_connection, + boost::asio::placeholders::error)); + } + + void handle_accept(connection::pointer new_connection, + const boost::system::error_code& error) + { + if (!error) + { + new_connection->start(); + } + + start_accept(); + } + + tcp::acceptor acceptor_; + std::string filename_; +}; + +int main(int argc, char* argv[]) +{ + try + { + if (argc != 3) + { + std::cerr << "Usage: transmit_file <port> <filename>\n"; + return 1; + } + + boost::asio::io_service io_service; + + using namespace std; // For atoi. + server s(io_service, atoi(argv[1]), argv[2]); + + io_service.run(); + } + catch (std::exception& e) + { + std::cerr << e.what() << std::endl; + } + + return 0; +} + +#else // defined(BOOST_ASIO_HAS_WINDOWS_OVERLAPPED_PTR) +# error Overlapped I/O not available on this platform +#endif // defined(BOOST_ASIO_HAS_WINDOWS_OVERLAPPED_PTR) |