1 files changed, 362 insertions, 0 deletions
diff --git a/kopete/protocols/msn/msnsocket.h b/kopete/protocols/msn/msnsocket.h
new file mode 100644
index 00000000..85df08c7
--- /dev/null
+++ b/kopete/protocols/msn/msnsocket.h
@@ -0,0 +1,362 @@
+/*
+    msnsocket.h - Base class for the sockets used in MSN
+
+    Copyright (c) 2002      by Martijn Klingens       <klingens@kde.org>
+    Copyright (c) 2002-2004 by Olivier Goffart        <ogoffart @ kde.org>
+    Copyright (c) 2005		by Gregg Edghill 		  <gregg.edghill@gmail.com>
+    Kopete    (c) 2002      by the Kopete developers  <kopete-devel@kde.org>
+
+    Portions of this code are taken from KMerlin,
+              (c) 2001 by Olaf Lueg              <olueg@olsd.de>
+
+    *************************************************************************
+    *                                                                       *
+    * 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.                                   *
+    *                                                                       *
+    *************************************************************************
+*/
+
+#ifndef MSNSOCKET_H
+#define MSNSOCKET_H
+
+#include <qobject.h>
+#include <qdatastream.h>
+#include <qstringlist.h>
+#include <qtimer.h>
+#include <qvaluelist.h>
+
+#include "kopete_export.h"
+
+namespace KNetwork {
+  class KBufferedSocket;
+  class KServerSocket;
+}
+
+class MimeMessage;
+
+/**
+ * @author Martijn Klingens <klingens@kde.org>
+ *
+ * MSNSocket encapsulates the common functionality shared by the Dispatch
+ * Server, the Notification Server and the Switchboard Server. It is
+ * inherited by the various specialized classes.
+ */
+class KOPETE_EXPORT MSNSocket : public QObject
+{
+	Q_OBJECT
+
+public:
+	MSNSocket(QObject* parent=0l);
+	~MSNSocket();
+
+	/**
+	 * Asynchronously read a block of data of the specified size. When the
+	 * data is available, the blockRead() signal will be emitted with the
+	 * data as parameter.
+	 *
+	 * NOTE: As the block queue takes precedence over the line-based
+	 *       command-processing this method can effectively block all
+	 *       communications when passed a wrong length!
+	 */
+	void readBlock( uint len );
+
+	/**
+	 * OnlineStatus encapsulates the 4 states a connection can be in,
+	 * Connecting, Connected, Disconnecting, Disconnected. Connecting
+	 * and Disconnecting are in the default implementation not used,
+	 * because the socket connect is an atomic operation and not yet
+	 * performed asynchronously.
+	 * In derived classes, like the Notification Server, this state is
+	 * actively used, because merely having a socket connection established
+	 * by no means indicates we're actually online - the rest of the
+	 * handshake likely has to follow first.
+	 */
+	enum OnlineStatus { Connecting, Connected, Disconnecting, Disconnected };
+	enum LookupStatus { Processing, Success, Failed };
+	enum Transport { TcpTransport, HttpTransport };
+	enum ErrorType { ErrorConnectionLost, ErrorConnectionError, ErrorCannotConnect, ErrorServerError, ErrorInformation};
+
+	OnlineStatus onlineStatus() { return m_onlineStatus; }
+
+	/*
+	 * return the local ip.
+	 * Used for filetransfer
+	 */
+	QString getLocalIP();
+
+	//BEGIN Http
+
+	virtual bool setUseHttpMethod( bool useHttp );
+	bool useHttpMethod() const;
+
+	//END
+
+public slots:
+	void connect( const QString &server, uint port );
+	virtual void disconnect();
+
+
+	/**
+	 * Send an MSN command to the socket
+	 *
+	 * For debugging it's convenient to have this method public, but using
+	 * it outside this class is deprecated for any other use!
+	 *
+	 * The size of the body (if any) is automatically added to the argument
+	 * list and shouldn't be explicitly specified! This size is in bytes
+	 * instead of characters to reflect what actually goes over the wire.
+	 *
+	 * if the param binary is set to true, then, the body is send as a binary message
+	 *
+	 * return the id
+	 */
+	int sendCommand( const QString &cmd, const QString &args = QString::null,
+		bool addId = true, const QByteArray &body = QByteArray() , bool binary=false );
+
+signals:
+	/**
+	 * A block read is ready.
+	 * After this the normal line-based reads go on again
+	 */
+	void blockRead( const QByteArray &block );
+
+	/**
+	 * The online status has changed
+	 */
+	void onlineStatusChanged( MSNSocket::OnlineStatus status );
+
+	/**
+	 * The connection failed
+	 */
+	void connectionFailed();
+
+	/**
+	 * The connection was closed
+	 */
+	void socketClosed();
+
+	/**
+	 * A error has occured. Handle the display of the message.
+	 */
+	void errorMessage( int type, const QString &msg );
+
+protected:
+	/**
+	 * Convenience method: escape spaces with '%20' for use in the protocol.
+	 * Doesn't escape any other sequence.
+	 */
+	QString escape( const QString &str );
+
+	/**
+	 * And the other way round...
+	 */
+	QString unescape( const QString &str );
+
+	/**
+	 * Set the online status. Emits onlineStatusChanged.
+	 */
+	void setOnlineStatus( OnlineStatus status );
+
+	/**
+	 * This method is called directly before the socket will actually connect.
+	 * Override in derived classes to setup whatever is needed before connect.
+	 */
+	virtual void aboutToConnect();
+
+	/**
+	 * Directly after the connect, this method is called. The default
+	 * implementation sets the OnlineStatus to Connected, be sure to override
+	 * this if a handshake is required.
+	 */
+	virtual void doneConnect();
+
+	/**
+	 * Directly after the disconnect, this method is called before the actual
+	 * cleanup takes place. The socket is close here. Cleanup internal
+	 * variables here.
+	 */
+	virtual void doneDisconnect();
+
+	/**
+	 * Handle an MSN error condition.
+	 * The default implementation displays a generic error message and
+	 * closes the connection. Override to allow more graceful handling and
+	 * possibly recovery.
+	 */
+	virtual void handleError( uint code, uint id );
+
+	/**
+	 * Handle an MSN command response line.
+	 * This method is pure virtual and *must* be overridden in derived
+	 * classes.
+	 */
+	virtual void parseCommand( const QString &cmd, uint id,
+		const QString &data ) = 0;
+
+	/**
+	 * Used in MSNFileTransferSocket
+	 */
+	virtual void bytesReceived( const QByteArray & );
+	bool accept( KNetwork::KServerSocket * );
+	void sendBytes( const QByteArray &data );
+
+	const QString &server() { return m_server; }
+	uint port() { return m_port; }
+
+	/**
+	 * The last confirmed ID by the server
+	 */
+	//uint m_lastId;
+
+private slots:
+	void slotDataReceived();
+	/**
+	 * If the socket emits a connectionFailed() then this slot is called
+	 * to handle the error.
+	 */
+	void slotSocketError( int error );
+
+	/*
+	 * Calls connectDone() when connection is successfully established.
+	 */
+	void slotConnectionSuccess();
+
+	/**
+	 * Sets m_lookupProgress to 'Finished' if count > 0 or 'Failed' if count = 0.
+	 */
+	void slotHostFound( );
+
+	/**
+	 * Check if new lines of data are available and process the first line
+	 */
+	void slotReadLine();
+
+	void slotSocketClosed();
+
+	//BEGIN Http
+
+	/**
+	 * Sends a poll request to the msn gateway when using HttpTransport.
+	 * equivalent to sending a PNG command over TcpTransport.
+	 */
+	void slotHttpPoll();
+
+	//END
+
+protected slots:
+	virtual void slotReadyWrite();
+
+private:
+	/**
+	 * Check if we're waiting for a block of raw data. Emits blockRead()
+	 * when the data is available.
+	 * Returns true when still waiting and false when there is no pending
+	 * read, or when the read is successfully handled.
+	 */
+	bool pollReadBlock();
+
+	/**
+	 * The id of the message sent to the MSN server. This ID will increment
+	 * for each subsequent message sent.
+	 */
+	uint m_id;
+
+	/**
+	 * Queue of pending commands (should be mostly empty, but is needed to
+	 * send more than one command to the server)
+	 */
+	QValueList<QByteArray> m_sendQueue;
+
+	/**
+	 * Parse a single line of data.
+	 * Will call either parseCommand or handleError depending on the type of
+	 * data received.
+	 */
+	void parseLine( const QString &str );
+
+	KNetwork::KBufferedSocket *m_socket;
+	OnlineStatus m_onlineStatus;
+
+	QString m_server;
+	uint m_port;
+
+	/**
+	 * The size of the requested block for block-based reads
+	 */
+	uint m_waitBlockSize;
+
+	class Buffer : public QByteArray
+	{
+	public:
+		Buffer( unsigned size = 0 );
+		~Buffer();
+		void add( char *str, unsigned size );
+		QByteArray take( unsigned size );
+	};
+
+	Buffer m_buffer;
+
+	//BEGIN Http
+
+	/**
+	 * Makes a http request headers string using the specified, host, query, and content length.
+	 * return: The string containing the http request headers.
+	 */
+	QString makeHttpRequestString(const QString& host, const QString& query, uint contentLength);
+
+	bool m_useHttp; 			// Indicates whether to use the msn http gateway to connect to the msn service.
+	bool m_bCanPoll; 			// Indicates whether polling of the http server is allowed.
+	bool m_bIsFirstInTransaction; 		// Indicates whether pending message to be sent is the first in the transaction.
+						// If so, the gateway is used.
+						// Use the gateway only for initial connected state; Otherwise, use the host.
+	QString m_gateway;			// Msn http gateway domain name.
+	QString m_gwip;				// The ip address of the msn gateway.
+	QString m_sessionId; 			// session id.
+	QTimer *m_timer; 			// Msn http poll timer.
+	QString m_type;				// Indicates the type of socket being used.  NS or SB
+	bool m_pending; 			// Indicates whether a http response is pending.
+	int m_remaining;			// Indicates how many bytes of content data remain
+						// to be received if the content bytes are sent in
+						// a seperate packet(s).
+
+	/**
+	 * Provides access to information returned from a URI request.
+	 */
+	class WebResponse
+	{
+	public:
+		 WebResponse(const QByteArray& bytes);
+		~WebResponse();
+
+		/**
+		 * Gets the headers associated with this response from the server.
+		 */
+		MimeMessage* getHeaders();
+		/**
+		 * Gets the data stream used to read the body of the response from the server.
+		 */
+		QDataStream* getResponseStream();
+		/**
+		 * Gets the status code of the response.
+		 */
+		int getStatusCode();
+		/**
+		 * Gets the status description returned with the response.
+		 */
+		QString getStatusDescription();
+
+	private:
+		MimeMessage *m_headers;
+		QDataStream *m_stream;
+		int m_statusCode;
+		QString m_statusDescription;
+	};
+
+	//END
+};
+
+#endif
+