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 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296
| #define _GNU_SOURCE #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <sys/file.h> #include <fcntl.h> #include <errno.h> #include <string.h> #include <sys/wait.h> #include <sys/stat.h>
int main() { int fd; pid_t pid; printf("=== Flock函数示例 ===\n"); // 创建测试文件 fd = open("test_flock.txt", O_CREAT | O_RDWR | O_TRUNC, 0644); if (fd == -1) { perror("创建测试文件失败"); exit(EXIT_FAILURE); } printf("成功创建测试文件,文件描述符: %d\n", fd); // 写入初始数据 const char *initial_data = "Initial data for flock test\n"; if (write(fd, initial_data, strlen(initial_data)) == -1) { perror("写入初始数据失败"); close(fd); exit(EXIT_FAILURE); } // 示例1: 基本的排他锁操作 printf("\n示例1: 基本的排他锁操作\n"); // 获取排他锁 if (flock(fd, LOCK_EX) == -1) { perror("获取排他锁失败"); } else { printf(" 成功获取排他锁\n"); // 在锁保护下写入数据 const char *exclusive_data = "Data written with exclusive lock\n"; if (write(fd, exclusive_data, strlen(exclusive_data)) == -1) { perror(" 写入数据失败"); } else { printf(" 成功写入数据到文件\n"); } // 释放锁 if (flock(fd, LOCK_UN) == -1) { perror(" 释放锁失败"); } else { printf(" 成功释放排他锁\n"); } } // 示例2: 共享锁操作 printf("\n示例2: 共享锁操作\n"); // 获取共享锁 if (flock(fd, LOCK_SH) == -1) { perror("获取共享锁失败"); } else { printf(" 成功获取共享锁\n"); // 在锁保护下读取数据 char buffer[200]; lseek(fd, 0, SEEK_SET); ssize_t bytes_read = read(fd, buffer, sizeof(buffer) - 1); if (bytes_read > 0) { buffer[bytes_read] = '\0'; printf(" 读取文件内容:\n%s", buffer); } // 释放锁 if (flock(fd, LOCK_UN) == -1) { perror(" 释放共享锁失败"); } else { printf(" 成功释放共享锁\n"); } } // 示例3: 非阻塞锁操作 printf("\n示例3: 非阻塞锁操作\n"); // 先获取一个排他锁 if (flock(fd, LOCK_EX) == -1) { perror("获取排他锁失败"); } else { printf(" 进程已持有排他锁\n"); // 尝试非阻塞获取另一个排他锁(应该失败) if (flock(fd, LOCK_EX | LOCK_NB) == -1) { if (errno == EAGAIN || errno == EWOULDBLOCK) { printf(" 非阻塞获取排他锁: 立即返回EAGAIN(锁被占用)\n"); } else { perror(" 非阻塞获取锁失败"); } } else { printf(" 非阻塞获取排他锁成功(不应该发生)\n"); flock(fd, LOCK_UN); // 释放意外获取的锁 } // 尝试非阻塞获取共享锁(也应该失败,因为有排他锁) if (flock(fd, LOCK_SH | LOCK_NB) == -1) { if (errno == EAGAIN || errno == EWOULDBLOCK) { printf(" 非阻塞获取共享锁: 立即返回EAGAIN(有排他锁)\n"); } else { perror(" 非阻塞获取共享锁失败"); } } else { printf(" 非阻塞获取共享锁成功(不应该发生)\n"); flock(fd, LOCK_UN); // 释放意外获取的锁 } // 释放锁 flock(fd, LOCK_UN); printf(" 释放排他锁\n"); } // 示例4: 多进程锁演示 printf("\n示例4: 多进程锁演示\n"); pid = fork(); if (pid == -1) { perror("fork失败"); close(fd); exit(EXIT_FAILURE); } else if (pid == 0) { // 子进程 printf(" 子进程 %d 开始执行\n", getpid()); // 子进程打开同一个文件 int child_fd = open("test_flock.txt", O_RDWR); if (child_fd == -1) { perror(" 子进程打开文件失败"); exit(EXIT_FAILURE); } printf(" 子进程尝试获取排他锁...\n"); if (flock(child_fd, LOCK_EX) == -1) { perror(" 子进程获取排他锁失败"); } else { printf(" 子进程成功获取排他锁\n"); // 写入子进程数据 char child_data[100]; sprintf(child_data, "Data from child process %d\n", getpid()); if (write(child_fd, child_data, strlen(child_data)) == -1) { perror(" 子进程写入数据失败"); } else { printf(" 子进程写入数据成功\n"); } sleep(3); // 持有锁3秒 // 释放锁 if (flock(child_fd, LOCK_UN) == -1) { perror(" 子进程释放锁失败"); } else { printf(" 子进程释放排他锁\n"); } } close(child_fd); exit(EXIT_SUCCESS); } else { // 父进程 sleep(1); // 让子进程先运行 printf(" 父进程 %d 尝试获取排他锁...\n", getpid()); // 父进程尝试获取排他锁(会被阻塞直到子进程释放) printf(" 父进程获取排他锁(会被阻塞)...\n"); if (flock(fd, LOCK_EX) == -1) { perror(" 父进程获取排他锁失败"); } else { printf(" 父进程成功获取排他锁(子进程已释放)\n"); // 写入父进程数据 char parent_data[100]; sprintf(parent_data, "Data from parent process %d\n", getpid()); if (write(fd, parent_data, strlen(parent_data)) == -1) { perror(" 父进程写入数据失败"); } else { printf(" 父进程写入数据成功\n"); } // 释放锁 if (flock(fd, LOCK_UN) == -1) { perror(" 父进程释放锁失败"); } else { printf(" 父进程释放排他锁\n"); } } // 等待子进程结束 wait(NULL); } // 示例5: 共享锁并发演示 printf("\n示例5: 共享锁并发演示\n"); // 创建多个子进程同时获取共享锁 for (int i = 0; i < 3; i++) { pid = fork(); if (pid == 0) { // 子进程 int child_fd = open("test_flock.txt", O_RDONLY); if (child_fd != -1) { printf(" 子进程 %d 尝试获取共享锁...\n", getpid()); if (flock(child_fd, LOCK_SH) == 0) { printf(" 子进程 %d 成功获取共享锁\n", getpid()); // 读取数据 char read_buffer[200]; lseek(child_fd, 0, SEEK_SET); ssize_t bytes_read = read(child_fd, read_buffer, sizeof(read_buffer) - 1); if (bytes_read > 0) { read_buffer[bytes_read] = '\0'; printf(" 子进程 %d 读取数据成功\n", getpid()); } sleep(2); // 持有共享锁2秒 flock(child_fd, LOCK_UN); printf(" 子进程 %d 释放共享锁\n", getpid()); } close(child_fd); } exit(EXIT_SUCCESS); } } // 等待所有子进程结束 for (int i = 0; i < 3; i++) { wait(NULL); } // 示例6: 锁的继承和关闭行为 printf("\n示例6: 锁的继承和关闭行为\n"); if (flock(fd, LOCK_EX) == 0) { printf(" 进程持有排他锁\n"); // 复制文件描述符 int dup_fd = dup(fd); if (dup_fd != -1) { printf(" 复制文件描述符: %d -> %d\n", fd, dup_fd); // 使用复制的fd释放锁 if (flock(dup_fd, LOCK_UN) == 0) { printf(" 使用复制的fd释放锁成功\n"); } close(dup_fd); } // 重新获取锁 if (flock(fd, LOCK_EX) == 0) { printf(" 重新获取锁成功\n"); flock(fd, LOCK_UN); } } // 示例7: 错误处理演示 printf("\n示例7: 错误处理演示\n"); // 尝试对无效文件描述符加锁 if (flock(999, LOCK_EX) == -1) { printf(" 对无效文件描述符加锁: %s\n", strerror(errno)); } // 清理资源 printf("\n清理资源...\n"); if (close(fd) == -1) { perror("关闭文件失败"); } else { printf("成功关闭文件描述符 %d\n", fd); } // 删除测试文件 if (unlink("test_flock.txt") == -1) { perror("删除测试文件失败"); } else { printf("成功删除测试文件\n"); } return 0; }
}
|