diff options
Diffstat (limited to 'kftpgrabber/src/engine/ftpsocket.h')
-rw-r--r-- | kftpgrabber/src/engine/ftpsocket.h | 154 |
1 files changed, 154 insertions, 0 deletions
diff --git a/kftpgrabber/src/engine/ftpsocket.h b/kftpgrabber/src/engine/ftpsocket.h new file mode 100644 index 0000000..1762bfd --- /dev/null +++ b/kftpgrabber/src/engine/ftpsocket.h @@ -0,0 +1,154 @@ +/* + * This file is part of the KFTPGrabber project + * + * Copyright (C) 2003-2006 by the KFTPGrabber developers + * Copyright (C) 2003-2006 Jernej Kos <kostko@jweb-network.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 + * is provided AS IS, WITHOUT ANY WARRANTY; without even the implied + * warranty of MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, and + * NON-INFRINGEMENT. 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., 51 Franklin Steet, Fifth Floor, Boston, + * MA 02110-1301, USA. + * + * In addition, as a special exception, the copyright holders give + * permission to link the code of portions of this program with the + * OpenSSL library under certain conditions as described in each + * individual source file, and distribute linked combinations + * including the two. + * + * You must obey the GNU General Public License in all respects + * for all of the code used other than OpenSSL. If you modify + * file(s) with this exception, you may extend this exception to your + * version of the file(s), but you are not obligated to do so. If you + * do not wish to do so, delete this exception statement from your + * version. If you delete this exception statement from all source + * files in the program, then also delete it here. + */ + +#ifndef KFTPENGINEFTPSOCKET_H +#define KFTPENGINEFTPSOCKET_H + +#include <kstreamsocket.h> +#include <kserversocket.h> +#include <kssl.h> + +#include <qguardedptr.h> +#include <qfile.h> + +#include "speedlimiter.h" +#include "socket.h" + +namespace KFTPEngine { + +class FtpDirectoryParser; +class Ssl; + +/** + * @author Jernej Kos <kostko@jweb-network.net> + */ +class FtpSocket : public KNetwork::KStreamSocket, public Socket, public SpeedLimiterItem +{ +Q_OBJECT +friend class Commands::Base; +friend class FtpCommandConnect; +friend class FtpCommandNegotiateData; +friend class FtpCommandList; +public: + FtpSocket(Thread *thread); + ~FtpSocket(); + + void protoConnect(const KURL &url); + void protoDisconnect(); + void protoAbort(); + void protoGet(const KURL &source, const KURL &destination); + void protoPut(const KURL &source, const KURL &destination); + void protoRemove(const KURL &path); + void protoRename(const KURL &source, const KURL &destination); + void protoChmodSingle(const KURL &path, int mode); + void protoMkdir(const KURL &path); + void protoList(const KURL &path); + void protoRaw(const QString &raw); + void protoSiteToSite(Socket *socket, const KURL &source, const KURL &destination); + void protoKeepAlive(); + + void changeWorkingDirectory(const QString &path, bool shouldCreate = false); + + void poll(); + + int features() { return SF_FXP_TRANSFER | SF_RAW_COMMAND; } + + bool isConnected() { return m_login; } + bool isEncrypted() { return isConnected() && getConfigInt("ssl"); } + + void setSslClientCertificate(KSSLPKCS12 *certificate) { m_clientCert = certificate; } + + bool isResponse(const QString &code); + QString getResponse() { return m_response; } + bool isMultiline() { return !m_multiLineCode.isEmpty(); } + + void sendCommand(const QString &command); + void resetCommandClass(ResetCode code = Ok); + + void setupPassiveTransferSocket(const QString &host, int port); + KNetwork::KSocketAddress setupActiveTransferSocket(); + + QFile *getTransferFile() { return &m_transferFile; } + + void checkTransferEnd(); + void checkTransferStart(); + void resetTransferStart() { m_transferStart = 0; } +protected: + void processBuffer(); + void parseLine(const QString &line); + void variableBufferUpdate(Q_LONG size); + void closeDataTransferSocket(); + void initializeTransferSocket(); + void transferCompleted(); +private: + bool m_login; + + QString m_buffer; + QString m_multiLineCode; + QString m_response; + + QGuardedPtr<KNetwork::KStreamSocket> m_transferSocket; + QGuardedPtr<KNetwork::KServerSocket> m_serverSocket; + FtpDirectoryParser *m_directoryParser; + + char m_controlBuffer[1024]; + + QFile m_transferFile; + char *m_transferBuffer; + int m_transferBufferSize; + int m_transferStart; + int m_transferEnd; + + bool m_transferConnecting; + bool m_controlConnecting; + + Ssl *m_controlSsl; + Ssl *m_dataSsl; + KSSLPKCS12 *m_clientCert; +protected slots: + void slotConnected(); + void slotControlTryRead(); + void slotError(); + + void slotDataAccept(KNetwork::KStreamSocket *socket); + void slotDataConnected(); + void slotDataTryRead(); + void slotDataTryWrite(); +}; + +} + +#endif |