/*. ******* coding:utf-8 AUTOHEADER START v1.1 *******
 *. vim: fileencoding=utf-8 syntax=c sw=2 ts=2 et
 *.
 *. © 2007-2009 Nima Talebi <nima@autonomy.net.au>
 *. © 2009      David Sommerseth <davids@redhat.com>
 *. © 2002-2008 Jean Delvare <khali@linux-fr.org>
 *. © 2000-2002 Alan Cox <alan@redhat.com>
 *.
 *. This file is part of Python DMI-Decode.
 *.
 *.     Python DMI-Decode is free software: you can redistribute it and/or modify
 *.     it under the terms of the GNU General Public License as published by
 *.     the Free Software Foundation, either version 2 of the License, or
 *.     (at your option) any later version.
 *.
 *.     Python DMI-Decode is distributed in the hope that it will be useful,
 *.     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *.     GNU General Public License for more details.
 *.
 *.     You should have received a copy of the GNU General Public License
 *.     along with Python DMI-Decode.  If not, see <http://www.gnu.org/licenses/>.
 *.
 *. THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
 *. WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 *. MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO
 *. EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 *. INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *. LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 *. PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 *. LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 *. OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 *. ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *.
 *. ADAPTED M. STONE & T. PARKER DISCLAIMER: THIS SOFTWARE COULD RESULT IN INJURY
 *. AND/OR DEATH, AND AS SUCH, IT SHOULD NOT BE BUILT, INSTALLED OR USED BY ANYONE.
 *.
 *. $AutoHeaderSerial::20090522                                                 $
 *. ******* AUTOHEADER END v1.1 ******* */

#include <Python.h>

#include <libxml/tree.h>

#include "xmlpythonizer.h"
#include "dmidecodemodule.h"
#include "dmixml.h"
#include <mcheck.h>

options opt;

static void init(void)
{
        /* sanity check */
        if(sizeof(u8) != 1 || sizeof(u16) != 2 || sizeof(u32) != 4 || '\0' != 0)
                fprintf(stderr, "%s: compiler incompatibility\n", "dmidecodemodule");

        opt.devmem = DEFAULT_MEM_DEV;
        opt.dumpfile = NULL;
        opt.flags = 0;
        opt.type = NULL;
        opt.mappingxml = NULL;
        opt.dmiversion_n = NULL;
        opt.python_xml_map = strdup(PYTHON_XML_MAP);
}

u8 *parse_opt_type(u8 * p, const char *arg)
{

        /* Allocate memory on first call only */
        if(p == NULL) {
                if(!(p = (u8 *) calloc(256, sizeof(u8)))) {
                        perror("calloc");
                        return NULL;
                }
        }

        unsigned int i, j;

        /* First try as a keyword */
        for(i = 0; i < ARRAY_SIZE(opt_type_keyword); i++) {
                if(!strcasecmp(arg, opt_type_keyword[i].keyword)) {
                        j = 0;
                        while(opt_type_keyword[i].type[j] != 255)
                                p[opt_type_keyword[i].type[j++]] = 1;
                        return p;
                }
        }

        /* Else try as a number */
        while(*arg != '\0') {
                unsigned long val;
                char *next;

                val = strtoul(arg, &next, 0);
                if(next == arg) {
                        fprintf(stderr, "Invalid type keyword: %s\n", arg);
                        free(p);
                        return NULL;
                }
                if(val > 0xff) {
                        fprintf(stderr, "Invalid type number: %lu\n", val);
                        free(p);
                        return NULL;
                }

                p[val] = 1;
                arg = next;
                while(*arg == ',' || *arg == ' ')
                        arg++;
        }

        return p;
}


