small compression
[carveJwlIkooP6JGAAIwe30JlM.git] / gameserver.c
index 3cbf87791fc0167a1c5d91c7a0f44c6fbb35040d..b7b809087c8e16c2a348891e0876ed7dfdea38b9 100644 (file)
@@ -16,6 +16,7 @@ volatile sig_atomic_t sig_stop;
 #include "network_common.h"
 #include "gameserver_db.h"
 #include "vg/vg_m.h"
+#include "vg/vg_msg.h"
 
 static void inthandler( int signum ) {
    sig_stop = 1;
@@ -320,6 +321,24 @@ static int packet_minsize( SteamNetworkingMessage_t *msg, u32 size ){
    }
 }
 
+struct db_set_username_thread_data {
+   u64 steamid;
+   char username[ NETWORK_USERNAME_MAX ];
+};
+
+static void gameserver_update_db_username( db_request *db_req ){
+   struct db_set_username_thread_data *inf = (void *)db_req->data;
+
+   if( inf->steamid == k_connection_unauthorized )
+      return;
+
+   int admin = 0;
+   if( inf->steamid == 76561198072130043 )
+      admin = 2;
+
+   db_updateuser( inf->steamid, inf->username, admin );
+}
+
 static void gameserver_rx_200_300( SteamNetworkingMessage_t *msg ){
    netmsg_blank *tmp = msg->m_pData;
 
@@ -352,6 +371,16 @@ static void gameserver_rx_200_300( SteamNetworkingMessage_t *msg ){
       u32 propsize = sizeof(netmsg_playerusername) + chs + 1;
       gameserver_send_to_all( client_id, prop, propsize,
                               k_nSteamNetworkingSend_Reliable );
+
+      /* update database about this */
+      db_request *call = db_alloc_request( 
+                           sizeof(struct db_set_username_thread_data) );
+      struct db_set_username_thread_data *inf = (void *)call->data;
+      inf->steamid = get_connection_authsteamid( msg );
+      vg_strncpy( client->username, inf->username, 
+                  sizeof(inf->username), k_strncpy_always_add_null );
+      call->handler = gameserver_update_db_username;
+      db_send_request( call );
    }
    else if( tmp->inetmsg_id == k_inetmsg_playerframe ){
       /* propogate */
@@ -393,6 +422,220 @@ static void gameserver_rx_200_300( SteamNetworkingMessage_t *msg ){
       gameserver_send_to_all( client_id, prop, msg->m_cbSize, 
                               k_nSteamNetworkingSend_Reliable );
    }
+   else {
+      vg_warn( "Unknown inetmsg_id recieved from client. (%u)\n",
+               tmp->inetmsg_id );
+   }
+}
+
+static void gameserver_request_respond( enum request_status status,
+                                        netmsg_request *res, vg_msg *body,
+                                        SteamNetworkingMessage_t *msg ){
+   int client_id = gameserver_client_index( msg->m_conn );
+   u32 len = 0;
+   if( body ){
+      len = body->cur.co;
+      vg_low( "[%d#%d] Response: %d\n", client_id, (i32)res->id, status );
+      vg_msg_print( body, len );
+   }
+
+   res->status = status;
+
+   SteamAPI_ISteamNetworkingSockets_SendMessageToConnection(
+         hSteamNetworkingSockets, msg->m_conn,
+         res, sizeof(netmsg_request) + len,
+         k_nSteamNetworkingSend_Reliable, NULL );
+   SteamAPI_SteamNetworkingMessage_t_Release( msg );
+}
+
+struct user_request_thread_data {
+   SteamNetworkingMessage_t *msg;
+};
+
+static u32 gameserver_get_current_week(void){
+   return time(NULL) / (7*24*60*60);
+}
+
+static enum request_status gameserver_cat_table( 
+      vg_msg *msg, 
+      const char *mod, const char *route, u32 week, const char *alias )
+{
+   char table_name[ DB_TABLE_UID_MAX ];
+   if( !db_get_highscore_table_name( mod, route, week, table_name ) )
+      return k_request_status_out_of_memory;
+
+   char buf[512];
+   vg_str q;
+   vg_strnull( &q, buf, 512 );
+   vg_strcat( &q, "SELECT * FROM \"" );
+   vg_strcat( &q, table_name );
+   vg_strcat( &q, "\" ORDER BY time DESC LIMIT 10;" );
+   if( !vg_strgood(&q) )
+      return k_request_status_out_of_memory;
+
+   sqlite3_stmt *stmt = db_stmt( q.buffer );
+   if( !stmt )
+      return k_request_status_database_error;
+
+   vg_msg_frame( msg, alias );
+   for( u32 i=0; i<10; i ++ ){
+      int fc = sqlite3_step( stmt );
+
+      if( fc == SQLITE_ROW ){
+         i32 time = sqlite3_column_int( stmt, 1 );
+         i64 steamid_i64 = sqlite3_column_int64( stmt, 0 );
+         u64 steamid = *((u64 *)&steamid_i64);
+
+         if( steamid == k_connection_unauthorized )
+            continue;
+
+         vg_msg_frame( msg, "" );
+         vg_msg_wkvu32( msg, "time", time );
+         vg_msg_wkvu64( msg, "steamid", steamid );
+
+         char username[32];
+         if( db_getuserinfo( steamid, username, sizeof(username), NULL ) )
+            vg_msg_wkvstr( msg, "username", username );
+         vg_msg_end_frame( msg );
+      }
+      else if( fc == SQLITE_DONE ){
+         break;
+      }
+      else {
+         log_sqlite3( fc );
+         break;
+      }
+   }
+
+   sqlite3_finalize( stmt );
+   vg_msg_end_frame( msg );
+   return k_request_status_ok;
+}
+
+static void gameserver_process_user_request( db_request *db_req ){
+   struct user_request_thread_data *inf = (void *)db_req->data;
+   SteamNetworkingMessage_t *msg = inf->msg;
+
+   int client_id = gameserver_client_index( msg->m_conn );
+   if( client_id == -1 ){
+      SteamAPI_SteamNetworkingMessage_t_Release( msg );
+      return;
+   }
+
+   netmsg_request *req = (netmsg_request *)msg->m_pData;
+   vg_msg data;
+   vg_msg_init( &data, req->q, msg->m_cbSize - sizeof(netmsg_request) );
+
+   /* create response packet */
+   netmsg_request *res = alloca( sizeof(netmsg_request) + 512 );
+   res->inetmsg_id = k_inetmsg_response;
+   res->id = req->id;
+   vg_msg body;
+   vg_msg_init( &body, res->q, 512 );
+
+   const char *endpoint = vg_msg_getkvstr( &data, "endpoint" );
+
+   if( !endpoint ){
+      gameserver_request_respond( k_request_status_invalid_endpoint,
+                                  res, NULL, msg );
+      return;
+   }
+
+   if( !strcmp( endpoint, "scoreboard" ) ){
+      const char *mod = vg_msg_getkvstr( &data, "mod" );
+      const char *route = vg_msg_getkvstr( &data, "route" );
+      u32 week = vg_msg_getkvu32( &data, "week", 0 );
+      
+      if( week == NETWORK_LEADERBOARD_CURRENT_WEEK ){
+         gameserver_cat_table( &body, mod, route, 
+                               gameserver_get_current_week(), "rows_weekly" );
+      }
+      else if( week == NETWORK_LEADERBOARD_ALLTIME_AND_CURRENT_WEEK ){
+         gameserver_cat_table( &body, mod, route, 0, "rows" );
+         gameserver_cat_table( &body, mod, route, 
+                               gameserver_get_current_week(), "rows_weekly" );
+      }
+      else 
+         gameserver_cat_table( &body, mod, route, week, "rows" );
+
+      if( body.error != k_vg_msg_error_OK ){
+         gameserver_request_respond( k_request_status_out_of_memory,
+                                     res, NULL, msg );
+         return;
+      }
+
+      gameserver_request_respond( k_request_status_ok, res, &body, msg );
+   }
+   else if( !strcmp( endpoint, "setlap" ) ){
+      /* TODO: we can change userdata to be the client ID and store that in
+       *       the client structure. it also would save us scanning the 
+       *       client array over and over..? 
+       */
+      u64 steamid = get_connection_authsteamid( msg );
+      if( steamid == k_connection_unauthorized ){
+         gameserver_request_respond( k_request_status_unauthorized,
+                                     res, NULL, msg );
+         return;
+      }
+
+      const char *mod = vg_msg_getkvstr( &data, "mod" );
+      const char *route = vg_msg_getkvstr( &data, "route" );
+      
+      char weekly_table[ DB_TABLE_UID_MAX ],
+           alltime_table[ DB_TABLE_UID_MAX ];
+
+      u32 week = gameserver_get_current_week();
+
+      if( !db_get_highscore_table_name( mod, route, 0, alltime_table ) ||
+          !db_get_highscore_table_name( mod, route, week, weekly_table ) ){
+         gameserver_request_respond( k_request_status_out_of_memory,
+                                     res, NULL, msg );
+         return;
+      }
+
+      i32 centiseconds = vg_msg_getkvi32( &data, "time", -1 );
+      if( centiseconds < 5*100 ){
+         gameserver_request_respond( k_request_status_client_error,
+                                     res, NULL, msg );
+         return;
+      }
+
+      db_writeusertime( alltime_table, steamid, centiseconds, 1 );
+      db_writeusertime( weekly_table, steamid, centiseconds, 1 );
+
+      gameserver_request_respond( k_request_status_ok, res, NULL, msg );
+   }
+   else{
+      gameserver_request_respond( k_request_status_invalid_endpoint,
+                                  res, NULL, msg );
+   }
+}
+
+static void gameserver_rx_300_400( SteamNetworkingMessage_t *msg ){
+   netmsg_blank *tmp = msg->m_pData;
+
+   int client_id = gameserver_client_index( msg->m_conn );
+   if( client_id == -1 ){
+      SteamAPI_SteamNetworkingMessage_t_Release( msg );
+      return;
+   }
+
+   if( tmp->inetmsg_id == k_inetmsg_request ){
+      if( !packet_minsize( msg, sizeof(netmsg_request)+1 ))
+         return;
+
+      db_request *call = db_alloc_request( 
+                           sizeof(struct user_request_thread_data) );
+      struct user_request_thread_data *inf = (void *)call->data;
+      inf->msg = msg;
+      call->handler = gameserver_process_user_request;
+      db_send_request( call );
+   }
+   else {
+      vg_warn( "Unknown inetmsg_id recieved from client. (%u)\n",
+               tmp->inetmsg_id );
+      SteamAPI_SteamNetworkingMessage_t_Release( msg );
+   }
 }
 
 static void poll_connections(void){
@@ -420,6 +663,10 @@ static void poll_connections(void){
 
          if( (tmp->inetmsg_id >= 200) && (tmp->inetmsg_id < 300) ){
             gameserver_rx_200_300( msg );
+            SteamAPI_SteamNetworkingMessage_t_Release( msg );
+         }
+         else if( (tmp->inetmsg_id >= 300) && (tmp->inetmsg_id < 400) ){
+            gameserver_rx_300_400( msg );
          }
          else{
             if( tmp->inetmsg_id == k_inetmsg_auth )
@@ -428,9 +675,8 @@ static void poll_connections(void){
                vg_warn( "Unknown inetmsg_id recieved from client. (%u)\n",
                         tmp->inetmsg_id );
             }
+            SteamAPI_SteamNetworkingMessage_t_Release( msg );
          }
-
-         SteamAPI_SteamNetworkingMessage_t_Release( msg );
       }
    }
 }
