/*
* Hedgewars, a free turn based strategy game
* Copyright (C) 2012 Simeon Maxein <smaxein@googlemail.com>
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include "ipcprotocol.h"
#include "../util/util.h"
#include "../util/logging.h"
#include "../md5/md5.h"
#include <stdio.h>
#include <stdbool.h>
#include <string.h>
#include <inttypes.h>
#include <stdlib.h>
int flib_ipc_append_message(flib_vector *vec, const char *fmt, ...) {
int result = -1;
if(!log_badargs_if2(vec==NULL, fmt==NULL)) {
// 1 byte size prefix, 255 bytes max message length, 1 0-byte for vsnprintf
char msgbuffer[257];
// Format the message, leaving one byte at the start for the length
va_list argp;
va_start(argp, fmt);
int msgSize = vsnprintf(msgbuffer+1, 256, fmt, argp);
va_end(argp);
if(!log_e_if(msgSize > 255, "Message too long (%u bytes)", (unsigned)msgSize)
&& !log_e_if(msgSize < 0, "printf error")) {
// Add the length prefix
((uint8_t*)msgbuffer)[0] = msgSize;
// Append it to the vector
result = flib_vector_append(vec, msgbuffer, msgSize+1);
}
}
return result;
}
int flib_ipc_append_mapconf(flib_vector *vec, const flib_map *map, bool mappreview) {
int result = -1;
flib_vector *tempvector = flib_vector_create();
if(!log_badargs_if2(vec==NULL, map==NULL)) {
bool error = false;
if(map->mapgen == MAPGEN_NAMED) {
error |= log_e_if(!map->name, "Missing map name")
|| flib_ipc_append_message(tempvector, "emap %s", map->name);
}
if(!mappreview) {
error |= log_e_if(!map->theme, "Missing map theme")
|| flib_ipc_append_message(tempvector, "etheme %s", map->theme);
}
error |= flib_ipc_append_seed(tempvector, map->seed);
error |= flib_ipc_append_message(tempvector, "e$template_filter %i", map->templateFilter);
error |= flib_ipc_append_message(tempvector, "e$mapgen %i", map->mapgen);
if(map->mapgen == MAPGEN_MAZE) {
error |= flib_ipc_append_message(tempvector, "e$maze_size %i", map->mazeSize);
}
if(map->mapgen == MAPGEN_DRAWN) {
/*
* We have to split the drawn map data into several edraw messages here because
* it can be longer than the maximum message size.
*/
const char *edraw = "edraw ";
int edrawlen = strlen(edraw);
for(size_t offset=0; offset<map->drawDataSize; offset+=200) {
size_t bytesRemaining = map->drawDataSize-offset;
int fragmentsize = bytesRemaining < 200 ? bytesRemaining : 200;
uint8_t messagesize = edrawlen + fragmentsize;
error |= flib_vector_append(tempvector, &messagesize, 1);
error |= flib_vector_append(tempvector, edraw, edrawlen);
error |= flib_vector_append(tempvector, map->drawData+offset, fragmentsize);
}
}
if(!log_e_if(error, "Error generating map config")) {
// Message created, now we can copy everything.
flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);
if(!flib_vector_append(vec, constbuf.data, constbuf.size)) {
result = 0;
}
}
}
flib_vector_destroy(tempvector);
return result;
}
int flib_ipc_append_seed(flib_vector *vec, const char *seed) {
if(log_badargs_if2(vec==NULL, seed==NULL)) {
return -1;
}
return flib_ipc_append_message(vec, "eseed %s", seed);
}
int flib_ipc_append_script(flib_vector *vec, const char *script) {
int result = -1;
if(!log_badargs_if2(vec==NULL, script==NULL)) {
result = flib_ipc_append_message(vec, "escript %s", script);
}
return result;
}
int flib_ipc_append_style(flib_vector *vec, const char *style) {
int result = -1;
char *copy = flib_strdupnull(style);
if(!log_badargs_if(vec==NULL) && copy) {
if(!strcmp("Normal", copy)) {
// "Normal" means no gametype script
// TODO if an empty script called "Normal" is added to the scripts directory this can be removed
result = 0;
} else {
size_t len = strlen(copy);
for(size_t i=0; i<len; i++) {
if(copy[i] == ' ') {
copy[i] = '_';
}
}
result = flib_ipc_append_message(vec, "escript %s%s.lua", MULTIPLAYER_SCRIPT_PATH, copy);
}
}
free(copy);
return result;
}
static uint32_t buildModFlags(const flib_scheme *scheme) {
uint32_t result = 0;
for(int i=0; i<flib_meta.modCount; i++) {
if(scheme->mods[i]) {
int bitmaskIndex = flib_meta.mods[i].bitmaskIndex;
result |= (UINT32_C(1) << bitmaskIndex);
}
}
return result;
}
int flib_ipc_append_gamescheme(flib_vector *vec, const flib_scheme *scheme) {
int result = -1;
flib_vector *tempvector = flib_vector_create();
if(!log_badargs_if2(vec==NULL, scheme==NULL) && tempvector) {
bool error = false;
error |= flib_ipc_append_message(tempvector, "e$gmflags %"PRIu32, buildModFlags(scheme));
for(int i=0; i<flib_meta.settingCount; i++) {
if(flib_meta.settings[i].engineCommand) {
int value = scheme->settings[i];
if(flib_meta.settings[i].maxMeansInfinity) {
value = value>=flib_meta.settings[i].max ? 9999 : value;
}
if(flib_meta.settings[i].times1000) {
value *= 1000;
}
error |= flib_ipc_append_message(tempvector, "%s %i", flib_meta.settings[i].engineCommand, value);
}
}
if(!error) {
// Message created, now we can copy everything.
flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);
if(!flib_vector_append(vec, constbuf.data, constbuf.size)) {
result = 0;
}
}
}
flib_vector_destroy(tempvector);
return result;
}
static int appendWeaponSet(flib_vector *vec, flib_weaponset *set) {
return flib_ipc_append_message(vec, "eammloadt %s", set->loadout)
|| flib_ipc_append_message(vec, "eammprob %s", set->crateprob)
|| flib_ipc_append_message(vec, "eammdelay %s", set->delay)
|| flib_ipc_append_message(vec, "eammreinf %s", set->crateammo);
}
static void calculateMd5Hex(const char *in, char out[33]) {
md5_state_t md5state;
uint8_t md5bytes[16];
md5_init(&md5state);
md5_append(&md5state, (unsigned char*)in, strlen(in));
md5_finish(&md5state, md5bytes);
for(int i=0;i<sizeof(md5bytes); i++) {
snprintf(out+i*2, 3, "%02x", (unsigned)md5bytes[i]);
}
}
static int flib_ipc_append_addteam(flib_vector *vec, const flib_team *team, bool perHogAmmo, bool noAmmoStore) {
int result = -1;
flib_vector *tempvector = flib_vector_create();
if(!log_badargs_if2(vec==NULL, team==NULL) && tempvector) {
bool error = false;
if(!perHogAmmo && !noAmmoStore) {
error = error
|| appendWeaponSet(tempvector, team->hogs[0].weaponset)
|| flib_ipc_append_message(tempvector, "eammstore");
}
char md5Hex[33];
calculateMd5Hex(team->ownerName ? team->ownerName : "", md5Hex);
if(team->colorIndex<0 || team->colorIndex>=flib_teamcolor_count) {
flib_log_e("Color index out of bounds for team %s: %i", team->name, team->colorIndex);
error = true;
} else {
error |= flib_ipc_append_message(tempvector, "eaddteam %s %"PRIu32" %s", md5Hex, flib_teamcolors[team->colorIndex], team->name);
}
if(team->remoteDriven) {
error |= flib_ipc_append_message(tempvector, "erdriven");
}
error |= flib_ipc_append_message(tempvector, "egrave %s", team->grave);
error |= flib_ipc_append_message(tempvector, "efort %s", team->fort);
error |= flib_ipc_append_message(tempvector, "evoicepack %s", team->voicepack);
error |= flib_ipc_append_message(tempvector, "eflag %s", team->flag);
for(int i=0; i<team->bindingCount; i++) {
error |= flib_ipc_append_message(tempvector, "ebind %s %s", team->bindings[i].binding, team->bindings[i].action);
}
for(int i=0; i<team->hogsInGame; i++) {
if(perHogAmmo && !noAmmoStore) {
error |= appendWeaponSet(tempvector, team->hogs[i].weaponset);
}
error |= flib_ipc_append_message(tempvector, "eaddhh %i %i %s", team->hogs[i].difficulty, team->hogs[i].initialHealth, team->hogs[i].name);
error |= flib_ipc_append_message(tempvector, "ehat %s", team->hogs[i].hat);
}
if(!error) {
// Message created, now we can copy everything.
flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);
if(!flib_vector_append(vec, constbuf.data, constbuf.size)) {
result = 0;
}
}
}
flib_vector_destroy(tempvector);
return result;
}
int flib_ipc_append_fullconfig(flib_vector *vec, const flib_gamesetup *setup, bool netgame) {
int result = -1;
flib_vector *tempvector = flib_vector_create();
if(!log_badargs_if2(vec==NULL, setup==NULL) && tempvector) {
bool error = false;
bool perHogAmmo = false;
bool sharedAmmo = false;
error |= flib_ipc_append_message(vec, netgame ? "TN" : "TL");
if(setup->map) {
error |= flib_ipc_append_mapconf(tempvector, setup->map, false);
}
if(setup->style) {
error |= flib_ipc_append_style(tempvector, setup->style);
}
if(setup->gamescheme) {
error |= flib_ipc_append_gamescheme(tempvector, setup->gamescheme);
sharedAmmo = flib_scheme_get_mod(setup->gamescheme, "sharedammo");
// Shared ammo has priority over per-hog ammo
perHogAmmo = !sharedAmmo && flib_scheme_get_mod(setup->gamescheme, "perhogammo");
}
if(setup->teamlist->teams && setup->teamlist->teamCount>0) {
int *clanColors = flib_calloc(setup->teamlist->teamCount, sizeof(int));
if(!clanColors) {
error = true;
} else {
int clanCount = 0;
for(int i=0; !error && i<setup->teamlist->teamCount; i++) {
flib_team *team = setup->teamlist->teams[i];
// Find the clan index of this team (clans are identified by color).
bool newClan = false;
int clan = 0;
while(clan<clanCount && clanColors[clan] != team->colorIndex) {
clan++;
}
if(clan==clanCount) {
newClan = true;
clanCount++;
clanColors[clan] = team->colorIndex;
}
// If shared ammo is active, only add an ammo store for the first team in each clan.
bool noAmmoStore = sharedAmmo&&!newClan;
error |= flib_ipc_append_addteam(tempvector, setup->teamlist->teams[i], perHogAmmo, noAmmoStore);
}
}
free(clanColors);
}
error |= flib_ipc_append_message(tempvector, "!");
if(!error) {
// Message created, now we can copy everything.
flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);
if(!flib_vector_append(vec, constbuf.data, constbuf.size)) {
result = 0;
}
}
}
return result;
}