xmlNode *dmidecode_set_version()
{
        int found = 0;
        size_t fp;
        int efi;
        u8 *buf = NULL;
        xmlNode *ver_n = NULL;

        /* Set default option values */
        opt.devmem = DEFAULT_MEM_DEV;

        /* Read from dump if so instructed */
        if(opt.dumpfile != NULL) {
                const char *dumpfile = PyString_AS_STRING(opt.dumpfile);

                //. printf("Reading SMBIOS/DMI data from file %s.\n", dumpfile);
                if((buf = mem_chunk(0, 0x20, dumpfile)) != NULL) {
                        if(memcmp(buf, "_SM_", 4) == 0) {
                                ver_n = smbios_decode_set_version(buf, dumpfile);
                                if( dmixml_GetAttrValue(ver_n, "unknown") == NULL ) {
                                        found++;
                                }
                        } else if(memcmp(buf, "_DMI_", 5) == 0) {
                                ver_n = legacy_decode_set_version(buf, dumpfile);
                                if( dmixml_GetAttrValue(ver_n, "unknown") == NULL ) {
                                        found++;
                                }
                        }
                }
        } else {          /* Read from /dev/mem */
                /* First try EFI (ia64, Intel-based Mac) */
                efi = address_from_efi(&fp);
                if(efi == EFI_NOT_FOUND) {
                        /* Fallback to memory scan (x86, x86_64) */
                        if((buf = mem_chunk(0xF0000, 0x10000, opt.devmem)) != NULL) {
                                for(fp = 0; fp <= 0xFFF0; fp += 16) {
                                        if(memcmp(buf + fp, "_SM_", 4) == 0 && fp <= 0xFFE0) {
                                                ver_n = smbios_decode_set_version(buf + fp, opt.devmem);
                                                if( dmixml_GetAttrValue(ver_n, "unknown") == NULL ) {
                                                        found++;
                                                }
                                                fp += 16;
                                        } else if(memcmp(buf + fp, "_DMI_", 5) == 0) {
                                                ver_n = legacy_decode_set_version (buf + fp, opt.devmem);
                                                if( dmixml_GetAttrValue(ver_n, "unknown") == NULL ) {
                                                        found++;
                                                }
                                        }
                                }
                        }
                } else if(efi == EFI_NO_SMBIOS) {
                        ver_n = NULL;
                } else {
                        // Process as EFI
                        if((buf = mem_chunk(fp, 0x20, opt.devmem)) != NULL) {
                                ver_n = smbios_decode_set_version(buf, opt.devmem);
                                if( dmixml_GetAttrValue(ver_n, "unknown") == NULL ) {
                                        found++;
                                }
                                //. TODO: dmixml_AddAttribute(dmixml_n, "efi_address", efiAddress);
                        }
                }
        }
        if( buf != NULL ) {
                free(buf);
        }

        if( !found ) {
                fprintf(stderr, "No SMBIOS nor DMI entry point found, sorry.");
        }
        free(opt.type);
        return ver_n;
}

xmlNode *dmidecode_get_xml(PyObject *self, const char *section)
{
        //mtrace();

        int ret = 0;
        int found = 0;
        size_t fp;
        int efi;
        u8 *buf;

        xmlNode *dmixml_n = xmlNewNode(NULL, (xmlChar *) "dmidecode");
        assert( dmixml_n != NULL );

        // Append DMI version info
        if( opt.dmiversion_n != NULL ) {
                xmlAddChild(dmixml_n, xmlCopyNode(opt.dmiversion_n, 1));
        }

        const char *f = opt.dumpfile ? PyString_AsString(opt.dumpfile) : opt.devmem;

        if(access(f, R_OK) < 0)
                PyErr_SetString(PyExc_IOError, "Permission denied to memory file/device");

        /* Read from dump if so instructed */
        if(opt.dumpfile != NULL) {
                const char *dumpfile = PyString_AS_STRING(opt.dumpfile);

                //  printf("Reading SMBIOS/DMI data from file %s.\n", dumpfile);
                if((buf = mem_chunk(0, 0x20, dumpfile)) != NULL) {
                        if(memcmp(buf, "_SM_", 4) == 0) {
                                if(smbios_decode(buf, dumpfile, dmixml_n))
                                        found++;
                        } else if(memcmp(buf, "_DMI_", 5) == 0) {
                                if(legacy_decode(buf, dumpfile, dmixml_n))
                                        found++;
                        }
                } else {
                        ret = 1;
                }
        } else {                /* Read from /dev/mem */
                /* First try EFI (ia64, Intel-based Mac) */
                efi = address_from_efi(&fp);
                if(efi == EFI_NOT_FOUND) {
                        /* Fallback to memory scan (x86, x86_64) */
                        if((buf = mem_chunk(0xF0000, 0x10000, opt.devmem)) != NULL) {
                                for(fp = 0; fp <= 0xFFF0; fp += 16) {
                                        if(memcmp(buf + fp, "_SM_", 4) == 0 && fp <= 0xFFE0) {
                                                if(smbios_decode(buf + fp, opt.devmem, dmixml_n)) {
                                                        found++;
                                                        fp += 16;
                                                }
                                        } else if(memcmp(buf + fp, "_DMI_", 5) == 0) {
                                                if(legacy_decode(buf + fp, opt.devmem, dmixml_n))
                                                        found++;
                                        }
                                }
                        } else
                                ret = 1;
                } else if(efi == EFI_NO_SMBIOS) {
                        ret = 1;
                } else {
                        if((buf = mem_chunk(fp, 0x20, opt.devmem)) == NULL)
                                ret = 1;
                        else if(smbios_decode(buf, opt.devmem, dmixml_n))
                                found++;
                        //  TODO: dmixml_AddAttribute(dmixml_n, "efi_address", "0x%08x", efiAddress);
                }
        }

        free(opt.type);
        if(ret == 0) {
                free(buf);
        } else {
                xmlFreeNode(dmixml_n);
                dmixml_n = NULL;
        }

        //muntrace();
        return dmixml_n;
}

