Logo Search packages:      
Sourcecode: poco version File versions  Download package

POP3ClientSession.cpp

//
// POP3ClientSession.cpp
//
// $Id: //poco/1.2/Net/src/POP3ClientSession.cpp#1 $
//
// 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 <ctype.h>


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)
{
}


00124 POP3ClientSession::POP3ClientSession(const std::string& host, Poco::UInt16 port):   
      _socket(SocketAddress(host, port)),
      _isOpen(true)
{
}


00131 POP3ClientSession::~POP3ClientSession()
{
      try
      {
            close();
      }
      catch (...)
      {
      }
}


00143 void POP3ClientSession::setTimeout(const Poco::Timespan& timeout)
{
      _socket.setReceiveTimeout(timeout);
}

      
00149 Poco::Timespan POP3ClientSession::getTimeout() const
{
      return _socket.getReceiveTimeout();
}


00155 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);
}


00167 void POP3ClientSession::close()
{
      if (_isOpen)
      {
            std::string response;
            sendCommand("QUIT", response);
            _socket.close();
            _isOpen = false;
      }
}


00179 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 && !isspace(*it)) ++it;
      while (it != end && isspace(*it)) ++it;
      while (it != end && isdigit(*it)) count = count*10 + *it++ - '0';
      return count;
}


00194 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 && isdigit(*it)) info.id = info.id*10 + *it++ - '0';
            while (it != end && isspace(*it)) ++it;
            while (it != end && isdigit(*it)) info.size = info.size*10 + *it++ - '0';
            messages.push_back(info);
            _socket.receiveMessage(response);
      }
}


00215 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);
}


00226 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);
}


00237 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);
}


00248 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
}


00262 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);
}


00270 bool POP3ClientSession::sendCommand(const std::string& command, std::string& response)
{
      _socket.sendMessage(command);
      _socket.receiveMessage(response);
      return isPositive(response);
}


00278 bool POP3ClientSession::sendCommand(const std::string& command, const std::string& arg, std::string& response)
{
      _socket.sendMessage(command, arg);
      _socket.receiveMessage(response);
      return isPositive(response);
}


00286 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);
}


00294 bool POP3ClientSession::isPositive(const std::string& response)
{
      return response.length() > 0 && response[0] == '+';
}


} } // namespace Poco::Net

Generated by  Doxygen 1.6.0   Back to index