int main(int argc, char **argv) { // Avoids the Error: Got a packet bigger than 'max_allowed_packet' bytes printf("Important: Requires that you first set the DB schema and the max packet size on the server.\n"); printf("See DependentExtensions/AutopatcherMySQLRepository/readme.txt\n"); // // MySQL is extremely slow in AutopatcherMySQLRepository::GetFilePart printf("WARNING: MySQL is an order of magnitude slower than PostgreSQL.\nRecommended you use AutopatcherServer_PostgreSQL instead."); printf("Server starting... "); RakNet::AutopatcherServer autopatcherServer; // RakNet::FLP_Printf progressIndicator; RakNet::FileListTransfer fileListTransfer; // So only one thread runs per connection, we create an array of connection objects, and tell the autopatcher server to use one thread per item static const int workerThreadCount=4; // Used for checking patches only static const int sqlConnectionObjectCount=32; // Used for both checking patches and downloading RakNet::AutopatcherMySQLRepository connectionObject[sqlConnectionObjectCount]; RakNet::AutopatcherRepositoryInterface *connectionObjectAddresses[sqlConnectionObjectCount]; for (int i=0; i < sqlConnectionObjectCount; i++) connectionObjectAddresses[i]=&connectionObject[i]; // fileListTransfer.AddCallback(&progressIndicator); autopatcherServer.SetFileListTransferPlugin(&fileListTransfer); // MySQL is extremely slow in AutopatcherMySQLRepository::GetFilePart so running tho incremental reads in a thread allows multiple concurrent users to read // Without this, only one user would be sent files at a time basically fileListTransfer.StartIncrementalReadThreads(sqlConnectionObjectCount); autopatcherServer.SetMaxConurrentUsers(MAX_INCOMING_CONNECTIONS); // More users than this get queued up RakNet::AutopatcherServerLoadNotifier_Printf loadNotifier; autopatcherServer.SetLoadManagementCallback(&loadNotifier); #ifdef USE_TCP RakNet::PacketizedTCP packetizedTCP; if (packetizedTCP.Start(LISTEN_PORT,MAX_INCOMING_CONNECTIONS)==false) { printf("Failed to start TCP. Is the port already in use?"); return 1; } packetizedTCP.AttachPlugin(&autopatcherServer); packetizedTCP.AttachPlugin(&fileListTransfer); #else RakNet::RakPeerInterface *rakPeer; rakPeer = RakNet::RakPeerInterface::GetInstance(); RakNet::SocketDescriptor socketDescriptor(LISTEN_PORT,0); rakPeer->Startup(MAX_INCOMING_CONNECTIONS,&socketDescriptor, 1); rakPeer->SetMaximumIncomingConnections(MAX_INCOMING_CONNECTIONS); rakPeer->AttachPlugin(&autopatcherServer); rakPeer->AttachPlugin(&fileListTransfer); #endif printf("started.\n"); printf("Enter database password:\n"); char password[128]; char username[256]; strcpy(username, "root"); Gets(password,sizeof(password)); if (password[0]==0) strcpy(password,"aaaa"); char db[256]; printf("Enter DB schema: "); // To create the schema, go to the command line client and type create schema autopatcher; // You also have to add // max_allowed_packet=128M // Where 128 is the maximum size file in megabytes you'll ever add // to MySQL\MySQL Server 5.1\my.ini in the [mysqld] section // Be sure to restart the service after doing so Gets(db,sizeof(db)); if (db[0]==0) strcpy(db,"autopatcher"); for (int conIdx=0; conIdx < sqlConnectionObjectCount; conIdx++) { if (!connectionObject[conIdx].Connect("localhost", username, password, db, 0, NULL, 0)) { printf("Database connection failed.\n"); return 1; } } printf("Database connection suceeded.\n"); printf("Starting threads\n"); autopatcherServer.StartThreads(workerThreadCount, sqlConnectionObjectCount, connectionObjectAddresses); printf("System ready for connections\n"); printf("(D)rop database\n(C)reate database.\n(A)dd application\n(U)pdate revision.\n(R)emove application\n(Q)uit\n"); char ch; RakNet::Packet *p; while (1) { #ifdef USE_TCP RakNet::SystemAddress notificationAddress; notificationAddress=packetizedTCP.HasCompletedConnectionAttempt(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_CONNECTION_REQUEST_ACCEPTED\n"); notificationAddress=packetizedTCP.HasNewIncomingConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_NEW_INCOMING_CONNECTION\n"); notificationAddress=packetizedTCP.HasLostConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_CONNECTION_LOST\n"); p=packetizedTCP.Receive(); while (p) { packetizedTCP.DeallocatePacket(p); p=packetizedTCP.Receive(); } #else p=rakPeer->Receive(); while (p) { if (p->data[0]==ID_NEW_INCOMING_CONNECTION) printf("ID_NEW_INCOMING_CONNECTION\n"); else if (p->data[0]==ID_DISCONNECTION_NOTIFICATION) printf("ID_DISCONNECTION_NOTIFICATION\n"); else if (p->data[0]==ID_CONNECTION_LOST) printf("ID_CONNECTION_LOST\n"); rakPeer->DeallocatePacket(p); p=rakPeer->Receive(); } #endif if (kbhit()) { ch=getch(); if (ch=='q') break; else if (ch=='c') { if (connectionObject[0].CreateAutopatcherTables()==false) printf("Error: %s\n", connectionObject[0].GetLastError()); else printf("Created\n"); } else if (ch=='d') { if (connectionObject[0].DestroyAutopatcherTables()==false) printf("Error: %s\n", connectionObject[0].GetLastError()); else printf("Destroyed\n"); } else if (ch=='a') { printf("Enter application name to add: "); char appName[512]; Gets(appName,sizeof(appName)); if (appName[0]==0) strcpy(appName, "TestApp"); if (connectionObject[0].AddApplication(appName, username)==false) printf("Error: %s\n", connectionObject[0].GetLastError()); else printf("Done\n"); } else if (ch=='r') { printf("Enter application name to remove: "); char appName[512]; Gets(appName,sizeof(appName)); if (appName[0]==0) strcpy(appName, "TestApp"); if (connectionObject[0].RemoveApplication(appName)==false) printf("Error: %s\n", connectionObject[0].GetLastError()); else printf("Done\n"); } else if (ch=='u') { printf("Enter application name: "); char appName[512]; Gets(appName,sizeof(appName)); if (appName[0]==0) strcpy(appName, "TestApp"); printf("Enter application directory: "); char appDir[512]; Gets(appDir,sizeof(appName)); if (appDir[0]==0) strcpy(appDir, "C:/temp"); if (connectionObject[0].UpdateApplicationFiles(appName, appDir, username, 0)==false) { printf("Error: %s\n", connectionObject[0].GetLastError()); } else { printf("Update success.\n"); } } } RakSleep(30); } #ifdef USE_TCP packetizedTCP.Stop(); #else RakNet::RakPeerInterface::DestroyInstance(rakPeer); #endif }
int main(int argc, char **argv) { printf("A simple client interface for the advanced autopatcher.\n"); printf("Use DirectoryDeltaTransfer for a simpler version of an autopatcher.\n"); printf("Difficulty: Intermediate\n\n"); printf("Client starting..."); RakNet::SystemAddress serverAddress=RakNet::UNASSIGNED_SYSTEM_ADDRESS; RakNet::AutopatcherClient autopatcherClient; RakNet::FileListTransfer fileListTransfer; autopatcherClient.SetFileListTransferPlugin(&fileListTransfer); unsigned short localPort=0; if (argc>=6) { localPort=atoi(argv[5]); } #ifdef USE_TCP RakNet::PacketizedTCP packetizedTCP; if (packetizedTCP.Start(localPort,1)==false) { printf("Failed to start TCP. Is the port already in use?"); return 1; } packetizedTCP.AttachPlugin(&autopatcherClient); packetizedTCP.AttachPlugin(&fileListTransfer); #else RakNet::RakPeerInterface *rakPeer; rakPeer = RakNet::RakPeerInterface::GetInstance(); RakNet::SocketDescriptor socketDescriptor(localPort,0); rakPeer->Startup(1,&socketDescriptor, 1); // Plugin will send us downloading progress notifications if a file is split to fit under the MTU 10 or more times rakPeer->SetSplitMessageProgressInterval(10); rakPeer->AttachPlugin(&autopatcherClient); rakPeer->AttachPlugin(&fileListTransfer); #endif printf("started\n"); char buff[512]; if (argc<2) { printf("Enter server IP: "); Gets(buff,sizeof(buff)); if (buff[0]==0) //strcpy(buff, "94.198.81.195"); strcpy(buff, "127.0.0.1"); } else strcpy(buff, argv[1]); #ifdef USE_TCP packetizedTCP.Connect(buff,60000,false); #else rakPeer->Connect(buff, 60000, 0, 0); #endif printf("Connecting...\n"); char appDir[512]; if (argc<3) { printf("Enter application directory: "); Gets(appDir,sizeof(appDir)); if (appDir[0]==0) { strcpy(appDir, "D:/temp2"); } } else strcpy(appDir, argv[2]); char appName[512]; if (argc<4) { printf("Enter application name: "); Gets(appName,sizeof(appName)); if (appName[0]==0) strcpy(appName, "TestApp"); } else strcpy(appName, argv[3]); bool patchImmediately=argc>=5 && argv[4][0]=='1'; if (patchImmediately==false) printf("Hit 'q' to quit, 'p' to patch, 'c' to cancel the patch. 'r' to reconnect. 'd' to disconnect.\n"); else printf("Hit 'q' to quit, 'c' to cancel the patch.\n"); char ch; RakNet::Packet *p; while (1) { #ifdef USE_TCP RakNet::SystemAddress notificationAddress; notificationAddress=packetizedTCP.HasCompletedConnectionAttempt(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) { printf("ID_CONNECTION_REQUEST_ACCEPTED\n"); serverAddress=notificationAddress; } notificationAddress=packetizedTCP.HasNewIncomingConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_NEW_INCOMING_CONNECTION\n"); notificationAddress=packetizedTCP.HasLostConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_CONNECTION_LOST\n"); p=packetizedTCP.Receive(); while (p) { if (p->data[0]==ID_AUTOPATCHER_REPOSITORY_FATAL_ERROR) { char buff[256]; RakNet::BitStream temp(p->data, p->length, false); temp.IgnoreBits(8); RakNet::StringCompressor::Instance()->DecodeString(buff, 256, &temp); printf("ID_AUTOPATCHER_REPOSITORY_FATAL_ERROR\n"); printf("%s\n", buff); } else if (p->data[0]==ID_AUTOPATCHER_FINISHED) { printf("ID_AUTOPATCHER_FINISHED with server time %f\n", autopatcherClient.GetServerDate()); double srvDate=autopatcherClient.GetServerDate(); FILE *fp = fopen("srvDate", "wb"); fwrite(&srvDate,sizeof(double),1,fp); fclose(fp); } else if (p->data[0]==ID_AUTOPATCHER_RESTART_APPLICATION) printf("Launch \"AutopatcherClientRestarter.exe autopatcherRestart.txt\"\nQuit this application immediately after to unlock files.\n"); packetizedTCP.DeallocatePacket(p); p=packetizedTCP.Receive(); } #else p=rakPeer->Receive(); while (p) { if (p->data[0]==ID_DISCONNECTION_NOTIFICATION) printf("ID_DISCONNECTION_NOTIFICATION\n"); else if (p->data[0]==ID_CONNECTION_LOST) printf("ID_CONNECTION_LOST\n"); else if (p->data[0]==ID_CONNECTION_REQUEST_ACCEPTED) { printf("ID_CONNECTION_REQUEST_ACCEPTED\n"); serverAddress=p->systemAddress; } else if (p->data[0]==ID_CONNECTION_ATTEMPT_FAILED) printf("ID_CONNECTION_ATTEMPT_FAILED\n"); else if (p->data[0]==ID_NO_FREE_INCOMING_CONNECTIONS) printf("ID_NO_FREE_INCOMING_CONNECTIONS\n"); else if (p->data[0]==ID_AUTOPATCHER_REPOSITORY_FATAL_ERROR) { char buff[256]; RakNet::BitStream temp(p->data, p->length, false); temp.IgnoreBits(8); RakNet::StringCompressor::Instance()->DecodeString(buff, 256, &temp); printf("ID_AUTOPATCHER_REPOSITORY_FATAL_ERROR\n"); printf("%s\n", buff); } else if (p->data[0]==ID_AUTOPATCHER_FINISHED) { printf("ID_AUTOPATCHER_FINISHED with server time %f\n", autopatcherClient.GetServerDate()); double srvDate=autopatcherClient.GetServerDate(); FILE *fp = fopen("srvDate", "wb"); fwrite(&srvDate,sizeof(double),1,fp); fclose(fp); } else if (p->data[0]==ID_AUTOPATCHER_RESTART_APPLICATION) printf("Launch \"AutopatcherClientRestarter.exe autopatcherRestart.txt\"\nQuit this application immediately after to unlock files.\n"); rakPeer->DeallocatePacket(p); p=rakPeer->Receive(); } #endif if (kbhit()) ch=getch(); else ch=0; if (ch=='q') break; else if (ch=='r') { #ifdef USE_TCP packetizedTCP.Connect(buff,60000,false); #else rakPeer->Connect(buff, 60000, 0, 0); #endif } else if (ch=='d') { #ifdef USE_TCP packetizedTCP.CloseConnection(serverAddress); #else rakPeer->CloseConnection(serverAddress, true); #endif } else if (ch=='p' || (serverAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS && patchImmediately==true)) { patchImmediately=false; char restartFile[512]; strcpy(restartFile, appDir); strcat(restartFile, "/autopatcherRestart.txt"); double lastUpdateDate; FILE *fp = fopen("srvDate", "rb"); if (fp) { fread(&lastUpdateDate, sizeof(lastUpdateDate), 1, fp); fclose(fp); } else lastUpdateDate=0; if (autopatcherClient.PatchApplication(appName, appDir, lastUpdateDate, serverAddress, &transferCallback, restartFile, argv[0])) { printf("Patching process starting.\n"); } else { printf("Failed to start patching.\n"); } } else if (ch=='c') { autopatcherClient.Clear(); printf("Autopatcher cleared.\n"); } RakSleep(30); } // Dereference so the destructor doesn't crash autopatcherClient.SetFileListTransferPlugin(0); #ifdef USE_TCP packetizedTCP.Stop(); #else rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); #endif return 1; }
int main(int argc, char **argv) { if (argc<8) { printf("Arguments: serverIP, pathToGame, gameName, patchImmediately, localPort, serverPort, fullScan"); return 0; } RakNet::SystemAddress TCPServerAddress=RakNet::UNASSIGNED_SYSTEM_ADDRESS; RakNet::AutopatcherClient autopatcherClient; RakNet::FileListTransfer fileListTransfer; RakNet::CloudClient cloudClient; autopatcherClient.SetFileListTransferPlugin(&fileListTransfer); bool didRebalance=false; // So we only reconnect to a lower load server once, for load balancing bool fullScan = argv[7][0]=='1'; unsigned short localPort; localPort=atoi(argv[5]); unsigned short serverPort=atoi(argv[6]); RakNet::PacketizedTCP packetizedTCP; if (packetizedTCP.Start(localPort,1)==false) { printf("Failed to start TCP. Is the port already in use?"); return 1; } packetizedTCP.AttachPlugin(&autopatcherClient); packetizedTCP.AttachPlugin(&fileListTransfer); RakNet::RakPeerInterface *rakPeer; rakPeer = RakNet::RakPeerInterface::GetInstance(); RakNet::SocketDescriptor socketDescriptor(localPort,0); rakPeer->Startup(1,&socketDescriptor, 1); rakPeer->AttachPlugin(&cloudClient); DataStructures::List<RakNet::RakNetSocket2* > sockets; rakPeer->GetSockets(sockets); printf("Started on port %i\n", sockets[0]->GetBoundAddress().GetPort()); char buff[512]; strcpy(buff, argv[1]); rakPeer->Connect(buff, serverPort, 0, 0); printf("Connecting...\n"); char appDir[512]; strcpy(appDir, argv[2]); char appName[512]; strcpy(appName, argv[3]); bool patchImmediately=argc>=5 && argv[4][0]=='1'; RakNet::Packet *p; while (1) { RakNet::SystemAddress notificationAddress; notificationAddress=packetizedTCP.HasCompletedConnectionAttempt(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) { printf("ID_CONNECTION_REQUEST_ACCEPTED\n"); TCPServerAddress=notificationAddress; } notificationAddress=packetizedTCP.HasNewIncomingConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_NEW_INCOMING_CONNECTION\n"); notificationAddress=packetizedTCP.HasLostConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_CONNECTION_LOST\n"); notificationAddress=packetizedTCP.HasFailedConnectionAttempt(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) { printf("ID_CONNECTION_ATTEMPT_FAILED TCP\n"); autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 0; } p=packetizedTCP.Receive(); while (p) { if (p->data[0]==ID_AUTOPATCHER_REPOSITORY_FATAL_ERROR) { char buff[256]; RakNet::BitStream temp(p->data, p->length, false); temp.IgnoreBits(8); RakNet::StringCompressor::Instance()->DecodeString(buff, 256, &temp); printf("ID_AUTOPATCHER_REPOSITORY_FATAL_ERROR\n"); printf("%s\n", buff); autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 0; } else if (p->data[0]==ID_AUTOPATCHER_CANNOT_DOWNLOAD_ORIGINAL_UNMODIFIED_FILES) { printf("ID_AUTOPATCHER_CANNOT_DOWNLOAD_ORIGINAL_UNMODIFIED_FILES\n"); autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 0; } else if (p->data[0]==ID_AUTOPATCHER_FINISHED) { printf("ID_AUTOPATCHER_FINISHED with server time %f\n", autopatcherClient.GetServerDate()); double srvDate=autopatcherClient.GetServerDate(); FILE *fp = fopen("srvDate", "wb"); fwrite(&srvDate,sizeof(double),1,fp); fclose(fp); autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 0; } else if (p->data[0]==ID_AUTOPATCHER_RESTART_APPLICATION) { printf("ID_AUTOPATCHER_RESTART_APPLICATION"); autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 0; } // Launch \"AutopatcherClientRestarter.exe autopatcherRestart.txt\"\nQuit this application immediately after to unlock files.\n"); packetizedTCP.DeallocatePacket(p); p=packetizedTCP.Receive(); } p=rakPeer->Receive(); while (p) { if (p->data[0]==ID_CONNECTION_REQUEST_ACCEPTED) { // UploadInstanceToCloud(&cloudClient, p->guid); // GetClientSubscription(&cloudClient, p->guid); GetServers(&cloudClient, p->guid); break; } else if (p->data[0]==ID_CONNECTION_ATTEMPT_FAILED) { printf("ID_CONNECTION_ATTEMPT_FAILED UDP\n"); autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 0; } else if (p->data[0]==ID_CLOUD_GET_RESPONSE) { RakNet::CloudQueryResult cloudQueryResult; cloudClient.OnGetReponse(&cloudQueryResult, p); unsigned int rowIndex; const bool wasCallToGetServers=cloudQueryResult.cloudQuery.keys[0].primaryKey=="CloudConnCount"; printf("\n"); if (wasCallToGetServers) printf("Downloaded server list. %i servers.\n", cloudQueryResult.rowsReturned.Size()); unsigned short connectionsOnOurServer=65535; unsigned short lowestConnectionsServer=65535; RakNet::SystemAddress lowestConnectionAddress; for (rowIndex=0; rowIndex < cloudQueryResult.rowsReturned.Size(); rowIndex++) { RakNet::CloudQueryRow *row = cloudQueryResult.rowsReturned[rowIndex]; if (wasCallToGetServers) { unsigned short connCount; RakNet::BitStream bsIn(row->data, row->length, false); bsIn.Read(connCount); printf("%i. Server found at %s with %i connections\n", rowIndex+1, row->serverSystemAddress.ToString(true), connCount); unsigned short connectionsExcludingOurselves; if (row->serverGUID==p->guid) connectionsExcludingOurselves=connCount-1; else connectionsExcludingOurselves=connCount; // Find the lowest load server (optional) if (p->guid==row->serverGUID) { connectionsOnOurServer=connectionsExcludingOurselves; } else if (connectionsExcludingOurselves < lowestConnectionsServer) { lowestConnectionsServer=connectionsExcludingOurselves; lowestConnectionAddress=row->serverSystemAddress; } } } // Do load balancing by reconnecting to lowest load server (optional) if (didRebalance==false && wasCallToGetServers) { if (cloudQueryResult.rowsReturned.Size()>0 && connectionsOnOurServer>lowestConnectionsServer) { printf("Reconnecting to lower load server %s\n", lowestConnectionAddress.ToString(false)); rakPeer->CloseConnection(p->guid, true); // Wait for the thread to close, otherwise will immediately get back ID_CONNECTION_ATTEMPT_FAILED because no free outgoing connection slots // Alternatively, just call Startup() with 2 slots instead of 1 RakSleep(500); rakPeer->Connect(lowestConnectionAddress.ToString(false), lowestConnectionAddress.GetPort(), 0, 0); // TCP Connect to new IP address packetizedTCP.Connect(lowestConnectionAddress.ToString(false),serverPort,false); } else { // TCP Connect to original IP address packetizedTCP.Connect(buff,serverPort,false); } didRebalance=true; } cloudClient.DeallocateWithDefaultAllocator(&cloudQueryResult); } rakPeer->DeallocatePacket(p); p=rakPeer->Receive(); } if (TCPServerAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS && patchImmediately==true) { patchImmediately=false; char restartFile[512]; strcpy(restartFile, appDir); strcat(restartFile, "/autopatcherRestart.txt"); double lastUpdateDate; if (fullScan==false) { FILE *fp = fopen("srvDate", "rb"); if (fp) { fread(&lastUpdateDate, sizeof(lastUpdateDate), 1, fp); fclose(fp); } else lastUpdateDate=0; } else lastUpdateDate=0; if (autopatcherClient.PatchApplication(appName, appDir, lastUpdateDate, TCPServerAddress, &transferCallback, restartFile, argv[0])) { printf("Patching process starting.\n"); } else { printf("Failed to start patching.\n"); autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 0; } } RakSleep(30); } // Dereference so the destructor doesn't crash autopatcherClient.SetFileListTransferPlugin(0); autopatcherClient.Clear(); packetizedTCP.Stop(); rakPeer->Shutdown(500,0); RakNet::RakPeerInterface::DestroyInstance(rakPeer); return 1; }
int main(int argc, char **argv) { printf("Server starting... "); RakNet::AutopatcherServer autopatcherServer; // RakNet::FLP_Printf progressIndicator; RakNet::FileListTransfer fileListTransfer; static const int workerThreadCount=4; // Used for checking patches only static const int sqlConnectionObjectCount=32; // Used for both checking patches and downloading AutopatcherPostgreRepository2_WithXDelta connectionObject[sqlConnectionObjectCount]; RakNet::AutopatcherRepositoryInterface *connectionObjectAddresses[sqlConnectionObjectCount]; for (int i=0; i < sqlConnectionObjectCount; i++) connectionObjectAddresses[i]=&connectionObject[i]; // fileListTransfer.AddCallback(&progressIndicator); autopatcherServer.SetFileListTransferPlugin(&fileListTransfer); // PostgreSQL is fast, so this may not be necessary, or could use fewer threads // This is used to read increments of large files concurrently, thereby serving users downloads as other users read from the DB fileListTransfer.StartIncrementalReadThreads(sqlConnectionObjectCount); autopatcherServer.SetMaxConurrentUsers(MAX_INCOMING_CONNECTIONS); // More users than this get queued up RakNet::AutopatcherServerLoadNotifier_Printf loadNotifier; autopatcherServer.SetLoadManagementCallback(&loadNotifier); #ifdef USE_TCP RakNet::PacketizedTCP packetizedTCP; if (packetizedTCP.Start(LISTEN_PORT,MAX_INCOMING_CONNECTIONS)==false) { printf("Failed to start TCP. Is the port already in use?"); return 1; } packetizedTCP.AttachPlugin(&autopatcherServer); packetizedTCP.AttachPlugin(&fileListTransfer); #else RakNet::RakPeerInterface *rakPeer; rakPeer = RakNet::RakPeerInterface::GetInstance(); RakNet::SocketDescriptor socketDescriptor(LISTEN_PORT,0); rakPeer->Startup(MAX_INCOMING_CONNECTIONS,&socketDescriptor, 1); rakPeer->SetMaximumIncomingConnections(MAX_INCOMING_CONNECTIONS); rakPeer->AttachPlugin(&autopatcherServer); rakPeer->AttachPlugin(&fileListTransfer); #endif printf("started.\n"); printf("Enter database password:\n"); char connectionString[256],password[128]; char username[256]; strcpy(username, "postgres"); gets(password); if (password[0]==0) strcpy(password, "aaaa"); strcpy(connectionString, "user="******" password="******"Database connection failed.\n"); return 1; } } printf("Database connection suceeded.\n"); printf("Starting threads\n"); // 4 Worker threads, which is CPU intensive // A greater number of SQL connections, which read files incrementally for large downloads autopatcherServer.StartThreads(workerThreadCount,sqlConnectionObjectCount, connectionObjectAddresses); autopatcherServer.CacheMostRecentPatch(0); // autopatcherServer.SetAllowDownloadOfOriginalUnmodifiedFiles(false); printf("System ready for connections\n"); // https://code.google.com/p/xdelta/downloads/list printf("Optional: Enter path to xdelta.exe: "); Gets(PATH_TO_XDELTA_EXE, sizeof(PATH_TO_XDELTA_EXE)); if (PATH_TO_XDELTA_EXE[0]==0) strcpy(PATH_TO_XDELTA_EXE, "c:/xdelta3-3.0.6-win32.exe"); if (PATH_TO_XDELTA_EXE[0]) { printf("Enter working directory to store temporary files: "); Gets(WORKING_DIRECTORY, sizeof(WORKING_DIRECTORY)); if (WORKING_DIRECTORY[0]==0) GetTempPath(MAX_PATH, WORKING_DIRECTORY); if (WORKING_DIRECTORY[strlen(WORKING_DIRECTORY)-1]=='\\' || WORKING_DIRECTORY[strlen(WORKING_DIRECTORY)-1]=='/') WORKING_DIRECTORY[strlen(WORKING_DIRECTORY)-1]=0; } printf("(D)rop database\n(C)reate database.\n(A)dd application\n(U)pdate revision.\n(R)emove application\n(Q)uit\n"); char ch; RakNet::Packet *p; while (1) { #ifdef USE_TCP RakNet::SystemAddress notificationAddress; notificationAddress=packetizedTCP.HasCompletedConnectionAttempt(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_CONNECTION_REQUEST_ACCEPTED\n"); notificationAddress=packetizedTCP.HasNewIncomingConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_NEW_INCOMING_CONNECTION\n"); notificationAddress=packetizedTCP.HasLostConnection(); if (notificationAddress!=RakNet::UNASSIGNED_SYSTEM_ADDRESS) printf("ID_CONNECTION_LOST\n"); p=packetizedTCP.Receive(); while (p) { packetizedTCP.DeallocatePacket(p); p=packetizedTCP.Receive(); } #else p=rakPeer->Receive(); while (p) { if (p->data[0]==ID_NEW_INCOMING_CONNECTION) printf("ID_NEW_INCOMING_CONNECTION\n"); else if (p->data[0]==ID_DISCONNECTION_NOTIFICATION) printf("ID_DISCONNECTION_NOTIFICATION\n"); else if (p->data[0]==ID_CONNECTION_LOST) printf("ID_CONNECTION_LOST\n"); rakPeer->DeallocatePacket(p); p=rakPeer->Receive(); } #endif if (kbhit()) { ch=getch(); if (ch=='q') break; else if (ch=='c') { if (connectionObject[0].CreateAutopatcherTables()==false) printf("%s", connectionObject[0].GetLastError()); } else if (ch=='d') { if (connectionObject[0].DestroyAutopatcherTables()==false) printf("%s", connectionObject[0].GetLastError()); } else if (ch=='a') { printf("Enter application name to add: "); char appName[512]; Gets(appName,sizeof(appName)); if (appName[0]==0) strcpy(appName, "TestApp"); if (connectionObject[0].AddApplication(appName, username)==false) printf("%s", connectionObject[0].GetLastError()); else printf("Done\n"); } else if (ch=='r') { printf("Enter application name to remove: "); char appName[512]; Gets(appName,sizeof(appName)); if (appName[0]==0) strcpy(appName, "TestApp"); if (connectionObject[0].RemoveApplication(appName)==false) printf("%s", connectionObject[0].GetLastError()); else printf("Done\n"); } else if (ch=='u') { printf("Enter application name: "); char appName[512]; Gets(appName,sizeof(appName)); if (appName[0]==0) strcpy(appName, "TestApp"); printf("Enter application directory: "); char appDir[512]; Gets(appDir,sizeof(appDir)); if (appDir[0]==0) strcpy(appDir, "D:\\temp"); if (connectionObject[0].UpdateApplicationFiles(appName, appDir, username, 0)==false) { printf("%s", connectionObject[0].GetLastError()); } else { printf("Update success.\n"); autopatcherServer.CacheMostRecentPatch(appName); } } } RakSleep(30); } #ifdef USE_TCP packetizedTCP.Stop(); #else RakNet::RakPeerInterface::DestroyInstance(rakPeer); #endif return 0; }