121 lines
2.8 KiB
C#
Executable File
121 lines
2.8 KiB
C#
Executable File
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2011 The OpenRA Developers (see AUTHORS)
|
|
* This file is part of OpenRA, which is free software. It is made
|
|
* available to you under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation. For more information,
|
|
* see COPYING.
|
|
*/
|
|
#endregion
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.IO;
|
|
|
|
namespace OpenRA.Network
|
|
{
|
|
public class ReplayConnection : IConnection
|
|
{
|
|
//uint nextFrame = 1;
|
|
FileStream replayStream;
|
|
|
|
public ReplayConnection( string replayFilename )
|
|
{
|
|
replayStream = File.OpenRead( replayFilename );
|
|
}
|
|
|
|
public int LocalClientId
|
|
{
|
|
get { return 0; }
|
|
}
|
|
|
|
public ConnectionState ConnectionState
|
|
{
|
|
get { return ConnectionState.Connected; }
|
|
}
|
|
|
|
// do nothing; ignore locally generated orders
|
|
public void Send( int frame, List<byte[]> orders ) { }
|
|
public void SendImmediate( List<byte[]> orders ) { }
|
|
public void SendSync( int frame, byte[] syncData )
|
|
{
|
|
var ms = new MemoryStream();
|
|
ms.Write( BitConverter.GetBytes( frame ) );
|
|
ms.Write( syncData );
|
|
sync.Add( ms.ToArray() );
|
|
}
|
|
|
|
List<byte[]> sync = new List<byte[]>();
|
|
|
|
public void Receive( Action<int, byte[]> packetFn )
|
|
{
|
|
while( sync.Count != 0 )
|
|
{
|
|
packetFn( LocalClientId, sync[ 0 ] );
|
|
sync.RemoveAt( 0 );
|
|
}
|
|
if( replayStream == null ) return;
|
|
|
|
var reader = new BinaryReader( replayStream );
|
|
while( replayStream.Position < replayStream.Length )
|
|
{
|
|
var client = reader.ReadInt32();
|
|
var packetLen = reader.ReadInt32();
|
|
var packet = reader.ReadBytes( packetLen );
|
|
packetFn( client, packet );
|
|
}
|
|
replayStream = null;
|
|
}
|
|
|
|
public void Dispose() { }
|
|
}
|
|
|
|
class ReplayRecorderConnection : IConnection
|
|
{
|
|
IConnection inner;
|
|
BinaryWriter writer;
|
|
|
|
public ReplayRecorderConnection( IConnection inner, FileStream replayFile )
|
|
{
|
|
this.inner = inner;
|
|
this.writer = new BinaryWriter( replayFile );
|
|
}
|
|
|
|
public int LocalClientId { get { return inner.LocalClientId; } }
|
|
public ConnectionState ConnectionState { get { return inner.ConnectionState; } }
|
|
|
|
public void Send( int frame, List<byte[]> orders ) { inner.Send( frame, orders ); }
|
|
public void SendImmediate( List<byte[]> orders ) { inner.SendImmediate( orders ); }
|
|
public void SendSync( int frame, byte[] syncData ) { inner.SendSync( frame, syncData ); }
|
|
|
|
public void Receive( Action<int, byte[]> packetFn )
|
|
{
|
|
inner.Receive( ( client, data ) =>
|
|
{
|
|
writer.Write( client );
|
|
writer.Write( data.Length );
|
|
writer.Write( data );
|
|
packetFn( client, data );
|
|
} );
|
|
}
|
|
|
|
bool disposed;
|
|
|
|
public void Dispose()
|
|
{
|
|
if( disposed )
|
|
return;
|
|
|
|
writer.Close();
|
|
disposed = true;
|
|
}
|
|
|
|
~ReplayRecorderConnection()
|
|
{
|
|
Dispose();
|
|
}
|
|
}
|
|
}
|