diff options
author | Allan McRae <allan@archlinux.org> | 2010-11-16 16:15:21 +1000 |
---|---|---|
committer | Dan McGee <dan@archlinux.org> | 2010-12-12 20:29:43 -0600 |
commit | f4e9deb6d74614ac427b37513359d91588dd1542 (patch) | |
tree | 3f707d2046a12ee0808a3e0cbd3eed8a1d036767 /lib/libalpm | |
parent | adb10c3ab244cc00eb863d5b036fa7bdd76f8152 (diff) |
Add function for listing system mount points
Add a mount_point_list() function that attempts to portably obtain
a list of system mount points and a struct to hold needed mount point
information.
Abort the transaction if we are unable to determine the mount points.
Signed-off-by: Allan McRae <allan@archlinux.org>
Signed-off-by: Dan McGee <dan@archlinux.org>
Diffstat (limited to 'lib/libalpm')
-rw-r--r-- | lib/libalpm/diskspace.c | 119 | ||||
-rw-r--r-- | lib/libalpm/diskspace.h | 20 |
2 files changed, 139 insertions, 0 deletions
diff --git a/lib/libalpm/diskspace.c b/lib/libalpm/diskspace.c index 31a2760d..a5beb84f 100644 --- a/lib/libalpm/diskspace.c +++ b/lib/libalpm/diskspace.c @@ -17,11 +17,130 @@ * along with this program. If not, see <http://www.gnu.org/licenses/>. */ +#include "config.h" + +#if defined HAVE_GETMNTENT +#include <mntent.h> +#include <sys/statvfs.h> +#elif defined HAVE_GETMNTINFO_STATFS +#include <sys/param.h> +#include <sys/mount.h> +#if HAVE_SYS_UCRED_H +#include <sys/ucred.h> +#endif +#elif defined HAVE_GETMNTINFO_STATVFS +#include <sys/types.h> +#include <sys/statvfs.h> +#endif + /* libalpm */ #include "diskspace.h" +#include "alpm_list.h" +#include "util.h" +#include "log.h" + +static alpm_list_t *mount_point_list() +{ + alpm_list_t *mount_points = NULL; + alpm_mountpoint_t *mp; + +#if defined HAVE_GETMNTENT + struct mntent *mnt; + FILE *fp; + struct statvfs fsp; + + fp = setmntent(MOUNTED, "r"); + + if (fp == NULL) { + return NULL; + } + + while((mnt = getmntent (fp))) { + if(statvfs(mnt->mnt_dir, &fsp) != 0) { + _alpm_log(PM_LOG_WARNING, "could not get filesystem information for %s\n", mnt->mnt_dir); + continue; + } + + MALLOC(mp, sizeof(alpm_mountpoint_t), RET_ERR(PM_ERR_MEMORY, NULL)); + mp->mount_dir = strdup(mnt->mnt_dir); + + MALLOC(mp->fsp, sizeof(struct statvfs), RET_ERR(PM_ERR_MEMORY, NULL)); + memcpy((void *)(mp->fsp), (void *)(&fsp), sizeof(struct statvfs)); + + mp->blocks_needed = 0; + mp->max_blocks_needed = 0; + mp->used = 0; + + mount_points = alpm_list_add(mount_points, mp); + } + + endmntent(fp); +#elif defined HAVE_GETMNTINFO_STATFS + int entries; + struct statfs *fsp; + + entries = getmntinfo(&fsp, MNT_NOWAIT); + + if (entries < 0) { + return NULL; + } + + for(; entries-- > 0; fsp++) { + MALLOC(mp, sizeof(alpm_mountpoint_t), RET_ERR(PM_ERR_MEMORY, NULL)); + mp->mount_dir = strdup(fsp->f_mntonname); + + MALLOC(mp->fsp, sizeof(struct statfs), RET_ERR(PM_ERR_MEMORY, NULL)); + memcpy((void *)(mp->fsp), (void *)fsp, sizeof(struct statfs)); + + mp->blocks_needed = 0; + mp->max_blocks_needed = 0; + + mount_points = alpm_list_add(mount_points, mp); + } +#elif defined HAVE_GETMNTINFO_STATVFS + int entries; + struct statvfs *fsp; + + entries = getmntinfo(&fsp, MNT_NOWAIT); + + if (entries < 0) { + return NULL; + } + + for (; entries-- > 0; fsp++) { + MALLOC(mp, sizeof(alpm_mountpoint_t), RET_ERR(PM_ERR_MEMORY, NULL)); + mp->mount_dir = strdup(fsp->f_mntonname); + + MALLOC(mp->fsp, sizeof(struct statvfs), RET_ERR(PM_ERR_MEMORY, NULL)); + memcpy((void *)(mp->fsp), (void *)fsp, sizeof(struct statvfs)); + + mp->blocks_needed = 0; + mp->max_blocks_needed = 0; + + mount_points = alpm_list_add(mount_points, mp); + } +#endif + + return(mount_points); +} int _alpm_check_diskspace(pmtrans_t *trans, pmdb_t *db) { + alpm_list_t *mount_points; + + mount_points = mount_point_list(); + if(mount_points == NULL) { + _alpm_log(PM_LOG_ERROR, _("count not determine filesystem mount points")); + return -1; + } + + for(i = mount_points; i; i = alpm_list_next(i)) { + alpm_mountpoint_t *data = i->data; + FREE(data->mount_dir); + FREE(data->fsp); + } + FREELIST(mount_points); + return 0; } diff --git a/lib/libalpm/diskspace.h b/lib/libalpm/diskspace.h index 69e39d8e..15ff2f67 100644 --- a/lib/libalpm/diskspace.h +++ b/lib/libalpm/diskspace.h @@ -20,8 +20,28 @@ #ifndef _ALPM_DISKSPACE_H #define _ALPM_DISKSPACE_H +#if defined HAVE_GETMNTINFO_STATFS +#include <sys/mount.h> +#else +#include <sys/statvfs.h> +#endif + #include "alpm.h" +typedef struct __alpm_mountpoint_t { + /* mount point information */ + char *mount_dir; +#if defined HAVE_GETMNTINFO_STATFS + struct statfs *fsp; +#else + struct statvfs *fsp; +#endif + /* storage for additional disk usage calculations */ + long blocks_needed; + long max_blocks_needed; + int used; +} alpm_mountpoint_t; + int _alpm_check_diskspace(pmtrans_t *trans, pmdb_t *db); #endif /* _ALPM_DISKSPACE_H */ |