527945f75ecf5320f04d8c1dea03fc86135f2652
[reactos.git] / rosapps / net / telnet / src / tnetwork.h
1 // This is a simple class to handle socket connections
2 // (Paul Brannan 6/15/98)
3
4 #ifndef __TNETWORK_H
5 #define __TNETWORK_H
6
7 #include <windows.h>
8
9 // Mingw32 doesn't use winsock.h (Paul Brannan 9/4/98)
10 #ifdef __MINGW32__
11 #ifdef __CYGWIN__
12 #include <winsock.h>
13 #else
14 #include <Windows32/sockets.h>
15 #endif
16 #else
17 #include <winsock.h>
18 #endif
19
20 enum NetworkType {TN_NETSOCKET, TN_NETPIPE};
21
22 typedef int(*Naws_func_t)(char *, int, int);
23
24 class TNetwork {
25 private:
26 SOCKET socket;
27 BOOL local_echo; // Paul Brannan 8/25/98
28 BOOL line_mode; // Paul Brannan 12/31/98
29 NetworkType net_type; // Paul Brannan 3/18/99
30 HANDLE pipeIn, pipeOut; // Paul Brannan 3/18/99
31 Naws_func_t naws_func;
32 char *local_address;
33
34 public:
35 TNetwork(SOCKET s = 0): socket(s), local_echo(1), line_mode(1),
36 net_type(TN_NETSOCKET), naws_func((Naws_func_t)NULL),
37 local_address((char *)NULL) {}
38 ~TNetwork() {if(local_address) delete local_address;}
39
40 void SetSocket(SOCKET s);
41 SOCKET GetSocket() {return socket;}
42 void SetPipe(HANDLE pIn, HANDLE pOut);
43 void SetNawsFunc(Naws_func_t func) {naws_func = func;}
44 void SetLocalAddress(char *buf);
45 const char* GetLocalAddress() {return local_address;}
46
47 NetworkType get_net_type() {return net_type;}
48
49 int WriteString(const char *str, const int length);
50 int ReadString (char *str, const int length);
51
52 BOOL get_local_echo() {return local_echo;}
53 void set_local_echo(BOOL b) {local_echo = b;}
54
55 BOOL get_line_mode() {return line_mode;}
56 void set_line_mode(BOOL b) {line_mode = b;}
57
58 void do_naws(int width, int height);
59 };
60
61 #endif