📄 clearlog.cpp

📁 一个短小精悍的MS SQL Server数据库日志清除工具程序。
💻 CPP
字号:
#include "stdafx.h"
#include "resource.h"

#define DBSETHOST 1
#define DBSETUSER 2
#define DBSETPWD  3
#define DBSETAPP  4
#define DBSETID   5
#define DBSETLANG 6

#define ID_SENDMAIL	13579
#define ID_ABOUT	24680

#define SUCCEED  1
#define FAIL     0
#define SUCCEED_ABORT 2

#define EXCOMM          9

#define DBNOERR         -1

#define INT_EXIT        0
#define INT_CONTINUE    1
#define INT_CANCEL      2

#define DBERRHANDLE_PROC FARPROC

#define STRINGBIND 10

#define NO_MORE_ROWS -2

typedef char DBCHAR;
typedef long DBINT;

#define LOGINREC void
typedef LOGINREC * PLOGINREC;

#define DBPROCESS void
typedef DBPROCESS * PDBPROCESS;

#define RETCODE INT
#define STATUS INT

typedef  PLOGINREC (DBLOGIN)(void);
typedef DBLOGIN *PDBLOGIN;

typedef  PLOGINREC (DBFREELOGIN)(PLOGINREC);
typedef DBFREELOGIN *PDBFREELOGIN;

typedef RETCODE (DBSETLNAME)(PLOGINREC, LPCSTR, INT);
typedef DBSETLNAME *PDBSETLNAME;

typedef PDBPROCESS (DBOPEN)(PLOGINREC, LPCSTR);
typedef DBOPEN *PDBOPEN;

typedef RETCODE (DBCLOSE)(PDBPROCESS);
typedef DBCLOSE *PDBCLOSE;

typedef RETCODE (DBCMD)(PDBPROCESS, LPCSTR);
typedef DBCMD *PDBCMD;

typedef RETCODE (DBUSE)(PDBPROCESS, LPCSTR);
typedef DBUSE *PDBUSE;

typedef RETCODE (DBSQLEXEC)(PDBPROCESS);
typedef DBSQLEXEC *PDBSQLEXEC;

typedef RETCODE (DBRESULTS)(PDBPROCESS);
typedef DBRESULTS *PDBRESULTS;

typedef RETCODE (DBBIND)(PDBPROCESS, INT, INT, DBINT, LPBYTE);
typedef DBBIND *PDBBIND;

typedef STATUS (DBNEXTROW)(PDBPROCESS);
typedef DBNEXTROW *PDBNEXTROW;

typedef DBINT (DBDATLEN)(PDBPROCESS, INT);
typedef DBDATLEN *PDBDATLEN;

typedef DBERRHANDLE_PROC DBERRHANDLE(DBERRHANDLE_PROC);
typedef DBERRHANDLE *PDBERRHANDLE;

typedef BOOL DBDEAD(PDBPROCESS);
typedef DBDEAD *PDBDEAD;

ATOM MyRegisterClass(HINSTANCE hInstance);
LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);

HWND g_hWnd, g_hwndServer, g_hwndUserName, g_hwndPassword,
	g_hwndConnect, g_hwndClear, g_hwndSelAll, g_hwndUnselect,
	g_hwndDatabase, g_hwndProgress;

HINSTANCE g_hInst, g_hDLL;

PDBLOGIN dblogin;
PDBFREELOGIN dbfreelogin;
PDBSETLNAME dbsetlname;
PDBOPEN dbopen;
PDBCLOSE dbclose;
PDBCMD dbcmd;
PDBUSE dbuse;
PDBSQLEXEC dbsqlexec;
PDBRESULTS dbresults;
PDBBIND dbbind;
PDBNEXTROW dbnextrow;
PDBDATLEN dbdatlen;
PDBERRHANDLE dberrhandle;
PDBDEAD dbdead;

char g_szServer[16], g_szUserName[128], g_szPassword[128];
char szError[] = "Can not load library 'ntwdblib.dll'.";

int Error(DBPROCESS *dbproc, int severity, int dberr, int oserr, char *dberrstr, char *oserrstr)
{
	MessageBox(g_hWnd, dberrstr, "Error", MB_ICONERROR);

	if (dbproc == NULL || dbdead(dbproc))
		return INT_EXIT;
	else
		return INT_CANCEL;
}

