1,067
edits
Changes
Jump to navigation
Jump to search
Line 8:
Line 8: − + − #include <sys/socket.h> − #include <linux/types.h> − #include <sys/types.h> − #include <arpa/inet.h> − #include <unistd.h> − #include <string.h> − #include <stdlib.h> − #include <stdio.h> − #include <signal.h> − − static int serve_new_conn(int sk) − − int rd, wr; − char buf[1024]; − − printf("New connection\n"); − − while (1) { − rd = read(sk, buf, sizeof(buf)); − if (!rd) − break; − − if (rd < 0) { − perror("Can't read socket"); − return 1; − } − − wr = 0; − while (wr < rd) { − int w; − − w = write(sk, buf + wr, rd - wr); − if (w <= 0) { − perror("Can't write socket"); − return 1; − } − − wr += w; − } − } − − printf("Done\n"); − return 0; − } − − static int main_srv(int argc, char **argv) − { − int sk, port, ret; − struct sockaddr_in addr; − − /* − * Let kids die themselves − */ − − signal(SIGCHLD, SIG_IGN); − − sk = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); − if (sk < 0) { − perror("Can't create socket"); − return -1; − } − − port = atoi(argv[1]); − memset(&addr, 0, sizeof(addr)); − addr.sin_family = AF_INET; − addr.sin_addr.s_addr = htonl(INADDR_ANY); − addr.sin_port = htons(port); − − printf("Binding to port %d\n", port); − − ret = bind(sk, (struct sockaddr *)&addr, sizeof(addr)); − if (ret < 0) { − perror("Can't bind socket"); − return -1; − } − − ret = listen(sk, 16); − if (ret < 0) { − perror("Can't put sock to listen"); − return -1; − } − − printf("Waiting for connections\n"); − while (1) { − int ask, pid; − − ask = accept(sk, NULL, NULL); − if (ask < 0) { − perror("Can't accept new conn"); − return -1; − } − − pid = fork(); − if (pid < 0) { − perror("Can't fork"); − return -1; − } − − if (pid > 0) − close(ask); − else { − close(sk); − ret = serve_new_conn(ask); − exit(ret); − } − } − } − − static int main_cl(int argc, char **argv) − { − int sk, port, ret, val = 1, rval; − struct sockaddr_in addr; − − sk = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); − if (sk < 0) { − perror("Can't create socket"); − return -1; − } − − port = atoi(argv[2]); − printf("Connecting to %s:%d\n", argv[1], port); − memset(&addr, 0, sizeof(addr)); − addr.sin_family = AF_INET; − ret = inet_aton(argv[1], &addr.sin_addr); − if (ret < 0) { − perror("Can't convert addr"); − return -1; − } − addr.sin_port = htons(port); − − ret = connect(sk, (struct sockaddr *)&addr, sizeof(addr)); − if (ret < 0) { − perror("Can't connect"); − return -1; − } − − while (1) { − write(sk, &val, sizeof(val)); − rval = -1; − read(sk, &rval, sizeof(rval)); − printf("PP %d -> %d\n", val, rval); − sleep(2); − val++; − } − } − − int main(int argc, char **argv) − { − if (argc == 2) − return main_srv(argc, argv); − else if (argc == 3) − return main_cl(argc, argv); − − printf("Bad usage\n"); − return 1; − } − </syntaxhighlight>
move the code out to a subpage, include it
! <code>tcp-howto.c</code>
! <code>tcp-howto.c</code>
|-
|-
| <syntaxhighlight lang="c">
| {{:Simple_TCP_pair/tcp-howto.c}}
{
|}
|}