home *** CD-ROM | disk | FTP | other *** search
/ PCNET 2006 September - Disc 1 / PCNET_CD_2006_09.iso / surpriz / MSRMesh-VirtualWIFI.MSI / virtualwifi.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2005-06-24  |  19.1 KB  |  626 lines
  1. /*
  2.  * Author   : Ranveer Chandra
  3.  * Directory: VirtualWiFi_Root\install
  4.  * File Name: VirtualWiFi.cpp
  5.  * Purpose  : This is the main VirtualWiFi installer file
  6.  */
  7.  
  8. //#include <nt.h>
  9. //#include <ntrtl.h>
  10. //#include <nturtl.h>
  11. #include <windows.h>
  12. #include <stdio.h>
  13. #include <stdlib.h>
  14. #include <winsock2.h>
  15. #include <ws2tcpip.h>
  16. #include <math.h>
  17. #include <string.h>
  18. #include <setupapi.h>
  19. #include "precomp.h"
  20.  
  21. //#include "types.h"
  22. //#include "ether.h"
  23. #include "ntddndis.h"
  24. #include "ioctl.h"
  25.  
  26. extern HRESULT AddOrRemoveVirtualWiFi(bool AddOrRemove, const char *SetupKitDir);
  27.  
  28. extern int ControlDeviceClass(DWORD Action);
  29.  
  30. extern VOID AddNetwork (HWND hwndDlg, BOOL addNetwork);
  31.  
  32. extern VOID RenameConnection ( LPCWSTR ConnectionName );
  33.  
  34. extern BOOL DoesConnectionExist ( LPCWSTR ConnectionName );
  35.  
  36. extern BOOL DoesEnabledConnectionExist ( LPCWSTR ConnectionName );
  37.  
  38. extern 
  39. HRESULT EnableDisableConnection ( LPCWSTR ConnectionName, BOOL enableConnection );
  40.  
  41. extern 
  42. int ServiceInstaller(char *action, char *pSvcPath);
  43.  
  44. HANDLE Handle;
  45.  
  46. bool Persistent = FALSE;
  47. bool Verbose = FALSE;
  48. extern bool AdminAccess = TRUE;
  49. bool ConvertHostName = TRUE; // Whether to query names or just print addrs.
  50.  
  51.  
  52. char *
  53. FormatGuid(const GUID *Guid)
  54. {
  55.     static char buffer[40];
  56.  
  57.     sprintf(buffer,
  58.             "{%08lX-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
  59.             Guid->Data1,
  60.             Guid->Data2,
  61.             Guid->Data3,
  62.             Guid->Data4[0],
  63.             Guid->Data4[1],
  64.             Guid->Data4[2],
  65.             Guid->Data4[3],
  66.             Guid->Data4[4],
  67.             Guid->Data4[5],
  68.             Guid->Data4[6],
  69.             Guid->Data4[7]);
  70.  
  71.     return buffer;
  72. }
  73.  
  74. WCHAR *
  75. MapAdapterGuidToFriendlyName(const GUID *Guid)
  76. {
  77.     static HMODULE hModule;
  78.     static HRESULT (WINAPI *pHrLanConnectionNameFromGuidOrPath)
  79.         (const GUID *pGuid, LPCWSTR pszwPath,
  80.          LPWSTR pszwName, LPDWORD pcchMax);
  81.  
  82.     //
  83.     // Get the conversion function, if we don't already have it.
  84.     //
  85.     if (hModule == NULL) {
  86.         char Path[MAX_PATH + 64];
  87.  
  88.         //
  89.         // Use an absolute path with LoadLibrary
  90.         // to prevent Prefast warning.
  91.         //
  92.         if (GetSystemDirectory(Path, MAX_PATH + 1) == 0)
  93.             return NULL;
  94.         strcat(Path, "\\netman.dll");
  95.  
  96.         hModule = LoadLibrary(Path);
  97.         if (hModule != NULL)
  98.             pHrLanConnectionNameFromGuidOrPath =
  99.                 (HRESULT (WINAPI *)(const GUID *, LPCWSTR, LPWSTR, LPDWORD))
  100.                 GetProcAddress(hModule, "HrLanConnectionNameFromGuidOrPath");
  101.     }
  102.  
  103.     if (pHrLanConnectionNameFromGuidOrPath != NULL) {
  104.         static WCHAR FriendlyName[2048];
  105.         DWORD FriendlyNameLength = 2048;
  106.  
  107.         if ((*pHrLanConnectionNameFromGuidOrPath)(Guid, NULL,
  108.                                                   FriendlyName,
  109.                                                   &FriendlyNameLength) == 0)
  110.             return FriendlyName;
  111.     }
  112.  
  113.     return NULL;
  114. }
  115.  
  116. DWORD WINAPI LoopOnce( LPVOID lpParam ) 
  117. { 
  118.     ULONG numVirtualWiFiNodes = *(ULONG *)lpParam;
  119.     char *adapterNum="1\0";
  120.     ULONG k = 0;
  121.  
  122.     Sleep(3000);
  123.     if(Verbose)
  124.         fprintf(stderr, "Number of virtual nodes %ld\n", numVirtualWiFiNodes);
  125.     for(k = 0; k < 3*numVirtualWiFiNodes; k++) {
  126.         ioctlSet(IOCTL_SWITCH_SSID, adapterNum, sizeof(adapterNum));
  127.         ioctlSet(IOCTL_SEND_BUFFERED_PACKETS, adapterNum, sizeof(adapterNum));
  128.     }
  129.  
  130.     return 0; 
  131. } 
  132.  
  133. void
  134. usage(void)
  135. {
  136.     fprintf(stdout, "usage: VirtualWiFi install\n"); //switch_time wait_time is_adaptive]\n");
  137.     fprintf(stdout, "       VirtualWiFi uninstall\n");
  138.     fprintf(stdout, "       VirtualWiFi addnetwork     -ssid <ssid> -mode <AH | IS>\n");// [-switch <switch_time> -wait <wait_time> -isadapt <is_adaptive>]\n");
  139.     fprintf(stdout, "       VirtualWiFi removenetwork  -ssid <ssid> -mode <AH | IS>\n");
  140.     fprintf(stdout, "       VirtualWiFi changeparams   -ssid <ssid> -mode <AH | IS>  -<param> <value>\n");
  141.     fprintf(stdout, "                  <param>: switch  <switch_time in ms>\n");
  142.     fprintf(stdout, "                           wait    <wait_time in ms>\n");
  143.     fprintf(stdout, "                           isadapt <is_adaptive 0 or 1>\n");
  144.  
  145.     exit(1);
  146. }
  147.  
  148. int __cdecl
  149. main(int argc, char **argv)
  150. {
  151.     WSADATA wsaData;
  152.     int Count;
  153.     FILE *fpConfig; 
  154.     HRESULT hr = S_OK;
  155.    
  156.     if (WSAStartup(MAKEWORD(2, 0), &wsaData)) {
  157.         fprintf(stderr, "WSAStartup failed\n");
  158.         exit(1);
  159.     }
  160.  
  161.     //
  162.     // Parse any global options.
  163.     //
  164.     for (Count = 1; Count < argc; Count++) {
  165.         if (!_stricmp(argv[Count], "-v")) {
  166.             Verbose = TRUE;
  167.             fprintf(stderr, "Set Verbose mode\n");
  168.         }
  169.         else if (!_stricmp(argv[Count], "-p"))
  170.             Persistent = TRUE;
  171.         else if (!_stricmp(argv[Count], "-d"))
  172.             ConvertHostName = FALSE;
  173.         else
  174.             break;
  175.     }
  176.  
  177.     argc -= Count;
  178.     argv += Count;
  179.  
  180.  
  181.     if (argc < 1)
  182.         usage();
  183.  
  184.     //
  185.     // Start with the commands which don't access the device.
  186.     //
  187.  
  188.     fprintf(stdout, " \n");
  189.  
  190.     if (!_stricmp(argv[0], "install")) {
  191.         char currentSSID[MAX_SSID_LENGTH];
  192.         char dataFileName[100];
  193.         char currentMode[3];
  194.         char connectionName[100];
  195.         wchar_t wConnectionName[100];
  196.         SetVirtualWiFiTimers setVirtualWiFiTimers;
  197.         char *stopstring;
  198.         char *temp = ".";
  199.  
  200.         if(Verbose) fprintf(stderr, "argc is %d\n", argc);
  201.  
  202.         fflush(stdout);
  203.  
  204.         if(isInstalledVirtualWiFi()) {
  205.             fprintf(stderr, "VirtualWiFi is already installed!\n");
  206.             exit(0);
  207.         }
  208.         /*
  209.         if(argc == 5) {
  210.             setVirtualWiFiTimers.adapterNum = "1";
  211.             setVirtualWiFiTimers.switchTime = strtol(argv[2], &stopstring, 10);
  212.             setVirtualWiFiTimers.waitTime = atol(argv[3]);
  213.             setVirtualWiFiTimers.isAdaptiveScheduling = atoi(argv[4]);
  214.         }
  215.         */
  216.  
  217.         if (GetAssoc() == TRUE) {
  218.             if(Verbose)
  219.                 fprintf(stderr, "Wireless card is associated\n");
  220.         }
  221.         else {
  222.             fprintf(stderr, "Your card is not connected to any network.\n");
  223.             fprintf(stderr, "Please connect your card any try again\n");
  224.             exit(0);
  225.         }    
  226.         
  227.         switch (argc) {
  228.         case 1:
  229.             fprintf(stdout, "Installing the VirtualWiFi Driver...\n");
  230.             hr = AddOrRemoveVirtualWiFi(TRUE, ".");
  231.             if (hr == S_OK) {
  232.                 fprintf(stdout, "Successfully installed the VirtualWiFi Driver.\n");
  233.                 fprintf(stdout, "Installing the VirtualWiFi Service ...\n");
  234.                 if (ServiceInstaller("install", temp) > 0)
  235.                     if(isInstalledVirtualWiFi())
  236.                         fprintf(stdout, "VirtualWiFi install complete.\n");
  237.                     else {
  238.                         fprintf(stdout, "Error: Check that name of your wireless adapter is in the keywords, FAQ#1\n");
  239.                         exit(0);
  240.                     }
  241.                 else {
  242.                     fprintf(stdout, "Error: Check that name of your wireless adapter is in the keywords, FAQ#1\n");
  243.                     exit(0);
  244.                 }
  245.             } else {
  246.                 fprintf(stdout, "Problem %d installing the VirtualWiFi Driver.\n", hr);
  247.                 exit(0);
  248.             }
  249.             break;
  250.  
  251.         case 2:
  252.             fprintf(stdout, "Installing the VirtualWiFi Driver...\n");
  253.             hr = AddOrRemoveVirtualWiFi(TRUE, argv[1]);
  254.             if (hr == S_OK) {
  255.                 fprintf(stdout, "Successfully installed the VirtualWiFi Driver.\n");
  256.                 fprintf(stdout, "Installing the VirtualWiFi Service ...\n");
  257.                 if(ServiceInstaller("install", argv[1]) > 0) {
  258.                     if(isInstalledVirtualWiFi())
  259.                         fprintf(stdout, "VirtualWiFi install complete.\n");
  260.                     else {
  261.                         fprintf(stdout, "Error: Check that name of your wireless adapter is in the keywords, FAQ#1\n");
  262.                         exit(0);
  263.                     }
  264.                 }
  265.                 else {
  266.                     fprintf(stdout, "Error: Check that name of your wireless adapter is in the keywords, FAQ#1\n");
  267.                     exit(0);
  268.                 }
  269.             } else {
  270.                 fprintf(stdout, "Problem %d installing the VirtualWiFi Driver.\n", hr);
  271.                 exit(0);
  272.             }
  273.             break;
  274.  
  275.         /*
  276.         case 5:
  277.             AddOrRemoveVirtualWiFi(TRUE, argv[1]);
  278.             ServiceInstaller("install", argv[1]);
  279.  
  280.  
  281.             if (Verbose)
  282.                 fprintf(stderr, "Setting switch time to %d, wait time to %d\n", 
  283.                     setVirtualWiFiTimers.switchTime, setVirtualWiFiTimers.waitTime);
  284.  
  285.             ioctlSet(IOCTL_SET_SWITCHING_PARAMETERS, &setVirtualWiFiTimers, 
  286.                 (DWORD)sizeof(SetVirtualWiFiTimers));
  287.  
  288.             break;
  289.  
  290.         
  291.         case 1:
  292.             AddOrRemoveVirtualWiFi(TRUE, NULL);
  293.             ServiceInstaller("install", NULL);
  294.             break;
  295.         */
  296.         default:
  297.             usage();
  298.         }
  299.         Sleep(1000);
  300.     
  301.         // Set the name of the connection
  302.         GetSSID(currentSSID);
  303.         if (Verbose)
  304.             fprintf(stderr, "Got the SSID %s\n", currentSSID);
  305.  
  306.         GetMode(currentMode);
  307.         if(Verbose)
  308.             fprintf(stderr, "Current Mode is %s\n", currentMode);
  309.         
  310.         sprintf(connectionName, "VirtualWiFi %s %s\0", currentMode, currentSSID);
  311.  
  312.         //fpConfig = fopen("C:\\Windows\\VirtualWiFiData.txt", "w");
  313.         sprintf(dataFileName, "%s\\VirtualWiFiData.txt", getenv("windir"));
  314.         fpConfig = fopen(dataFileName, "w");
  315.         fprintf(fpConfig, "%s\n", connectionName);
  316.         fclose(fpConfig);
  317.  
  318.         if(Verbose)
  319.             printf("Name the adapter to: %s\n", connectionName);
  320.         mbstowcs(wConnectionName, connectionName, strlen(connectionName)+1);
  321.  
  322.         RenameConnection(wConnectionName);
  323.  
  324.         exit(0);
  325.     }
  326.     
  327.     if (!_stricmp(argv[0], "uninstall")) {
  328.         HANDLE hThread; 
  329.         DWORD dwThreadId, dwThrdParam=1; 
  330.         ULONG numNodes=0;
  331.         char *adapterNum="1\0";
  332.  
  333.         if (argc != 1)
  334.             usage();
  335.         fprintf(stdout, "Uninstalling VirtualWiFi ...\n");
  336.         fprintf(stdout, "Stopping the VirtualWiFi Service.\n");
  337.         ServiceInstaller("uninstall", NULL);
  338.  
  339.         numNodes = ioctlGetFromAdapter(IOCTL_GET_NUM_NETWORKS, adapterNum, (DWORD)strlen(adapterNum), &numNodes, sizeof(ULONG));
  340.         hThread = CreateThread( 
  341.             NULL,                        // no security attributes 
  342.             0,                           // use default stack size  
  343.             LoopOnce,                     // thread function 
  344.             &numNodes,                     // argument to thread function
  345.             0,                           // use default creation flags 
  346.             &dwThreadId);                // returns the thread identifier 
  347.         if (hThread == NULL) {
  348.             printf("CreateThread failed." ); 
  349.         } else {
  350.             hr = AddOrRemoveVirtualWiFi(FALSE, NULL);
  351.             Sleep(1000);
  352.             CloseHandle( hThread );
  353.         }
  354.  
  355.         if (hr == S_OK) {
  356.             fprintf(stdout, "VirtualWiFi uninstalled.\n");
  357.         } else {
  358.             fprintf(stderr, "Problems uninstalling the VirtualWiFi driver.\n");
  359.         }
  360.         exit(0);
  361.     }
  362.     
  363.     if (!_stricmp(argv[0], "enable")) {
  364.         if (argc != 1)
  365.             usage();
  366.         Count = ControlDeviceClass(DICS_ENABLE);
  367.         printf("%u Adapter%s enabled.\n", Count, (Count == 1) ? "" : "s");
  368.         exit(0);
  369.     }
  370.  
  371.     if (!_stricmp(argv[0], "disable")) {
  372.         if (argc != 1)
  373.             usage();
  374.         Count = ControlDeviceClass(DICS_DISABLE);
  375.         printf("%u Adapter%s disabled.\n", Count, (Count == 1) ? "" : "s");
  376.         exit(0);
  377.     }
  378.  
  379.     if (!_stricmp(argv[0], "restart")) {
  380.         if (argc != 1)
  381.             usage();
  382.         Count = ControlDeviceClass(DICS_PROPCHANGE);
  383.         printf("%u Adapter%s restarted.\n", Count, (Count == 1) ? "" : "s");
  384.         exit(0);
  385.     }
  386.  
  387.     if (!_stricmp(argv[0], "changeparams")) {
  388.         SetVirtualWiFiTimersForConnection setVirtualWiFiTimersForConnection;
  389.         int count;
  390.         BOOL foundParamToSet = FALSE;
  391.         BOOL foundNetworkModeToSet = FALSE;
  392.         BOOL foundNetworkSSIDToSet = FALSE;
  393.  
  394.         if(!isInstalledVirtualWiFi()) {
  395.             fprintf(stderr, "Please install VirtualWiFi before this operation\n");
  396.             exit(0);
  397.         }
  398.         setVirtualWiFiTimersForConnection.adapterNum = "1";
  399.         setVirtualWiFiTimersForConnection.switchTime = 0;
  400.         setVirtualWiFiTimersForConnection.waitTime = 0;
  401.         setVirtualWiFiTimersForConnection.isAdaptiveScheduling = 2;
  402.  
  403.         for(count = 1; count < argc; count++) 
  404.         {
  405.             if(count == argc-1) {
  406.                 fprintf(stderr, "Incorrect arguments\n");
  407.                 usage();
  408.             }
  409.             if(!_stricmp(argv[count], "-switch")) {
  410.                 setVirtualWiFiTimersForConnection.switchTime = atol(argv[count+1]);
  411.                 foundParamToSet = TRUE;
  412.             } else if(!_stricmp(argv[count], "-wait")) {
  413.                 setVirtualWiFiTimersForConnection.waitTime = atol(argv[count+1]);
  414.                 foundParamToSet = TRUE;
  415.             } else if(!_stricmp(argv[count], "-isadapt")) {
  416.                 setVirtualWiFiTimersForConnection.isAdaptiveScheduling = atoi(argv[count+1]);
  417.                 foundParamToSet = TRUE;
  418.             }
  419.             else if(!_stricmp(argv[count], "-mode")) {
  420.                 if(!_stricmp(argv[count+1], "ah"))
  421.                     setVirtualWiFiTimersForConnection.networkMode = MODE_AH;
  422.                 else if(!_stricmp(argv[count+1], "is"))
  423.                     setVirtualWiFiTimersForConnection.networkMode = MODE_IS;
  424.                 else 
  425.                     usage();
  426.                 foundNetworkModeToSet = TRUE;
  427.             } else if(!_stricmp(argv[count], "-ssid")) {
  428.                 strcpy(setVirtualWiFiTimersForConnection.networkSSID, argv[count+1]);
  429.                 setVirtualWiFiTimersForConnection.ssidLength = (ULONG)strlen(setVirtualWiFiTimersForConnection.networkSSID);
  430.                 foundNetworkSSIDToSet = TRUE;
  431.             } else {
  432.                 fprintf(stderr, "Incorrect arguments\n");
  433.                 usage();
  434.             }
  435.             count+=1;
  436.         }
  437.         if(foundParamToSet && foundNetworkModeToSet && foundNetworkSSIDToSet) {
  438.             char connectionName[100];
  439.             wchar_t wConnectionName[100];
  440.             if(setVirtualWiFiTimersForConnection.networkMode == MODE_AH)
  441.                 sprintf(connectionName, "VirtualWiFi AH %s\0", setVirtualWiFiTimersForConnection.networkSSID);
  442.             else
  443.                 sprintf(connectionName, "VirtualWiFi IS %s\0", setVirtualWiFiTimersForConnection.networkSSID);
  444.             mbstowcs(wConnectionName, connectionName, strlen(connectionName)+1);
  445.                     
  446.             if(!DoesConnectionExist(wConnectionName)) {
  447.                 fprintf(stderr, "The specified connection does not exist\n");
  448.                 exit(0);
  449.             }
  450.             ServiceInstaller("stop", NULL);
  451.             Sleep(1000);
  452.             ioctlSet(IOCTL_SET_SWITCHING_PARAMETERS_FOR_CONNECTION, &setVirtualWiFiTimersForConnection, 
  453.                 (DWORD)sizeof(SetVirtualWiFiTimersForConnection));
  454.             Sleep(2000);
  455.             fprintf(stderr, "Set parameters successful\n");
  456.             ServiceInstaller("start", NULL);
  457.         } else {
  458.             fprintf(stderr, "Incorrect arguments. Specify SSID, mode and parameter\n");
  459.             usage();
  460.         }
  461.         exit(0);
  462.     }
  463.  
  464.     if (!_stricmp(argv[0], "addnetwork") || !_stricmp(argv[0], "removenetwork")) {
  465.         char newSSID[MAX_SSID_LENGTH];
  466.         char response[50];
  467.         char networkSSID[MAX_SSID_LENGTH];
  468.         char connectionName[100];
  469.         wchar_t wConnectionName[100];
  470.         USHORT mode;
  471.         int count = 0;
  472.         SetVirtualWiFiTimers setVirtualWiFiTimers;
  473.         bool userParameters = FALSE;
  474.         NetworkModeAndSSID networkModeAndSSID;
  475.  
  476.         if(!isInstalledVirtualWiFi()) {
  477.             fprintf(stderr, "Please install VirtualWiFi before this operation\n");
  478.             exit(0);
  479.         }
  480.         // take as input the parameters of the network
  481.         if(argc == 1) {
  482.             printf("Enter the SSID: ");
  483.             scanf("%s", newSSID);
  484.             printf("Enter Mode, ADHOC (0), INFRASTRUCTURE (1): ");
  485.             scanf("%d", &mode);
  486.         } else if(argc == 5) {
  487.             for(count = 1; count < argc; count++) {
  488.                 if(!_stricmp(argv[count], "-mode")) {
  489.                     if(!_stricmp(argv[count+1], "ah"))
  490.                         mode = MODE_AH;
  491.                     else if(!_stricmp(argv[count+1], "is"))
  492.                         mode = MODE_IS;
  493.                     else 
  494.                         usage();
  495.                 } else if(!_stricmp(argv[count], "-ssid")) {
  496.                     strcpy(newSSID, argv[count+1]);
  497.                 } else {
  498.                     fprintf(stderr, "Incorrect args\n");
  499.                     exit(0);
  500.                 }
  501.                 count+=1;
  502.             }
  503.         } else if(argc == 11 && !_stricmp(argv[0], "addnetwork")) {
  504.             for(count = 1; count < argc; count++) {
  505.                 if(!_stricmp(argv[count], "-mode")) {
  506.                     if(!_stricmp(argv[count+1], "ah"))
  507.                         mode = MODE_AH;
  508.                     else if(!_stricmp(argv[count+1], "is"))
  509.                         mode = MODE_IS;
  510.                     else 
  511.                         usage();
  512.                 } else if(!_stricmp(argv[count], "-ssid")) {
  513.                     strcpy(newSSID, argv[count+1]);
  514.                 } else if(!_stricmp(argv[count], "-switch")) {
  515.                     setVirtualWiFiTimers.switchTime = atol(argv[count+1]);
  516.                 } else if(!_stricmp(argv[count], "-wait")) {
  517.                     setVirtualWiFiTimers.waitTime = atol(argv[count+1]);
  518.                 } else if(!_stricmp(argv[count], "-isadapt")) {
  519.                     setVirtualWiFiTimers.isAdaptiveScheduling = atoi(argv[count+1]);
  520.                 } else {
  521.                     fprintf(stderr, "Incorrect args\n");
  522.                     exit(0);
  523.                 }
  524.                 count+=1;
  525.             }
  526.  
  527.             userParameters = TRUE;
  528.             setVirtualWiFiTimers.adapterNum = "1";
  529.         } else 
  530.             usage();
  531.  
  532.         count = 0;
  533.         if(mode == MODE_AH)
  534.             sprintf(connectionName, "VirtualWiFi AH %s\0", newSSID);
  535.         else
  536.             sprintf(connectionName, "VirtualWiFi IS %s\0", newSSID);
  537.  
  538.         if(Verbose) 
  539.             fprintf(stderr, "Check Name of the adapter: %s\n", connectionName);
  540.         mbstowcs(wConnectionName, connectionName, strlen(connectionName)+1);
  541.                 
  542.         if(!DoesEnabledConnectionExist(wConnectionName)) {
  543.             if(!_stricmp(argv[0], "addnetwork")) {
  544.                 ServiceInstaller("stop", NULL);
  545.                 /* This worked with WMI. Needed to change this for SP2 */
  546.                 /*    SetMode(mode);
  547.                     Sleep(50);
  548.                     SetSSID(newSSID);
  549.                     while((GetAssoc()!= S_OK) && count < 5) {
  550.                         SetSSID(newSSID);
  551.                         count++;
  552.                         Sleep(1000);
  553.                     }
  554.                 */
  555.                 strcpy(networkModeAndSSID.networkSSID, newSSID);
  556.                 networkModeAndSSID.adapterNum = "1";
  557.                 networkModeAndSSID.ssidLength = (ULONG)strlen(networkModeAndSSID.networkSSID);
  558.                 networkModeAndSSID.networkMode = mode;
  559.  
  560.                 if(!DoesConnectionExist(wConnectionName)) {
  561.                     char dataFileName[100];
  562.                     GetSSID(networkSSID);
  563.                     while(_stricmp(networkSSID, newSSID) && count < 5) {
  564.                         ioctlSet(IOCTL_SET_CURRENT_SSID, &networkModeAndSSID, (DWORD) sizeof(networkModeAndSSID));
  565.                         count++;
  566.                         Sleep(1000);
  567.                         GetSSID(networkSSID);
  568.                     }
  569.     
  570.                     if(_stricmp(networkSSID, newSSID)) {
  571.                         fprintf(stderr, "Unable to associate to the network\n");
  572.                         fprintf(stdout, "Do you want to continue with the installation? (Y/N) \n");
  573.                         scanf("%s", response);
  574.                         if(!_stricmp(response, "y")) {
  575.                             fprintf(stdout, "Continuing with the installation.\n"); 
  576.                         } else {
  577.                             fprintf(stderr, "EXITING .....\n");
  578.                             ServiceInstaller("start", NULL);
  579.                             exit(0);
  580.                         }
  581.                     }
  582.  
  583.                     AddNetwork(GetDesktopWindow(), TRUE);
  584.                     if (Verbose)
  585.                         fprintf(stderr, "Trying to rename\n");
  586.                     fprintf(stdout, "Successfully added network.\n");
  587.                     RenameConnection(wConnectionName);
  588.                     //fpConfig = fopen("C:\\Windows\\VirtualWiFiData.txt", "a");
  589.                     //printf("Windir is %s\n", getenv("windir"));
  590.                     sprintf(dataFileName, "%s\\VirtualWiFiData.txt", getenv("windir"));
  591.                     fpConfig = fopen(dataFileName, "a");
  592.                     fprintf(fpConfig, "%s\n", connectionName);
  593.                     fclose(fpConfig);
  594.                 } else {
  595.                     ioctlSet(IOCTL_SET_CURRENT_SSID, &networkModeAndSSID, (DWORD) sizeof(networkModeAndSSID));
  596.                     EnableDisableConnection(wConnectionName, TRUE);
  597.                 }    
  598.                 if(userParameters) {
  599.                     fprintf(stderr, "Trying to set\n");
  600.                     ioctlSet(IOCTL_SET_SWITCHING_PARAMETERS, &setVirtualWiFiTimers, 
  601.                         (DWORD)sizeof(SetVirtualWiFiTimers));
  602.                     fprintf(stderr, "Set parameters successful\n");
  603.                 }
  604.             } else { /* remove a network */
  605.                 fprintf(stderr, "Network to remove is not installed\n");
  606.             }
  607.         } else {
  608.             if(!_stricmp(argv[0], "removenetwork")) {
  609.                 //AddNetwork(GetDesktopWindow(), FALSE);
  610.                 EnableDisableConnection(wConnectionName, FALSE);
  611.                 ServiceInstaller("stop", NULL);
  612.             } else {
  613.                 fprintf(stderr, "Network to add is already installed\n");
  614.                 exit(0);
  615.             }
  616.         }
  617.         if(Verbose)
  618.             fprintf(stderr, "Starting Service ...\n");
  619.  
  620.         ServiceInstaller("start", NULL);
  621.         exit(0);
  622.     }
  623.  
  624.     usage();
  625. }
  626.