libast/port/mnt.c

/***********************************************************************
*                                                                      *
*               This software is part of the ast package               *
*          Copyright (c) 1985-2011 AT&T Intellectual Property          *
*                      and is licensed under the                       *
*                 Eclipse Public License, Version 1.0                  *
*                    by AT&T Intellectual Property                     *
*                                                                      *
*                A copy of the License is available at                 *
*          http://www.eclipse.org/org/documents/epl-v10.html           *
*         (with md5 checksum b35adb5213ca9657e911e9befb180842)         *
*                                                                      *
*              Information and Software Systems Research               *
*                            AT&T Research                             *
*                           Florham Park NJ                            *
*                                                                      *
*                 Glenn Fowler <gsf@research.att.com>                  *
*                  David Korn <dgk@research.att.com>                   *
*                   Phong Vo <kpv@research.att.com>                    *
*                                                                      *
***********************************************************************/
#pragma prototyped
/*
 * Glenn Fowler
 * AT&T Research
 *
 * mounted filesystem scan support
 * where are the standards when you really need them
 */

#include <ast.h>
#include <mnt.h>
#include <ls.h>

#if _lib_mntopen && _lib_mntread && _lib_mntclose

NoN(mnt)

#else

/*
 * the original interface just had mode
 */

#define FIXARGS(p,m,s)      do {                    \
                    if ((p)&&*(p)!='/') {       \
                        mode = p;       \
                        path = 0;       \
                    }               \
                    if (!path)          \
                        path = s;       \
                } while (0)
typedef struct
{
    Mnt_t   mnt;
    char    buf[128];
#if __CYGWIN__
    char    typ[128];
    char    opt[128];
#endif
} Header_t;

#if __CYGWIN__
#include <ast_windows.h>
#endif

static void
set(register Header_t* hp, const char* fs, const char* dir, const char* type, const char* options)
{
    const char* x;

    hp->mnt.flags = 0;
    if (x = (const char*)strchr(fs, ':'))
    {
        if (*++x && *x != '\\')
        {
            hp->mnt.flags |= MNT_REMOTE;
            if (*x == '(')
            {
                fs = x;
                type = "auto";
            }
        }
    }
    else if (x = (const char*)strchr(fs, '@'))
    {
        hp->mnt.flags |= MNT_REMOTE;
        sfsprintf(hp->buf, sizeof(hp->buf) - 1, "%s:%*.*s", x + 1, x - fs, x - fs, fs);
        fs = (const char*)hp->buf;
    }
    else if (strmatch(type, "[aAnN][fF][sS]*"))
        hp->mnt.flags |= MNT_REMOTE;
    if (streq(fs, "none"))
        fs = dir;
    hp->mnt.fs = (char*)fs;
    hp->mnt.dir = (char*)dir;
    hp->mnt.type = (char*)type;
    hp->mnt.options = (char*)options;
#if __CYGWIN__
    if (streq(type, "system") || streq(type, "user"))
    {
        char*   s;
        int mode;
        DWORD   vser;
        DWORD   flags;
        DWORD   len;
        char    drive[4];

        mode = SetErrorMode(SEM_FAILCRITICALERRORS);
        drive[0] = fs[0];
        drive[1] = ':';
        drive[2] = '\\';
        drive[3] = 0;
        if (GetVolumeInformation(drive, 0, 0, &vser, &len, &flags, hp->typ, sizeof(hp->typ) - 1))
            hp->mnt.type = hp->typ;
        else
            flags = 0;
        SetErrorMode(mode);
        s = strcopy(hp->mnt.options = hp->opt, type);
        s = strcopy(s, ",ignorecase");
        if (options)
        {
            *s++ = ',';
            strcpy(s, options);
        }
    }
#endif
}

#undef  MNT_REMOTE

#if _sys_mount && ( _lib_getfsstat || _lib_getmntinfo )

/*
 * 4.4 bsd getmntinfo
 *
 * what a crappy interface
 * data returned in static buffer -- ok
 * big chunk of allocated memory that cannot be freed -- come on
 * *and* netbsd changed the interface somewhere along the line
 * private interface? my bad -- public interface? par for the bsd course
 *
 * we assume getfsstat may suffer the same statfs/statvfs confusion
 */

