-
Notifications
You must be signed in to change notification settings - Fork 0
/
Client.cpp
136 lines (93 loc) · 2.38 KB
/
Client.cpp
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#define WINVER 0x600
#include <iostream>
#include <thread>
#include <mutex>
#include <string>
#include <winsock2.h>
#include <ws2tcpip.h>
#pragma comment(lib, "ws2_32.lib");
using namespace std;
void recv_message(const SOCKET&);
std::mutex m1;
std::mutex m2;
int main() {
string ipAddress = "127.0.0.1"; //ip address of the server
int port = 54000; //Listening port number on server
// Initialize Winsock
WSAData data;
WORD ver = MAKEWORD(2,2);
int wsResult = WSAStartup(ver, &data);
if (wsResult != 0)
{
cerr << "Can't start winsock, Err #" << wsResult << endl;
}
//Create socket
SOCKET sock = socket(AF_INET, SOCK_STREAM, 0);
//run separate thread for receiving message!
std::thread recv_thread(recv_message, std::ref(sock));
recv_thread.detach(); //this thread will run freely on its own
if (sock == INVALID_SOCKET)
{
cerr << "Can't create socket, Err #" << WSAGetLastError() << endl;
}
// Fill in a hint structure
sockaddr_in hint;
hint.sin_family = AF_INET;
hint.sin_port = htons(port);
hint.sin_addr.s_addr = inet_addr("127.0.0.1");
//inet_pton mysteriously does not work for some reason?
//inet_pton(AF_INET, ipAddress.c_str(), &hint.sin_addr);
// connect to server
int connResult = connect(sock, (sockaddr*) &hint, sizeof(hint));
if(connResult == SOCKET_ERROR)
{
cerr << "Can't connect to server, Err #" << WSAGetLastError() << endl;
closesocket(sock);
WSACleanup();
}
//Do-while loop to send and recieve data
char buf[4096];
int bytesIn = recv(sock, buf, 4096, 0);
//welcome message to client
if (bytesIn > 0) {
cout << buf << endl;
}
/*
string userName = "";
/*
while(userName == "") {
cout << "Enter your username: " << endl;
getline(cin, userName);
}
string header = userName + " says: ";
*/
string userInput;
do {
//prompt user for some text
//std::lock_guard<std::mutex> guard1(m1);
//guard
cout << "> ";
getline(cin, userInput);
//header += userInput;
if (userInput.size() > 0) {
//send text
int sendResult = send(sock, userInput.c_str(), userInput.size() + 1, 0);
cin.clear();
}
} while (userInput.size() > 0);
// Gracefully close down everything
closesocket(sock);
WSACleanup();
}
void recv_message(const SOCKET& s) {
char buf[4096];
ZeroMemory(buf, 4096);
for (;;) {
int bytesReceived = recv(s, buf, 4096, 0);
if (bytesReceived > 0)
{
cout << buf << endl;
ZeroMemory(buf, 4096);
}
}
}