在另一个子进程写入文本文件时将其锁定

时间:2018-07-21 22:08:14

标签: c linux fork file-writing file-locking

我在c中实现了分叉的多个客户端,它们都应该写入一个公共文件。到目前为止,这已经失败了,因为这些套接字中的信息都在文件目标中被弄乱了。 这是我的代码

    FILE *Ufptr;
    Ufptr = fopen("Unsorted_busy_list.txt","a+");

    fprintf(Ufptr, "%s:%d\n",inet_ntoa(newAddr.sin_addr),ntohs(newAddr.sin_port));
    fclose(Ufptr);

有人告诉我,在文件上使用fcntl和互斥锁可以做到这一点,但这是新手,并且不知道如何在文件写入过程中实现这一点。 任何帮助

1 个答案:

答案 0 :(得分:2)

正如我在评论中提到的那样,如果父级使用子级的输出,则通常更容易使用Unix域数据报套接字对(或每个子进程对)。 Unix域数据报套接字保留消息边界,以便使用send()成功发送的每个数据报都在一个recv()中接收。您甚至可以将数据作为二进制结构发送。不需要锁。如果您为每个孩子使用一个套接字对,则可以轻松地将父端设置为非阻塞,并使用select()poll()在单个循环中读取所有数据报。


回到正确的问题。

这是append_file(filename, format, ...)的示例实现,该实现使用POSIX.1-2008 vdprintf()通过基于fcntl()的咨询记录锁来写入文件:

#define  _POSIX_C_SOURCE  200809L
#include <stdlib.h>
#include <unistd.h>
#include <stdarg.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <errno.h>

int append_file(const char *filename, const char *format, ...)
{
    struct flock  lock;
    va_list       args;
    int           fd, cause;

    /* Sanity checks. */
    if (!filename || !*filename)
        return errno = EINVAL;

    /* Open the file for appending. Create if necessary. */
    fd = open(filename, O_WRONLY | O_APPEND | O_CREAT, 0666);
    if (fd == -1)
        return errno;

    /* Lock the entire file exclusively.
       Because we use record locks, append_file() to the same
       file is NOT thread safe: whenever the first descriptor
       is closed, all record locks to the same file in the process
       are dropped. */
    lock.l_type = F_WRLCK;
    lock.l_whence = SEEK_SET;
    lock.l_start = 0;
    lock.l_len = 0;
    if (fcntl(fd, F_SETLKW, &lock) == -1) {
        cause = errno;
        close(fd);
        return errno = cause;
    }

    if (format && *format) {
        cause = 0;
        va_start(args, format);
        if (vdprintf(fd, format, args) < 0)
            cause = errno;
        va_end(args);
        if (cause) {
            close(fd);
            return errno = cause;
        }
    }

    /* Note: This releases ALL record locks to this file
             in this process! */
    if (close(fd) == -1)
        return errno;

    /* Success! */
    return 0;
}

int main(int argc, char *argv[])
{
    int arg = 1;

    if (argc < 3) {
        fprintf(stderr, "\n");
        fprintf(stderr, "Usage: %s [ -h | --help ]\n", argv[0]);
        fprintf(stderr, "       %s FILENAME STRING [ STRING ... ]\n", argv[0]);
        fprintf(stderr, "\n");
    }

    for (arg = 2; arg < argc; arg++)
        if (append_file(argv[1], "%s\n", argv[arg])) {
            fprintf(stderr, "%s: %s.\n", argv[1], strerror(errno));
            return EXIT_FAILURE;
        }

    return EXIT_SUCCESS;
}

如果所有作者都使用上述append_file()附加到文件,则可以随时重命名文件。 (请注意,如果一个或多个进程正在等待重命名过程中释放记录锁,则可以在重命名之后对 文件进行最后的追加。)

要截断文件,请先对其进行排他锁,然后调用ftruncate(fd, 0)

要读取文件,请使用基于fcntl()的共享锁F_RDLCK(允许其他读者同时使用;如果打算“原子”截断文件,则可以使用F_WRLCK)。阅读完当前内容后),否则您可能会在结尾看到部分最终记录。