llinux网络编程-UDP编程

函数`recvfrom()`用于从一个已连接或未连接的套接字接收数据，并将数据存储到指定的缓冲区中。

具体作用如下：
- 接收数据：`recvfrom()`函数用于从套接字接收数据。它阻塞当前线程，直到从套接字接收到数据为止。如果有数据可用，则将数据读取到提供的缓冲区中，并返回接收到的字节数。
- 获取发送方地址：`recvfrom()`函数还可以获取发送方的地址信息。传入的地址指针会被填充为发送方的地址，可以在后续代码中使用这个地址进行回复或其他处理。

函数原型为：
```c
ssize_t recvfrom(int sockfd, void *buf, size_t len, int flags, struct sockaddr *src_addr, socklen_t *addrlen);
```
- `sockfd`：套接字描述符
- `buf`：接收数据的缓冲区指针
- `len`：缓冲区的大小
- `flags`：可选的标志参数，通常为0
- `src_addr`：发送方地址的缓冲区指针
- `addrlen`：发送方地址缓冲区的大小，接收后会被修改为实际地址大小

使用`recvfrom()`函数可以实现对套接字的数据接收操作，并获取发送方的地址信息，适用于网络通信中需要接收数据并处理的场景。

#include "net.h"

int main(void)
{

        int fd = -1;
        struct sockaddr_in sin;

        /* 1. 创建socket fd */
        if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
        { //udp程序
                perror("socket");
                exit(1);
        }

        /* 2. 允许绑定地址快速重用 */
        int b_reuse = 1;
        setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &b_reuse, sizeof(int));

        /*2. 绑定 */
/*2.1 填充struct sockaddr_in结构体变量 */
        bzero(&sin, sizeof(sin));
        sin.sin_family = AF_INET;
        sin.sin_port = htons(SERV_PORT);       //网络字节序的端口号

        /* 让服务器程序能绑定在任意的IP上 */
#if 1
        sin.sin_addr.s_addr = htonl(INADDR_ANY);
#else
        if (inet_pton(AF_INET, SERV_IP_ADDR, (void*)&sin.sin_addr) != 1)
        {
                perror("inet_pton");
                exit(1);
        }
#endif
        /*2.2 绑定 */
        if (bind(fd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
        {
                perror("bind");
                exit(1);
        }

        char buf[BUFSIZ];
        struct sockaddr_in cin;
        socklen_t addrlen = sizeof(cin);
        printf("\nUDP server started!\n");
        while (1)
        {
                //接手前将缓冲区清零
                bzero(buf, BUFSIZ);
                //阻塞接收数据 从一个已连接或未连接的套接字接收数据，并将数据存储到指定的缓冲区中
                if (recvfrom(fd, buf, BUFSIZ - 1, 0, (struct sockaddr*)&cin, &addrlen) < 0)
                {
                        perror("recvfrom");
                        continue;
                }

                char ipv4_addr[16];
                if (!inet_ntop(AF_INET, (void*)&cin.sin_addr, ipv4_addr, sizeof(cin)))
                {
                        perror("inet_ntop");
                        exit(1);
                }

                printf("Recived from(%s:%d), data:%s", ipv4_addr, ntohs(cin.sin_port), buf);

                if (!strncasecmp(buf, QUIT_STR, strlen(QUIT_STR)))
                {  //用户输入了quit字符
                        printf("Client(%s:%d) is exiting!\n", ipv4_addr, ntohs(cin.sin_port));
                }

        }

        close(fd);

        return 0;
}

/*udp demo */
/* usage:
 * ./client serv_ip serv_port
*/
#include "net.h"
//用户输入ip地址和端口号
void usage(char* s)
{
	printf("\nThis is udp demo!\n");
	printf("\nUsage:\n\t %s serv_ip serv_port", s);
	printf("\n\t serv_ip: udp server ip address");
	printf("\n\t serv_port: udp server port(serv_port > 5000)\n\n");
}

int main(int argc, char* argv[])
{
	int fd = -1;
	int port = SERV_PORT;
	//字符串转整形
	port = atoi(argv[2]);
	if (port < 0 || (port > 0 && port <= 5000))
	{
		usage(argv[0]);
		exit(1);
	}
	struct sockaddr_in sin;
	if (argc != 3)
	{
		usage(argv[0]);
		exit(1);
	}

	/* 1. 创建socket fd*/
	if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
	{ //UDP编程
		perror("socket");
		exit(1);
	}

	/*2.1 填充struct sockaddr_in结构体变量 */
	bzero(&sin, sizeof(sin));

	sin.sin_family = AF_INET;
	sin.sin_port = htons(SERV_PORT); //网络字节序的端口号
#if 0
	sin.sin_addr.s_addr = inet_addr(argv[1]);
#else
	if (inet_pton(AF_INET, argv[1], (void*)&sin.sin_addr) != 1)
	{
		perror("inet_pton");
		exit(1);
	}
#endif	
	printf("UDP client started!\n");
	char buf[BUFSIZ];
	while (1)
	{
		fprintf(stderr, "pls input string:");
		bzero(buf, BUFSIZ);
		if (fgets(buf, BUFSIZ - 1, stdin) == NULL)
		{
			perror("fgets");
			continue;
		}

		sendto(fd, buf, strlen(buf), 0, (struct sockaddr*)&sin, sizeof(sin));

		if (!strncasecmp(buf, QUIT_STR, strlen(QUIT_STR)))
		{  //用户输入了quit字符
			printf("Client is exited!\n");
			break;
		}

	}
	close(fd);
	return 0;
}

#ifndef __MAKEU_NET_H__
#define __MAKEU_NET_H__

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/types.h>                  /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>                 /* superset of previous */

#define SERV_PORT 5001
#define SERV_IP_ADDR "192.168.7.246"

#define QUIT_STR "quit"

#endif