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
|
// minimal client implementation
#include <arpa/inet.h>
#include <assert.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "chatty.h"
#include "protocol.h"
int
main(int argc, char** argv)
{
if (argc < 3) {
fprintf(stderr, "usage: send <author> <msg>\n");
return 1;
}
s32 err, serverfd, nsend, nrecv;
serverfd = socket(AF_INET, SOCK_STREAM, 0);
assert(serverfd != -1);
const struct sockaddr_in address = {AF_INET, htons(PORT), {0}, {0}};
err = connect(serverfd, (struct sockaddr*)&address, sizeof(address));
assert(err == 0);
// Get our ID
ID id = 0;
{
// get author len
u32 author_len = strlen(argv[1]);
assert(author_len + 1 <= AUTHOR_LEN); // add 1 for null terminator
// Introduce ourselves
HeaderMessage header = HEADER_INIT(HEADER_TYPE_INTRODUCTION);
IntroductionMessage message;
memcpy(message.author, argv[1], author_len);
nsend = send(serverfd, &header, sizeof(header), 0);
assert(nsend != -1);
nsend = send(serverfd, &message, sizeof(message), 0);
assert(nsend != -1);
// Get id
nrecv = recv(serverfd, &header, sizeof(header), 0);
assert(nrecv != -1);
if (header.type == HEADER_TYPE_ERROR) {
ErrorMessage message;
nrecv = recv(serverfd, &message, sizeof(message), 0);
fprintf(stderr, "Got '%s' error.\n'", errorTypeString(message.type));
close(serverfd);
return 1;
}
assert(header.type == HEADER_TYPE_ID);
IDMessage idmessage;
nrecv = recv(serverfd, &idmessage, sizeof(idmessage), 0);
assert(nrecv != -1);
fprintf(stderr, "Got id: %lu\n", idmessage.id);
}
// convert text to wide string
u32 text_len = strlen(argv[2]) + 1;
u32 text_wide[text_len];
u32 size = mbstowcs((wchar_t*)text_wide, argv[2], text_len - 1);
assert(size == text_len - 1);
text_wide[text_len - 1] = 0;
HeaderMessage header = HEADER_INIT(HEADER_TYPE_TEXT);
TextMessage message;
bzero(&message, TEXTMESSAGE_SIZE);
message = (TextMessage){.id = id, .timestamp = time(NULL), .len = text_len};
nsend = send(serverfd, &header, sizeof(header), 0);
assert(nsend != -1);
fprintf(stderr, "header bytes sent: %d\n", nsend);
nsend = send(serverfd, &message, TEXTMESSAGE_SIZE, 0);
assert(nsend != -1);
fprintf(stderr, "message bytes sent: %d\n", nsend);
u32 text_size = message.len * sizeof(*message.text);
nsend = send(serverfd, text_wide, text_size, 0);
fprintf(stderr, "text bytes sent: %d\n", nsend);
return 0;
}
|