BOOL LoadDLLProc()
{
	BOOL bRet = FALSE;
	g_hDLL = LoadLibrary("ntwdblib.dll");
	if (g_hDLL)
	{
		dblogin = (PDBLOGIN)GetProcAddress(g_hDLL, "dblogin");
		dbfreelogin = (PDBFREELOGIN)GetProcAddress(g_hDLL, "dbfreelogin");
		dbsetlname = (PDBSETLNAME)GetProcAddress(g_hDLL, "dbsetlname");
		dbopen = (PDBOPEN)GetProcAddress(g_hDLL, "dbopen");
		dbclose = (PDBCLOSE)GetProcAddress(g_hDLL, "dbclose");
		dbcmd = (PDBCMD)GetProcAddress(g_hDLL, "dbcmd");
		dbuse = (PDBUSE)GetProcAddress(g_hDLL, "dbuse");
		dbsqlexec = (PDBSQLEXEC)GetProcAddress(g_hDLL, "dbsqlexec");
		dbresults = (PDBRESULTS)GetProcAddress(g_hDLL, "dbresults");
		dbbind = (PDBBIND)GetProcAddress(g_hDLL, "dbbind");
		dbnextrow = (PDBNEXTROW)GetProcAddress(g_hDLL, "dbnextrow");
		dbdatlen = (PDBDATLEN)GetProcAddress(g_hDLL, "dbdatlen");
		dberrhandle = (PDBERRHANDLE)GetProcAddress(g_hDLL, "dberrhandle");
		dbdead = (PDBDEAD)GetProcAddress(g_hDLL, "dbdead");
		dberrhandle((DBERRHANDLE_PROC)Error);
		bRet = TRUE;
	}
	return bRet;
}

void FreeDLLProc()
{
	if (g_hDLL)
	{
		FreeLibrary(g_hDLL);
		g_hDLL = NULL;
	}
}

void AddDatabase()
{
	LOGINREC *logrec = dblogin();
	if (logrec)
	{
		dbsetlname(logrec, g_szUserName, DBSETUSER);
		dbsetlname(logrec, g_szPassword, DBSETPWD);
		dbsetlname(logrec, "LZHXZY", DBSETAPP);

		DBPROCESS *dbproc = dbopen(logrec, g_szServer);
		if (dbproc)
		{
			dbuse(dbproc, "master");
			dbcmd(dbproc, "SELECT name FROM sysdatabases WHERE name NOT IN ('master','model','msdb','tempdb')");
			dbsqlexec(dbproc);
			if (dbresults(dbproc) == SUCCEED)
			{
				DBCHAR szName[128];
				dbbind(dbproc, 1, STRINGBIND, (DBINT)128, (BYTE*)szName);
				while (dbnextrow(dbproc) != NO_MORE_ROWS)
				{
					szName[dbdatlen(dbproc, 1)] = '\0';
					SendMessage(g_hwndDatabase, LB_ADDSTRING, 0, (LPARAM)szName);
				}
			}
			dbclose(dbproc);
		}
		dbfreelogin(logrec);
	}
}

BOOL ClearLogFile(const char *szDBName, char *szDBFile, char *szLogFile)
{
	BOOL bRet = FALSE;
	LOGINREC *logrec = dblogin();
	if (logrec)
	{
		dbsetlname(logrec, g_szUserName, DBSETUSER);
		dbsetlname(logrec, g_szPassword, DBSETPWD);
		dbsetlname(logrec, "LZHXZY", DBSETAPP);

		DBPROCESS *dbproc = dbopen(logrec, g_szServer);
		if (dbproc)
		{
			if (dbuse(dbproc, szDBName) == SUCCEED)
			{
				dbcmd(dbproc, "sp_helpfile");
				dbsqlexec(dbproc);
				dbresults(dbproc);
				DBCHAR szName[MAX_PATH];
				dbbind(dbproc, 3, STRINGBIND, (DBINT)MAX_PATH, (BYTE*)szName);
				dbnextrow(dbproc);
				szName[dbdatlen(dbproc, 3)] = '\0';
				strcpy(szDBFile, szName);
				dbnextrow(dbproc);
				szName[dbdatlen(dbproc, 3)] = '\0';
				strcpy(szLogFile, szName);
				bRet = TRUE;
			}
			dbclose(dbproc);
			if (bRet)
			{
				bRet = FALSE;
				dbproc = dbopen(logrec, g_szServer);
				if (dbproc)
				{
					if (dbuse(dbproc, "master") == SUCCEED)
					{
						char szCmd[MAX_PATH + 256];
						sprintf(szCmd, "sp_detach_db '%s'", szDBName);
						dbcmd(dbproc, szCmd);
						dbsqlexec(dbproc);
						ZeroMemory(szCmd, MAX_PATH + 256);
						sprintf(szCmd, "xp_cmdshell 'del \"%s\"'", szLogFile);
						dbcmd(dbproc, szCmd);
						dbsqlexec(dbproc);
						bRet = TRUE;
					}
					dbclose(dbproc);
					if (bRet)
					{
						bRet = FALSE;
						dbproc = dbopen(logrec, g_szServer);
						if (dbproc)
						{
							if (dbuse(dbproc, "master") == SUCCEED)
							{
								char szCmd[MAX_PATH + 256];
								sprintf(szCmd, "sp_attach_single_file_db '%s','%s'", szDBName, szDBFile);
								dbcmd(dbproc, szCmd);
								dbsqlexec(dbproc);
								bRet = TRUE;
							}
							dbclose(dbproc);
						}
					}
				}
			}
		}
		dbfreelogin(logrec);
	}
	return bRet;
}

