X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=lib%2Finit.c;h=fe40f91ccc36c1d4fd5c52595116ab78899f1340;hb=0766bfa0f557430a6dd3256d45a5b756499f9f7b;hp=167c893131ac04e63a1fe7c2382866bb41b7a90b;hpb=98ccf6d67b6dd95c7471fd4e5a6d6452a62394b9;p=pciutils.git diff --git a/lib/init.c b/lib/init.c index 167c893..fe40f91 100644 --- a/lib/init.c +++ b/lib/init.c @@ -1,7 +1,7 @@ /* * The PCI Library -- Initialization and related things * - * Copyright (c) 1997--2008 Martin Mares + * Copyright (c) 1997--2018 Martin Mares * * Can be freely distributed and used under the terms of the GNU GPL. */ @@ -13,6 +13,63 @@ #include "internal.h" +#ifdef PCI_OS_DJGPP +#include /* for __dos_argv0 */ +#endif + +#ifdef PCI_OS_WINDOWS + +#include + +/* Force usage of ANSI (char*) variant of GetModuleFileName() function */ +#ifdef _WIN32 +#ifdef GetModuleFileName +#undef GetModuleFileName +#endif +#define GetModuleFileName GetModuleFileNameA +#endif + +/* Define __ImageBase for all linkers */ +#ifdef _WIN32 +/* GNU LD provides __ImageBase symbol since 2.19, in previous versions it is + * under name _image_base__, so add weak alias for compatibility. */ +#ifdef __GNUC__ +asm(".weak\t" PCI_STRINGIFY(__MINGW_USYMBOL(__ImageBase)) "\n\t" + ".set\t" PCI_STRINGIFY(__MINGW_USYMBOL(__ImageBase)) "," PCI_STRINGIFY(__MINGW_USYMBOL(_image_base__))); +#endif +/* + * MSVC link.exe provides __ImageBase symbol since 12.00 (MSVC 6.0), for + * previous versions resolve it at runtime via GetModuleHandleA() which + * returns base for main executable or via VirtualQuery() for DLL builds. + */ +#if defined(_MSC_VER) && _MSC_VER < 1200 +static HMODULE +get_current_module_handle(void) +{ +#ifdef PCI_SHARED_LIB + MEMORY_BASIC_INFORMATION info; + size_t len = VirtualQuery(&get_current_module_handle, &info, sizeof(info)); + if (len != sizeof(info)) + return NULL; + return (HMODULE)info.AllocationBase; +#else + return GetModuleHandleA(NULL); +#endif +} +#define __ImageBase (*(IMAGE_DOS_HEADER *)get_current_module_handle()) +#else +extern IMAGE_DOS_HEADER __ImageBase; +#endif +#endif + +#if defined(_WINDLL) +extern HINSTANCE _hModule; +#elif defined(_WINDOWS) +extern HINSTANCE _hInstance; +#endif + +#endif + static struct pci_methods *pci_methods[PCI_ACCESS_MAX] = { NULL, #ifdef PCI_HAVE_PM_LINUX_SYSFS @@ -57,54 +114,75 @@ static struct pci_methods *pci_methods[PCI_ACCESS_MAX] = { #else NULL, #endif -}; - -struct pci_access * -pci_alloc(void) -{ - struct pci_access *a = malloc(sizeof(struct pci_access)); - int i; - - memset(a, 0, sizeof(*a)); - pci_set_name_list_path(a, PCI_PATH_IDS_DIR "/" PCI_IDS, 0); -#ifdef PCI_USE_DNS - pci_define_param(a, "net.domain", PCI_ID_DOMAIN); - pci_define_param(a, "net.cache_path", "~/.pciids-cache"); - a->id_lookup_mode = PCI_LOOKUP_CACHE; +#ifdef PCI_HAVE_PM_DARWIN_DEVICE + &pm_darwin, +#else + NULL, #endif - for(i=0; iconfig) - pci_methods[i]->config(a); - return a; -} - -void * -pci_malloc(struct pci_access *a, int size) -{ - void *x = malloc(size); - - if (!x) - a->error("Out of memory (allocation of %d bytes failed)", size); - return x; -} - -void -pci_mfree(void *x) -{ - if (x) - free(x); -} +#ifdef PCI_HAVE_PM_SYLIXOS_DEVICE + &pm_sylixos_device, +#else + NULL, +#endif +#ifdef PCI_HAVE_PM_HURD_CONF + &pm_hurd, +#else + NULL, +#endif +#ifdef PCI_HAVE_PM_WIN32_CFGMGR32 + &pm_win32_cfgmgr32, +#else + NULL, +#endif +#ifdef PCI_HAVE_PM_WIN32_KLDBG + &pm_win32_kldbg, +#else + NULL, +#endif +#ifdef PCI_HAVE_PM_WIN32_SYSDBG + &pm_win32_sysdbg, +#else + NULL, +#endif +#ifdef PCI_HAVE_PM_MMIO_CONF + &pm_mmio_conf1, + &pm_mmio_conf1_ext, +#else + NULL, + NULL, +#endif +#if defined(PCI_HAVE_PM_ECAM) + &pm_ecam, +#else + NULL, +#endif +}; -char * -pci_strdup(struct pci_access *a, char *s) -{ - int len = strlen(s) + 1; - char *t = pci_malloc(a, len); - memcpy(t, s, len); - return t; -} +// If PCI_ACCESS_AUTO is selected, we probe the access methods in this order +static int probe_sequence[] = { + // System-specific methods + PCI_ACCESS_SYS_BUS_PCI, + PCI_ACCESS_PROC_BUS_PCI, + PCI_ACCESS_FBSD_DEVICE, + PCI_ACCESS_AIX_DEVICE, + PCI_ACCESS_NBSD_LIBPCI, + PCI_ACCESS_OBSD_DEVICE, + PCI_ACCESS_DARWIN, + PCI_ACCESS_SYLIXOS_DEVICE, + PCI_ACCESS_HURD, + PCI_ACCESS_WIN32_CFGMGR32, + PCI_ACCESS_WIN32_KLDBG, + PCI_ACCESS_WIN32_SYSDBG, + // Low-level methods poking the hardware directly + PCI_ACCESS_ECAM, + PCI_ACCESS_I386_TYPE1, + PCI_ACCESS_I386_TYPE2, + PCI_ACCESS_MMIO_TYPE1_EXT, + PCI_ACCESS_MMIO_TYPE1, + -1, +}; -static void +static void PCI_NONRET pci_generic_error(char *msg, ...) { va_list args; @@ -112,6 +190,7 @@ pci_generic_error(char *msg, ...) va_start(args, msg); fputs("pcilib: ", stderr); vfprintf(stderr, msg, args); + va_end(args); fputc('\n', stderr); exit(1); } @@ -124,6 +203,7 @@ pci_generic_warn(char *msg, ...) va_start(args, msg); fputs("pcilib: ", stderr); vfprintf(stderr, msg, args); + va_end(args); fputc('\n', stderr); } @@ -142,81 +222,212 @@ pci_null_debug(char *msg UNUSED, ...) { } -char * -pci_get_param(struct pci_access *acc, char *param) +// Memory allocation functions are safe to call if pci_access is not fully initalized or even NULL + +void * +pci_malloc(struct pci_access *a, int size) { - struct pci_param *p; + void *x = malloc(size); - for (p=acc->params; p; p=p->next) - if (!strcmp(p->param, param)) - return p->value; - return NULL; + if (!x) + (a && a->error ? a->error : pci_generic_error)("Out of memory (allocation of %d bytes failed)", size); + return x; } void -pci_define_param(struct pci_access *acc, char *param, char *value) +pci_mfree(void *x) { - struct pci_param *p = pci_malloc(acc, sizeof(*p)); + if (x) + free(x); +} - p->next = acc->params; - acc->params = p; - p->param = param; - p->value = value; - p->value_malloced = 0; +char * +pci_strdup(struct pci_access *a, const char *s) +{ + int len = strlen(s) + 1; + char *t = pci_malloc(a, len); + memcpy(t, s, len); + return t; } int -pci_set_param_internal(struct pci_access *acc, char *param, char *value, int copy) +pci_lookup_method(char *name) { - struct pci_param *p; - - for (p=acc->params; p; p=p->next) - if (!strcmp(p->param, param)) - { - if (p->value_malloced) - pci_mfree(p->value); - p->value_malloced = copy; - if (copy) - p->value = pci_strdup(acc, value); - else - p->value = value; - return 0; - } + int i; + + for (i=0; iname, name)) + return i; return -1; } -int -pci_set_param(struct pci_access *acc, char *param, char *value) +char * +pci_get_method_name(int index) { - return pci_set_param_internal(acc, param, value, 1); + if (index < 0 || index >= PCI_ACCESS_MAX) + return NULL; + else if (!pci_methods[index]) + return ""; + else + return pci_methods[index]->name; } +#if defined(PCI_OS_WINDOWS) || defined(PCI_OS_DJGPP) + static void -pci_free_params(struct pci_access *acc) +pci_init_name_list_path(struct pci_access *a) { - struct pci_param *p; - - while (p = acc->params) + if ((PCI_PATH_IDS_DIR)[0]) + pci_set_name_list_path(a, PCI_PATH_IDS_DIR "\\" PCI_IDS, 0); + else { - acc->params = p->next; - if (p->value_malloced) - pci_mfree(p->value); - pci_mfree(p); + char *path, *sep; + size_t len; + +#if defined(PCI_OS_WINDOWS) && (defined(_WIN32) || defined(_WINDLL) || defined(_WINDOWS)) + + HMODULE module; + size_t size; + +#if defined(_WIN32) + module = (HINSTANCE)&__ImageBase; +#elif defined(_WINDLL) + module = _hModule; +#elif defined(_WINDOWS) + module = _hInstance; +#endif + + /* + * Module file name can have arbitrary length despite all MS examples say + * about MAX_PATH upper limit. This limit does not apply for example when + * executable is running from network disk with very long UNC paths or + * when using "\\??\\" prefix for specifying executable binary path. + * Function GetModuleFileName() returns passed size argument when passed + * buffer is too small and does not signal any error. In this case retry + * again with larger buffer. + */ + size = 256; /* initial buffer size (more than sizeof(PCI_IDS)-4) */ +retry: + path = pci_malloc(a, size); + len = GetModuleFileName(module, path, size-sizeof(PCI_IDS)-4); /* 4 for "\\\\?\\" */ + if (len >= size-sizeof(PCI_IDS)-4) + { + free(path); + size *= 2; + goto retry; + } + else if (len == 0) + path[0] = '\0'; + + /* + * GetModuleFileName() has bugs. On Windows 10 it prepends current drive + * letter if path is just pure NT namespace (with "\\??\\" prefix). Such + * extra drive letter makes path fully invalid and unusable. So remove + * extra drive letter to make path valid again. + * Reproduce: CreateProcessW("\\??\\C:\\lspci.exe", ...) + */ + if (((path[0] >= 'a' && path[0] <= 'z') || + (path[0] >= 'A' && path[0] <= 'Z')) && + strncmp(path+1, ":\\??\\", 5) == 0) + { + memmove(path, path+2, len-2); + len -= 2; + path[len] = '\0'; + } + + /* + * GetModuleFileName() has bugs. On Windows 10 it does not add "\\\\?\\" + * prefix when path is in native NT UNC namespace. Such path is treated by + * WinAPI/DOS functions as standard DOS path relative to the current + * directory, hence something completely different. So prepend missing + * "\\\\?\\" prefix to make path valid again. + * Reproduce: CreateProcessW("\\??\\UNC\\10.0.2.4\\qemu\\lspci.exe", ...) + * + * If path starts with DOS drive letter and with appended PCI_IDS is + * longer than 260 bytes and is without "\\\\?\\" prefix then append it. + * This prefix is required for paths and file names with DOS drive letter + * longer than 260 bytes. + */ + if (strncmp(path, "\\UNC\\", 5) == 0 || + strncmp(path, "UNC\\", 4) == 0 || + (((path[0] >= 'a' && path[0] <= 'z') || (path[0] >= 'A' && path[0] <= 'Z')) && + len + sizeof(PCI_IDS) >= 260)) + { + memmove(path+4, path, len); + memcpy(path, "\\\\?\\", 4); + len += 4; + path[len] = '\0'; + } + +#elif defined(PCI_OS_DJGPP) || defined(PCI_OS_WINDOWS) + + const char *exe_path; + +#ifdef PCI_OS_DJGPP + exe_path = __dos_argv0; +#else + exe_path = _pgmptr; +#endif + + len = strlen(exe_path); + path = pci_malloc(a, len+sizeof(PCI_IDS)); + memcpy(path, exe_path, len+1); + +#endif + + sep = strrchr(path, '\\'); + if (!sep) + { + /* + * If current module path (current executable for static builds or + * current DLL library for shared build) cannot be determined then + * fallback to the current directory. + */ + free(path); + pci_set_name_list_path(a, PCI_IDS, 0); + } + else + { + memcpy(sep+1, PCI_IDS, sizeof(PCI_IDS)); + pci_set_name_list_path(a, path, 1); + } } } -struct pci_param * -pci_walk_params(struct pci_access *acc, struct pci_param *prev) +#else + +static void +pci_init_name_list_path(struct pci_access *a) { - /* So far, the params form a simple linked list, but this can change in the future */ - if (!prev) - return acc->params; - else - return prev->next; + pci_set_name_list_path(a, PCI_PATH_IDS_DIR "/" PCI_IDS, 0); } -void -pci_init(struct pci_access *a) +#endif + +struct pci_access * +pci_alloc(void) +{ + struct pci_access *a = pci_malloc(NULL, sizeof(struct pci_access)); + int i; + + memset(a, 0, sizeof(*a)); + pci_init_name_list_path(a); +#ifdef PCI_USE_DNS + pci_define_param(a, "net.domain", PCI_ID_DOMAIN, "DNS domain used for resolving of ID's"); + pci_define_param(a, "net.cache_name", "~/.pciids-cache", "Name of the ID cache file"); + a->id_lookup_mode = PCI_LOOKUP_CACHE; +#endif +#ifdef PCI_HAVE_HWDB + pci_define_param(a, "hwdb.disable", "0", "Do not look up names in UDEV's HWDB if non-zero"); +#endif + for (i=0; iconfig) + pci_methods[i]->config(a); + return a; +} + +int +pci_init_internal(struct pci_access *a, int skip_method) { if (!a->error) a->error = pci_generic_error; @@ -227,7 +438,7 @@ pci_init(struct pci_access *a) if (!a->debugging) a->debug = pci_null_debug; - if (a->method) + if (a->method != PCI_ACCESS_AUTO) { if (a->method >= PCI_ACCESS_MAX || !pci_methods[a->method]) a->error("This access method is not supported."); @@ -236,24 +447,56 @@ pci_init(struct pci_access *a) else { unsigned int i; - for(i=0; idebug("Trying method %d...", i); - if (pci_methods[i]->detect(a)) - { - a->debug("...OK\n"); - a->methods = pci_methods[i]; - a->method = i; - break; - } - a->debug("...No.\n"); - } + for (i=0; probe_sequence[i] >= 0; i++) + { + struct pci_methods *m = pci_methods[probe_sequence[i]]; + if (!m) + continue; + if (skip_method == probe_sequence[i]) + continue; + a->debug("Trying method %s...", m->name); + if (m->detect(a)) + { + a->debug("...OK\n"); + a->methods = m; + a->method = probe_sequence[i]; + break; + } + a->debug("...No.\n"); + } if (!a->methods) - a->error("Cannot find any working access method."); + return 0; } a->debug("Decided to use %s\n", a->methods->name); a->methods->init(a); + return 1; +} + +void +pci_init_v35(struct pci_access *a) +{ + if (!pci_init_internal(a, -1)) + a->error("Cannot find any working access method."); +} + +STATIC_ALIAS(void pci_init(struct pci_access *a), pci_init_v35(a)); +DEFINE_ALIAS(void pci_init_v30(struct pci_access *a), pci_init_v35); +SYMBOL_VERSION(pci_init_v30, pci_init@LIBPCI_3.0); +SYMBOL_VERSION(pci_init_v35, pci_init@@LIBPCI_3.5); + +struct pci_access * +pci_clone_access(struct pci_access *a) +{ + struct pci_access *b = pci_alloc(); + + b->writeable = a->writeable; + b->buscentric = a->buscentric; + b->debugging = a->debugging; + b->error = a->error; + b->warning = a->warning; + b->debug = a->debug; + + return b; } void @@ -261,7 +504,7 @@ pci_cleanup(struct pci_access *a) { struct pci_dev *d, *e; - for(d=a->devices; d; d=e) + for (d=a->devices; d; d=e) { e = d->next; pci_free_dev(d);