static PyObject *dmidecode_get(PyObject *self, const char *section)
{
        PyObject *pydata = NULL;
        xmlNode *dmixml_n = NULL;

        /* Set default option values */
        opt.devmem = DEFAULT_MEM_DEV;
        opt.flags = 0;
        opt.type = NULL;
        opt.type = parse_opt_type(opt.type, section);

        if(opt.type == NULL) {
                return NULL;
        }

        dmixml_n = dmidecode_get_xml(self, section);
        if( dmixml_n != NULL ) {
                ptzMAP *mapping = NULL;

                // Convert the retrieved XML nodes to Python dicts
                if( opt.mappingxml == NULL ) {
                        // Load mapping into memory
                        opt.mappingxml = xmlReadFile(opt.python_xml_map, NULL, 0);
                        assert( opt.mappingxml != NULL );
                }

                mapping = dmiMAP_ParseMappingXML(opt.mappingxml, section);
                if( mapping == NULL ) {
                        return NULL;
                }

                // Generate Python dict out of XML node
                pydata = pythonizeXMLnode(mapping, dmixml_n);

#if 0  // DEBUG - will dump generated XML to stdout
                xmlDoc *doc = xmlNewDoc((xmlChar *) "1.0");
                xmlDocSetRootElement(doc, xmlCopyNode(dmixml_n, 1));
                xmlSaveFormatFileEnc("-", doc, "UTF-8", 1);
                xmlFreeDoc(doc);
#endif
                ptzmap_Free(mapping);
                xmlFreeNode(dmixml_n);
        } else {
                return NULL;
        }

        return pydata;
}



static PyObject *dmidecode_get_bios(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "bios");
}
static PyObject *dmidecode_get_system(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "system");
}
static PyObject *dmidecode_get_baseboard(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "baseboard");
}
static PyObject *dmidecode_get_chassis(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "chassis");
}
static PyObject *dmidecode_get_processor(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "processor");
}
static PyObject *dmidecode_get_memory(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "memory");
}
static PyObject *dmidecode_get_cache(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "cache");
}
static PyObject *dmidecode_get_connector(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "connector");
}
static PyObject *dmidecode_get_slot(PyObject * self, PyObject * args)
{
        return dmidecode_get(self, "slot");
}
static PyObject *dmidecode_get_type(PyObject * self, PyObject * args)
{
        long unsigned int lu;

        if(PyArg_ParseTuple(args, (char *)"i", &lu)) {
                if(lu < 256) {
                        char s[8];

                        sprintf(s, "%lu", lu);
                        return dmidecode_get(self, s);
                }
                return Py_False;
        }
        return Py_None;
}

static PyObject *dmidecode_dump(PyObject * self, PyObject * null)
{
        const char *f;

        f = opt.dumpfile ? PyString_AsString(opt.dumpfile) : opt.devmem;
        struct stat _buf;

        stat(f, &_buf);

        if((access(f, F_OK) != 0) || ((access(f, W_OK) == 0) && S_ISREG(_buf.st_mode)))
                if(dump(PyString_AS_STRING(opt.dumpfile)))
                        Py_RETURN_TRUE;
        Py_RETURN_FALSE;
}

