146 lines
3.3 KiB
C++
146 lines
3.3 KiB
C++
#pragma once
|
|
using namespace std;
|
|
|
|
#include "stdafx.h"
|
|
#include <queue>
|
|
#include "System.h"
|
|
#include "DataInputStream.h"
|
|
#include "DataOutputStream.h"
|
|
#include "net.minecraft.network.packet.h"
|
|
#include "C4JThread.h"
|
|
|
|
#include "Socket.h"
|
|
|
|
// 4J JEV, size of the threads (bytes).
|
|
#define READ_STACK_SIZE 0
|
|
#define WRITE_STACK_SIZE 0
|
|
|
|
class ByteArrayOutputStream;
|
|
|
|
class Connection
|
|
{
|
|
friend DWORD WINAPI runRead(LPVOID lpParam);
|
|
friend DWORD WINAPI runWrite(LPVOID lpParam);
|
|
friend DWORD WINAPI runSendAndQuit(LPVOID lpParam);
|
|
friend DWORD WINAPI runClose(LPVOID lpParam);
|
|
|
|
private:
|
|
static const int SEND_BUFFER_SIZE = 1024 * 5;
|
|
|
|
public:
|
|
static int readThreads, writeThreads;
|
|
|
|
private:
|
|
static const int MAX_TICKS_WITHOUT_INPUT = 20 * 60;
|
|
|
|
public:
|
|
static const int IPTOS_LOWCOST = 0x02;
|
|
static const int IPTOS_RELIABILITY = 0x04;
|
|
static const int IPTOS_THROUGHPUT = 0x08;
|
|
static const int IPTOS_LOWDELAY = 0x10;
|
|
|
|
private:
|
|
Socket *socket;
|
|
const SocketAddress *address;
|
|
DataInputStream *dis;
|
|
DataOutputStream *bufferedDos; // 4J This is the same type of dos the java game has
|
|
|
|
// 4J Added
|
|
DataOutputStream *byteArrayDos; // 4J This dos allows us to write individual packets to the socket
|
|
ByteArrayOutputStream *baos;
|
|
Socket::SocketOutputStream *sos;
|
|
|
|
bool running;
|
|
|
|
queue<shared_ptr<Packet> > incoming; // 4J - was using synchronizedList...
|
|
CRITICAL_SECTION incoming_cs; // ... now has this critical section
|
|
queue<shared_ptr<Packet> > outgoing; // 4J - was using synchronizedList - but don't think it is required as usage is wrapped in writeLock critical section
|
|
queue<shared_ptr<Packet> > outgoing_slow; // 4J - was using synchronizedList - but don't think it is required as usage is wrapped in writeLock critical section
|
|
|
|
|
|
PacketListener *packetListener;
|
|
bool quitting;
|
|
|
|
|
|
C4JThread* readThread;
|
|
C4JThread* writeThread;
|
|
|
|
C4JThread::Event* m_hWakeReadThread;
|
|
C4JThread::Event* m_hWakeWriteThread;
|
|
|
|
DWORD saqThreadID, closeThreadID;
|
|
|
|
bool disconnected;
|
|
DisconnectPacket::eDisconnectReason disconnectReason;
|
|
void **disconnectReasonObjects; // 4J a pointer to an array.
|
|
|
|
int noInputTicks;
|
|
int estimatedRemaining;
|
|
|
|
int tickCount; // 4J Added
|
|
|
|
public:
|
|
static int readSizes[256];
|
|
static int writeSizes[256];
|
|
|
|
int fakeLag;
|
|
|
|
private:
|
|
void _init();
|
|
|
|
// 4J Jev, these might be better of as private
|
|
CRITICAL_SECTION threadCounterLock;
|
|
CRITICAL_SECTION writeLock;
|
|
|
|
public:
|
|
// 4J Jev, need to delete the critical section.
|
|
~Connection();
|
|
Connection(Socket *socket, const wstring& id, PacketListener *packetListener); // throws IOException
|
|
|
|
void setListener(PacketListener *packetListener);
|
|
void send(shared_ptr<Packet> packet);
|
|
|
|
public:
|
|
void queueSend(shared_ptr<Packet> packet);
|
|
|
|
private:
|
|
int slowWriteDelay;
|
|
|
|
bool writeTick();
|
|
|
|
public:
|
|
void flush();
|
|
|
|
private:
|
|
bool readTick();
|
|
|
|
private:
|
|
|
|
/* 4J JEV, removed try/catch
|
|
void handleException(Exception e)
|
|
{
|
|
e.printStackTrace();
|
|
close("disconnect.genericReason", "Internal exception: " + e.toString());
|
|
}*/
|
|
|
|
public:
|
|
void close(DisconnectPacket::eDisconnectReason reason, ...);
|
|
|
|
void tick();
|
|
|
|
SocketAddress *getRemoteAddress();
|
|
|
|
void sendAndQuit();
|
|
|
|
int countDelayedPackets();
|
|
|
|
Socket *getSocket() { return socket; }
|
|
|
|
private:
|
|
static int runRead(void* lpParam);
|
|
static int runWrite(void* lpParam);
|
|
static int runClose(void* lpParam);
|
|
static int runSendAndQuit(void* lpParam);
|
|
|
|
};
|