int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
	MSG msg;
	UINT nWidth = 260;
	UINT nHeight = 295;

	MyRegisterClass(hInstance);

	g_hInst = hInstance;
	int iSX, iSY;
	iSX = GetSystemMetrics(SM_CXSCREEN);
	iSY = GetSystemMetrics(SM_CYSCREEN);
	HWND hWnd = CreateWindowEx(0, "LZHXZY", "Clear Log File", WS_VISIBLE | WS_SYSMENU,
		(iSX - nWidth) / 2, (iSY - nHeight) / 2, nWidth, nHeight, NULL, NULL, hInstance, NULL);

	if (!hWnd)
		return FALSE;

	g_hWnd = hWnd;

	ShowWindow(hWnd, nCmdShow);
	UpdateWindow(hWnd);

	while (GetMessage(&msg, NULL, 0, 0)) 
	{
		if (msg.message == WM_CHAR)
		{
			switch (msg.wParam)
			{
			case VK_TAB:
				SetFocus(GetNextDlgTabItem(g_hWnd, GetFocus(), HIBYTE(GetKeyState(VK_SHIFT)) != 0));
				break;
			case VK_RETURN:
				hWnd = GetFocus();
				if (GetWindowLong(hWnd, GWL_ID) == 2468)
					SendMessage(g_hWnd, WM_COMMAND, MAKELPARAM(BN_CLICKED, 0), (LPARAM)hWnd);
			}
		}
		TranslateMessage(&msg);
		DispatchMessage(&msg);
	}
	return msg.wParam;
}

ATOM MyRegisterClass(HINSTANCE hInstance)
{
	WNDCLASSEX wcex;

	wcex.cbSize = sizeof(WNDCLASSEX); 

	wcex.style			= CS_HREDRAW | CS_VREDRAW;
	wcex.lpfnWndProc	= (WNDPROC)WndProc;
	wcex.cbClsExtra		= 0;
	wcex.cbWndExtra		= 0;
	wcex.hInstance		= hInstance;
	wcex.hIcon			= LoadIcon(hInstance, (LPCTSTR)IDI_CLEARLOG);
	wcex.hCursor		= LoadCursor(NULL, IDC_ARROW);
	wcex.hbrBackground	= GetSysColorBrush(COLOR_BTNFACE);
	wcex.lpszMenuName	= NULL;
	wcex.lpszClassName	= "LZHXZY";
	wcex.hIconSm		= 0;

	return RegisterClassEx(&wcex);
}

LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
	MENUITEMINFO mii;
	MSGBOXPARAMS mp;
	int i, iCount;
	
	switch (message) 
	{
	case WM_CREATE:
		mii.cbSize = sizeof(MENUITEMINFO);
		mii.fMask = MIIM_TYPE;
		mii.fType = MFT_SEPARATOR;
		InsertMenuItem(GetSystemMenu(hWnd, FALSE), 5, TRUE, &mii);
		mii.fMask |= MIIM_ID;
		mii.fType = MFT_STRING;
		mii.wID = ID_SENDMAIL;
		mii.dwTypeData = "Send mail...";
		mii.cch = 10;
		InsertMenuItem(GetSystemMenu(hWnd, FALSE), 6, TRUE, &mii);
		mii.wID = ID_ABOUT;
		mii.dwTypeData = "About ClearLog...";
		mii.cch = 17;
		InsertMenuItem(GetSystemMenu(hWnd, FALSE), 7, TRUE, &mii);
		InitCommonControls();
		CreateWindowEx(WS_EX_RIGHT, "STATIC", "SQL Server:", WS_VISIBLE | WS_CHILD, 10, 7, 80, 24, hWnd, 0, g_hInst, NULL);
		CreateWindowEx(WS_EX_RIGHT, "STATIC", "User Name:", WS_VISIBLE | WS_CHILD, 10, 34, 80, 24, hWnd, 0, g_hInst, NULL);
		CreateWindowEx(WS_EX_RIGHT, "STATIC", "Password:", WS_VISIBLE | WS_CHILD, 10, 61, 80, 24, hWnd, 0, g_hInst, NULL);
		g_hwndServer = CreateWindowEx(WS_EX_CLIENTEDGE, "EDIT", "", WS_VISIBLE | WS_CHILD | WS_TABSTOP, 95, 5, 150, 24, hWnd, 0, g_hInst, NULL);
		g_hwndUserName = CreateWindowEx(WS_EX_CLIENTEDGE, "EDIT", "", WS_VISIBLE | WS_CHILD | WS_TABSTOP, 95, 32, 150, 24, hWnd, 0, g_hInst, NULL);
		g_hwndPassword = CreateWindowEx(WS_EX_CLIENTEDGE, "EDIT", "", WS_VISIBLE | WS_CHILD | ES_PASSWORD | WS_TABSTOP, 95, 59, 150, 24, hWnd, 0, g_hInst, NULL);
		g_hwndDatabase = CreateWindowEx(WS_EX_CLIENTEDGE, "LISTBOX", "", WS_VISIBLE | WS_CHILD | WS_VSCROLL | LBS_MULTIPLESEL | LBS_NOTIFY | LBS_EXTENDEDSEL | LBS_NOINTEGRALHEIGHT | WS_TABSTOP, 10, 88, 138, 150, hWnd, 0, g_hInst, NULL);
		g_hwndConnect = CreateWindowEx(0, "BUTTON", "Connect", WS_VISIBLE | WS_CHILD | WS_TABSTOP, 155, 89, 90, 28, hWnd, (HMENU)2468, g_hInst, NULL);
		g_hwndSelAll = CreateWindowEx(0, "BUTTON", "Select All", WS_VISIBLE | WS_CHILD | WS_TABSTOP, 155, 129, 90, 28, hWnd, (HMENU)2468, g_hInst, NULL);
		g_hwndUnselect = CreateWindowEx(0, "BUTTON", "Unselect", WS_VISIBLE | WS_CHILD | WS_TABSTOP, 155, 169, 90, 28, hWnd, (HMENU)2468, g_hInst, NULL);
		g_hwndClear = CreateWindowEx(0, "BUTTON", "Clear Log", WS_VISIBLE | WS_CHILD | WS_TABSTOP, 155, 209, 90, 28, hWnd, (HMENU)2468, g_hInst, NULL);
		g_hwndProgress = CreateWindowEx(0, "msctls_progress32", "", WS_VISIBLE | WS_CHILD | PBS_SMOOTH, 10, 244, 235, 20, hWnd, 0, g_hInst, NULL);
		SendMessage(g_hwndProgress, PBM_SETSTEP, 1, 0);
		SendMessage(g_hwndProgress, PBM_SETBARCOLOR, 0, RGB(0, 0, 255));
		SendMessage(g_hwndServer, EM_LIMITTEXT, MAX_COMPUTERNAME_LENGTH, 0);
		SendMessage(g_hwndUserName, EM_LIMITTEXT, MAX_PATH, 0);
		SendMessage(g_hwndPassword, EM_LIMITTEXT, MAX_PATH, 0);
		EnableWindow(g_hwndConnect, FALSE);
		EnableWindow(g_hwndClear, FALSE);
		EnableWindow(g_hwndSelAll, FALSE);
		EnableWindow(g_hwndUnselect, FALSE);
		SetFocus(g_hwndServer);
		break;
	case WM_SYSCOMMAND:
		switch (LOWORD(wParam))
		{
		case ID_SENDMAIL:
			ShellExecute(hWnd, "open", "mailto:xzylzh@163.net", NULL, NULL, SW_SHOW);
			break;
		case ID_ABOUT:
			mp.cbSize = sizeof(MSGBOXPARAMS);
			mp.dwStyle = MB_USERICON;
			mp.hInstance = g_hInst;
			mp.hwndOwner = g_hWnd;
			mp.lpszCaption = "About ClearLog";
			mp.lpszIcon = MAKEINTRESOURCE(IDI_CLEARLOG);
			mp.lpszText = "ClearLog Version 1.0.0\n\nCopyright (C) 2003-2004 XZY. All rights reserved.";
			MessageBoxIndirect(&mp);
		}
		break;
	case WM_COMMAND:
		switch (HIWORD(wParam))
		{
		case EN_CHANGE:
			EnableWindow(g_hwndConnect, GetWindowTextLength(g_hwndServer) > 0 && GetWindowTextLength(g_hwndUserName) > 0);
			break;
		case BN_CLICKED:
			if ((HWND)lParam == g_hwndConnect)
			{
				GetWindowText(g_hwndServer, g_szServer, 16);
				GetWindowText(g_hwndUserName, g_szUserName, 128);
				GetWindowText(g_hwndPassword, g_szPassword, 128);
				SendMessage(g_hwndDatabase, LB_RESETCONTENT, 0, 0);
				EnableWindow(g_hwndClear, FALSE);
				EnableWindow(g_hwndUnselect, FALSE);
				if (LoadDLLProc())
				{
					AddDatabase();
					FreeDLLProc();
				}
				else
					MessageBox(g_hWnd, szError, "Error", MB_ICONERROR);
				EnableWindow(g_hwndSelAll, SendMessage(g_hwndDatabase, LB_GETCOUNT, 0, 0) > 0);
			}
			else if ((HWND)lParam == g_hwndSelAll)
			{
				iCount = SendMessage(g_hwndDatabase, LB_GETCOUNT, 0, 0);
				for (i = 0; i < iCount; i ++)
					SendMessage(g_hwndDatabase, LB_SETSEL, 1, i);
				EnableWindow(g_hwndUnselect, TRUE);
				EnableWindow(g_hwndSelAll, FALSE);
				EnableWindow(g_hwndClear, TRUE);
			}
			else if ((HWND)lParam == g_hwndUnselect)
			{
				iCount = SendMessage(g_hwndDatabase, LB_GETCOUNT, 0, 0);
				for (i = 0; i < iCount; i ++)
					SendMessage(g_hwndDatabase, LB_SETSEL, 0, i);
				EnableWindow(g_hwndSelAll, TRUE);
				EnableWindow(g_hwndUnselect, FALSE);
				EnableWindow(g_hwndClear, FALSE);
			}
			else if ((HWND)lParam == g_hwndClear)
			{
				SendMessage(g_hwndProgress, PBM_SETRANGE, 0, MAKELPARAM(0, SendMessage(g_hwndDatabase, LB_GETSELCOUNT, 0, 0)));
				iCount = SendMessage(g_hwndDatabase, LB_GETCOUNT, 0, 0);
				BOOL bSuccess = FALSE;
				if (LoadDLLProc())
				{
					for (i = 0; i < iCount; i ++)
					{
						if (SendMessage(g_hwndDatabase, LB_GETSEL, i, 0) > 0)
						{
							char szDBFile[MAX_PATH], szLogFile[MAX_PATH], szDBName[128];
							DWORD dwLen = SendMessage(g_hwndDatabase, LB_GETTEXT, i, (LPARAM)szDBName);
							if (ClearLogFile(szDBName, szDBFile, szLogFile))
								SendMessage(g_hwndDatabase, LB_SETSEL, 0, i);
							SendMessage(g_hwndProgress, PBM_STEPIT, 0, 0);
						}
					}
					FreeDLLProc();
					bSuccess = TRUE;
				}
				else
					MessageBox(g_hWnd, szError, "Error", MB_ICONERROR);
				if (bSuccess)
					MessageBox(g_hWnd, "Clear log file successfully.", "Information", MB_ICONINFORMATION);
				SendMessage(g_hwndProgress, PBM_SETPOS, 0, 0);
			}
			break;
		case LBN_SELCHANGE:
			EnableWindow(g_hwndClear, SendMessage(g_hwndDatabase, LB_GETSELCOUNT, 0, 0) > 0);
			EnableWindow(g_hwndUnselect, SendMessage(g_hwndDatabase, LB_GETSELCOUNT, 0, 0) > 0);
			EnableWindow(g_hwndSelAll, SendMessage(g_hwndDatabase, LB_GETSELCOUNT, 0, 0) != SendMessage(g_hwndDatabase, LB_GETCOUNT, 0, 0));
		}
		break;
	case WM_DESTROY:
		PostQuitMessage(0);
   }
   return DefWindowProc(hWnd, message, wParam, lParam);
}
💿 文件大小 8 K
👤 上传用户 a1a1J0
📂 所属分类 SQL Server
🏷️ 相关标签

#Server #SQL #数据库 #日志
⌨️ 快捷键说明

复制代码 Ctrl + C
搜索代码 Ctrl + F
全屏模式 F11
切换主题 Ctrl + Shift + D
显示快捷键 ?
增大字号 Ctrl + =
减小字号 Ctrl + -