#include <windows.h>
#include <stdio.h>
#include <string.h>
#include <winlib/winlib.h>

#include "myping.h"


BOOL CALLBACK MainWndProc(HWND hwnd, UINT msg, WPARAM wParam,
               LPARAM lParam)
{
       WORD id;
       static VARS *v;
       char s[82];

       switch (msg) {
               case WM_INITDIALOG:
                       v = (VARS*)lParam;

                       memset(s, 0, 82);
                       if (! GetRegVal("Software\\MyPing", "host", (void*)s, 80))
                               strcpy(s, "localhost");
                       SetDlgItemText(hwnd, IDC_HOST, (LPCTSTR)s);

                       memset(s, 0, 82);
                       if (! GetRegVal("Software\\MyPing", "interval", (void*)s, 80))
                               strcpy(s, "1000");
                       SetDlgItemText(hwnd, IDC_INT, (LPCTSTR)s);

                       SetStatusParts(GetDlgItem(hwnd, IDC_STATUS), 2);
                       break;

               case WM_COMMAND:
                       id = LOWORD(wParam);

                       switch (id) {
                               case IDC_START:
                                       if (v->hIP)
                                               break;

                                       memset(s, 0, 80);
                                       GetDlgItemText(hwnd, IDC_HOST, s, 80);
                                       if (! (v->phostent = gethostbyname(s))) {
                                               MessageBox(hwnd, "Couldn't resolve host", "MyPing",
                                                               MB_OK | MB_ICONERROR);
                                               break;
                                       }

                                       v->dwIPAddr = (DWORD *)(*(v->phostent->h_addr_list));

                                       v->hIP = v->pIcmpCreateFile();
                                       v->I.Ttl = 6;

                                       memset(s, 0, 80);
                                       GetDlgItemText(hwnd, IDC_INT, s, 80);
                                       SetTimer(hwnd, 1, atoi(s), NULL);

                                       ShowStatus(GetDlgItem(hwnd, IDC_STATUS), 0, "Running");
                                       break;

                               case IDC_STOP:
                                       if (! v->hIP)
                                               break;

                                       KillTimer(hwnd, 1);
                                       v->pIcmpCloseHandle(v->hIP);
                                       v->hIP = 0;

                                       ShowStatus(GetDlgItem(hwnd, IDC_STATUS), 0, "");
                                       break;

                               default:
                                       return FALSE;
                       }
                       break;

               case WM_TIMER:
                       v->pIcmpSendEcho(v->hIP, *(v->dwIPAddr), 0, 0, &v->I, &v->E,
                                       sizeof(v->E), 8000);
                       break;

               case WM_CLOSE:
                       DestroyWindow(hwnd);
                       break;

               case WM_DESTROY:
                       if (v->hIP) {
                               KillTimer(hwnd, 1);
                               v->pIcmpCloseHandle(v->hIP);
                       }

                       memset(s, 0, 82);
                       GetDlgItemText(hwnd, IDC_HOST, s, 80);
                       SetRegVal("Software\\MyPing", "host", (void*)s, strlen(s));

                       memset(s, 0, 82);
                       GetDlgItemText(hwnd, IDC_INT, s, 80);
                       SetRegVal("Software\\MyPing", "interval", (void*)s, strlen(s));

                       FreeLibrary(v->hIcmp);
                       WSACleanup();
                       PostQuitMessage(0);
                       break;

               default:
                       return FALSE;
       }
       return TRUE;
}


static VARS* Init(HINSTANCE hInst, LPSTR cmdLine, INT nCmdShow)
{
       WNDCLASS wc;
       VARS *v;

       v = (VARS*)malloc(sizeof(VARS));

       v->hInst = hInst;

       v->hIcmp = LoadLibrary("ICMP.DLL");
       v->hIP = 0;

       v->pIcmpCreateFile = GetProcAddress(v->hIcmp, "IcmpCreateFile");
       v->pIcmpCloseHandle = GetProcAddress(v->hIcmp, "IcmpCloseHandle");
       v->pIcmpSendEcho = GetProcAddress(v->hIcmp, "IcmpSendEcho");

       WSAStartup(0x0101, &v->wsa);

       GetClassInfo(NULL, "#32770", &wc);
       wc.lpszClassName = "MainWndClass";
       if (! RegisterClass(&wc))
               return NULL;

       v->hwndMain = CreateDialogParam(v->hInst, "MainWnd", 0,
                       MainWndProc, (LPARAM)v);

       ShowWindow(v->hwndMain, nCmdShow);
       return v;
}


int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
               LPSTR lpCmdLine, INT nCmdShow)
{
       MSG msg;
       VARS *v;
       HACCEL hAccel;

       if (! (v = Init(hInstance, lpCmdLine, nCmdShow)))
               return 0;

       while (GetMessage(&msg, NULL, NULL, NULL)) {
               if (! IsDialogMessage(v->hwndMain, &msg)) {
                       TranslateMessage(&msg);
                       DispatchMessage(&msg);
               }
       }

       return msg.wParam;
}

You are viewing proxied material from bitreich.org. The copyright of proxied material belongs to its original authors. Any comments or complaints in relation to proxied material should be directed to the original authors of the content concerned. Please see the disclaimer for more details.