@@ -442,8 +688,8 @@ static u64 seconds_to_server_ticks( double s ){
 static void test_runner( db_request *req ){
    vg_warn( "RUNNER\n" );
    char table[DB_TABLE_UID_MAX];
-   if( db_get_highscore_table_name( "sr003-local-mp_mtzero", 
-                                    "megapark-yellow", 302, table ) ){
+   if( db_get_highscore_table_name( "sr002-local-mp_mtzero", 
+                                    "Coastal Run", 0, table ) ){
       if( db_writeusertime( table, 76561198072130043, 232, 1 ) ){
          vg_success( "Written time\n" );
          i32 v = db_readusertime( table, 76561198072130043 );
@@ -490,7 +736,7 @@ int main( int argc, char *argv[] ){
       vg_warn( "Running without user authentication.\n" );
    }
 
-   if( !SteamGameServer_Init( 0, 27400, 27401, 
+   if( !SteamGameServer_Init( 0, NETWORK_PORT, NETWORK_PORT+1, 
                               gameserver.auth_mode, "1.0.0.0" ) ){
       vg_error( "SteamGameServer_Init failed\n" );
       return 0;
@@ -517,7 +763,7 @@ int main( int argc, char *argv[] ){
    HSteamListenSocket listener;
    SteamNetworkingIPAddr localAddr;
    SteamAPI_SteamNetworkingIPAddr_Clear( &localAddr );
-   localAddr.m_port = 27402;
+   localAddr.m_port = NETWORK_PORT;
 
    listener = SteamAPI_ISteamNetworkingSockets_CreateListenSocketIP(
                   hSteamNetworkingSockets, &localAddr, 0, NULL );
@@ -554,6 +800,7 @@ int main( int argc, char *argv[] ){
    
    vg_info( "Shutting down\n..." );
    SteamGameServer_Shutdown();
+   db_kill();
    db_free();
 
    return 0;