1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#ifndef NLNETPACKETRECEIVERTCP_H
#define NLNETPACKETRECEIVERTCP_H
#include "Aria.h"
#include "ArNetPacket.h"
/**
This class receives TCP packets from a socket, you need to have an
open socket and give it to the socket with setSocket, then you need
to set up a callback to process packets with setProcessPacketCB,
finally call readData which will read in all the data and call the
processPacketCB.
**/
class ArNetPacketReceiverTcp
{
public:
/// Constructor
AREXPORT ArNetPacketReceiverTcp();
/// Destructor
AREXPORT ~ArNetPacketReceiverTcp();
/// Sets the socket this receiver uses
AREXPORT void setSocket(ArSocket *socket);
/// Gets the socket this receiver uses
AREXPORT ArSocket *getSocket(void);
/// Sets the callback for use when a packet is received
AREXPORT void setProcessPacketCB(ArFunctor1<ArNetPacket *> *functor);
/// Gets the callback used when a packet is received
AREXPORT ArFunctor1<ArNetPacket *> *getProcessPacketCB(void);
/// Sets the logging prefix
AREXPORT void setLoggingPrefix(const char *loggingPrefix);
/// Reads in all the data available calling the processPacketCB
AREXPORT bool readData(void);
/// Sets whether we're quiet about errors or not
void setQuiet(bool quiet) { myQuiet = quiet; }
/// Gets whether we're quiet about errors or not
bool getQuiet(void) { return myQuiet; }
protected:
enum Ret {
RET_CONN_CLOSED, // the connection was closed (in a good manner)
RET_CONN_ERROR, // the connection was has an error (so close it)
RET_GOT_PACKET, // we got a good packet
RET_BAD_PACKET, // we got a bad packet (checksum wrong)
RET_FAILED_READ, // our read failed (no data)
RET_TIMED_OUT}; // we were reading and timed out
/// Reads in a single packet, returns NULL if not one
AREXPORT Ret readPacket(int msWait);
enum State { STATE_SYNC1, STATE_SYNC2, STATE_LENGTH1,
STATE_LENGTH2, STATE_ACQUIRE_DATA };
enum {
TOTAL_PACKET_LENGTH = ArNetPacket::MAX_LENGTH+ArNetPacket::HEADER_LENGTH+ArNetPacket::FOOTER_LENGTH
};
State myState;
ArFunctor1<ArNetPacket *> *myProcessPacketCB;
bool myQuiet;
ArSocket *mySocket;
ArTime myLastPacket;
ArNetPacket myPacket;
char myReadBuff[TOTAL_PACKET_LENGTH];
int myReadCount;
int myReadLength;
int myReadCommand;
unsigned char mySync1;
unsigned char mySync2;
std::string myLoggingPrefix;
};
#endif // NLNETPACKETRECEIVERTCP
|