#include <sys/param.h>      /* expect some macro redefinitions here */
#include <sys/mount.h>

#if _lib_getfsstat
#if _lib_getfsstat_statvfs
#define statfs      statvfs
#define f_flags     f_flag
#endif
#else
#if _lib_getmntinfo_statvfs
#define statfs      statvfs
#define f_flags     f_flag
#endif
#endif

typedef struct
{
    Header_t    hdr;
    struct statfs*  next;
    struct statfs*  last;
    char        opt[256];
#if _lib_getfsstat
    struct statfs   buf[1];
#endif
} Handle_t;

#ifdef MFSNAMELEN
#define TYPE(f)     ((f)->f_fstypename)
#else
#ifdef INITMOUNTNAMES
#define TYPE(f)     ((char*)type[(f)->f_type])
static const char*  type[] = INITMOUNTNAMES;
#else
#if _sys_fs_types
#define TYPE(f)     ((char*)mnt_names[(f)->f_type])
#include <sys/fs_types.h>
#else
#define TYPE(f)     (strchr((f)->f_mntfromname,':')?"nfs":"ufs")
#endif
#endif
#endif

static struct Mnt_options_t
{
    unsigned long   flag;
    const char* name;
}
options[] =
{
#ifdef MNT_RDONLY
    MNT_RDONLY, "rdonly",
#endif
#ifdef MNT_SYNCHRONOUS
    MNT_SYNCHRONOUS,"synchronous",
#endif
#ifdef MNT_NOEXEC
    MNT_NOEXEC, "noexec",
#endif
#ifdef MNT_NOSUID
    MNT_NOSUID, "nosuid",
#endif
#ifdef MNT_NODEV
    MNT_NODEV,  "nodev",
#endif
#ifdef MNT_UNION
    MNT_UNION,  "union",
#endif
#ifdef MNT_ASYNC
    MNT_ASYNC,  "async",
#endif
#ifdef MNT_NOCOREDUMP
    MNT_NOCOREDUMP, "nocoredump",
#endif
#ifdef MNT_NOATIME
    MNT_NOATIME,    "noatime",
#endif
#ifdef MNT_SYMPERM
    MNT_SYMPERM,    "symperm",
#endif
#ifdef MNT_NODEVMTIME
    MNT_NODEVMTIME, "nodevmtime",
#endif
#ifdef MNT_SOFTDEP
    MNT_SOFTDEP,    "softdep",
#endif
#ifdef MNT_EXRDONLY
    MNT_EXRDONLY,   "exrdonly",
#endif
#ifdef MNT_EXPORTED
    MNT_EXPORTED,   "exported",
#endif
#ifdef MNT_DEFEXPORTED
    MNT_DEFEXPORTED,"defexported",
#endif
#ifdef MNT_EXPORTANON
    MNT_EXPORTANON, "exportanon",
#endif
#ifdef MNT_EXKERB
    MNT_EXKERB, "exkerb",
#endif
#ifdef MNT_EXNORESPORT
    MNT_EXNORESPORT,"exnoresport",
#endif
#ifdef MNT_EXPUBLIC
    MNT_EXPUBLIC,   "expublic",
#endif
#ifdef MNT_LOCAL
    MNT_LOCAL,  "local",
#endif
#ifdef MNT_QUOTA
    MNT_QUOTA,  "quota",
#endif
#ifdef MNT_ROOTFS
    MNT_ROOTFS, "rootfs",
#endif
    0,      "unknown",
};

void*
mntopen(const char* path, const char* mode)
{
    register Handle_t*  mp;
    register int        n;

    FIXARGS(path, mode, 0);
#if _lib_getfsstat
    if ((n = getfsstat(NiL, 0, MNT_WAIT)) <= 0)
        return 0;
    n = (n - 1) * sizeof(struct statfs);
#else
    n = 0;
#endif
    if (!(mp = newof(0, Handle_t, 1, n)))
        return 0;
#if _lib_getfsstat
    n = getfsstat(mp->next = mp->buf, n + sizeof(struct statfs), MNT_WAIT);
#else
    n = getmntinfo(&mp->next, 0);
#endif
    if (n <= 0)
    {
        free(mp);
        return 0;
    }
    mp->last = mp->next + n;
    return (void*)mp;
}

