//*****************************************************************************
//* (C) Copyright 1995 Gregory Consulting Limited
//*
//* This file is Copyrighted and is subject to the rules outlined in the
//* software agreement accompanying this CDROM and book. You are free to
//* modify this code as you see fit according to those rules.
//*
//* THIS IS SAMPLE CODE! No warranties are expressed or implied, you're
//* using it at your own risk. In addition, the resulting program may
//* not be fully functional. Support for some features are left out so
//* that they can be added by you, the reader.
//*
//*****************************************************************************

// socket.h : header file
//
#ifndef _SOCKET_H
#define _SOCKET_H

#include "winsock.h"

#define WS_VERSION_REQUIRED 0x0101
#define WS_VERSION_MAJOR 1
#define WS_VERSION_MINOR 1
#define MIN_SOCKETS_REQUIRED 10
#define ID_CONNECT_TIMER 1
#define ID_RECEIVE_TIMER 2                                         

#define WM_SOCKETS_MESSAGE 				WM_USER+1
#define WM_CONNECT_GET_HOST_MESSAGE 	WM_USER+2
#define WM_GET_SERVER_PORT_MESSAGE 		WM_USER+3
                               
// at least as big as the largest possible line.(currently 512 bytes)                               
const int SOCK_BLOCK_SIZE = 520;

/////////////////////////////////////////////////////////////////////////////
// QSocket window       

enum SocketStatus {UNINITIALIZED, DISCONNECTED, CONNECTING, LISTENING, DISCONNECTING, CONNECTED, 
			ERRORSTATE, TIMEDOUT } ;

enum SocketReceiveCmd { SocketStatusChanged = -100, NewSocketAccepted = -101, ReadyToSend = -102 };

typedef struct SendData {
	char buffer[SOCK_BLOCK_SIZE];
	int len;
	int amt;
} SendData;
                        
class QSocket : public CSocket
{   

// Construction
public:
	QSocket(BOOL create_socket = TRUE);

// Attributes
private:                  

	SocketStatus CurrentStatus;
	CString ErrorString;
	
	CStringList ReceiveLines;
	CString RemainingReceive;
	CStringList SendLines;
	char *RawSendData;
	int RawSendDataLength;
	
	CWnd *ReceiveWindow;
	UINT ReceiveMessage;
// Operations
public:

// Implementation
public:
	virtual ~QSocket();
	
	BOOL Disconnect(void);
	                             
	BOOL SetReceiveTarget(CWnd *window, UINT message);                             
	void Send(const CString& data);
	void SendRaw(const void *data, const int dataLen);
	CString GetLine(void);
	BOOL Listen(int back_log = 5);
	QSocket *Accept();
	void WaitUntilSendBufferEmpty();
   	void Linger(void);
	void StartSending(void);

	SocketStatus GetStatus(void) { return CurrentStatus; }
	CString GetErrorString(void) { return ErrorString; }
protected:
	// helper functions
 	int SetErrorVars(char *file, int line);
	void DisplayError();
 	BOOL Start(void);
	void AddToReceive(const char *buffer, int amt);
	void OnReceive(int error);
	void OnSend(int error);
	void OnAccept(int error);
	void OnConnect(int error);
	void OnClose(int error);
	BOOL ConnectHelper(const SOCKADDR* lpSockAddr, int nSockAddrLen);
};

/////////////////////////////////////////////////////////////////////////////
#endif