uidl.c

   1 #include <fcntl.h>
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include <unistd.h>
   6 #include <sys/types.h>
   7 #include <sys/stat.h>
   8 #include "uidl.h"
   9
  10 static int file_size(int fd)
  11 {
  12         struct stat st;
  13         fstat(fd, &st);
  14         return st.st_size;
  15 }
  16
  17 static int xread(int fd, char *buf, int len)
  18 {
  19         int nr = 0;
  20         while (nr < len) {
  21                 int cr = read(fd, buf + nr, len - nr);
  22                 if (cr == -1)
  23                         break;
  24                 nr += cr;
  25         }
  26         return nr;
  27 }
  28
  29 struct uidl *uidl_read(char *filename)
  30 {
  31         struct uidl *uidl = malloc(sizeof(*uidl));
  32         int len;
  33         uidl->fd = open(filename, O_RDWR | O_CREAT, 0600);
  34         len = file_size(uidl->fd);
  35         lseek(uidl->fd, 0, SEEK_SET);
  36         uidl->txt = malloc(len + 1);
  37         xread(uidl->fd, uidl->txt, len);
  38         uidl->txt[len] = '\0';
  39         lseek(uidl->fd, 0, SEEK_END);
  40         return uidl;
  41 }
  42
  43 int uidl_find(struct uidl *uidl, char *id)
  44 {
  45         char kw[256];
  46         snprintf(kw, sizeof(kw), "%s\n", id);
  47         return !!strstr(uidl->txt, kw);
  48 }
  49
  50 void uidl_add(struct uidl *uidl, char *id)
  51 {
  52         char kw[256];
  53         snprintf(kw, sizeof(kw), "%s\n", id);
  54         write(uidl->fd, kw, strlen(kw));
  55 }
  56
  57 void uidl_save(struct uidl *uidl)
  58 {
  59         close(uidl->fd);
  60         free(uidl->txt);
  61         free(uidl);
  62 }