rpmio/rpmmg.c

Go to the documentation of this file.

#include "system.h"

#if defined(HAVE_MAGIC_H)
#include "magic.h"
#endif

#include <rpmiotypes.h>
#include <rpmio.h>      /* for *Pool methods */
#include <rpmlog.h>
#include <rpmurl.h>
#define _RPMMG_INTERNAL
#include <rpmmg.h>

#include "debug.h"

/*@unchecked@*/
int _rpmmg_debug = 0;

/*@-mustmod@*/  /* XXX splint on crack */
static void rpmmgFini(void * _mg)
        /*@globals fileSystem @*/
        /*@modifies *_mg, fileSystem @*/
{
    rpmmg mg = _mg;

#if defined(HAVE_MAGIC_H)
    if (mg->ms) {
        magic_close(mg->ms);
        mg->ms = NULL;
    }
#endif
    mg->fn = _free(mg->fn);
}
/*@=mustmod@*/

/*@unchecked@*/ /*@only@*/ /*@null@*/
rpmioPool _rpmmgPool = NULL;

static rpmmg rpmmgGetPool(/*@null@*/ rpmioPool pool)
        /*@globals _rpmmgPool, fileSystem @*/
        /*@modifies pool, _rpmmgPool, fileSystem @*/
{
    rpmmg mg;

    if (_rpmmgPool == NULL) {
        _rpmmgPool = rpmioNewPool("mg", sizeof(*mg), -1, _rpmmg_debug,
                        NULL, NULL, rpmmgFini);
        pool = _rpmmgPool;
    }
    return (rpmmg) rpmioGetPool(pool, sizeof(*mg));
}

rpmmg rpmmgNew(const char * fn, int flags)
{
    rpmmg mg = rpmmgGetPool(_rpmmgPool);
    int xx;

    if (fn)
        mg->fn = xstrdup(fn);
#if defined(HAVE_MAGIC_H)
    mg->flags = (flags ? flags : MAGIC_CHECK);/* XXX MAGIC_COMPRESS flag? */
    mg->ms = magic_open(flags);
    if (mg->ms == NULL) {
        rpmlog(RPMLOG_ERR, _("magic_open(0x%x) failed: %s\n"),
                flags, strerror(errno));
        return rpmmgFree(mg);
    }
    xx = magic_load(mg->ms, mg->fn);
    if (xx == -1) {
        rpmlog(RPMLOG_ERR, _("magic_load(ms, %s) failed: %s\n"),
                (fn ? fn : "(nil)"), magic_error(mg->ms));
        return rpmmgFree(mg);
    }
#endif

    return rpmmgLink(mg);
}

const char * rpmmgFile(rpmmg mg, const char *fn)
{
    const char * t = NULL;

if (_rpmmg_debug)
fprintf(stderr, "--> rpmmgFile(%p, %s)\n", mg, (fn ? fn : "(nil)"));
#if defined(HAVE_MAGIC_H)
    if (mg->ms) {
        const char * lpath = NULL;
        int ut = urlPath(fn, &lpath);

        switch (ut) {
        case URL_IS_FTP:
        case URL_IS_HKP:
        case URL_IS_HTTP:
        case URL_IS_HTTPS:
        {   char b[512];
            size_t nb = 0;
            FD_t fd;
            
            fd = Fopen(fn, "r.ufdio");
            if (fd != NULL && !Ferror(fd)) {
                nb = Fread(b, 1, sizeof(b), fd);
                (void) Fclose(fd);
            }
            if (nb > 0)
                return rpmmgBuffer(mg, b, nb);
        }   break;
        case URL_IS_DASH:
        case URL_IS_MONGO:      /* XXX FIXME */
            break;
        case URL_IS_PATH:
            fn = lpath;
            /*@fallthrough@*/
        case URL_IS_UNKNOWN:
        default:
            t = magic_file(mg->ms, fn);
            /* XXX HACK: libmagic compiled without <pcreposix.h> spews here. */
            if (t == NULL) {
                const char * msg = magic_error(mg->ms);
                if (strstr(msg, "regexec error 17, (match failed)") == NULL)
                    rpmlog(RPMLOG_ERR, _("magic_file(ms, %s) failed: %s\n"),
                            (fn ? fn : "(nil)"), msg);
            }
            break;
        }
    }
#endif

    if (t == NULL) t = "";
    t = xstrdup(t);

if (_rpmmg_debug)
fprintf(stderr, "<-- rpmmgFile(%p, %s) %s\n", mg, (fn ? fn : "(nil)"), t);
    return t;
}

const char * rpmmgBuffer(rpmmg mg, const char * b, size_t nb)
{
    const char * t = NULL;

if (_rpmmg_debug)
fprintf(stderr, "--> rpmmgBuffer(%p, %p[%d])\n", mg, b, (int)nb);
    if (nb == 0) nb = strlen(b);
#if defined(HAVE_MAGIC_H)
    if (mg->ms) {
        t = magic_buffer(mg->ms, b, nb);
        /* XXX HACK: libmagic compiled without <pcreposix.h> spews here. */
        if (t == NULL) {
            const char * msg = magic_error(mg->ms);
            if (strstr(msg, "regexec error 17, (match failed)") == NULL)
                rpmlog(RPMLOG_ERR, _("magic_buffer(ms, %p[%u]) failed: %s\n"),
                        b, (unsigned)nb, msg);
        }
    }
#endif

    if (t == NULL) t = "";
    t = xstrdup(t);

if (_rpmmg_debug)
fprintf(stderr, "<-- rpmmgBuffer(%p, %p[%d]) %s\n", mg, b, (int)nb, t);
    return t;
}