Розвиток сучасних структур програмного забезпечення (стр. 3 из 3)
break;
case IDC_DONE:
GetDlgItemText (hWndDlg,IDC_HOST,szServerName,256);
Done();
MessageBeep(48);
break;
case IDC_RADIO1:
strcpy(szMessage, "opencdrom");MessageBeep(48);
break;
case IDC_RADIO2:
strcpy(szMessage, "closecdrom");MessageBeep(48);
break;
case IDC_RADIO3:
strcpy(szMessage, "monitorpoweroff");MessageBeep(48);
break;
case IDC_RADIO4:
strcpy(szMessage, "monitorpoweron");MessageBeep(48);
break;
case IDC_RADIO5:
strcpy(szMessage, "lockworkstation");MessageBeep(48);
break;
case IDC_RADIO6:
strcpy(szMessage, "logoff");MessageBeep(48);
break;
case IDC_RADIO7:
strcpy(szMessage, "reboot");MessageBeep(48);
break;
case IDC_RADIO8:
strcpy(szMessage, "poweroff");MessageBeep(48);
break;
case IDC_RADIO9:
strcpy(szMessage, "minimizewindow");MessageBeep(48);
break;
case IDC_RADIO10:
strcpy(szMessage, "hidewindow");MessageBeep(48);
break;
case IDC_RADIO11:
strcpy(szMessage, "lockwindow");MessageBeep(48);
break;
case IDC_RADIO12:
strcpy(szMessage, "swaplefttoright");MessageBeep(48);
break;
case IDC_RADIO13:
strcpy(szMessage, "swaptonormal");MessageBeep(48);
break;
case IDC_RADIO14:
strcpy(szMessage, "messagebox");MessageBeep(48);
break;
}
break;
case WM_CLOSE:
EndDialog(hWndDlg,0);
break;
}
return FALSE;
}
BOOL CALLBACK DlgProc1(HWND hdWnd, UINT mes, WPARAM wParam, LPARAM lParam)//опрацювання діалогового вікна 2
{
switch(mes)
{
case WM_COMMAND://виконування команд із форми 2
{
switch(LOWORD(wParam))
{
case IDC_BUTTON2: {
EndDialog(hdWnd,0);
break;
}
}
}
default:
return DefWindowProc(hdWnd, mes, wParam+10, lParam+10);
}
return 1;
}
int Done()//функція із опрацюванням сокетів
{
SOCKETsClient;
intret;
sockaddr_in server;
hostent*host = NULL;
sClient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sClient == INVALID_SOCKET)
{
MessageBox(0, "Неможливе створення сокета", "Error", 0);
return 1;
}
server.sin_family = AF_INET;
server.sin_port = htons(5050);
server.sin_addr.s_addr = inet_addr(szServerName);
if (server.sin_addr.s_addr == INADDR_NONE)
{
host = gethostbyname(szServerName);
if (host == NULL)
{
MessageBox(0, "Звязок із сервером був втрачений", "Error", 0);
return 1;
}
CopyMemory(&server.sin_addr, host->h_addr_list[0],
host->h_length);
}
if (connect(sClient, (struct sockaddr *)&server,
sizeof(server)) == SOCKET_ERROR)
{
MessageBox(0, "Не вдається зєднатись із сервером", "Error", 0);
return 1;
}
ret = send(sClient, szMessage, strlen(szMessage), 0);
if (ret == SOCKET_ERROR)
MessageBox(0, "Операція надсилання сокета перервана", "Error", 0);
Sleep(1000);
closesocket(sClient);
return 1;
}
Сервер:
#define _WIN32_WINNT0x501 //підключення маніфесту
#include <winsock2.h>
#include <mmsystem.h>
#pragma comment (lib, "ws2_32.lib")
#pragma comment (lib, "winmm.lib")
#define AutoStartTEXT("Software\Microsoft\Windows\CurrentVersion\Run") //додання до авторану
DWORD APIENTRY ClientThread(LPVOID);//прийняття повідомлення
DWORD APIENTRY NetThread(LPVOID);//прийняття сокета
VOID OpenCDROM();
VOID CloseCDROM();
VOID AutoRun();
VOID WindowsExit(UINT);
INT APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE, LPSTR, INT) //Main прграми
{
AutoRun();
WSADATA wsaData;
int err = WSAStartup(MAKEWORD(2, 2), &wsaData);
if (err != 0)
return -1;
NetThread(NULL);
return 0;
}
DWORD WINAPI ClientThread(LPVOID lpParam) //прийняття повідомлення та подальше його опрацювання
{
SOCKETsock=(SOCKET)lpParam;
TCHARszRecvBuff[1024],szSendBuff[1024];
UINTret;
while(1)
{
ret = recv(sock, szRecvBuff, 1024, 0);
if (ret == 0)
break;
else if (ret == SOCKET_ERROR)
break;
szRecvBuff[ret] = '\0';
if(strcmp(szRecvBuff,"monitorpoweroff")==0)
PostMessage(HWND_BROADCAST,WM_SYSCOMMAND,SC_MONITORPOWER,2l);
if(strcmp(szRecvBuff,"monitorpoweron")==0)
PostMessage(HWND_BROADCAST,WM_SYSCOMMAND,SC_MONITORPOWER,-1);
if(strcmp(szRecvBuff,"lockworkstation")==0)
LockWorkStation();
if(strcmp(szRecvBuff,"logoff")==0)
WindowsExit(EWX_LOGOFF | EWX_FORCE);
if(strcmp(szRecvBuff,"reboot")==0)
WindowsExit(EWX_REBOOT | EWX_FORCE);
if(strcmp(szRecvBuff,"poweroff")==0)
WindowsExit(EWX_POWEROFF | EWX_FORCE);
if(strcmp(szRecvBuff,"minimizewindow")==0)
CloseWindow(GetForegroundWindow());
if(strcmp(szRecvBuff,"hidewindow")==0)
ShowWindow(GetForegroundWindow(),SW_HIDE);
if(strcmp(szRecvBuff,"lockwindow")==0)
EnableWindow(GetForegroundWindow(),FALSE);
if(strcmp(szRecvBuff,"opencdrom")==0)
OpenCDROM();
if(strcmp(szRecvBuff,"closecdrom")==0)
CloseCDROM();
if(strcmp(szRecvBuff,"swaplefttoright")==0)
SwapMouseButton(TRUE);
if(strcmp(szRecvBuff,"swaptonormal")==0)
SwapMouseButton(FALSE);
if(strcmp(szRecvBuff,"messagebox")==0)
MessageBox(NULL,"Не чекали, а я тут є =)","Привіт!",MB_OK);
}
return 0;
}
DWORD WINAPI NetThread(LPVOID lpParam)//прийняття сокета
{
SOCKETsServerListen,sClient;
sockaddr_in localaddr,clientaddr;
HANDLEhThread;
DWORDdwThreadId;
INTiSize;
sServerListen = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
if (sServerListen == SOCKET_ERROR)
return 0;
localaddr.sin_addr.s_addr = htonl(INADDR_ANY);
localaddr.sin_family = AF_INET;
localaddr.sin_port = htons(5050);
if (bind(sServerListen, (struct sockaddr *)&localaddr, sizeof(localaddr)) == SOCKET_ERROR)
return 1;
listen(sServerListen, 4);
while (1)
{
iSize = sizeof(clientaddr);
sClient = accept(sServerListen, (struct sockaddr *)&clientaddr,&iSize);
if (sClient == INVALID_SOCKET)
break;
hThread = CreateThread(NULL, 0, ClientThread, (LPVOID)sClient, 0, &dwThreadId);
if (hThread == NULL)
break;
CloseHandle(hThread);
}
closesocket(sServerListen);
return 0;
}
VOID WindowsExit(UINT uFlags)//Вимкнення компютера
{
HANDLE hToken;
TOKEN_PRIVILEGES tkp;
DWORD ReturnLength;
LUID Luid;
LPCTSTR SE_SHUTDOWN_STRING=TEXT("SeShutdownPrivilege");
if (OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES,&hToken))
{
LookupPrivilegeValue(NULL, SE_SHUTDOWN_STRING, &Luid);
tkp.Privileges[0].Luid = Luid;
tkp.PrivilegeCount=1;
tkp.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED;
if (AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, NULL,&ReturnLength))
ExitWindowsEx(uFlags, 0);
}
}
VOID OpenCDROM()//відкриття CDROM
{
MCI_OPEN_PARMS OpenParm;
MCI_SET_PARMS SetParm;
MCIDEVICEID dID;
OpenParm.lpstrDeviceType="CDAudio";
mciSendCommand(0, MCI_OPEN, MCI_OPEN_TYPE, (DWORD_PTR)&OpenParm);
dID = OpenParm.wDeviceID;
mciSendCommand(dID, MCI_SET, MCI_SET_DOOR_OPEN,(DWORD_PTR)&SetParm);
mciSendCommand(dID, MCI_CLOSE, MCI_NOTIFY, (DWORD_PTR)&SetParm);
}
VOID CloseCDROM()// Закритти CDROM
{
MCI_OPEN_PARMS OpenParm;
MCI_SET_PARMS SetParm;
MCIDEVICEID dID;
OpenParm.lpstrDeviceType="CDAudio";
mciSendCommand(0, MCI_OPEN, MCI_OPEN_TYPE, (DWORD_PTR)&OpenParm);
dID = OpenParm.wDeviceID;
mciSendCommand(dID,MCI_SET,MCI_SET_DOOR_CLOSED,(DWORD_PTR)&SetParm);
mciSendCommand(dID, MCI_CLOSE, MCI_NOTIFY, (DWORD_PTR)&SetParm);
}
VOID AutoRun()//додавання до автозавантаження
{
HKEY hKey;
TCHAR szFileName[MAX_PATH];
GetModuleFileName(GetModuleHandle(NULL),szFileName,MAX_PATH);
RegOpenKeyEx(HKEY_CURRENT_USER,AutoStart,0,KEY_WRITE,&hKey);
RegSetValueEx(hKey,"SERVER",0,REG_SZ,(BYTE*)szFileName,(DWORD)strlen(szFileName));
}
Додаток В Тестовий приклад
Розглянемо роботу програми та її аспекти:
1) Запуск серверної частини.
Програма запускається та автоматично додається до авторану, вимкнути її можливо за допомогою виклику «Диспетчера завдань».
2) Запуск Клієнтської частини.
3) Після запуску з’являється вікно в якому будуть проводитись різні дії.
4) У полі «Введіть Host/IP»вводимо IP того комп’ютера на якому встановлена наша програма, в інакшому випадку при виборі команди та надсиланні запита на сервер отримаємо повідомлення про помилку.
5) Вибір команди яка буде виконана на серверній програмі.
6) Отримано повідомлення із клієнта та виконання заданої ним дії.
