--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/nettools/conntest/inc/SocketsWrite.h	Thu Dec 17 08:39:25 2009 +0200
@@ -0,0 +1,174 @@
+/*
+* Copyright (c) 2006 Nokia Corporation and/or its subsidiary(-ies). 
+* All rights reserved.
+* This component and the accompanying materials are made available
+* under the terms of "Eclipse Public License v1.0"
+* which accompanies this distribution, and is available
+* at the URL "http://www.eclipse.org/legal/epl-v10.html".
+*
+* Initial Contributors:
+* Nokia Corporation - initial contribution.
+*
+* Contributors:
+*
+* Description: CSocketsWrite is an active object which implements data 
+* sending through an already open socket using UDP or TCP
+*
+*/
+
+#ifndef __SOCKETSWRITE_H__
+#define __SOCKETSWRITE_H__
+
+#include <in_sock.h>
+#include "TimeOutNotify.h"
+#include "conntest.hrh"
+
+// FORWARD DECLARATIONS
+class CTimeOutTimer;
+class MUINotify;
+class MDataSendNotifyHandler;
+
+// CLASS DECLARATION
+
+/**
+*  This class handles writing data to the socket.
+*  Data to be written is accumulated in iTransferBuffer,
+*  and is then transferred to iWriteBuffer for the actual
+*  write to the socket.
+*/
+class CSocketsWrite : public CActive, public MTimeOutNotify
+{
+public:
+
+    /**
+    * Create a CSocketsWrite object
+    * @param aConsole console to use for ui output
+    * @param aSocket socket to write to
+    * @param aSendHandler handler to notify UDP packet sends
+    * @result a pointer to the created instance of CSocketsWrite
+    */
+    static CSocketsWrite* NewL( MUINotify& aConsole, 
+                                RSocket& aSocket, 
+                                MDataSendNotifyHandler& aSendHandler );
+    
+    /**
+    * Create a CSocketsWrite object
+    * @param aConsole console to use for ui output
+    * @param aSocket socket to write to
+    * @param aSendHandler handler to notify UDP packet sends
+    * @result a pointer to the created instance of CSocketsWrite
+    */
+    static CSocketsWrite* NewLC( MUINotify& aConsole, 
+                                 RSocket& aSocket, 
+                                 MDataSendNotifyHandler& aSendHandler );
+    
+    /**
+    * Destroy the object and release all memory objects
+    */
+    ~CSocketsWrite();
+    
+    /**
+    * Write the data to the socket (buffered)
+    * @param aData the data to be written
+    * @param aAddress Address where data is sent (needed by UDP)
+    * @param aProtocol Protocol to be used (TCP or UDP)
+    */
+    void IssueWriteL(const TDesC8& aData, TInetAddr* aAddress, TUint aProtocol);
+    
+    /**
+    * Write the data to the socket flooded
+    * @param aData the data to be written
+    * @param aAddress Address where data is sent (needed by UDP)
+    * @param aProtocol Protocol to be used (TCP or UDP)
+    * @param aCount The packet will be sent aCount times
+    */
+    void IssueWriteL(const TDesC8& aData, TInetAddr* aAddress, TUint aProtocol, TInt aCount);
+    
+    /**
+    * Write the data to the socket (buffered)
+    * @param aData the data to be written
+    * @param aAddress Address where data is sent (needed by UDP)
+    * @param aProtocol Protocol to be used (TCP or UDP)
+    */
+    void IssueWriteL( const TDesC8& aData, 
+                      TInetAddr* aAddress, 
+                      TUint aProtocol, 
+                      TInt aPacketSize, 
+                      TInt aPackets );
+    
+protected: // from CActive
+          
+    /**
+    * Cancel any outstanding operation
+    */
+    void DoCancel();
+    
+    /**
+    * Called when operation complete
+    */
+    void RunL();	
+    
+    // From MTimeOutNotify
+    void TimerExpired(); 
+    
+private:
+
+    /**
+    * Perform the first phase of two phase construction 
+    * @param aConsole console to use for ui output
+    * @param aSocket socket to write to
+    * @param aSendHandler handler to notify UDP packet sends
+    */
+    CSocketsWrite( MUINotify& aConsole, 
+                   RSocket& aSocket, 
+                   MDataSendNotifyHandler& aSendHandler );
+    
+    /**
+    * Perform the second phase construction of a CSocketsWrite 
+    */
+    void ConstructL();
+    
+    /**
+    * Handle a 'write buffer empty' situation.
+    */    
+    void SendNextPacket();
+
+    /**
+    * Calls the actual write when flooding
+    */
+    void DoFloodWrite();
+    
+    
+private: // Member data
+    enum TWriteState 
+    {
+        ESending, EWaiting ,ECommsFailed
+    };
+    
+    
+    // Member variables
+    RSocket&                iSocket;
+    MUINotify&              iConsole; // console for displaying text etc
+    //TBuf8<KMaxSendBuffer>   iTransferBuffer; // Accumulate data to send in here
+    //TBuf8<KMaxSendBuffer>   iWriteBuffer; // Holds data currently being sent to socket
+    HBufC8*                 iTransferBuffer;
+    HBufC8*                 iWriteBuffer;
+    CTimeOutTimer*          iTimer;
+    TInt                    iTimeOut;
+    TWriteState             iWriteStatus;
+    TInetAddr*              iAddress;
+    TUint                   iProtocol;
+	TInt                    iPackets;
+	TInt                    iPacketSize;
+    TInt                    iSentBytes;
+	TTime                   iStartTime;
+	TInt                    iDataChunkCount;
+	HBufC8*                 iReqBodySubmitBuffer;
+	TPtr8                   iReqBodySubmitBufferPtr;
+	MDataSendNotifyHandler& iSendHandler;
+	TBool                   iFlood; // Flood mode sending
+	TInt                    iCount; // Times to flood the packet
+};
+
+#endif // __SOCKETSWRITE_H__
+