00001
00002
00003
00004
00005
00006
00007
00008
00009
00012 #ifdef ENABLE_NETWORK
00013
00014 #include "../stdafx.h"
00015 #include "../debug.h"
00016 #include "network_gui.h"
00017 #include "../saveload/saveload.h"
00018 #include "../saveload/saveload_filter.h"
00019 #include "../command_func.h"
00020 #include "../console_func.h"
00021 #include "../strings_func.h"
00022 #include "../window_func.h"
00023 #include "../company_func.h"
00024 #include "../company_base.h"
00025 #include "../company_gui.h"
00026 #include "../core/random_func.hpp"
00027 #include "../date_func.h"
00028 #include "../gfx_func.h"
00029 #include "../error.h"
00030 #include "../rev.h"
00031 #include "network.h"
00032 #include "network_base.h"
00033 #include "network_client.h"
00034 #include "../core/backup_type.hpp"
00035
00036 #include "table/strings.h"
00037
00038
00039
00040
00042 struct PacketReader : LoadFilter {
00043 static const size_t CHUNK = 32 * 1024;
00044
00045 AutoFreeSmallVector<byte *, 16> blocks;
00046 byte *buf;
00047 byte *bufe;
00048 byte **block;
00049 size_t written_bytes;
00050 size_t read_bytes;
00051
00053 PacketReader() : LoadFilter(NULL), buf(NULL), bufe(NULL), block(NULL), written_bytes(0), read_bytes(0)
00054 {
00055 }
00056
00061 void AddPacket(const Packet *p)
00062 {
00063 assert(this->read_bytes == 0);
00064
00065 size_t in_packet = p->size - p->pos;
00066 size_t to_write = min((size_t)(this->bufe - this->buf), in_packet);
00067 const byte *pbuf = p->buffer + p->pos;
00068
00069 this->written_bytes += in_packet;
00070 if (to_write != 0) {
00071 memcpy(this->buf, pbuf, to_write);
00072 this->buf += to_write;
00073 }
00074
00075
00076 if (to_write == in_packet) return;
00077
00078
00079 pbuf += to_write;
00080 to_write = in_packet - to_write;
00081 this->buf = *this->blocks.Append() = CallocT<byte>(CHUNK);
00082 this->bufe = this->buf + CHUNK;
00083
00084 memcpy(this->buf, pbuf, to_write);
00085 this->buf += to_write;
00086 }
00087
00088 size_t Read(byte *rbuf, size_t size)
00089 {
00090
00091 size_t ret_size = size = min(this->written_bytes - this->read_bytes, size);
00092 this->read_bytes += ret_size;
00093 const byte *rbufe = rbuf + ret_size;
00094
00095 while (rbuf != rbufe) {
00096 if (this->buf == this->bufe) {
00097 this->buf = *this->block++;
00098 this->bufe = this->buf + CHUNK;
00099 }
00100
00101 size_t to_write = min(this->bufe - this->buf, rbufe - rbuf);
00102 memcpy(rbuf, this->buf, to_write);
00103 rbuf += to_write;
00104 this->buf += to_write;
00105 }
00106
00107 return ret_size;
00108 }
00109
00110 void Reset()
00111 {
00112 this->read_bytes = 0;
00113
00114 this->block = this->blocks.Begin();
00115 this->buf = *this->block++;
00116 this->bufe = this->buf + CHUNK;
00117 }
00118 };
00119
00120
00125 ClientNetworkGameSocketHandler::ClientNetworkGameSocketHandler(SOCKET s) : NetworkGameSocketHandler(s), savegame(NULL), status(STATUS_INACTIVE)
00126 {
00127 assert(ClientNetworkGameSocketHandler::my_client == NULL);
00128 ClientNetworkGameSocketHandler::my_client = this;
00129 }
00130
00132 ClientNetworkGameSocketHandler::~ClientNetworkGameSocketHandler()
00133 {
00134 assert(ClientNetworkGameSocketHandler::my_client == this);
00135 ClientNetworkGameSocketHandler::my_client = NULL;
00136
00137 delete this->savegame;
00138 }
00139
00140 NetworkRecvStatus ClientNetworkGameSocketHandler::CloseConnection(NetworkRecvStatus status)
00141 {
00142 assert(status != NETWORK_RECV_STATUS_OKAY);
00143
00144
00145
00146
00147
00148
00149
00150 if (this->sock == INVALID_SOCKET) return status;
00151
00152 DEBUG(net, 1, "Closed client connection %d", this->client_id);
00153
00154 this->SendPackets(true);
00155
00156
00157
00158
00159
00160 CSleep(3 * MILLISECONDS_PER_TICK);
00161
00162 delete this->GetInfo();
00163 delete this;
00164
00165 return status;
00166 }
00167
00172 void ClientNetworkGameSocketHandler::ClientError(NetworkRecvStatus res)
00173 {
00174
00175
00176 NetworkErrorCode errorno;
00177
00178
00179 if (res == NETWORK_RECV_STATUS_CLOSE_QUERY) {
00180 this->NetworkSocketHandler::CloseConnection();
00181 this->CloseConnection(res);
00182 _networking = false;
00183
00184 DeleteWindowById(WC_NETWORK_STATUS_WINDOW, 0);
00185 return;
00186 }
00187
00188 switch (res) {
00189 case NETWORK_RECV_STATUS_DESYNC: errorno = NETWORK_ERROR_DESYNC; break;
00190 case NETWORK_RECV_STATUS_SAVEGAME: errorno = NETWORK_ERROR_SAVEGAME_FAILED; break;
00191 case NETWORK_RECV_STATUS_NEWGRF_MISMATCH: errorno = NETWORK_ERROR_NEWGRF_MISMATCH; break;
00192 default: errorno = NETWORK_ERROR_GENERAL; break;
00193 }
00194
00195
00196 if (res != NETWORK_RECV_STATUS_SERVER_ERROR && res != NETWORK_RECV_STATUS_SERVER_FULL &&
00197 res != NETWORK_RECV_STATUS_SERVER_BANNED) {
00198 SendError(errorno);
00199 }
00200
00201 _switch_mode = SM_MENU;
00202 this->CloseConnection(res);
00203 _networking = false;
00204 }
00205
00206
00212 bool ClientNetworkGameSocketHandler::Receive()
00213 {
00214 if (my_client->CanSendReceive()) {
00215 NetworkRecvStatus res = my_client->ReceivePackets();
00216 if (res != NETWORK_RECV_STATUS_OKAY) {
00217
00218
00219 my_client->ClientError(res);
00220 return false;
00221 }
00222 }
00223 return _networking;
00224 }
00225
00227 void ClientNetworkGameSocketHandler::Send()
00228 {
00229 my_client->SendPackets();
00230 my_client->CheckConnection();
00231 }
00232
00237 bool ClientNetworkGameSocketHandler::GameLoop()
00238 {
00239 _frame_counter++;
00240
00241 NetworkExecuteLocalCommandQueue();
00242
00243 extern void StateGameLoop();
00244 StateGameLoop();
00245
00246
00247 if (_sync_frame != 0) {
00248 if (_sync_frame == _frame_counter) {
00249 #ifdef NETWORK_SEND_DOUBLE_SEED
00250 if (_sync_seed_1 != _random.state[0] || _sync_seed_2 != _random.state[1]) {
00251 #else
00252 if (_sync_seed_1 != _random.state[0]) {
00253 #endif
00254 NetworkError(STR_NETWORK_ERROR_DESYNC);
00255 DEBUG(desync, 1, "sync_err: %08x; %02x", _date, _date_fract);
00256 DEBUG(net, 0, "Sync error detected!");
00257 my_client->ClientError(NETWORK_RECV_STATUS_DESYNC);
00258 return false;
00259 }
00260
00261
00262
00263
00264 if (_network_first_time) {
00265 _network_first_time = false;
00266 SendAck();
00267 }
00268
00269 _sync_frame = 0;
00270 } else if (_sync_frame < _frame_counter) {
00271 DEBUG(net, 1, "Missed frame for sync-test (%d / %d)", _sync_frame, _frame_counter);
00272 _sync_frame = 0;
00273 }
00274 }
00275
00276 return true;
00277 }
00278
00279
00281 ClientNetworkGameSocketHandler * ClientNetworkGameSocketHandler::my_client = NULL;
00282
00284 static uint32 last_ack_frame;
00285
00287 static uint32 _password_game_seed;
00289 static char _password_server_id[NETWORK_SERVER_ID_LENGTH];
00290
00292 static uint8 _network_server_max_companies;
00294 static uint8 _network_server_max_spectators;
00295
00297 CompanyID _network_join_as;
00298
00300 const char *_network_join_server_password = NULL;
00302 const char *_network_join_company_password = NULL;
00303
00305 assert_compile(NETWORK_SERVER_ID_LENGTH == 16 * 2 + 1);
00306
00307
00308
00309
00310
00311
00313 NetworkRecvStatus ClientNetworkGameSocketHandler::SendCompanyInformationQuery()
00314 {
00315 my_client->status = STATUS_COMPANY_INFO;
00316 _network_join_status = NETWORK_JOIN_STATUS_GETTING_COMPANY_INFO;
00317 SetWindowDirty(WC_NETWORK_STATUS_WINDOW, 0);
00318
00319 Packet *p = new Packet(PACKET_CLIENT_COMPANY_INFO);
00320 my_client->SendPacket(p);
00321 return NETWORK_RECV_STATUS_OKAY;
00322 }
00323
00325 NetworkRecvStatus ClientNetworkGameSocketHandler::SendJoin()
00326 {
00327 my_client->status = STATUS_JOIN;
00328 _network_join_status = NETWORK_JOIN_STATUS_AUTHORIZING;
00329 SetWindowDirty(WC_NETWORK_STATUS_WINDOW, 0);
00330
00331 Packet *p = new Packet(PACKET_CLIENT_JOIN);
00332 p->Send_string(_openttd_revision);
00333 p->Send_string(_settings_client.network.client_name);
00334 p->Send_uint8 (_network_join_as);
00335 p->Send_uint8 (NETLANG_ANY);
00336 my_client->SendPacket(p);
00337 return NETWORK_RECV_STATUS_OKAY;
00338 }
00339
00341 NetworkRecvStatus ClientNetworkGameSocketHandler::SendNewGRFsOk()
00342 {
00343 Packet *p = new Packet(PACKET_CLIENT_NEWGRFS_CHECKED);
00344 my_client->SendPacket(p);
00345 return NETWORK_RECV_STATUS_OKAY;
00346 }
00347
00352 NetworkRecvStatus ClientNetworkGameSocketHandler::SendGamePassword(const char *password)
00353 {
00354 Packet *p = new Packet(PACKET_CLIENT_GAME_PASSWORD);
00355 p->Send_string(password);
00356 my_client->SendPacket(p);
00357 return NETWORK_RECV_STATUS_OKAY;
00358 }
00359
00364 NetworkRecvStatus ClientNetworkGameSocketHandler::SendCompanyPassword(const char *password)
00365 {
00366 Packet *p = new Packet(PACKET_CLIENT_COMPANY_PASSWORD);
00367 p->Send_string(GenerateCompanyPasswordHash(password, _password_server_id, _password_game_seed));
00368 my_client->SendPacket(p);
00369 return NETWORK_RECV_STATUS_OKAY;
00370 }
00371
00373 NetworkRecvStatus ClientNetworkGameSocketHandler::SendGetMap()
00374 {
00375 my_client->status = STATUS_MAP_WAIT;
00376
00377 Packet *p = new Packet(PACKET_CLIENT_GETMAP);
00378
00379
00380
00381
00382
00383
00384 if (IsReleasedVersion()) p->Send_uint32(_openttd_newgrf_version);
00385 my_client->SendPacket(p);
00386 return NETWORK_RECV_STATUS_OKAY;
00387 }
00388
00390 NetworkRecvStatus ClientNetworkGameSocketHandler::SendMapOk()
00391 {
00392 my_client->status = STATUS_ACTIVE;
00393
00394 Packet *p = new Packet(PACKET_CLIENT_MAP_OK);
00395 my_client->SendPacket(p);
00396 return NETWORK_RECV_STATUS_OKAY;
00397 }
00398
00400 NetworkRecvStatus ClientNetworkGameSocketHandler::SendAck()
00401 {
00402 Packet *p = new Packet(PACKET_CLIENT_ACK);
00403
00404 p->Send_uint32(_frame_counter);
00405 p->Send_uint8 (my_client->token);
00406 my_client->SendPacket(p);
00407 return NETWORK_RECV_STATUS_OKAY;
00408 }
00409
00414 NetworkRecvStatus ClientNetworkGameSocketHandler::SendCommand(const CommandPacket *cp)
00415 {
00416 Packet *p = new Packet(PACKET_CLIENT_COMMAND);
00417 my_client->NetworkGameSocketHandler::SendCommand(p, cp);
00418
00419 my_client->SendPacket(p);
00420 return NETWORK_RECV_STATUS_OKAY;
00421 }
00422
00424 NetworkRecvStatus ClientNetworkGameSocketHandler::SendChat(NetworkAction action, DestType type, int dest, const char *msg, int64 data)
00425 {
00426 Packet *p = new Packet(PACKET_CLIENT_CHAT);
00427
00428 p->Send_uint8 (action);
00429 p->Send_uint8 (type);
00430 p->Send_uint32(dest);
00431 p->Send_string(msg);
00432 p->Send_uint64(data);
00433
00434 my_client->SendPacket(p);
00435 return NETWORK_RECV_STATUS_OKAY;
00436 }
00437
00439 NetworkRecvStatus ClientNetworkGameSocketHandler::SendError(NetworkErrorCode errorno)
00440 {
00441 Packet *p = new Packet(PACKET_CLIENT_ERROR);
00442
00443 p->Send_uint8(errorno);
00444 my_client->SendPacket(p);
00445 return NETWORK_RECV_STATUS_OKAY;
00446 }
00447
00452 NetworkRecvStatus ClientNetworkGameSocketHandler::SendSetPassword(const char *password)
00453 {
00454 Packet *p = new Packet(PACKET_CLIENT_SET_PASSWORD);
00455
00456 p->Send_string(GenerateCompanyPasswordHash(password, _password_server_id, _password_game_seed));
00457 my_client->SendPacket(p);
00458 return NETWORK_RECV_STATUS_OKAY;
00459 }
00460
00465 NetworkRecvStatus ClientNetworkGameSocketHandler::SendSetName(const char *name)
00466 {
00467 Packet *p = new Packet(PACKET_CLIENT_SET_NAME);
00468
00469 p->Send_string(name);
00470 my_client->SendPacket(p);
00471 return NETWORK_RECV_STATUS_OKAY;
00472 }
00473
00477 NetworkRecvStatus ClientNetworkGameSocketHandler::SendQuit()
00478 {
00479 Packet *p = new Packet(PACKET_CLIENT_QUIT);
00480
00481 my_client->SendPacket(p);
00482 return NETWORK_RECV_STATUS_OKAY;
00483 }
00484
00490 NetworkRecvStatus ClientNetworkGameSocketHandler::SendRCon(const char *pass, const char *command)
00491 {
00492 Packet *p = new Packet(PACKET_CLIENT_RCON);
00493 p->Send_string(pass);
00494 p->Send_string(command);
00495 my_client->SendPacket(p);
00496 return NETWORK_RECV_STATUS_OKAY;
00497 }
00498
00504 NetworkRecvStatus ClientNetworkGameSocketHandler::SendMove(CompanyID company, const char *password)
00505 {
00506 Packet *p = new Packet(PACKET_CLIENT_MOVE);
00507 p->Send_uint8(company);
00508 p->Send_string(GenerateCompanyPasswordHash(password, _password_server_id, _password_game_seed));
00509 my_client->SendPacket(p);
00510 return NETWORK_RECV_STATUS_OKAY;
00511 }
00512
00517 bool ClientNetworkGameSocketHandler::IsConnected()
00518 {
00519 return my_client != NULL && my_client->status == STATUS_ACTIVE;
00520 }
00521
00522
00523
00524
00525
00526
00527
00528 extern bool SafeLoad(const char *filename, int mode, GameMode newgm, Subdirectory subdir, struct LoadFilter *lf = NULL);
00529
00530 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_FULL(Packet *p)
00531 {
00532
00533 ShowErrorMessage(STR_NETWORK_ERROR_SERVER_FULL, INVALID_STRING_ID, WL_CRITICAL);
00534 DeleteWindowById(WC_NETWORK_STATUS_WINDOW, 0);
00535
00536 return NETWORK_RECV_STATUS_SERVER_FULL;
00537 }
00538
00539 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_BANNED(Packet *p)
00540 {
00541
00542 ShowErrorMessage(STR_NETWORK_ERROR_SERVER_BANNED, INVALID_STRING_ID, WL_CRITICAL);
00543 DeleteWindowById(WC_NETWORK_STATUS_WINDOW, 0);
00544
00545 return NETWORK_RECV_STATUS_SERVER_BANNED;
00546 }
00547
00548 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_COMPANY_INFO(Packet *p)
00549 {
00550 if (this->status != STATUS_COMPANY_INFO) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00551
00552 byte company_info_version = p->Recv_uint8();
00553
00554 if (!this->HasClientQuit() && company_info_version == NETWORK_COMPANY_INFO_VERSION) {
00555
00556 if (!p->Recv_bool()) return NETWORK_RECV_STATUS_CLOSE_QUERY;
00557
00558 CompanyID current = (Owner)p->Recv_uint8();
00559 if (current >= MAX_COMPANIES) return NETWORK_RECV_STATUS_CLOSE_QUERY;
00560
00561 NetworkCompanyInfo *company_info = GetLobbyCompanyInfo(current);
00562 if (company_info == NULL) return NETWORK_RECV_STATUS_CLOSE_QUERY;
00563
00564 p->Recv_string(company_info->company_name, sizeof(company_info->company_name));
00565 company_info->inaugurated_year = p->Recv_uint32();
00566 company_info->company_value = p->Recv_uint64();
00567 company_info->money = p->Recv_uint64();
00568 company_info->income = p->Recv_uint64();
00569 company_info->performance = p->Recv_uint16();
00570 company_info->use_password = p->Recv_bool();
00571 for (uint i = 0; i < NETWORK_VEH_END; i++) {
00572 company_info->num_vehicle[i] = p->Recv_uint16();
00573 }
00574 for (uint i = 0; i < NETWORK_VEH_END; i++) {
00575 company_info->num_station[i] = p->Recv_uint16();
00576 }
00577 company_info->ai = p->Recv_bool();
00578
00579 p->Recv_string(company_info->clients, sizeof(company_info->clients));
00580
00581 SetWindowDirty(WC_NETWORK_WINDOW, 0);
00582
00583 return NETWORK_RECV_STATUS_OKAY;
00584 }
00585
00586 return NETWORK_RECV_STATUS_CLOSE_QUERY;
00587 }
00588
00589
00590
00591
00592 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_CLIENT_INFO(Packet *p)
00593 {
00594 NetworkClientInfo *ci;
00595 ClientID client_id = (ClientID)p->Recv_uint32();
00596 CompanyID playas = (CompanyID)p->Recv_uint8();
00597 char name[NETWORK_NAME_LENGTH];
00598
00599 p->Recv_string(name, sizeof(name));
00600
00601 if (this->status < STATUS_AUTHORIZED) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00602 if (this->HasClientQuit()) return NETWORK_RECV_STATUS_CONN_LOST;
00603
00604 ci = NetworkClientInfo::GetByClientID(client_id);
00605 if (ci != NULL) {
00606 if (playas == ci->client_playas && strcmp(name, ci->client_name) != 0) {
00607
00608 NetworkTextMessage(NETWORK_ACTION_NAME_CHANGE, CC_DEFAULT, false, ci->client_name, name);
00609 } else if (playas != ci->client_playas) {
00610
00611
00612 }
00613
00614
00615
00616 if (client_id == _network_own_client_id) SetLocalCompany(!Company::IsValidID(playas) ? COMPANY_SPECTATOR : playas);
00617
00618 ci->client_playas = playas;
00619 strecpy(ci->client_name, name, lastof(ci->client_name));
00620
00621 SetWindowDirty(WC_CLIENT_LIST, 0);
00622
00623 return NETWORK_RECV_STATUS_OKAY;
00624 }
00625
00626
00627
00628
00629
00630
00631 if (!NetworkClientInfo::CanAllocateItem()) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00632
00633
00634 ci = new NetworkClientInfo(client_id);
00635 ci->client_playas = playas;
00636 if (client_id == _network_own_client_id) this->SetInfo(ci);
00637
00638 strecpy(ci->client_name, name, lastof(ci->client_name));
00639
00640 SetWindowDirty(WC_CLIENT_LIST, 0);
00641
00642 return NETWORK_RECV_STATUS_OKAY;
00643 }
00644
00645 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_ERROR(Packet *p)
00646 {
00647 NetworkErrorCode error = (NetworkErrorCode)p->Recv_uint8();
00648
00649 switch (error) {
00650
00651 case NETWORK_ERROR_NOT_AUTHORIZED:
00652 case NETWORK_ERROR_NOT_EXPECTED:
00653 case NETWORK_ERROR_COMPANY_MISMATCH:
00654 ShowErrorMessage(STR_NETWORK_ERROR_SERVER_ERROR, INVALID_STRING_ID, WL_CRITICAL);
00655 break;
00656 case NETWORK_ERROR_FULL:
00657 ShowErrorMessage(STR_NETWORK_ERROR_SERVER_FULL, INVALID_STRING_ID, WL_CRITICAL);
00658 break;
00659 case NETWORK_ERROR_WRONG_REVISION:
00660 ShowErrorMessage(STR_NETWORK_ERROR_WRONG_REVISION, INVALID_STRING_ID, WL_CRITICAL);
00661 break;
00662 case NETWORK_ERROR_WRONG_PASSWORD:
00663 ShowErrorMessage(STR_NETWORK_ERROR_WRONG_PASSWORD, INVALID_STRING_ID, WL_CRITICAL);
00664 break;
00665 case NETWORK_ERROR_KICKED:
00666 ShowErrorMessage(STR_NETWORK_ERROR_KICKED, INVALID_STRING_ID, WL_CRITICAL);
00667 break;
00668 case NETWORK_ERROR_CHEATER:
00669 ShowErrorMessage(STR_NETWORK_ERROR_CHEATER, INVALID_STRING_ID, WL_CRITICAL);
00670 break;
00671 case NETWORK_ERROR_TOO_MANY_COMMANDS:
00672 ShowErrorMessage(STR_NETWORK_ERROR_TOO_MANY_COMMANDS, INVALID_STRING_ID, WL_CRITICAL);
00673 break;
00674 default:
00675 ShowErrorMessage(STR_NETWORK_ERROR_LOSTCONNECTION, INVALID_STRING_ID, WL_CRITICAL);
00676 }
00677
00678 DeleteWindowById(WC_NETWORK_STATUS_WINDOW, 0);
00679
00680 return NETWORK_RECV_STATUS_SERVER_ERROR;
00681 }
00682
00683 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_CHECK_NEWGRFS(Packet *p)
00684 {
00685 if (this->status != STATUS_JOIN) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00686
00687 uint grf_count = p->Recv_uint8();
00688 NetworkRecvStatus ret = NETWORK_RECV_STATUS_OKAY;
00689
00690
00691 for (; grf_count > 0; grf_count--) {
00692 GRFIdentifier c;
00693 this->ReceiveGRFIdentifier(p, &c);
00694
00695
00696 const GRFConfig *f = FindGRFConfig(c.grfid, FGCM_EXACT, c.md5sum);
00697 if (f == NULL) {
00698
00699 char buf[sizeof(c.md5sum) * 2 + 1];
00700 md5sumToString(buf, lastof(buf), c.md5sum);
00701 DEBUG(grf, 0, "NewGRF %08X not found; checksum %s", BSWAP32(c.grfid), buf);
00702 ret = NETWORK_RECV_STATUS_NEWGRF_MISMATCH;
00703 }
00704 }
00705
00706 if (ret == NETWORK_RECV_STATUS_OKAY) {
00707
00708 return SendNewGRFsOk();
00709 }
00710
00711
00712 ShowErrorMessage(STR_NETWORK_ERROR_NEWGRF_MISMATCH, INVALID_STRING_ID, WL_CRITICAL);
00713 return ret;
00714 }
00715
00716 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_NEED_GAME_PASSWORD(Packet *p)
00717 {
00718 if (this->status < STATUS_JOIN || this->status >= STATUS_AUTH_GAME) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00719 this->status = STATUS_AUTH_GAME;
00720
00721 const char *password = _network_join_server_password;
00722 if (!StrEmpty(password)) {
00723 return SendGamePassword(password);
00724 }
00725
00726 ShowNetworkNeedPassword(NETWORK_GAME_PASSWORD);
00727
00728 return NETWORK_RECV_STATUS_OKAY;
00729 }
00730
00731 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_NEED_COMPANY_PASSWORD(Packet *p)
00732 {
00733 if (this->status < STATUS_JOIN || this->status >= STATUS_AUTH_COMPANY) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00734 this->status = STATUS_AUTH_COMPANY;
00735
00736 _password_game_seed = p->Recv_uint32();
00737 p->Recv_string(_password_server_id, sizeof(_password_server_id));
00738 if (this->HasClientQuit()) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00739
00740 const char *password = _network_join_company_password;
00741 if (!StrEmpty(password)) {
00742 return SendCompanyPassword(password);
00743 }
00744
00745 ShowNetworkNeedPassword(NETWORK_COMPANY_PASSWORD);
00746
00747 return NETWORK_RECV_STATUS_OKAY;
00748 }
00749
00750 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_WELCOME(Packet *p)
00751 {
00752 if (this->status < STATUS_JOIN || this->status >= STATUS_AUTHORIZED) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00753 this->status = STATUS_AUTHORIZED;
00754
00755 _network_own_client_id = (ClientID)p->Recv_uint32();
00756
00757
00758 _password_game_seed = p->Recv_uint32();
00759 p->Recv_string(_password_server_id, sizeof(_password_server_id));
00760
00761
00762 return SendGetMap();
00763 }
00764
00765 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_WAIT(Packet *p)
00766 {
00767
00768 if (this->status != STATUS_MAP_WAIT) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00769
00770
00771 _network_join_status = NETWORK_JOIN_STATUS_WAITING;
00772 _network_join_waiting = p->Recv_uint8();
00773 SetWindowDirty(WC_NETWORK_STATUS_WINDOW, 0);
00774
00775 return NETWORK_RECV_STATUS_OKAY;
00776 }
00777
00778 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_MAP_BEGIN(Packet *p)
00779 {
00780 if (this->status < STATUS_AUTHORIZED || this->status >= STATUS_MAP) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00781 this->status = STATUS_MAP;
00782
00783 if (this->savegame != NULL) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00784
00785 this->savegame = new PacketReader();
00786
00787 _frame_counter = _frame_counter_server = _frame_counter_max = p->Recv_uint32();
00788
00789 _network_join_bytes = 0;
00790 _network_join_bytes_total = 0;
00791
00792 _network_join_status = NETWORK_JOIN_STATUS_DOWNLOADING;
00793 SetWindowDirty(WC_NETWORK_STATUS_WINDOW, 0);
00794
00795 return NETWORK_RECV_STATUS_OKAY;
00796 }
00797
00798 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_MAP_SIZE(Packet *p)
00799 {
00800 if (this->status != STATUS_MAP) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00801 if (this->savegame == NULL) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00802
00803 _network_join_bytes_total = p->Recv_uint32();
00804 SetWindowDirty(WC_NETWORK_STATUS_WINDOW, 0);
00805
00806 return NETWORK_RECV_STATUS_OKAY;
00807 }
00808
00809 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_MAP_DATA(Packet *p)
00810 {
00811 if (this->status != STATUS_MAP) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00812 if (this->savegame == NULL) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00813
00814
00815 this->savegame->AddPacket(p);
00816
00817 _network_join_bytes = (uint32)this->savegame->written_bytes;
00818 SetWindowDirty(WC_NETWORK_STATUS_WINDOW, 0);
00819
00820 return NETWORK_RECV_STATUS_OKAY;
00821 }
00822
00823 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_MAP_DONE(Packet *p)
00824 {
00825 if (this->status != STATUS_MAP) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00826 if (this->savegame == NULL) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00827
00828 _network_join_status = NETWORK_JOIN_STATUS_PROCESSING;
00829 SetWindowDirty(WC_NETWORK_STATUS_WINDOW, 0);
00830
00831
00832
00833
00834
00835
00836
00837
00838 LoadFilter *lf = this->savegame;
00839 this->savegame = NULL;
00840 lf->Reset();
00841
00842
00843 ClearErrorMessages();
00844 bool load_success = SafeLoad(NULL, SL_LOAD, GM_NORMAL, NO_DIRECTORY, lf);
00845
00846
00847 this->last_packet = _realtime_tick;
00848
00849 if (!load_success) {
00850 DeleteWindowById(WC_NETWORK_STATUS_WINDOW, 0);
00851 ShowErrorMessage(STR_NETWORK_ERROR_SAVEGAMEERROR, INVALID_STRING_ID, WL_CRITICAL);
00852 return NETWORK_RECV_STATUS_SAVEGAME;
00853 }
00854
00855
00856
00857
00858 SendMapOk();
00859
00860
00861
00862 if (_network_join_as == COMPANY_NEW_COMPANY || !Company::IsValidID(_network_join_as)) {
00863 SetLocalCompany(COMPANY_SPECTATOR);
00864
00865 if (_network_join_as != COMPANY_SPECTATOR) {
00866
00867
00868 _network_join_status = NETWORK_JOIN_STATUS_REGISTERING;
00869 ShowJoinStatusWindow();
00870 NetworkSendCommand(0, 0, 0, CMD_COMPANY_CTRL, NULL, NULL, _local_company);
00871 }
00872 } else {
00873
00874 SetLocalCompany(_network_join_as);
00875 }
00876
00877 return NETWORK_RECV_STATUS_OKAY;
00878 }
00879
00880 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_FRAME(Packet *p)
00881 {
00882 if (this->status != STATUS_ACTIVE) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00883
00884 _frame_counter_server = p->Recv_uint32();
00885 _frame_counter_max = p->Recv_uint32();
00886 #ifdef ENABLE_NETWORK_SYNC_EVERY_FRAME
00887
00888
00889 if (p->pos + 1 < p->size) {
00890 _sync_frame = _frame_counter_server;
00891 _sync_seed_1 = p->Recv_uint32();
00892 #ifdef NETWORK_SEND_DOUBLE_SEED
00893 _sync_seed_2 = p->Recv_uint32();
00894 #endif
00895 }
00896 #endif
00897
00898 if (p->pos != p->size) this->token = p->Recv_uint8();
00899
00900 DEBUG(net, 5, "Received FRAME %d", _frame_counter_server);
00901
00902
00903
00904 if (!_network_first_time && last_ack_frame < _frame_counter) {
00905 last_ack_frame = _frame_counter + DAY_TICKS;
00906 DEBUG(net, 4, "Sent ACK at %d", _frame_counter);
00907 SendAck();
00908 }
00909
00910 return NETWORK_RECV_STATUS_OKAY;
00911 }
00912
00913 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_SYNC(Packet *p)
00914 {
00915 if (this->status != STATUS_ACTIVE) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00916
00917 _sync_frame = p->Recv_uint32();
00918 _sync_seed_1 = p->Recv_uint32();
00919 #ifdef NETWORK_SEND_DOUBLE_SEED
00920 _sync_seed_2 = p->Recv_uint32();
00921 #endif
00922
00923 return NETWORK_RECV_STATUS_OKAY;
00924 }
00925
00926 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_COMMAND(Packet *p)
00927 {
00928 if (this->status != STATUS_ACTIVE) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00929
00930 CommandPacket cp;
00931 const char *err = this->ReceiveCommand(p, &cp);
00932 cp.frame = p->Recv_uint32();
00933 cp.my_cmd = p->Recv_bool();
00934
00935 if (err != NULL) {
00936 IConsolePrintF(CC_ERROR, "WARNING: %s from server, dropping...", err);
00937 return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00938 }
00939
00940 this->incoming_queue.Append(&cp);
00941
00942 return NETWORK_RECV_STATUS_OKAY;
00943 }
00944
00945 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_CHAT(Packet *p)
00946 {
00947 if (this->status != STATUS_ACTIVE) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00948
00949 char name[NETWORK_NAME_LENGTH], msg[NETWORK_CHAT_LENGTH];
00950 const NetworkClientInfo *ci = NULL, *ci_to;
00951
00952 NetworkAction action = (NetworkAction)p->Recv_uint8();
00953 ClientID client_id = (ClientID)p->Recv_uint32();
00954 bool self_send = p->Recv_bool();
00955 p->Recv_string(msg, NETWORK_CHAT_LENGTH);
00956 int64 data = p->Recv_uint64();
00957
00958 ci_to = NetworkClientInfo::GetByClientID(client_id);
00959 if (ci_to == NULL) return NETWORK_RECV_STATUS_OKAY;
00960
00961
00962 if (self_send) {
00963 switch (action) {
00964 case NETWORK_ACTION_CHAT_CLIENT:
00965
00966 snprintf(name, sizeof(name), "%s", ci_to->client_name);
00967 ci = NetworkClientInfo::GetByClientID(_network_own_client_id);
00968 break;
00969
00970
00971 case NETWORK_ACTION_GIVE_MONEY:
00972 if (!Company::IsValidID(ci_to->client_playas)) return NETWORK_RECV_STATUS_OKAY;
00973
00974 case NETWORK_ACTION_CHAT_COMPANY: {
00975 StringID str = Company::IsValidID(ci_to->client_playas) ? STR_COMPANY_NAME : STR_NETWORK_SPECTATORS;
00976 SetDParam(0, ci_to->client_playas);
00977
00978 GetString(name, str, lastof(name));
00979 ci = NetworkClientInfo::GetByClientID(_network_own_client_id);
00980 break;
00981 }
00982
00983 default: return NETWORK_RECV_STATUS_MALFORMED_PACKET;
00984 }
00985 } else {
00986
00987 snprintf(name, sizeof(name), "%s", ci_to->client_name);
00988 ci = ci_to;
00989 }
00990
00991 if (ci != NULL) {
00992 NetworkTextMessage(action, GetDrawStringCompanyColour(ci->client_playas), self_send, name, msg, data);
00993 }
00994 return NETWORK_RECV_STATUS_OKAY;
00995 }
00996
00997 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_ERROR_QUIT(Packet *p)
00998 {
00999 if (this->status < STATUS_AUTHORIZED) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01000
01001 ClientID client_id = (ClientID)p->Recv_uint32();
01002
01003 NetworkClientInfo *ci = NetworkClientInfo::GetByClientID(client_id);
01004 if (ci != NULL) {
01005 NetworkTextMessage(NETWORK_ACTION_LEAVE, CC_DEFAULT, false, ci->client_name, NULL, GetNetworkErrorMsg((NetworkErrorCode)p->Recv_uint8()));
01006 delete ci;
01007 }
01008
01009 SetWindowDirty(WC_CLIENT_LIST, 0);
01010
01011 return NETWORK_RECV_STATUS_OKAY;
01012 }
01013
01014 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_QUIT(Packet *p)
01015 {
01016 if (this->status < STATUS_AUTHORIZED) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01017
01018 ClientID client_id = (ClientID)p->Recv_uint32();
01019
01020 NetworkClientInfo *ci = NetworkClientInfo::GetByClientID(client_id);
01021 if (ci != NULL) {
01022 NetworkTextMessage(NETWORK_ACTION_LEAVE, CC_DEFAULT, false, ci->client_name, NULL, STR_NETWORK_MESSAGE_CLIENT_LEAVING);
01023 delete ci;
01024 } else {
01025 DEBUG(net, 0, "Unknown client (%d) is leaving the game", client_id);
01026 }
01027
01028 SetWindowDirty(WC_CLIENT_LIST, 0);
01029
01030
01031 return NETWORK_RECV_STATUS_OKAY;
01032 }
01033
01034 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_JOIN(Packet *p)
01035 {
01036 if (this->status < STATUS_AUTHORIZED) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01037
01038 ClientID client_id = (ClientID)p->Recv_uint32();
01039
01040 NetworkClientInfo *ci = NetworkClientInfo::GetByClientID(client_id);
01041 if (ci != NULL) {
01042 NetworkTextMessage(NETWORK_ACTION_JOIN, CC_DEFAULT, false, ci->client_name);
01043 }
01044
01045 SetWindowDirty(WC_CLIENT_LIST, 0);
01046
01047 return NETWORK_RECV_STATUS_OKAY;
01048 }
01049
01050 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_SHUTDOWN(Packet *p)
01051 {
01052
01053
01054 if (this->status >= STATUS_JOIN) {
01055 ShowErrorMessage(STR_NETWORK_MESSAGE_SERVER_SHUTDOWN, INVALID_STRING_ID, WL_CRITICAL);
01056 }
01057
01058 return NETWORK_RECV_STATUS_SERVER_ERROR;
01059 }
01060
01061 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_NEWGAME(Packet *p)
01062 {
01063
01064
01065 if (this->status >= STATUS_JOIN) {
01066
01067
01068
01069 _network_reconnect = _network_own_client_id % 16;
01070 ShowErrorMessage(STR_NETWORK_MESSAGE_SERVER_REBOOT, INVALID_STRING_ID, WL_CRITICAL);
01071 }
01072
01073 return NETWORK_RECV_STATUS_SERVER_ERROR;
01074 }
01075
01076 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_RCON(Packet *p)
01077 {
01078 if (this->status < STATUS_AUTHORIZED) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01079
01080 TextColour colour_code = (TextColour)p->Recv_uint16();
01081 if (!IsValidConsoleColour(colour_code)) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01082
01083 char rcon_out[NETWORK_RCONCOMMAND_LENGTH];
01084 p->Recv_string(rcon_out, sizeof(rcon_out));
01085
01086 IConsolePrint(colour_code, rcon_out);
01087
01088 return NETWORK_RECV_STATUS_OKAY;
01089 }
01090
01091 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_MOVE(Packet *p)
01092 {
01093 if (this->status < STATUS_AUTHORIZED) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01094
01095
01096 ClientID client_id = (ClientID)p->Recv_uint32();
01097 CompanyID company_id = (CompanyID)p->Recv_uint8();
01098
01099 if (client_id == 0) {
01100
01101 DEBUG(net, 0, "[move] received invalid client index = 0");
01102 return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01103 }
01104
01105 const NetworkClientInfo *ci = NetworkClientInfo::GetByClientID(client_id);
01106
01107 if (ci == NULL) return NETWORK_RECV_STATUS_OKAY;
01108
01109
01110 if (!Company::IsValidID(company_id)) company_id = COMPANY_SPECTATOR;
01111
01112 if (client_id == _network_own_client_id) {
01113 SetLocalCompany(company_id);
01114 }
01115
01116 return NETWORK_RECV_STATUS_OKAY;
01117 }
01118
01119 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_CONFIG_UPDATE(Packet *p)
01120 {
01121 if (this->status < STATUS_ACTIVE) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01122
01123 _network_server_max_companies = p->Recv_uint8();
01124 _network_server_max_spectators = p->Recv_uint8();
01125
01126 return NETWORK_RECV_STATUS_OKAY;
01127 }
01128
01129 NetworkRecvStatus ClientNetworkGameSocketHandler::Receive_SERVER_COMPANY_UPDATE(Packet *p)
01130 {
01131 if (this->status < STATUS_ACTIVE) return NETWORK_RECV_STATUS_MALFORMED_PACKET;
01132
01133 _network_company_passworded = p->Recv_uint16();
01134 SetWindowClassesDirty(WC_COMPANY);
01135
01136 return NETWORK_RECV_STATUS_OKAY;
01137 }
01138
01142 void ClientNetworkGameSocketHandler::CheckConnection()
01143 {
01144
01145 if (this->status < STATUS_AUTHORIZED) return;
01146
01147
01148 if (_realtime_tick < this->last_packet) this->last_packet = _realtime_tick;
01149
01150
01151
01152 uint lag = (_realtime_tick - this->last_packet) / 1000;
01153 if (lag < 5) return;
01154
01155
01156
01157 if (lag > 20) {
01158 this->NetworkGameSocketHandler::CloseConnection();
01159 ShowErrorMessage(STR_NETWORK_ERROR_LOSTCONNECTION, INVALID_STRING_ID, WL_CRITICAL);
01160 return;
01161 }
01162
01163
01164 static uint last_lag = 0;
01165 if (last_lag == lag) return;
01166
01167 last_lag = lag;
01168 SetDParam(0, lag);
01169 ShowErrorMessage(STR_NETWORK_ERROR_CLIENT_GUI_LOST_CONNECTION_CAPTION, STR_NETWORK_ERROR_CLIENT_GUI_LOST_CONNECTION, WL_INFO);
01170 }
01171
01172
01174 void NetworkClient_Connected()
01175 {
01176
01177 _frame_counter = 0;
01178 _frame_counter_server = 0;
01179 last_ack_frame = 0;
01180
01181 MyClient::SendJoin();
01182 }
01183
01189 void NetworkClientSendRcon(const char *password, const char *command)
01190 {
01191 MyClient::SendRCon(password, command);
01192 }
01193
01200 void NetworkClientRequestMove(CompanyID company_id, const char *pass)
01201 {
01202 MyClient::SendMove(company_id, pass);
01203 }
01204
01209 void NetworkClientsToSpectators(CompanyID cid)
01210 {
01211 Backup<CompanyByte> cur_company(_current_company, FILE_LINE);
01212
01213 if (cid == _local_company) SetLocalCompany(COMPANY_SPECTATOR);
01214
01215 NetworkClientInfo *ci;
01216 FOR_ALL_CLIENT_INFOS(ci) {
01217 if (ci->client_playas != cid) continue;
01218 NetworkTextMessage(NETWORK_ACTION_COMPANY_SPECTATOR, CC_DEFAULT, false, ci->client_name);
01219 ci->client_playas = COMPANY_SPECTATOR;
01220 }
01221
01222 cur_company.Restore();
01223 }
01224
01228 void NetworkUpdateClientName()
01229 {
01230 NetworkClientInfo *ci = NetworkClientInfo::GetByClientID(_network_own_client_id);
01231
01232 if (ci == NULL) return;
01233
01234
01235 if (strcmp(ci->client_name, _settings_client.network.client_name) != 0) {
01236 if (!_network_server) {
01237 MyClient::SendSetName(_settings_client.network.client_name);
01238 } else {
01239 if (NetworkFindName(_settings_client.network.client_name)) {
01240 NetworkTextMessage(NETWORK_ACTION_NAME_CHANGE, CC_DEFAULT, false, ci->client_name, _settings_client.network.client_name);
01241 strecpy(ci->client_name, _settings_client.network.client_name, lastof(ci->client_name));
01242 NetworkUpdateClientInfo(CLIENT_ID_SERVER);
01243 }
01244 }
01245 }
01246 }
01247
01256 void NetworkClientSendChat(NetworkAction action, DestType type, int dest, const char *msg, int64 data)
01257 {
01258 MyClient::SendChat(action, type, dest, msg, data);
01259 }
01260
01265 void NetworkClientSetCompanyPassword(const char *password)
01266 {
01267 MyClient::SendSetPassword(password);
01268 }
01269
01275 bool NetworkClientPreferTeamChat(const NetworkClientInfo *cio)
01276 {
01277
01278 if (!_settings_client.gui.prefer_teamchat || !Company::IsValidID(cio->client_playas)) return false;
01279
01280 const NetworkClientInfo *ci;
01281 FOR_ALL_CLIENT_INFOS(ci) {
01282 if (ci->client_playas == cio->client_playas && ci != cio) return true;
01283 }
01284
01285 return false;
01286 }
01287
01292 bool NetworkMaxCompaniesReached()
01293 {
01294 return Company::GetNumItems() >= (_network_server ? _settings_client.network.max_companies : _network_server_max_companies);
01295 }
01296
01301 bool NetworkMaxSpectatorsReached()
01302 {
01303 return NetworkSpectatorCount() >= (_network_server ? _settings_client.network.max_spectators : _network_server_max_spectators);
01304 }
01305
01306 #endif