sqlbackupservice.h

我写的SQL Server网络备份工具。（服务器端用VC开发

#pragma once
#include "ntservice.h"

class MySocket
{
public:
	MySocket():m_Socket(INVALID_SOCKET)
	{
	}
	MySocket(int af, int type, int protocol)
	{
		m_Socket = socket(af , type , protocol);
	}
	~MySocket()
	{
		if(m_Socket != INVALID_SOCKET)
		{
			closesocket(m_Socket);
		}
	}
	operator SOCKET ()
	{
		return m_Socket;
	}
	void Attach(SOCKET sock)
	{
        _ASSERTE(m_Socket == INVALID_SOCKET);
		m_Socket = sock;
	}
	void Close()
	{
		if(m_Socket != INVALID_SOCKET)
		{
			closesocket(m_Socket);
			m_Socket = INVALID_SOCKET;
		}
	}
protected:
	SOCKET m_Socket;
};

class CSQLBackupService :
	public CNTService
{
public:
	CSQLBackupService(void);
	~CSQLBackupService(void);
protected:
	TCHAR m_szAppPath[_MAX_PATH];
	TCHAR m_szAppName[_MAX_PATH];
	MySocket m_sockListen;
	
	virtual void Run(void);
	void OutputMessageToFile(char * pMessage);
	static DWORD WINAPI ClientThread(LPVOID lpParam);
	virtual void OnStop(void);
	void ResponseBackup(MySocket &ClientSocket , int nParamLenght);
	void SendErrorMessage(MySocket &ClientSocket, char * pMessage);
	BOOL SendFile(MySocket& ClientSocket , const CString &strFileName );
	BOOL Backup(const char *pDatabase ,const CString & strFileName, CString &strMsg);
	void ResponseQueryDatabase(MySocket& sockClient);
	void SendConnectSucceed(MySocket & sockClient);
};