src/file.c

   1 /*-*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2
   3 /*  Monkey HTTP Daemon
   4  *  ------------------
   5  *  Copyright (C) 2001-2010, Eduardo Silva P. <edsiper@gmail.com>
   6  *
   7  *  This program is free software; you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License as published by
   9  *  the Free Software Foundation; either version 2 of the License, or
  10  *  (at your option) any later version.
  11  *
  12  *  This program is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU Library General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU General Public License
  18  *  along with this program; if not, write to the Free Software
  19  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  20  */
  21
  22 #include <sys/types.h>
  23 #include <sys/stat.h>
  24 #include <unistd.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27
  28 #include "monkey.h"
  29 #include "file.h"
  30 #include "user.h"
  31 #include "memory.h"
  32
  33 struct file_info *mk_file_get_info(char *path)
  34 {
  35     struct file_info *f_info;
  36     struct stat f, target;
  37
  38     /* Stat right resource */
  39     if (lstat(path, &f) == -1) {
  40         return NULL;
  41     }
  42
  43     f_info = mk_mem_malloc(sizeof(struct file_info));
  44     f_info->is_link = MK_FILE_FALSE;
  45     f_info->is_directory = MK_FILE_FALSE;
  46     f_info->exec_access = MK_FILE_FALSE;
  47     f_info->read_access = MK_FILE_FALSE;
  48
  49     if (S_ISLNK(f.st_mode)) {
  50         f_info->is_link = MK_FILE_TRUE;
  51         if (stat(path, &target) == -1) {
  52             return NULL;
  53         }
  54     }
  55     else {
  56         target = f;
  57     }
  58
  59     f_info->size = target.st_size;
  60     f_info->last_modification = target.st_mtime;
  61
  62     if (S_ISDIR(target.st_mode)) {
  63         f_info->is_directory = MK_FILE_TRUE;
  64     }
  65
  66     /* Checking read access */
  67     if ((target.st_mode & S_IRUSR && target.st_uid == euid) ||
  68         (target.st_mode & S_IRGRP && target.st_gid == egid) ||
  69         (target.st_mode & S_IROTH)) {
  70         f_info->read_access = MK_FILE_TRUE;
  71     }
  72
  73     /* Checking execution access */
  74     if ((target.st_mode & S_IXUSR && target.st_uid == euid) ||
  75         (target.st_mode & S_IXGRP && target.st_gid == egid) ||
  76         (target.st_mode & S_IXOTH)) {
  77         f_info->exec_access = MK_FILE_TRUE;
  78
  79     }
  80     return f_info;
  81 }
  82
  83 /* Read file content to a memory buffer,
  84  * Use this function just for really SMALL files
  85  */
  86 char *mk_file_to_buffer(char *path)
  87 {
  88     FILE *fp;
  89     char *buffer;
  90     long bytes;
  91     struct file_info *finfo;
  92
  93     if (!(finfo = mk_file_get_info(path))) {
  94         return NULL;
  95     }
  96
  97     if (!(fp = fopen(path, "r"))) {
  98         return NULL;
  99     }
 100
 101     buffer = calloc(finfo->size + 1, sizeof(char));
 102     if (!buffer) {
 103         fclose(fp);
 104         return NULL;
 105     }
 106
 107     bytes = fread(buffer, finfo->size, 1, fp);
 108
 109     if (bytes < 1) {
 110         mk_mem_free(buffer);
 111         fclose(fp);
 112         return NULL;
 113     }
 114
 115     fclose(fp);
 116     return (char *) buffer;
 117
 118 }