Mnt_t*
mntread(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;
    register int        i;
    register int        n;
    register unsigned long  flags;

    if (mp->next < mp->last)
    {
        flags = mp->next->f_flags;
        n = 0;
        for (i = 0; i < elementsof(options); i++)
            if (flags & options[i].flag)
                n += sfsprintf(mp->opt + n, sizeof(mp->opt) - n - 1, ",%s", options[i].name);
        set(&mp->hdr, mp->next->f_mntfromname, mp->next->f_mntonname, TYPE(mp->next), n ? (mp->opt + 1) : (char*)0);
        mp->next++;
        return &mp->hdr.mnt;
    }
    return 0;
}

int
mntclose(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;

    if (!mp)
        return -1;
    free(mp);
    return 0;
}

#else

#if _lib_mntctl && _sys_vmount

/*
 * aix
 */

#include <sys/vmount.h>

#define SIZE        (16 * 1024)

static const char*  type[] =
{
    "aix", "aix#1", "nfs", "jfs", "aix#4", "cdrom"
};

typedef struct
{
    Header_t    hdr;
    long        count;
    struct vmount*  next;
    char        remote[128];
    char        type[16];
    struct vmount   info[1];
} Handle_t;

void*
mntopen(const char* path, const char* mode)
{
    register Handle_t*  mp;

    FIXARGS(path, mode, 0);
    if (!(mp = newof(0, Handle_t, 1, SIZE)))
        return 0;
    if ((mp->count = mntctl(MCTL_QUERY, sizeof(Handle_t) + SIZE, &mp->info)) <= 0)
    {
        free(mp);
        return 0;
    }
    mp->next = mp->info;
    return (void*)mp;
}

Mnt_t*
mntread(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;
    register char*      s;
    register char*      t;
    register char*      o;

    if (mp->count > 0)
    {
        if (vmt2datasize(mp->next, VMT_HOST) && (s = vmt2dataptr(mp->next, VMT_HOST)) && !streq(s, "-"))
        {
            sfsprintf(mp->remote, sizeof(mp->remote) - 1, "%s:%s", s, vmt2dataptr(mp->next, VMT_OBJECT));
            s = mp->remote;
        }
        else
            s = vmt2dataptr(mp->next, VMT_OBJECT);
        if (vmt2datasize(mp->next, VMT_ARGS))
            o = vmt2dataptr(mp->next, VMT_ARGS);
        else
            o = NiL;
        switch (mp->next->vmt_gfstype)
        {
#ifdef MNT_AIX
        case MNT_AIX:
            t = "aix";
            break;
#endif
#ifdef MNT_NFS
        case MNT_NFS:
            t = "nfs";
            break;
#endif
#ifdef MNT_JFS
        case MNT_JFS:
            t = "jfs";
            break;
#endif
#ifdef MNT_CDROM
        case MNT_CDROM:
            t = "cdrom";
            break;
#endif
#ifdef MNT_SFS
        case MNT_SFS:
            t = "sfs";
            break;
#endif
#ifdef MNT_CACHEFS
        case MNT_CACHEFS:
            t = "cachefs";
            break;
#endif
#ifdef MNT_NFS3
        case MNT_NFS3:
            t = "nfs3";
            break;
#endif
#ifdef MNT_AUTOFS
        case MNT_AUTOFS:
            t = "autofs";
            break;
#endif
        default:
            sfsprintf(t = mp->type, sizeof(mp->type), "aix%+d", mp->next->vmt_gfstype);
            break;
        }
        set(&mp->hdr, s, vmt2dataptr(mp->next, VMT_STUB), t, o);
        if (--mp->count > 0)
            mp->next = (struct vmount*)((char*)mp->next + mp->next->vmt_length);
        return &mp->hdr.mnt;
    }
    return 0;
}

int
mntclose(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;

    if (!mp)
        return -1;
    free(mp);
    return 0;
}

#else

#if !_lib_setmntent
#undef  _lib_getmntent
#if !_SCO_COFF && !_SCO_ELF && !_UTS
#undef  _hdr_mnttab
#endif
#endif

