change source dir

client-src/client.c

@@ -0,0 +1,105 @@
+#include <pthread.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <arpa/inet.h>
+#include <unistd.h>
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <stdint.h>
+#include <string.h>
+#include <errno.h>
+
+struct args {
+	int fd;
+};
+
+static void
+readall(int fd, void * buffer, size_t sz) {
+	for (;;) {
+		int err = recv(fd , buffer, sz, MSG_WAITALL);
+		if (err < 0) {
+			if (errno == EAGAIN || errno == EINTR)
+				continue;
+			break;
+		}
+		return;
+	}
+	perror("Socket error");
+	exit(1);
+}
+
+static void *
+_read(void *ud) {
+	struct args *p = ud;
+	int fd = p->fd;
+	fflush(stdout);
+	for (;;) {
+		uint8_t header[2];
+		fflush(stdout);
+		readall(fd, header, 2);
+		size_t len = header[0] << 8 | header[1];
+		if (len>0) {
+			char tmp[len+1];
+			readall(fd, tmp, len);
+			tmp[len]='\0';
+			printf("%s\n",tmp);
+		}
+	}
+	return NULL;
+}
+
+static void
+test(int fd) {
+	char tmp[1024];
+	while (!feof(stdin)) {
+		fgets(tmp,sizeof(tmp),stdin);
+		int n = strlen(tmp) -1;
+		uint8_t head[2];
+		head[0] = (n >> 8) & 0xff;
+		head[1] = n & 0xff;
+		int r;
+		r = send(fd, head, 2, 0);
+		if (r<0) {
+			perror("send head");
+		}
+		r = send(fd, tmp , n, 0);
+		if (r<0) {
+			perror("send data");
+		}
+	}
+}
+
+int 
+main(int argc, char * argv[]) {
+	if (argc < 3) {
+		printf("connect address port\n");
+		return 1;
+	}
+
+	int fd = socket(AF_INET,SOCK_STREAM,0);
+	struct sockaddr_in my_addr;
+
+	my_addr.sin_addr.s_addr=inet_addr(argv[1]);
+	my_addr.sin_family=AF_INET;
+	my_addr.sin_port=htons(strtol(argv[2],NULL,10));
+
+	int r = connect(fd,(struct sockaddr *)&my_addr,sizeof(struct sockaddr_in));
+
+	if (r == -1) {
+		perror("Connect failed:");
+		return 1;
+	}
+
+	struct args arg = { fd };
+	pthread_t pid ;
+	pthread_create(&pid, NULL, _read, &arg);
+
+	test(fd);
+
+	pthread_join(pid, NULL); 
+
+	close(fd);
+
+	return 0;
+}