pkmn-classic-framework/gts/pokemondpds.ashx.cs
2014-10-14 18:50:41 -04:00

553 lines
22 KiB
C#

using System;
using System.Collections.Generic;
using System.Linq;
using System.Web;
using System.Web.SessionState;
using PkmnFoundations.Data;
using PkmnFoundations.Structures;
using PkmnFoundations.Support;
using System.IO;
using System.Threading;
using GamestatsBase;
namespace PkmnFoundations.GTS
{
/// <summary>
/// Summary description for pokemondpds
/// </summary>
public class pokemondpds : GamestatsHandler
{
public pokemondpds()
: base("sAdeqWo3voLeC5r16DYv",
0x45, 0x1111, 0x80000000, 0x4a3b2c1d, "pokemondpds",
GamestatsRequestVersions.Version2, GamestatsResponseVersions.Version1)
{
}
public override void ProcessGamestatsRequest(byte[] data, MemoryStream response, string url, int pid, HttpContext context, GamestatsSession session)
{
switch (url)
{
default:
SessionManager.Remove(session);
// unrecognized page url
ShowError(context, 404);
return;
#region Common
// Called during startup. Seems to contain trainer profile stats.
case "/pokemondpds/common/setProfile.asp":
{
SessionManager.Remove(session);
if (data.Length != 100)
{
ShowError(context, 400);
return;
}
#if !DEBUG
try
{
#endif
// todo: Figure out what fun stuff is contained in this blob!
byte[] profileBinary = new byte[100];
Array.Copy(data, 0, profileBinary, 0, 100);
TrainerProfile4 profile = new TrainerProfile4(pid, profileBinary);
Database.Instance.GamestatsSetProfile4(profile);
#if !DEBUG
}
catch { }
#endif
response.Write(new byte[] { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
0, 8);
} break;
#endregion
#region GTS
// Called during startup. Unknown purpose.
case "/pokemondpds/worldexchange/info.asp":
SessionManager.Remove(session);
// todo: find out the meaning of this request.
// is it simply done to check whether the GTS is online?
// todo: add a mostly null record to GtsProfiles4 if the
// player is on D/P. We can still track last visit times
// and gather trainer profile data from GTS activity.
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
break;
// Called during startup and when you check your pokemon's status.
case "/pokemondpds/worldexchange/result.asp":
{
SessionManager.Remove(session);
/* After the above step(s) or performing any of
* the tasks below other than searching, the game
* makes a request to /pokemondpds/worldexchange/result.asp.
* If the game has had a Pokémon sent to it via a trade,
* the server responds with the entire encrypted Pokémon
* save struct. Otherwise, if there is a Pokémon deposited
* in the GTS, it responds with 0x0004; if not, it responds
* with 0x0005. */
GtsRecord4 record = Database.Instance.GtsDataForUser4(pid);
if (record == null)
{
// No pokemon in the system
response.Write(new byte[] { 0x05, 0x00 }, 0, 2);
}
else if (record.IsExchanged > 0)
{
// traded pokemon arriving!!!
response.Write(record.Save(), 0, 292);
}
else
{
// my existing pokemon is in the system, untraded
response.Write(new byte[] { 0x04, 0x00 }, 0, 2);
}
// other responses:
// 0-2 causes a BSOD but it flashes siezure. Scary
// 3 causes it to be "checking GTS's status" forever.
} break;
// Called after result.asp returns 4 when you check your pokemon's status
case "/pokemondpds/worldexchange/get.asp":
{
SessionManager.Remove(session);
// this is only called if result.asp returned 4.
// todo: what does this do if the contained pokemon is traded??
GtsRecord4 record = Database.Instance.GtsDataForUser4(pid);
if (record == null)
{
// No pokemon in the system
// what do here?
ShowError(context, 400);
return;
}
else
{
// just write the record whether traded or not...
response.Write(record.Save(), 0, 292);
}
} break;
// Called after result.asp returns an inbound pokemon record to delete it
case "/pokemondpds/worldexchange/delete.asp":
{
SessionManager.Remove(session);
GtsRecord4 record = Database.Instance.GtsDataForUser4(pid);
if (record == null)
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
}
else if (record.IsExchanged > 0)
{
// delete the arrived pokemon from the system
// todo: add transactions
// todo: log the successful trade?
// (either here or when the trade is done)
bool success = Database.Instance.GtsDeletePokemon4(pid);
if (success)
{
#if !DEBUG
try
{
#endif
Database.Instance.GtsLogTrade4(record, DateTime.UtcNow);
#if !DEBUG
}
catch { }
#endif
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
}
else
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
}
}
else
{
// own pokemon is there, fail. Use return.asp instead.
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
}
} break;
// called to delete your own pokemon after taking it back
case "/pokemondpds/worldexchange/return.asp":
{
SessionManager.Remove(session);
GtsRecord4 record = Database.Instance.GtsDataForUser4(pid);
if (record == null)
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
}
else if (record.IsExchanged > 0)
{
// a traded pokemon is there, fail. Use delete.asp instead.
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
}
else
{
// delete own pokemon
// todo: add transactions
bool success = Database.Instance.GtsDeletePokemon4(pid);
if (success)
{
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
}
else
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
}
}
} break;
// Called when you deposit a pokemon into the system.
case "/pokemondpds/worldexchange/post.asp":
{
if (data.Length != 292)
{
SessionManager.Remove(session);
ShowError(context, 400);
return;
}
// todo: add transaction
if (Database.Instance.GtsDataForUser4(pid) != null)
{
// there's already a pokemon inside
SessionManager.Remove(session);
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
break;
}
// keep the record in memory while we wait for post_finish.asp request
byte[] recordBinary = new byte[292];
Array.Copy(data, 0, recordBinary, 0, 292);
GtsRecord4 record = new GtsRecord4(recordBinary);
if (!record.Validate())
{
// hack check failed
SessionManager.Remove(session);
response.Write(new byte[] { 0x0c, 0x00 }, 0, 2);
break;
}
// the following two fields are blank in the uploaded record.
// The server must provide them instead.
record.TimeDeposited = DateTime.UtcNow;
record.TimeExchanged = null;
record.IsExchanged = 0;
record.PID = pid;
session.Tag = record;
// todo: delete any other post.asp sessions registered under this PID
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
} break;
case "/pokemondpds/worldexchange/post_finish.asp":
{
SessionManager.Remove(session);
if (data.Length != 8)
{
ShowError(context, 400);
return;
}
// todo: these _finish requests seem to come with a magic number of 4 bytes
// at offset 0. Find out what this is supposed to do and how to validate it.
// find a matching session which contains our record
GamestatsSession prevSession = SessionManager.FindSession(pid, "/pokemondpds/worldexchange/post.asp");
if (prevSession == null)
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
return;
}
SessionManager.Remove(prevSession);
if (prevSession.Tag == null)
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
return;
}
AssertHelper.Assert(prevSession.Tag is GtsRecord4);
GtsRecord4 record = (GtsRecord4)prevSession.Tag;
if (Database.Instance.GtsDepositPokemon4(record))
{
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
}
else
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
} break;
// the search request has a funny bit string request of search terms
// and just returns a chunk of records end to end.
case "/pokemondpds/worldexchange/search.asp":
{
SessionManager.Remove(session);
if (data.Length < 7 || data.Length > 8)
{
ShowError(context, 400);
return;
}
ushort species = BitConverter.ToUInt16(data, 0);
if (species < 1)
{
ShowError(context, 400);
return;
}
int resultsCount = (int)data[6];
if (resultsCount < 1) break; // optimize away requests for no rows
Genders gender = (Genders)data[2];
byte minLevel = data[3];
byte maxLevel = data[4];
// byte 5 unknown
byte country = 0;
if (data.Length > 7) country = data[7];
if (resultsCount > 7) resultsCount = 7; // stop DDOS
GtsRecord4[] records = Database.Instance.GtsSearch4(pid, species, gender, minLevel, maxLevel, country, resultsCount);
foreach (GtsRecord4 record in records)
{
response.Write(record.Save(), 0, 292);
}
} break;
// the exchange request uploads a record of the exchangee pokemon
// plus the desired PID to trade for at the very end.
case "/pokemondpds/worldexchange/exchange.asp":
{
if (data.Length != 296)
{
SessionManager.Remove(session);
ShowError(context, 400);
return;
}
byte[] uploadData = new byte[292];
Array.Copy(data, 0, uploadData, 0, 292);
GtsRecord4 upload = new GtsRecord4(uploadData);
upload.IsExchanged = 0;
int targetPid = BitConverter.ToInt32(data, 292);
GtsRecord4 result = Database.Instance.GtsDataForUser4(targetPid);
if (result == null || result.IsExchanged != 0)
{
// Pokémon is traded (or was never here to begin with)
// todo: I only checked this on GenV. Check that this
// is the correct response on GenIV.
SessionManager.Remove(session);
response.Write(new byte[] { 0x02, 0x00 }, 0, 2);
break;
}
// enforce request requirements server side
if (!upload.Validate() || !upload.CanTrade(result))
{
// todo: find the correct codes for these
SessionManager.Remove(session);
response.Write(new byte[] { 0x0c, 0x00 }, 0, 2);
return;
}
object[] tag = new GtsRecord4[2];
tag[0] = upload;
tag[1] = result;
session.Tag = tag;
GtsRecord4 tradedResult = result.Clone();
tradedResult.FlagTraded(upload); // only real purpose is to generate a proper response
// todo: we need a mechanism to "reserve" a pokemon being traded at this
// point in the process, but be able to relinquish it if exchange_finish
// never happens.
// Currently, if two people try to take the same pokemon, it will appear
// to work for both but then fail for the second after they've saved
// their game. This causes a hard crash and a "save file is corrupt,
// "previous will be loaded" error when restarting.
// the reservation can be done in application state and has no reason
// to touch the database. (exchange_finish won't work anyway if application
// state is lost.)
// I also have a hunch that failure to send the exchange_finish request
// is what causes the notorious GTS glitch where a pokemon is listed
// under the wrong species and you can't trade it
response.Write(result.Save(), 0, 292);
} break;
case "/pokemondpds/worldexchange/exchange_finish.asp":
{
SessionManager.Remove(session);
if (data.Length != 8)
{
ShowError(context, 400);
return;
}
// find a matching session which contains our record
GamestatsSession prevSession = SessionManager.FindSession(pid, "/pokemondpds/worldexchange/exchange.asp");
if (prevSession == null)
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
return;
}
SessionManager.Remove(prevSession);
if (prevSession.Tag == null)
{
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
return;
}
AssertHelper.Assert(prevSession.Tag is GtsRecord4[]);
GtsRecord4[] tag = (GtsRecord4[])prevSession.Tag;
AssertHelper.Assert(tag.Length == 2);
AssertHelper.Assert(tag[0] is GtsRecord4);
AssertHelper.Assert(tag[0] is GtsRecord4);
GtsRecord4 upload = (GtsRecord4)tag[0];
GtsRecord4 result = (GtsRecord4)tag[1];
if (Database.Instance.GtsTradePokemon4(upload, result))
{
#if !DEBUG
try
{
#endif
Database.Instance.GtsLogTrade4(result, null);
#if !DEBUG
}
catch { }
#endif
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
}
else
response.Write(new byte[] { 0x00, 0x00 }, 0, 2);
} break;
#endregion
#region Battle Tower
case "/pokemondpds/battletower/info.asp":
SessionManager.Remove(session);
// Probably an availability/status code.
// todo: See how the game reacts to various values.
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
break;
case "/pokemondpds/battletower/roomnum.asp":
SessionManager.Remove(session);
//byte rank = data[0x00];
response.Write(new byte[] { 0x32, 0x00 }, 0, 2);
break;
case "/pokemondpds/battletower/download.asp":
{
SessionManager.Remove(session);
if (data.Length != 2)
{
ShowError(context, 400);
return;
}
byte rank = data[0x00];
byte roomNum = data[0x01];
if (rank > 9 || roomNum > 49)
{
ShowError(context, 400);
return;
}
BattleTowerRecord4[] opponents = Database.Instance.BattleTowerGetOpponents4(pid, rank, roomNum);
BattleTowerProfile4[] leaders = Database.Instance.BattleTowerGetLeaders4(rank, roomNum);
BattleTowerRecord4[] fakeOpponents = FakeOpponentGenerator4.GenerateFakeOpponents(7 - opponents.Length);
foreach (BattleTowerRecord4 record in fakeOpponents)
{
response.Write(record.Save(), 0, 228);
}
foreach (BattleTowerRecord4 record in opponents)
{
response.Write(record.Save(), 0, 228);
}
foreach (BattleTowerProfile4 leader in leaders)
{
response.Write(leader.Save(), 0, 34);
}
// This is completely insane. The game crashes when you
// use Check Leaders if the response arrives too fast,
// so we artificially delay it.
// todo: This is slower than it needs to be if the
// database is slow to respond. We should sleep for a
// variable time based on when the request was received.
Thread.Sleep(500);
} break;
case "/pokemondpds/battletower/upload.asp":
{
SessionManager.Remove(session);
if (data.Length != 239)
{
ShowError(context, 400);
return;
}
BattleTowerRecord4 record = new BattleTowerRecord4(data, 0);
record.Rank = data[0xe4];
record.RoomNum = data[0xe5];
record.BattlesWon = data[0xe6];
record.Unknown5 = BitConverter.ToUInt64(data, 0xe7);
// todo: Do we want to store their record anyway if they lost the first round?
if (record.BattlesWon > 0)
Database.Instance.BattleTowerUpdateRecord4(record);
if (record.BattlesWon == 7)
Database.Instance.BattleTowerAddLeader4(record);
response.Write(new byte[] { 0x01, 0x00 }, 0, 2);
} break;
#endregion
}
}
}
}