#if _lib_getmntent && ( _hdr_mntent || _sys_mntent && !_sys_mnttab )

#if defined(__STDPP__directive) && defined(__STDPP__hide)
__STDPP__directive pragma pp:hide endmntent getmntent
#else
#define endmntent   ______endmntent
#define getmntent   ______getmntent
#endif

#include <stdio.h>
#if _hdr_mntent
#include <mntent.h>
#else
#include <sys/mntent.h>
#endif

#if defined(__STDPP__directive) && defined(__STDPP__hide)
__STDPP__directive pragma pp:nohide endmntent getmntent
#else
#undef  endmntent
#undef  getmntent
#endif

extern int      endmntent(FILE*);
extern struct mntent*   getmntent(FILE*);

#else

#undef  _lib_getmntent

#if _hdr_mnttab
#include <mnttab.h>
#else
#if _sys_mnttab
#include <sys/mnttab.h>
#endif
#endif

#endif

#ifndef MOUNTED
#ifdef  MNT_MNTTAB
#define MOUNTED     MNT_MNTTAB
#else
#if _hdr_mnttab || _sys_mnttab
#define MOUNTED     "/etc/mnttab"
#else
#define MOUNTED     "/etc/mtab"
#endif
#endif
#endif

#ifdef __Lynx__
#undef  MOUNTED
#define MOUNTED     "/etc/fstab"
#define SEP     ':'
#endif

#if _lib_getmntent

typedef struct
#if _mem_mnt_opts_mntent
#define OPTIONS(p)  ((p)->mnt_opts)
#else
#define OPTIONS(p)  NiL
#endif

{
    Header_t    hdr;
    FILE*       fp;
} Handle_t;

void*
mntopen(const char* path, const char* mode)
{
    register Handle_t*  mp;

    FIXARGS(path, mode, MOUNTED);
    if (!(mp = newof(0, Handle_t, 1, 0)))
        return 0;
    if (!(mp->fp = setmntent(path, mode)))
    {
        free(mp);
        return 0;
    }
    return (void*)mp;
}

Mnt_t*
mntread(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;
    register struct mntent* mnt;

    if (mnt = getmntent(mp->fp))
    {
        set(&mp->hdr, mnt->mnt_fsname, mnt->mnt_dir, mnt->mnt_type, OPTIONS(mnt));
        return &mp->hdr.mnt;
    }
    return 0;
}

int
mntclose(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;

    if (!mp)
        return -1;
    endmntent(mp->fp);
    free(mp);
    return 0;
}

#else

#if _sys_mntent && _lib_w_getmntent

#include <sys/mntent.h>

#define mntent      w_mntent

#define mnt_dir     mnt_mountpoint
#define mnt_type    mnt_fstname

#define MNTBUFSIZE  (sizeof(struct w_mnth)+16*sizeof(struct w_mntent))

#if _mem_mnt_opts_w_mntent
#define OPTIONS(p)  ((p)->mnt_opts)
#else
#define OPTIONS(p)  NiL
#endif

#else

#undef _lib_w_getmntent

#define MNTBUFSIZE  sizeof(struct mntent)

#if !_mem_mt_dev_mnttab || !_mem_mt_filsys_mnttab
#undef  _hdr_mnttab
#endif

#if _hdr_mnttab

#define mntent  mnttab

#define mnt_fsname  mt_dev
#define mnt_dir     mt_filsys
#if _mem_mt_fstyp_mnttab
#define mnt_type    mt_fstyp
#endif

#if _mem_mnt_opts_mnttab
#define OPTIONS(p)  ((p)->mnt_opts)
#else
#define OPTIONS(p)  NiL
#endif

#else

struct mntent
{
    char    mnt_fsname[256];
    char    mnt_dir[256];
    char    mnt_type[32];
    char    mnt_opts[64];
};

#define OPTIONS(p)  ((p)->mnt_opts)

#endif

#endif

typedef struct
{
    Header_t    hdr;
    Sfio_t*     fp;
    struct mntent*  mnt;
#if _lib_w_getmntent
    int     count;
#endif
    char        buf[MNTBUFSIZE];
} Handle_t;