static PyObject *dmidecode_get_dev(PyObject * self, PyObject * null)
{
        PyObject *dev;

        if(opt.dumpfile != NULL)
                dev = opt.dumpfile;
        else
                dev = PyString_FromString(opt.devmem);
        Py_INCREF(dev);
        return dev;
}

static PyObject *dmidecode_set_dev(PyObject * self, PyObject * arg)
{
        if(PyString_Check(arg)) {
                if(opt.dumpfile == arg)
                        Py_RETURN_TRUE;

                struct stat buf;
                char *f = PyString_AsString(arg);

                stat(f, &buf);
                if(opt.dumpfile) {
                        Py_DECREF(opt.dumpfile);
                }

                if(S_ISCHR(buf.st_mode)) {
                        if(memcmp(PyString_AsString(arg), "/dev/mem", 8) == 0) {
                                opt.dumpfile = NULL;
                                Py_RETURN_TRUE;
                        } else {
                                Py_RETURN_FALSE;
                        }
                } else if(!S_ISDIR(buf.st_mode)) {
                        opt.dumpfile = arg;
                        Py_INCREF(opt.dumpfile);
                        Py_RETURN_TRUE;
                }
        }
        Py_RETURN_FALSE;
        //PyErr_Occurred();
}

static PyObject *dmidecode_set_pythonxmlmap(PyObject * self, PyObject * arg)
{
        if(PyString_Check(arg)) {
                struct stat fileinfo;
                char *fname = PyString_AsString(arg);

                memset(&fileinfo, 0, sizeof(struct stat));

                if( stat(fname, &fileinfo) != 0 ) {
                        PyErr_SetString(PyExc_IOError, "Could not access the given python map XML file");
                        return NULL;
                }

                free(opt.python_xml_map);
                opt.python_xml_map = strdup(fname);
                Py_RETURN_TRUE;
        } else {
                Py_RETURN_FALSE;
        }
}


static PyMethodDef DMIDataMethods[] = {
        {(char *)"dump", dmidecode_dump, METH_NOARGS, (char *)"Dump dmidata to set file"},
        {(char *)"get_dev", dmidecode_get_dev, METH_NOARGS,
         (char *)"Get an alternative memory device file"},
        {(char *)"set_dev", dmidecode_set_dev, METH_O,
         (char *)"Set an alternative memory device file"},

        {(char *)"bios", dmidecode_get_bios, METH_VARARGS, (char *)"BIOS Data"},
        {(char *)"system", dmidecode_get_system, METH_VARARGS, (char *)"System Data"},
        {(char *)"baseboard", dmidecode_get_baseboard, METH_VARARGS, (char *)"Baseboard Data"},
        {(char *)"chassis", dmidecode_get_chassis, METH_VARARGS, (char *)"Chassis Data"},
        {(char *)"processor", dmidecode_get_processor, METH_VARARGS, (char *)"Processor Data"},
        {(char *)"memory", dmidecode_get_memory, METH_VARARGS, (char *)"Memory Data"},
        {(char *)"cache", dmidecode_get_cache, METH_VARARGS, (char *)"Cache Data"},
        {(char *)"connector", dmidecode_get_connector, METH_VARARGS, (char *)"Connector Data"},
        {(char *)"slot", dmidecode_get_slot, METH_VARARGS, (char *)"Slot Data"},

        {(char *)"type", dmidecode_get_type, METH_VARARGS, (char *)"By Type"},

        {(char *)"pythonmap", dmidecode_set_pythonxmlmap, METH_O,
         (char *) "Use another python dict map definition. The default file is " PYTHON_XML_MAP},

        {NULL, NULL, 0, NULL}
};

PyMODINIT_FUNC initdmidecode(void)
{
        char *dmiver = NULL;
        PyObject *module = NULL;
        PyObject *version = NULL;

        init();
        module = Py_InitModule3((char *)"dmidecode", DMIDataMethods,
                                "Python extension module for dmidecode");

        version = PyString_FromString("2.10");
        Py_INCREF(version);
        PyModule_AddObject(module, "version", version);

        opt.dmiversion_n = dmidecode_set_version();
        dmiver = dmixml_GetContent(opt.dmiversion_n);
        PyModule_AddObject(module, "dmi", dmiver ? PyString_FromString(dmiver) : Py_None);
}