mirror of
https://github.com/pocoproject/poco.git
synced 2025-01-08 11:02:17 +01:00
303 lines
7.5 KiB
C++
303 lines
7.5 KiB
C++
//
|
|
// POP3ClientSession.cpp
|
|
//
|
|
// $Id: //poco/svn/Net/src/POP3ClientSession.cpp#2 $
|
|
//
|
|
// Library: Net
|
|
// Package: Mail
|
|
// Module: POP3ClientSession
|
|
//
|
|
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person or organization
|
|
// obtaining a copy of the software and accompanying documentation covered by
|
|
// this license (the "Software") to use, reproduce, display, distribute,
|
|
// execute, and transmit the Software, and to prepare derivative works of the
|
|
// Software, and to permit third-parties to whom the Software is furnished to
|
|
// do so, all subject to the following:
|
|
//
|
|
// The copyright notices in the Software and this entire statement, including
|
|
// the above license grant, this restriction and the following disclaimer,
|
|
// must be included in all copies of the Software, in whole or in part, and
|
|
// all derivative works of the Software, unless such copies or derivative
|
|
// works are solely in the form of machine-executable object code generated by
|
|
// a source language processor.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
|
|
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
|
|
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
|
|
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
// DEALINGS IN THE SOFTWARE.
|
|
//
|
|
|
|
|
|
#include "Poco/Net/POP3ClientSession.h"
|
|
#include "Poco/Net/MailMessage.h"
|
|
#include "Poco/Net/MailStream.h"
|
|
#include "Poco/Net/SocketAddress.h"
|
|
#include "Poco/Net/NetException.h"
|
|
#include "Poco/StreamCopier.h"
|
|
#include "Poco/NumberFormatter.h"
|
|
#include "Poco/UnbufferedStreamBuf.h"
|
|
#include <istream>
|
|
#include <cctype>
|
|
|
|
|
|
using Poco::NumberFormatter;
|
|
using Poco::StreamCopier;
|
|
|
|
|
|
namespace Poco {
|
|
namespace Net {
|
|
|
|
|
|
class DialogStreamBuf: public Poco::UnbufferedStreamBuf
|
|
{
|
|
public:
|
|
DialogStreamBuf(DialogSocket& socket):
|
|
_socket(socket)
|
|
{
|
|
}
|
|
|
|
~DialogStreamBuf()
|
|
{
|
|
}
|
|
|
|
private:
|
|
int readFromDevice()
|
|
{
|
|
return _socket.get();
|
|
}
|
|
|
|
DialogSocket& _socket;
|
|
};
|
|
|
|
|
|
class DialogIOS: public virtual std::ios
|
|
{
|
|
public:
|
|
DialogIOS(DialogSocket& socket):
|
|
_buf(socket)
|
|
{
|
|
poco_ios_init(&_buf);
|
|
}
|
|
|
|
~DialogIOS()
|
|
{
|
|
}
|
|
|
|
DialogStreamBuf* rdbuf()
|
|
{
|
|
return &_buf;
|
|
}
|
|
|
|
protected:
|
|
DialogStreamBuf _buf;
|
|
};
|
|
|
|
|
|
class DialogInputStream: public DialogIOS, public std::istream
|
|
{
|
|
public:
|
|
DialogInputStream(DialogSocket& socket):
|
|
DialogIOS(socket),
|
|
std::istream(&_buf)
|
|
{
|
|
}
|
|
|
|
~DialogInputStream()
|
|
{
|
|
}
|
|
};
|
|
|
|
|
|
POP3ClientSession::POP3ClientSession(const StreamSocket& socket):
|
|
_socket(socket),
|
|
_isOpen(true)
|
|
{
|
|
}
|
|
|
|
|
|
POP3ClientSession::POP3ClientSession(const std::string& host, Poco::UInt16 port):
|
|
_socket(SocketAddress(host, port)),
|
|
_isOpen(true)
|
|
{
|
|
}
|
|
|
|
|
|
POP3ClientSession::~POP3ClientSession()
|
|
{
|
|
try
|
|
{
|
|
close();
|
|
}
|
|
catch (...)
|
|
{
|
|
}
|
|
}
|
|
|
|
|
|
void POP3ClientSession::setTimeout(const Poco::Timespan& timeout)
|
|
{
|
|
_socket.setReceiveTimeout(timeout);
|
|
}
|
|
|
|
|
|
Poco::Timespan POP3ClientSession::getTimeout() const
|
|
{
|
|
return _socket.getReceiveTimeout();
|
|
}
|
|
|
|
|
|
void POP3ClientSession::login(const std::string& username, const std::string& password)
|
|
{
|
|
std::string response;
|
|
_socket.receiveMessage(response);
|
|
if (!isPositive(response)) throw SMTPException("The POP3 service is unavailable", response);
|
|
sendCommand("USER", username, response);
|
|
if (!isPositive(response)) throw POP3Exception("Login rejected for user", response);
|
|
sendCommand("PASS", password, response);
|
|
if (!isPositive(response)) throw POP3Exception("Password rejected for user", response);
|
|
}
|
|
|
|
|
|
void POP3ClientSession::close()
|
|
{
|
|
if (_isOpen)
|
|
{
|
|
std::string response;
|
|
sendCommand("QUIT", response);
|
|
_socket.close();
|
|
_isOpen = false;
|
|
}
|
|
}
|
|
|
|
|
|
int POP3ClientSession::messageCount()
|
|
{
|
|
std::string response;
|
|
sendCommand("STAT", response);
|
|
if (!isPositive(response)) throw POP3Exception("Cannot determine message count", response);
|
|
std::string::const_iterator it = response.begin();
|
|
std::string::const_iterator end = response.end();
|
|
int count = 0;
|
|
while (it != end && !std::isspace(*it)) ++it;
|
|
while (it != end && std::isspace(*it)) ++it;
|
|
while (it != end && std::isdigit(*it)) count = count*10 + *it++ - '0';
|
|
return count;
|
|
}
|
|
|
|
|
|
void POP3ClientSession::listMessages(MessageInfoVec& messages)
|
|
{
|
|
messages.clear();
|
|
std::string response;
|
|
sendCommand("LIST", response);
|
|
if (!isPositive(response)) throw POP3Exception("Cannot get message list", response);
|
|
_socket.receiveMessage(response);
|
|
while (response != ".")
|
|
{
|
|
MessageInfo info = {0, 0};
|
|
std::string::const_iterator it = response.begin();
|
|
std::string::const_iterator end = response.end();
|
|
while (it != end && std::isdigit(*it)) info.id = info.id*10 + *it++ - '0';
|
|
while (it != end && std::isspace(*it)) ++it;
|
|
while (it != end && std::isdigit(*it)) info.size = info.size*10 + *it++ - '0';
|
|
messages.push_back(info);
|
|
_socket.receiveMessage(response);
|
|
}
|
|
}
|
|
|
|
|
|
void POP3ClientSession::retrieveMessage(int id, MailMessage& message)
|
|
{
|
|
std::string response;
|
|
sendCommand("RETR", NumberFormatter::format(id), response);
|
|
if (!isPositive(response)) throw POP3Exception("Cannot get message list", response);
|
|
DialogInputStream sis(_socket);
|
|
MailInputStream mis(sis);
|
|
message.read(mis);
|
|
while (mis.good()) mis.get(); // read any remaining junk
|
|
}
|
|
|
|
|
|
void POP3ClientSession::retrieveMessage(int id, MailMessage& message, PartHandler& handler)
|
|
{
|
|
std::string response;
|
|
sendCommand("RETR", NumberFormatter::format(id), response);
|
|
if (!isPositive(response)) throw POP3Exception("Cannot get message list", response);
|
|
DialogInputStream sis(_socket);
|
|
MailInputStream mis(sis);
|
|
message.read(mis, handler);
|
|
while (mis.good()) mis.get(); // read any remaining junk
|
|
}
|
|
|
|
|
|
void POP3ClientSession::retrieveMessage(int id, std::ostream& ostr)
|
|
{
|
|
std::string response;
|
|
sendCommand("RETR", NumberFormatter::format(id), response);
|
|
if (!isPositive(response)) throw POP3Exception("Cannot get message list", response);
|
|
DialogInputStream sis(_socket);
|
|
MailInputStream mis(sis);
|
|
StreamCopier::copyStream(mis, ostr);
|
|
}
|
|
|
|
|
|
void POP3ClientSession::retrieveHeader(int id, MessageHeader& header)
|
|
{
|
|
std::string response;
|
|
sendCommand("TOP", NumberFormatter::format(id), "0", response);
|
|
if (!isPositive(response)) throw POP3Exception("Cannot get message list", response);
|
|
DialogInputStream sis(_socket);
|
|
MailInputStream mis(sis);
|
|
header.read(mis);
|
|
// skip stuff following header
|
|
mis.get(); // \r
|
|
mis.get(); // \n
|
|
}
|
|
|
|
|
|
void POP3ClientSession::deleteMessage(int id)
|
|
{
|
|
std::string response;
|
|
sendCommand("DELE", NumberFormatter::format(id), response);
|
|
if (!isPositive(response)) throw POP3Exception("Cannot mark message for deletion", response);
|
|
}
|
|
|
|
|
|
bool POP3ClientSession::sendCommand(const std::string& command, std::string& response)
|
|
{
|
|
_socket.sendMessage(command);
|
|
_socket.receiveMessage(response);
|
|
return isPositive(response);
|
|
}
|
|
|
|
|
|
bool POP3ClientSession::sendCommand(const std::string& command, const std::string& arg, std::string& response)
|
|
{
|
|
_socket.sendMessage(command, arg);
|
|
_socket.receiveMessage(response);
|
|
return isPositive(response);
|
|
}
|
|
|
|
|
|
bool POP3ClientSession::sendCommand(const std::string& command, const std::string& arg1, const std::string& arg2, std::string& response)
|
|
{
|
|
_socket.sendMessage(command, arg1, arg2);
|
|
_socket.receiveMessage(response);
|
|
return isPositive(response);
|
|
}
|
|
|
|
|
|
bool POP3ClientSession::isPositive(const std::string& response)
|
|
{
|
|
return response.length() > 0 && response[0] == '+';
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|