void*
mntopen(const char* path, const char* mode)
{
    register Handle_t*  mp;

    FIXARGS(path, mode, MOUNTED);
    if (!(mp = newof(0, Handle_t, 1, 0)))
        return 0;
#if _lib_w_getmntent
    if ((mp->count = w_getmntent(mp->buf, sizeof(mp->buf))) > 0)
        mp->mnt = (struct mntent*)(((struct w_mnth*)mp->buf) + 1);
    else
#else
    mp->mnt = (struct mntent*)mp->buf;
    if (!(mp->fp = sfopen(NiL, path, mode)))
#endif
    {
        free(mp);
        return 0;
    }
    return (void*)mp;
}

Mnt_t*
mntread(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;

#if _lib_w_getmntent

    if (mp->count-- <= 0)
    {
        if ((mp->count = w_getmntent(mp->buf, sizeof(mp->buf))) <= 0)
            return 0;
        mp->count--;
        mp->mnt = (struct mntent*)(((struct w_mnth*)mp->buf) + 1);
    }
    set(&mp->hdr, mp->mnt->mnt_fsname, mp->mnt->mnt_dir, mp->mnt->mnt_type, OPTIONS(mp->mnt));
    mp->mnt++;
    return &mp->hdr.mnt;

#else

#if _hdr_mnttab

    while (sfread(mp->fp, &mp->buf, sizeof(mp->buf)) == sizeof(mp->buf))
        if (*mp->mnt->mnt_fsname && *mp->mnt->mnt_dir)
        {
#ifndef mnt_type
            struct stat st;

            static char typ[32];

            set(&mp->hdr, mp->mnt->mnt_fsname, mp->mnt->mnt_dir, stat(mp->mnt->mnt_dir, &st) ? FS_default : strlcpy(typ, fmtfs(&st), sizeof(typ)), OPTIONS(mp->mnt));
#else
            set(&mp->hdr, mp->mnt->mnt_fsname, mp->mnt->mnt_dir, mp->mnt->mnt_type, OPTIONS(mp->mnt));
#endif
            return &mp->hdr.mnt;
        }
    return 0;

#else

    register int        c;
    register char*      s;
    register char*      m;
    register char*      b;
    register int        q;
    register int        x;

 again:
    q = 0;
    x = 0;
    b = s = mp->mnt->mnt_fsname;
    m = s + sizeof(mp->mnt->mnt_fsname) - 1;
    for (;;) switch (c = sfgetc(mp->fp))
    {
    case EOF:
        return 0;
    case '"':
    case '\'':
        if (q == c)
            q = 0;
        else if (!q)
            q = c;
        break;
#ifdef SEP
    case SEP:
#else
    case ' ':
    case '\t':
#endif
        if (s != b && !q) switch (++x)
        {
        case 1:
            *s = 0;
            b = s = mp->mnt->mnt_dir;
            m = s + sizeof(mp->mnt->mnt_dir) - 1;
            break;
        case 2:
            *s = 0;
            b = s = mp->mnt->mnt_type;
            m = s + sizeof(mp->mnt->mnt_type) - 1;
            break;
        case 3:
            *s = 0;
            b = s = mp->mnt->mnt_opts;
            m = s + sizeof(mp->mnt->mnt_opts) - 1;
            break;
        case 4:
            *s = 0;
            b = s = m = 0;
            break;
        }
        break;
    case '\n':
        if (x >= 3)
        {
            set(&mp->hdr, mp->mnt->mnt_fsname, mp->mnt->mnt_dir, mp->mnt->mnt_type, OPTIONS(mp->mnt));
            return &mp->hdr.mnt;
        }
        goto again;
    default:
        if (s < m)
            *s++ = c;
        break;
    }

#endif

#endif

}

int
mntclose(void* handle)
{
    register Handle_t*  mp = (Handle_t*)handle;

    if (!mp)
        return -1;
    sfclose(mp->fp);
    free(mp);
    return 0;
}

#endif

#endif

#endif

/*
 * currently no write
 */

int
mntwrite(void* handle, const Mnt_t* mnt)
{
    NoP(handle);
    NoP(mnt);
    return -1;
}

#endif