X-Git-Url: http://deadsoftware.ru/gitweb?a=blobdiff_plain;f=src%2Fgame%2Fg_nethandler.pas;h=ccaddfc87b2b29e1d2966f78d080aa30db6f7ed1;hb=c55f7817903ddb9173e96819ec8723447b2d4c68;hp=53e6d8f8725aa5195d4b6851a58cf4f39e05096c;hpb=987c4a835a103345b59937e8e1be8524a6228712;p=d2df-sdl.git

diff --git a/src/game/g_nethandler.pas b/src/game/g_nethandler.pas
index 53e6d8f..ccaddfc 100644
--- a/src/game/g_nethandler.pas
+++ b/src/game/g_nethandler.pas
@@ -34,59 +34,85 @@ implementation
 
 uses sysutils, g_console;
 
+// TODO: Unify this with g_Net_Host_HandlePacket() somehow? They're almost the same.
 procedure g_Net_Client_HandlePacket(P: pENetPacket; Handler: TNetClientMsgHandler);
 var
   MNext: Integer;
   MSize: LongWord;
-  MHandled: Boolean;
+  MHandled: Boolean = false;
   NetMsg: TMsg;
+  Err: Boolean;
 begin
-  if not NetMsg.Init(P^.data, P^.dataLength, True) then
-    Exit;
+  if not NetMsg.Init(P^.data, P^.dataLength, True)
+    then exit;
 
+  Err := False;
   MNext := 0;
 
-  while NetMsg.BytesLeft() > 0 do
+  while (NetMsg.BytesLeft() > 0) and (not Err) do
   begin
-    MSize := NetMsg.ReadLongWord();
-    MNext := NetMsg.ReadCount + MSize;
-    MHandled := Handler(NetMsg); // TODO: maybe do something with this bool
-    NetMsg.Seek(MNext);
+    try
+      MSize := NetMsg.ReadLongWord();
+      MNext := NetMsg.ReadCount + MSize;
+      MHandled := Handler(NetMsg); // TODO: maybe do something with this bool
+      NetMsg.Seek(MNext);
+    except
+      Err := True;
+    end;
   end;
 
   enet_packet_destroy(P);
+  //if Err
+  //  then MC_MalformedPacket(S);
 end;
 
 procedure g_Net_Host_HandlePacket(S: pTNetClient; P: pENetPacket; Handler: TNetHostMsgHandler);
 var
   MNext: Integer;
   MSize: LongWord;
-  MHandled: Boolean;
+  MHandled: Boolean = false;
   NetMsg: TMsg;
+  Err: Boolean;
 begin
-  if not NetMsg.Init(P^.data, P^.dataLength, True) then
-    Exit;
+  if not NetMsg.Init(P^.data, P^.dataLength, True)
+    then exit;
 
+  Err := False;
   MNext := 0;
 
-  while NetMsg.BytesLeft() > 0 do
+  while (NetMsg.BytesLeft() > 0) and (not Err) do
   begin
-    MSize := NetMsg.ReadLongWord();
-    MNext := NetMsg.ReadCount + MSize;
-    MHandled := Handler(S, NetMsg); // TODO: maybe do something with this bool
-    NetMsg.Seek(MNext);
+    try
+      MSize := NetMsg.ReadLongWord();
+      MNext := NetMsg.ReadCount + MSize;
+      MHandled := Handler(S, NetMsg); // TODO: maybe do something with this bool
+      NetMsg.Seek(MNext);
+    except
+      Err := True;
+    end;
   end;
 
   enet_packet_destroy(P);
+  if Err
+    then MH_MalformedPacket(S);
 end;
 
 
 function g_Net_ClientMsgHandler(NetMsg: TMsg): Boolean;
 var
   MID: Byte;
+  Err: Boolean;
 begin
   Result := True;
-  MID := NetMsg.ReadByte();
+  Err := False;
+  try
+    MID := NetMsg.ReadByte();
+  except
+    MID := 0;
+    Err := True;
+  end;
+
+  //if Err then begin MC_MalformedPacket(S); Exit; end;
 
   case MID of
     NET_MSG_CHAT:   MC_RECV_Chat(NetMsg);
@@ -97,6 +123,7 @@ begin
     NET_MSG_GEVENT: MC_RECV_GameEvent(NetMsg);
     NET_MSG_FLAG:   MC_RECV_FlagEvent(NetMsg);
     NET_MSG_GSET:   MC_RECV_GameSettings(NetMsg);
+    NET_MSG_FLAGPOS:MC_RECV_FlagPos(NetMsg);
 
     NET_MSG_PLR:    MC_RECV_PlayerCreate(NetMsg);
     NET_MSG_PLRPOS: MC_RECV_PlayerPos(NetMsg);
@@ -119,6 +146,7 @@ begin
 
     NET_MSG_ISPAWN: MC_RECV_ItemSpawn(NetMsg);
     NET_MSG_IDEL:   MC_RECV_ItemDestroy(NetMsg);
+    NET_MSG_IPOS:   MC_RECV_ItemPos(NetMsg);
 
     NET_MSG_PSTATE: MC_RECV_PanelState(NetMsg);
     NET_MSG_PTEX:   MC_RECV_PanelTexture(NetMsg);
@@ -140,9 +168,18 @@ end;
 function g_Net_ClientLightMsgHandler(NetMsg: TMsg): Boolean;
 var
   MID: Byte;
+  Err: Boolean;
 begin
   Result := True;
-  MID := NetMsg.ReadByte();
+  Err := False;
+  try
+    MID := NetMsg.ReadByte();
+  except
+    MID := 0;
+    Err := True;
+  end;
+
+  //if Err then begin MC_MalformedPacket(S); Exit; end;
 
   case MID of
     NET_MSG_GEVENT: MC_RECV_GameEvent(NetMsg);
@@ -158,9 +195,18 @@ end;
 function g_Net_HostMsgHandler(S: pTNetClient; NetMsg: TMsg): Boolean;
 var
   MID: Byte;
+  Err: Boolean;
 begin
   Result := True;
-  MID := NetMsg.ReadByte();
+  Err := False;
+  try
+    MID := NetMsg.ReadByte();
+  except
+    MID := 0;
+    Err := True;
+  end;
+
+  if Err then begin MH_MalformedPacket(S); Exit; end;
 
   case MID of
     NET_MSG_INFO: MH_RECV_Info(S, NetMsg);
@@ -174,8 +220,8 @@ begin
     NET_MSG_RCON_AUTH: MH_RECV_RCONPassword(S, NetMsg);
     NET_MSG_RCON_CMD:  MH_RECV_RCONCommand(S, NetMsg);
 
-    NET_MSG_MAP_REQUEST: MH_RECV_MapRequest(S, NetMsg);
-    NET_MSG_RES_REQUEST: MH_RECV_ResRequest(S, NetMsg);
+    //NET_MSG_MAP_REQUEST: MH_RECV_MapRequest(S, NetMsg);
+    //NET_MSG_RES_REQUEST: MH_RECV_ResRequest(S, NetMsg);
 
     NET_MSG_VOTE_EVENT: MH_RECV_Vote(S, NetMsg);
   end;