#include <WinSock2.h>
#include <Windows.h>
#include <iostream>
#include <tchar.h>
#include <strsafe.h>

void ShowErrAndExit(PTSTR lpszFunction)
{
   LPVOID lpMsgBuf;
   LPVOID lpDisplayBuf;
   DWORD dw = GetLastError();

   FormatMessage(
      FORMAT_MESSAGE_ALLOCATE_BUFFER |
      FORMAT_MESSAGE_FROM_SYSTEM |
      FORMAT_MESSAGE_IGNORE_INSERTS,
      NULL,
      dw,
      MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
      (LPTSTR) &lpMsgBuf,
      0, NULL );

   lpDisplayBuf = (LPVOID)LocalAlloc(LMEM_ZEROINIT,
      (lstrlen((LPCTSTR)lpMsgBuf)+lstrlen((LPCTSTR)lpszFunction)+40)*sizeof(TCHAR));
   StringCchPrintf((LPTSTR)lpDisplayBuf,
      LocalSize(lpDisplayBuf) / sizeof(TCHAR),
      TEXT("%s failed with error %d: %s"),
      lpszFunction, dw, lpMsgBuf);
   std::cout << lpDisplayBuf << std::endl;

   LocalFree(lpMsgBuf);
   LocalFree(lpDisplayBuf);
   ExitProcess(1);
}

int _tmain()
{
   // startup network
   WSADATA wsaData;
   if (WSAStartup(MAKEWORD(2,2), &wsaData) != 0)
      ShowErrAndExit(TEXT("WSAStartup"));

   // create network socket
   SOCKET hSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
   if (hSocket == INVALID_SOCKET)
      ShowErrAndExit(TEXT("Socket"));

   // connect
   sockaddr_in modem;
   modem.sin_family = AF_INET;
   modem.sin_port=htons(80);
   modem.sin_addr.s_addr = inet_addr("192.168.1.254");
   if (connect(hSocket, (sockaddr*) &modem, sizeof(sockaddr_in)) == SOCKET_ERROR)
      ShowErrAndExit(TEXT("Connect"));

   // send reboot string
   char* lpData = "GET /doc/doreboot.htm HTTP/1.1\r\nHost: 192.168.1.254\r\nConnection: keep-alive\r\nAuthorization: Basic YWRtaW46cGFzc3dvcmQ=\r\n\r\n";
   if (send(hSocket, lpData, (int) strlen(lpData) + 1, NULL) == SOCKET_ERROR)
      ShowErrAndExit(TEXT("Send"));

   // cleanup
   closesocket(hSocket);

   bool reboot = false;
   do
   {
      // wait 50 seconds
      for (int x = 0; x < 50; x++)
         Sleep (1000);

      // check for connect twice
      if (gethostbyname("www.google.com") == NULL && gethostbyname("www.yahoo.com") == NULL)
      {
         Sleep(10000);
         if (gethostbyname("www.google.com") == NULL && gethostbyname("www.yahoo.com") == NULL)
            reboot = true;
         else
            reboot = false;
      }
      else
         reboot = false;

      // connection still not established, reboot modem
      if (reboot == true)
      {
         hSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

         modem.sin_family = AF_INET;
         modem.sin_port=htons(80);
         modem.sin_addr.s_addr = inet_addr("192.168.1.254");
         if (connect(hSocket, (sockaddr*) &modem, sizeof(sockaddr_in)) == SOCKET_ERROR)
            ShowErrAndExit(TEXT("Connect"));

         if (send(hSocket, lpData, (int) strlen(lpData) + 1, NULL) == SOCKET_ERROR)
            ShowErrAndExit(TEXT("Send"));

         closesocket(hSocket);
      }
   } while (reboot != false); // while reboot = true

   // cleanup
   WSACleanup();
   return 0;
}