Linux 高级编程 - 必须掌握的 5 个底层 IO 函数

// open 需要 3 个头文件
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

/*
 * filename: 要打开的文件名称
 * flags: 打开标记，例如：O_CREAT | ORDWR 表示文件不存在就创建，并且可读可写
 * mode: 打开权限
 * return: 成功，返回一个新的文件描述符 fd，失败返回 -1，并设置 errno
 */
int open (const char *filename, int flags, mode t mode);

#include <unistd.h>

/*
 * fd: 要关闭的文件的描述符
 * return: 成功返回 0，失败返回 -1，并设置 errno
 */
int close(int fd);

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

int main(int argc, char* argv[]) {
	if (argc < 2) {
		printf("please input filename:\n");
		exit(1);
	}

	// argv[1] = "1.txt", 打开的文件不存在就创建 | 可读可写，666（rw-rw-rw）
	int fd = open(argv[1], O_CREAT | O_RDWR, 666);

	if (fd < 0) {
		printf("file open fail.\n");
		exit(1);
	} else {
		printf("file open success.\n");
		// 必须关闭文件
		close(fd);
		printf("file closed.\n");
	}

	return 0;
}

#include <unistd.h>

/*
 * filedes: 要读取的文件描述符
 * buffer: 存储读取字节的缓冲区
 * size: 要读取的大小
 * return: 成功返回读取的字节数，失败返回 -1，并设置 errno
 */
ssize_t read (int filedes, void *buffer, size t size);

#include <unistd.h>

/*
 * filedes: 写入的文件描述符
 * buffer: 存储待写入数据的缓存区
 * size: 要写入的字节数
 * return: 成功返回写入的字节数，失败返回 -1，并设置 errno
 */
ssize_t write (int filedes, const void *buffer, size t size);

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <string.h>

int main(int argc, char* argv[]) {
	// 1. 打开两个文件
	int fd1 = open(argv[1], O_CREAT | O_RDWR, 0666);
	int fd2 = open(argv[2], O_CREAT | O_RDWR, 0666);

	if ((fd1 < 0) || (fd2 < 0)) {
		printf("file open fail.\n");
		exit(1);
	} else {
		printf("file1 open success: fd1 = %d\n", fd1);
		printf("file2 open success: fd2 = %d\n", fd2);

		char buf[1024];
		// clear
		memset(buf, 0, 1024);

		int read_length = 0;
		int write_length = 0;

		// 2. 将文件 1 的内容读取到 buf 中
		if ((read_length = read(fd1, buf, 1024)) != -1)  {
			// 3. 将 buf 的内容写入到文件 2 中
			if (read_length == write(fd2, buf, read_length)) 
				printf("write to fd2 success.\n");
			else 
				printf("write to fd2 falil.\n");
		} else {
			printf("read fd1 fail.\n");
		}
		
		// 必须关闭 2 个文件
		close(fd1);
		close(fd2);

		printf("file1 close success.\n");
		printf("file2 close success.\n");
	}

	return 0;
}

#include <sys/types.h>
#include <unistd.h>

/*
 * filedes: 要操作的文件描述符
 * offset: 根据当前 whence 的偏移量
 * whence: 指定当前文件指针的位置
 * return: 成功返回设置后的文件位置，可以使用 `SEEK_CUR` 查看当前文件指针位置，
 *         失败返回 -1 并设置 errno
 */
off_t lseek (int filedes, off t offset, int whence);

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/types.h>
#include <unistd.h>

int main(int argc, char* argv[]) {
	// 打开文件
	int fd = open(argv[1], O_CREAT | O_RDWR, 0666);

	if (fd < 0) {
		printf("file open fail.\n");
		exit(1);
	} else {
		printf("file1 open success: fd = %d\n", fd);

		// 得到文件长度
		int file_length = lseek(fd, 0, SEEK_END);
		
		// 输出文件长度
		printf("file_length = %d\n", file_length);

		// 关闭文件
		close(fd);
		printf("file1 close success.\n");
	}

	return 0;
}