Alc/helpers.c

   1 /**
   2  * OpenAL cross platform audio library
   3  * Copyright (C) 2011 by authors.
   4  * This library is free software; you can redistribute it and/or
   5  *  modify it under the terms of the GNU Library General Public
   6  *  License as published by the Free Software Foundation; either
   7  *  version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  *  Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  *  License along with this library; if not, write to the
  16  *  Free Software Foundation, Inc.,
  17  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  * Or go to http://www.gnu.org/copyleft/lgpl.html
  19  */
  20
  21 #ifdef _WIN32
  22 #ifdef __MINGW32__
  23 #define _WIN32_IE 0x501
  24 #else
  25 #define _WIN32_IE 0x400
  26 #endif
  27 #endif
  28
  29 #include "config.h"
  30
  31 #include <stdlib.h>
  32 #include <time.h>
  33 #include <errno.h>
  34 #include <stdarg.h>
  35 #ifdef HAVE_MALLOC_H
  36 #include <malloc.h>
  37 #endif
  38 #ifdef HAVE_DIRENT_H
  39 #include <dirent.h>
  40 #endif
  41
  42 #ifndef AL_NO_UID_DEFS
  43 #if defined(HAVE_GUIDDEF_H) || defined(HAVE_INITGUID_H)
  44 #define INITGUID
  45 #include <windows.h>
  46 #ifdef HAVE_GUIDDEF_H
  47 #include <guiddef.h>
  48 #else
  49 #include <initguid.h>
  50 #endif
  51
  52 DEFINE_GUID(KSDATAFORMAT_SUBTYPE_PCM,        0x00000001, 0x0000, 0x0010, 0x80,0x00, 0x00,0xaa,0x00,0x38,0x9b,0x71);
  53 DEFINE_GUID(KSDATAFORMAT_SUBTYPE_IEEE_FLOAT, 0x00000003, 0x0000, 0x0010, 0x80,0x00, 0x00,0xaa,0x00,0x38,0x9b,0x71);
  54
  55 DEFINE_GUID(IID_IDirectSoundNotify,   0xb0210783, 0x89cd, 0x11d0, 0xaf,0x08, 0x00,0xa0,0xc9,0x25,0xcd,0x16);
  56
  57 DEFINE_GUID(CLSID_MMDeviceEnumerator, 0xbcde0395, 0xe52f, 0x467c, 0x8e,0x3d, 0xc4,0x57,0x92,0x91,0x69,0x2e);
  58 DEFINE_GUID(IID_IMMDeviceEnumerator,  0xa95664d2, 0x9614, 0x4f35, 0xa7,0x46, 0xde,0x8d,0xb6,0x36,0x17,0xe6);
  59 DEFINE_GUID(IID_IAudioClient,         0x1cb9ad4c, 0xdbfa, 0x4c32, 0xb1,0x78, 0xc2,0xf5,0x68,0xa7,0x03,0xb2);
  60 DEFINE_GUID(IID_IAudioRenderClient,   0xf294acfc, 0x3146, 0x4483, 0xa7,0xbf, 0xad,0xdc,0xa7,0xc2,0x60,0xe2);
  61 DEFINE_GUID(IID_IAudioCaptureClient,  0xc8adbd64, 0xe71e, 0x48a0, 0xa4,0xde, 0x18,0x5c,0x39,0x5c,0xd3,0x17);
  62
  63 #ifdef HAVE_MMDEVAPI
  64 #include <devpropdef.h>
  65 #include <propkeydef.h>
  66 DEFINE_DEVPROPKEY(DEVPKEY_Device_FriendlyName, 0xa45c254e, 0xdf1c, 0x4efd, 0x80,0x20, 0x67,0xd1,0x46,0xa8,0x50,0xe0, 14);
  67 DEFINE_PROPERTYKEY(PKEY_AudioEndpoint_FormFactor, 0x1da5d803, 0xd492, 0x4edd, 0x8c,0x23, 0xe0,0xc0,0xff,0xee,0x7f,0x0e, 0);
  68 #endif
  69 #endif
  70 #endif /* AL_NO_UID_DEFS */
  71
  72 #ifdef HAVE_DLFCN_H
  73 #include <dlfcn.h>
  74 #endif
  75 #ifdef HAVE_INTRIN_H
  76 #include <intrin.h>
  77 #endif
  78 #ifdef HAVE_CPUID_H
  79 #include <cpuid.h>
  80 #endif
  81 #ifdef HAVE_SYS_SYSCONF_H
  82 #include <sys/sysconf.h>
  83 #endif
  84 #ifdef HAVE_FLOAT_H
  85 #include <float.h>
  86 #endif
  87 #ifdef HAVE_IEEEFP_H
  88 #include <ieeefp.h>
  89 #endif
  90
  91 #ifndef _WIN32
  92 #include <unistd.h>
  93 #elif defined(_WIN32_IE)
  94 #include <shlobj.h>
  95 #endif
  96
  97 #include "alMain.h"
  98 #include "alu.h"
  99 #include "atomic.h"
 100 #include "uintmap.h"
 101 #include "vector.h"
 102 #include "alstring.h"
 103 #include "compat.h"
 104 #include "threads.h"
 105
 106
 107 extern inline ALuint NextPowerOf2(ALuint value);
 108 extern inline ALint fastf2i(ALfloat f);
 109 extern inline ALuint fastf2u(ALfloat f);
 110
 111
 112 ALuint CPUCapFlags = 0;
 113
 114
 115 void FillCPUCaps(ALuint capfilter)
 116 {
 117     ALuint caps = 0;
 118
 119 /* FIXME: We really should get this for all available CPUs in case different
 120  * CPUs have different caps (is that possible on one machine?). */
 121 #if defined(HAVE_GCC_GET_CPUID) && (defined(__i386__) || defined(__x86_64__) || \
 122                                     defined(_M_IX86) || defined(_M_X64))
 123     union {
 124         unsigned int regs[4];
 125         char str[sizeof(unsigned int[4])];
 126     } cpuinf[3];
 127
 128     if(!__get_cpuid(0, &cpuinf[0].regs[0], &cpuinf[0].regs[1], &cpuinf[0].regs[2], &cpuinf[0].regs[3]))
 129         ERR("Failed to get CPUID\n");
 130     else
 131     {
 132         unsigned int maxfunc = cpuinf[0].regs[0];
 133         unsigned int maxextfunc = 0;
 134
 135         if(__get_cpuid(0x80000000, &cpuinf[0].regs[0], &cpuinf[0].regs[1], &cpuinf[0].regs[2], &cpuinf[0].regs[3]))
 136             maxextfunc = cpuinf[0].regs[0];
 137         TRACE("Detected max CPUID function: 0x%x (ext. 0x%x)\n", maxfunc, maxextfunc);
 138
 139         TRACE("Vendor ID: \"%.4s%.4s%.4s\"\n", cpuinf[0].str+4, cpuinf[0].str+12, cpuinf[0].str+8);
 140         if(maxextfunc >= 0x80000004 &&
 141            __get_cpuid(0x80000002, &cpuinf[0].regs[0], &cpuinf[0].regs[1], &cpuinf[0].regs[2], &cpuinf[0].regs[3]) &&
 142            __get_cpuid(0x80000003, &cpuinf[1].regs[0], &cpuinf[1].regs[1], &cpuinf[1].regs[2], &cpuinf[1].regs[3]) &&
 143            __get_cpuid(0x80000004, &cpuinf[2].regs[0], &cpuinf[2].regs[1], &cpuinf[2].regs[2], &cpuinf[2].regs[3]))
 144             TRACE("Name: \"%.16s%.16s%.16s\"\n", cpuinf[0].str, cpuinf[1].str, cpuinf[2].str);
 145
 146         if(maxfunc >= 1 &&
 147            __get_cpuid(1, &cpuinf[0].regs[0], &cpuinf[0].regs[1], &cpuinf[0].regs[2], &cpuinf[0].regs[3]))
 148         {
 149             if((cpuinf[0].regs[3]&(1<<25)))
 150             {
 151                 caps |= CPU_CAP_SSE;
 152                 if((cpuinf[0].regs[3]&(1<<26)))
 153                 {
 154                     caps |= CPU_CAP_SSE2;
 155                     if((cpuinf[0].regs[2]&(1<<0)))
 156                     {
 157                         caps |= CPU_CAP_SSE3;
 158                         if((cpuinf[0].regs[2]&(1<<19)))
 159                             caps |= CPU_CAP_SSE4_1;
 160                     }
 161                 }
 162             }
 163         }
 164     }
 165 #elif defined(HAVE_CPUID_INTRINSIC) && (defined(__i386__) || defined(__x86_64__) || \
 166                                         defined(_M_IX86) || defined(_M_X64))
 167     union {
 168         int regs[4];
 169         char str[sizeof(int[4])];
 170     } cpuinf[3];
 171
 172     (__cpuid)(cpuinf[0].regs, 0);
 173     if(cpuinf[0].regs[0] == 0)
 174         ERR("Failed to get CPUID\n");
 175     else
 176     {
 177         unsigned int maxfunc = cpuinf[0].regs[0];
 178         unsigned int maxextfunc;
 179
 180         (__cpuid)(cpuinf[0].regs, 0x80000000);
 181         maxextfunc = cpuinf[0].regs[0];
 182
 183         TRACE("Detected max CPUID function: 0x%x (ext. 0x%x)\n", maxfunc, maxextfunc);
 184
 185         TRACE("Vendor ID: \"%.4s%.4s%.4s\"\n", cpuinf[0].str+4, cpuinf[0].str+12, cpuinf[0].str+8);
 186         if(maxextfunc >= 0x80000004)
 187         {
 188             (__cpuid)(cpuinf[0].regs, 0x80000002);
 189             (__cpuid)(cpuinf[1].regs, 0x80000003);
 190             (__cpuid)(cpuinf[2].regs, 0x80000004);
 191             TRACE("Name: \"%.16s%.16s%.16s\"\n", cpuinf[0].str, cpuinf[1].str, cpuinf[2].str);
 192         }
 193
 194         if(maxfunc >= 1)
 195         {
 196             (__cpuid)(cpuinf[0].regs, 1);
 197             if((cpuinf[0].regs[3]&(1<<25)))
 198             {
 199                 caps |= CPU_CAP_SSE;
 200                 if((cpuinf[0].regs[3]&(1<<26)))
 201                 {
 202                     caps |= CPU_CAP_SSE2;
 203                     if((cpuinf[0].regs[2]&(1<<0)))
 204                     {
 205                         caps |= CPU_CAP_SSE3;
 206                         if((cpuinf[0].regs[2]&(1<<19)))
 207                             caps |= CPU_CAP_SSE4_1;
 208                     }
 209                 }
 210             }
 211         }
 212     }
 213 #else
 214     /* Assume support for whatever's supported if we can't check for it */
 215 #if defined(HAVE_SSE4_1)
 216 #warning "Assuming SSE 4.1 run-time support!"
 217     caps |= CPU_CAP_SSE | CPU_CAP_SSE2 | CPU_CAP_SSE3 | CPU_CAP_SSE4_1;
 218 #elif defined(HAVE_SSE3)
 219 #warning "Assuming SSE 3 run-time support!"
 220     caps |= CPU_CAP_SSE | CPU_CAP_SSE2 | CPU_CAP_SSE3;
 221 #elif defined(HAVE_SSE2)
 222 #warning "Assuming SSE 2 run-time support!"
 223     caps |= CPU_CAP_SSE | CPU_CAP_SSE2;
 224 #elif defined(HAVE_SSE)
 225 #warning "Assuming SSE run-time support!"
 226     caps |= CPU_CAP_SSE;
 227 #endif
 228 #endif
 229 #ifdef HAVE_NEON
 230     /* Assume Neon support if compiled with it */
 231     caps |= CPU_CAP_NEON;
 232 #endif
 233
 234     TRACE("Extensions:%s%s%s%s%s%s\n",
 235         ((capfilter&CPU_CAP_SSE)    ? ((caps&CPU_CAP_SSE)    ? " +SSE"    : " -SSE")    : ""),
 236         ((capfilter&CPU_CAP_SSE2)   ? ((caps&CPU_CAP_SSE2)   ? " +SSE2"   : " -SSE2")   : ""),
 237         ((capfilter&CPU_CAP_SSE3)   ? ((caps&CPU_CAP_SSE3)   ? " +SSE3"   : " -SSE3")   : ""),
 238         ((capfilter&CPU_CAP_SSE4_1) ? ((caps&CPU_CAP_SSE4_1) ? " +SSE4.1" : " -SSE4.1") : ""),
 239         ((capfilter&CPU_CAP_NEON)   ? ((caps&CPU_CAP_NEON)   ? " +Neon"   : " -Neon")   : ""),
 240         ((!capfilter) ? " -none-" : "")
 241     );
 242     CPUCapFlags = caps & capfilter;
 243 }
 244
 245
 246 void SetMixerFPUMode(FPUCtl *ctl)
 247 {
 248 #ifdef HAVE_FENV_H
 249     fegetenv(STATIC_CAST(fenv_t, ctl));
 250 #if defined(__GNUC__) && defined(HAVE_SSE)
 251     /* FIXME: Some fegetenv implementations can get the SSE environment too?
 252      * How to tell when it does? */
 253     if((CPUCapFlags&CPU_CAP_SSE))
 254         __asm__ __volatile__("stmxcsr %0" : "=m" (*&ctl->sse_state));
 255 #endif
 256
 257 #ifdef FE_TOWARDZERO
 258     fesetround(FE_TOWARDZERO);
 259 #endif
 260 #if defined(__GNUC__) && defined(HAVE_SSE)
 261     if((CPUCapFlags&CPU_CAP_SSE))
 262     {
 263         int sseState = ctl->sse_state;
 264         sseState |= 0x6000; /* set round-to-zero */
 265         sseState |= 0x8000; /* set flush-to-zero */
 266         if((CPUCapFlags&CPU_CAP_SSE2))
 267             sseState |= 0x0040; /* set denormals-are-zero */
 268         __asm__ __volatile__("ldmxcsr %0" : : "m" (*&sseState));
 269     }
 270 #endif
 271
 272 #elif defined(HAVE___CONTROL87_2)
 273
 274     int mode;
 275     __control87_2(0, 0, &ctl->state, NULL);
 276     __control87_2(_RC_CHOP, _MCW_RC, &mode, NULL);
 277 #ifdef HAVE_SSE
 278     if((CPUCapFlags&CPU_CAP_SSE))
 279     {
 280         __control87_2(0, 0, NULL, &ctl->sse_state);
 281         __control87_2(_RC_CHOP|_DN_FLUSH, _MCW_RC|_MCW_DN, NULL, &mode);
 282     }
 283 #endif
 284
 285 #elif defined(HAVE__CONTROLFP)
 286
 287     ctl->state = _controlfp(0, 0);
 288     (void)_controlfp(_RC_CHOP, _MCW_RC);
 289 #endif
 290 }
 291
 292 void RestoreFPUMode(const FPUCtl *ctl)
 293 {
 294 #ifdef HAVE_FENV_H
 295     fesetenv(STATIC_CAST(fenv_t, ctl));
 296 #if defined(__GNUC__) && defined(HAVE_SSE)
 297     if((CPUCapFlags&CPU_CAP_SSE))
 298         __asm__ __volatile__("ldmxcsr %0" : : "m" (*&ctl->sse_state));
 299 #endif
 300
 301 #elif defined(HAVE___CONTROL87_2)
 302
 303     int mode;
 304     __control87_2(ctl->state, _MCW_RC, &mode, NULL);
 305 #ifdef HAVE_SSE
 306     if((CPUCapFlags&CPU_CAP_SSE))
 307         __control87_2(ctl->sse_state, _MCW_RC|_MCW_DN, NULL, &mode);
 308 #endif
 309
 310 #elif defined(HAVE__CONTROLFP)
 311
 312     _controlfp(ctl->state, _MCW_RC);
 313 #endif
 314 }
 315
 316
 317 static int StringSortCompare(const void *str1, const void *str2)
 318 {
 319     return al_string_cmp(*(const_al_string*)str1, *(const_al_string*)str2);
 320 }
 321
 322 #ifdef _WIN32
 323
 324 static WCHAR *strrchrW(WCHAR *str, WCHAR ch)
 325 {
 326     WCHAR *ret = NULL;
 327     while(*str)
 328     {
 329         if(*str == ch)
 330             ret = str;
 331         ++str;
 332     }
 333     return ret;
 334 }
 335
 336 al_string GetProcPath(void)
 337 {
 338     al_string ret = AL_STRING_INIT_STATIC();
 339     WCHAR *pathname, *sep;
 340     DWORD pathlen;
 341     DWORD len;
 342
 343     pathlen = 256;
 344     pathname = malloc(pathlen * sizeof(pathname[0]));
 345     while(pathlen > 0 && (len=GetModuleFileNameW(NULL, pathname, pathlen)) == pathlen)
 346     {
 347         free(pathname);
 348         pathlen <<= 1;
 349         pathname = malloc(pathlen * sizeof(pathname[0]));
 350     }
 351     if(len == 0)
 352     {
 353         free(pathname);
 354         ERR("Failed to get process name: error %lu\n", GetLastError());
 355         return ret;
 356     }
 357
 358     pathname[len] = 0;
 359     if((sep = strrchrW(pathname, '\\')))
 360     {
 361         WCHAR *sep2 = strrchrW(pathname, '/');
 362         if(sep2) *sep2 = 0;
 363         else *sep = 0;
 364     }
 365     else if((sep = strrchrW(pathname, '/')))
 366         *sep = 0;
 367     al_string_copy_wcstr(&ret, pathname);
 368     free(pathname);
 369
 370     TRACE("Got: %s\n", al_string_get_cstr(ret));
 371     return ret;
 372 }
 373
 374
 375 static WCHAR *FromUTF8(const char *str)
 376 {
 377     WCHAR *out = NULL;
 378     int len;
 379
 380     if((len=MultiByteToWideChar(CP_UTF8, 0, str, -1, NULL, 0)) > 0)
 381     {
 382         out = calloc(sizeof(WCHAR), len);
 383         MultiByteToWideChar(CP_UTF8, 0, str, -1, out, len);
 384     }
 385     return out;
 386 }
 387
 388
 389 void *LoadLib(const char *name)
 390 {
 391     HANDLE hdl = NULL;
 392     WCHAR *wname;
 393
 394     wname = FromUTF8(name);
 395     if(!wname)
 396         ERR("Failed to convert UTF-8 filename: \"%s\"\n", name);
 397     else
 398     {
 399         hdl = LoadLibraryW(wname);
 400         free(wname);
 401     }
 402     return hdl;
 403 }
 404 void CloseLib(void *handle)
 405 { FreeLibrary((HANDLE)handle); }
 406 void *GetSymbol(void *handle, const char *name)
 407 {
 408     void *ret;
 409
 410     ret = (void*)GetProcAddress((HANDLE)handle, name);
 411     if(ret == NULL)
 412         ERR("Failed to load %s\n", name);
 413     return ret;
 414 }
 415
 416 WCHAR *strdupW(const WCHAR *str)
 417 {
 418     const WCHAR *n;
 419     WCHAR *ret;
 420     size_t len;
 421
 422     n = str;
 423     while(*n) n++;
 424     len = n - str;
 425
 426     ret = calloc(sizeof(WCHAR), len+1);
 427     if(ret != NULL)
 428         memcpy(ret, str, sizeof(WCHAR)*len);
 429     return ret;
 430 }
 431
 432 FILE *al_fopen(const char *fname, const char *mode)
 433 {
 434     WCHAR *wname=NULL, *wmode=NULL;
 435     FILE *file = NULL;
 436
 437     wname = FromUTF8(fname);
 438     wmode = FromUTF8(mode);
 439     if(!wname)
 440         ERR("Failed to convert UTF-8 filename: \"%s\"\n", fname);
 441     else if(!wmode)
 442         ERR("Failed to convert UTF-8 mode: \"%s\"\n", mode);
 443     else
 444         file = _wfopen(wname, wmode);
 445
 446     free(wname);
 447     free(wmode);
 448
 449     return file;
 450 }
 451
 452
 453 void al_print(const char *type, const char *func, const char *fmt, ...)
 454 {
 455     char str[1024];
 456     WCHAR *wstr;
 457     va_list ap;
 458
 459     va_start(ap, fmt);
 460     vsnprintf(str, sizeof(str), fmt, ap);
 461     va_end(ap);
 462
 463     str[sizeof(str)-1] = 0;
 464     wstr = FromUTF8(str);
 465     if(!wstr)
 466         fprintf(LogFile, "AL lib: %s %s: <UTF-8 error> %s", type, func, str);
 467     else
 468     {
 469         fprintf(LogFile, "AL lib: %s %s: %ls", type, func, wstr);
 470         free(wstr);
 471         wstr = NULL;
 472     }
 473     fflush(LogFile);
 474 }
 475
 476
 477 static inline int is_slash(int c)
 478 { return (c == '\\' || c == '/'); }
 479
 480 static void DirectorySearch(const char *path, const char *ext, vector_al_string *results)
 481 {
 482     al_string pathstr = AL_STRING_INIT_STATIC();
 483     WIN32_FIND_DATAW fdata;
 484     WCHAR *wpath;
 485     HANDLE hdl;
 486
 487     al_string_copy_cstr(&pathstr, path);
 488     al_string_append_cstr(&pathstr, "\\*");
 489     al_string_append_cstr(&pathstr, ext);
 490
 491     TRACE("Searching %s\n", al_string_get_cstr(pathstr));
 492
 493     wpath = FromUTF8(al_string_get_cstr(pathstr));
 494
 495     hdl = FindFirstFileW(wpath, &fdata);
 496     if(hdl != INVALID_HANDLE_VALUE)
 497     {
 498         size_t base = VECTOR_SIZE(*results);
 499         do {
 500             al_string str = AL_STRING_INIT_STATIC();
 501             al_string_copy_cstr(&str, path);
 502             al_string_append_char(&str, '\\');
 503             al_string_append_wcstr(&str, fdata.cFileName);
 504             TRACE("Got result %s\n", al_string_get_cstr(str));
 505             VECTOR_PUSH_BACK(*results, str);
 506         } while(FindNextFileW(hdl, &fdata));
 507         FindClose(hdl);
 508
 509         if(VECTOR_SIZE(*results) > base)
 510             qsort(VECTOR_BEGIN(*results)+base, VECTOR_SIZE(*results)-base,
 511                     sizeof(VECTOR_FRONT(*results)), StringSortCompare);
 512     }
 513
 514     free(wpath);
 515     al_string_deinit(&pathstr);
 516 }
 517
 518 vector_al_string SearchDataFiles(const char *ext, const char *subdir)
 519 {
 520     static const int ids[2] = { CSIDL_APPDATA, CSIDL_COMMON_APPDATA };
 521     static RefCount search_lock;
 522     vector_al_string results = VECTOR_INIT_STATIC();
 523     size_t i;
 524
 525     while(ATOMIC_EXCHANGE(uint, &search_lock, 1) == 1)
 526         althrd_yield();
 527
 528     /* If the path is absolute, use it directly. */
 529     if(isalpha(subdir[0]) && subdir[1] == ':' && is_slash(subdir[2]))
 530     {
 531         al_string path = AL_STRING_INIT_STATIC();
 532         al_string_copy_cstr(&path, subdir);
 533 #define FIX_SLASH(i) do { if(*(i) == '/') *(i) = '\\'; } while(0)
 534         VECTOR_FOR_EACH(char, path, FIX_SLASH);
 535 #undef FIX_SLASH
 536
 537         DirectorySearch(al_string_get_cstr(path), ext, &results);
 538
 539         al_string_deinit(&path);
 540     }
 541     else if(subdir[0] == '\\' && subdir[1] == '\\' && subdir[2] == '?' && subdir[3] == '\\')
 542         DirectorySearch(subdir, ext, &results);
 543     else
 544     {
 545         al_string path = AL_STRING_INIT_STATIC();
 546         WCHAR *cwdbuf;
 547
 548         /* Search the app-local directory. */
 549         if((cwdbuf=_wgetenv(L"ALSOFT_LOCAL_PATH")) && *cwdbuf != '\0')
 550         {
 551             al_string_copy_wcstr(&path, cwdbuf);
 552             if(is_slash(VECTOR_BACK(path)))
 553             {
 554                 VECTOR_POP_BACK(path);
 555                 *VECTOR_END(path) = 0;
 556             }
 557         }
 558         else if(!(cwdbuf=_wgetcwd(NULL, 0)))
 559             al_string_copy_cstr(&path, ".");
 560         else
 561         {
 562             al_string_copy_wcstr(&path, cwdbuf);
 563             if(is_slash(VECTOR_BACK(path)))
 564             {
 565                 VECTOR_POP_BACK(path);
 566                 *VECTOR_END(path) = 0;
 567             }
 568             free(cwdbuf);
 569         }
 570 #define FIX_SLASH(i) do { if(*(i) == '/') *(i) = '\\'; } while(0)
 571         VECTOR_FOR_EACH(char, path, FIX_SLASH);
 572 #undef FIX_SLASH
 573         DirectorySearch(al_string_get_cstr(path), ext, &results);
 574
 575         /* Search the local and global data dirs. */
 576         for(i = 0;i < COUNTOF(ids);i++)
 577         {
 578             WCHAR buffer[PATH_MAX];
 579             if(SHGetSpecialFolderPathW(NULL, buffer, ids[i], FALSE) != FALSE)
 580             {
 581                 al_string_copy_wcstr(&path, buffer);
 582                 if(!is_slash(VECTOR_BACK(path)))
 583                     al_string_append_char(&path, '\\');
 584                 al_string_append_cstr(&path, subdir);
 585 #define FIX_SLASH(i) do { if(*(i) == '/') *(i) = '\\'; } while(0)
 586                 VECTOR_FOR_EACH(char, path, FIX_SLASH);
 587 #undef FIX_SLASH
 588
 589                 DirectorySearch(al_string_get_cstr(path), ext, &results);
 590             }
 591         }
 592
 593         al_string_deinit(&path);
 594     }
 595
 596     ATOMIC_STORE(&search_lock, 0);
 597
 598     return results;
 599 }
 600
 601 #else
 602
 603 al_string GetProcPath(void)
 604 {
 605     al_string ret = AL_STRING_INIT_STATIC();
 606     const char *fname;
 607     char *pathname, *sep;
 608     size_t pathlen;
 609     ssize_t len;
 610
 611     pathlen = 256;
 612     pathname = malloc(pathlen);
 613
 614     fname = "/proc/self/exe";
 615     len = readlink(fname, pathname, pathlen);
 616     if(len == -1 && errno == ENOENT)
 617     {
 618         fname = "/proc/self/file";
 619         len = readlink(fname, pathname, pathlen);
 620     }
 621
 622     while(len > 0 && (size_t)len == pathlen)
 623     {
 624         free(pathname);
 625         pathlen <<= 1;
 626         pathname = malloc(pathlen);
 627         len = readlink(fname, pathname, pathlen);
 628     }
 629     if(len <= 0)
 630     {
 631         free(pathname);
 632         ERR("Failed to link %s: %s\n", fname, strerror(errno));
 633         return ret;
 634     }
 635
 636     pathname[len] = 0;
 637     sep = strrchr(pathname, '/');
 638     if(sep)
 639         al_string_copy_range(&ret, pathname, sep);
 640     else
 641         al_string_copy_cstr(&ret, pathname);
 642     free(pathname);
 643
 644     TRACE("Got: %s\n", al_string_get_cstr(ret));
 645     return ret;
 646 }
 647
 648
 649 #ifdef HAVE_DLFCN_H
 650
 651 void *LoadLib(const char *name)
 652 {
 653     const char *err;
 654     void *handle;
 655
 656     dlerror();
 657     handle = dlopen(name, RTLD_NOW);
 658     if((err=dlerror()) != NULL)
 659         handle = NULL;
 660     return handle;
 661 }
 662 void CloseLib(void *handle)
 663 { dlclose(handle); }
 664 void *GetSymbol(void *handle, const char *name)
 665 {
 666     const char *err;
 667     void *sym;
 668
 669     dlerror();
 670     sym = dlsym(handle, name);
 671     if((err=dlerror()) != NULL)
 672     {
 673         WARN("Failed to load %s: %s\n", name, err);
 674         sym = NULL;
 675     }
 676     return sym;
 677 }
 678
 679 #endif /* HAVE_DLFCN_H */
 680
 681 void al_print(const char *type, const char *func, const char *fmt, ...)
 682 {
 683     va_list ap;
 684
 685     va_start(ap, fmt);
 686     fprintf(LogFile, "AL lib: %s %s: ", type, func);
 687     vfprintf(LogFile, fmt, ap);
 688     va_end(ap);
 689
 690     fflush(LogFile);
 691 }
 692
 693
 694 static void DirectorySearch(const char *path, const char *ext, vector_al_string *results)
 695 {
 696     size_t extlen = strlen(ext);
 697     DIR *dir;
 698
 699     TRACE("Searching %s for *%s\n", path, ext);
 700     dir = opendir(path);
 701     if(dir != NULL)
 702     {
 703         size_t base = VECTOR_SIZE(*results);
 704         struct dirent *dirent;
 705         while((dirent=readdir(dir)) != NULL)
 706         {
 707             al_string str;
 708             size_t len;
 709             if(strcmp(dirent->d_name, ".") == 0 || strcmp(dirent->d_name, "..") == 0)
 710                 continue;
 711
 712             len = strlen(dirent->d_name);
 713             if(!(len > extlen))
 714                 continue;
 715             if(strcasecmp(dirent->d_name+len-extlen, ext) != 0)
 716                 continue;
 717
 718             AL_STRING_INIT(str);
 719             al_string_copy_cstr(&str, path);
 720             if(VECTOR_BACK(str) != '/')
 721                 al_string_append_char(&str, '/');
 722             al_string_append_cstr(&str, dirent->d_name);
 723             TRACE("Got result %s\n", al_string_get_cstr(str));
 724             VECTOR_PUSH_BACK(*results, str);
 725         }
 726         closedir(dir);
 727
 728         if(VECTOR_SIZE(*results) > base)
 729             qsort(VECTOR_BEGIN(*results)+base, VECTOR_SIZE(*results)-base,
 730                   sizeof(VECTOR_FRONT(*results)), StringSortCompare);
 731     }
 732 }
 733
 734 vector_al_string SearchDataFiles(const char *ext, const char *subdir)
 735 {
 736     static RefCount search_lock;
 737     vector_al_string results = VECTOR_INIT_STATIC();
 738
 739     while(ATOMIC_EXCHANGE(uint, &search_lock, 1) == 1)
 740         althrd_yield();
 741
 742     if(subdir[0] == '/')
 743         DirectorySearch(subdir, ext, &results);
 744     else
 745     {
 746         al_string path = AL_STRING_INIT_STATIC();
 747         const char *str, *next;
 748         char cwdbuf[PATH_MAX];
 749
 750         /* Search the app-local directory. */
 751         if((str=getenv("ALSOFT_LOCAL_PATH")) && *str != '\0')
 752             DirectorySearch(str, ext, &results);
 753         else if(getcwd(cwdbuf, sizeof(cwdbuf)))
 754             DirectorySearch(cwdbuf, ext, &results);
 755         else
 756             DirectorySearch(".", ext, &results);
 757
 758         // Search local data dir
 759         if((str=getenv("XDG_DATA_HOME")) != NULL && str[0] != '\0')
 760         {
 761             al_string_copy_cstr(&path, str);
 762             if(VECTOR_BACK(path) != '/')
 763                 al_string_append_char(&path, '/');
 764             al_string_append_cstr(&path, subdir);
 765             DirectorySearch(al_string_get_cstr(path), ext, &results);
 766         }
 767         else if((str=getenv("HOME")) != NULL && str[0] != '\0')
 768         {
 769             al_string_copy_cstr(&path, str);
 770             if(VECTOR_BACK(path) == '/')
 771             {
 772                 VECTOR_POP_BACK(path);
 773                 *VECTOR_END(path) = 0;
 774             }
 775             al_string_append_cstr(&path, "/.local/share/");
 776             al_string_append_cstr(&path, subdir);
 777             DirectorySearch(al_string_get_cstr(path), ext, &results);
 778         }
 779
 780         // Search global data dirs
 781         if((str=getenv("XDG_DATA_DIRS")) == NULL || str[0] == '\0')
 782             str = "/usr/local/share/:/usr/share/";
 783
 784         next = str;
 785         while((str=next) != NULL && str[0] != '\0')
 786         {
 787             next = strchr(str, ':');
 788             if(!next)
 789                 al_string_copy_cstr(&path, str);
 790             else
 791             {
 792                 al_string_copy_range(&path, str, next);
 793                 ++next;
 794             }
 795             if(!al_string_empty(path))
 796             {
 797                 if(VECTOR_BACK(path) != '/')
 798                     al_string_append_char(&path, '/');
 799                 al_string_append_cstr(&path, subdir);
 800
 801                 DirectorySearch(al_string_get_cstr(path), ext, &results);
 802             }
 803         }
 804
 805         al_string_deinit(&path);
 806     }
 807
 808     ATOMIC_STORE(&search_lock, 0);
 809
 810     return results;
 811 }
 812
 813 #endif
 814
 815
 816 void SetRTPriority(void)
 817 {
 818     ALboolean failed = AL_FALSE;
 819
 820 #ifdef _WIN32
 821     if(RTPrioLevel > 0)
 822         failed = !SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_TIME_CRITICAL);
 823 #elif defined(HAVE_PTHREAD_SETSCHEDPARAM) && !defined(__OpenBSD__)
 824     if(RTPrioLevel > 0)
 825     {
 826         struct sched_param param;
 827         /* Use the minimum real-time priority possible for now (on Linux this
 828          * should be 1 for SCHED_RR) */
 829         param.sched_priority = sched_get_priority_min(SCHED_RR);
 830         failed = !!pthread_setschedparam(pthread_self(), SCHED_RR, &param);
 831     }
 832 #else
 833     /* Real-time priority not available */
 834     failed = (RTPrioLevel>0);
 835 #endif
 836     if(failed)
 837         ERR("Failed to set priority level for thread\n");
 838 }
 839
 840
 841 ALboolean vector_reserve(char *ptr, size_t base_size, size_t obj_size, size_t obj_count, ALboolean exact)
 842 {
 843     vector_ *vecptr = (vector_*)ptr;
 844     if((*vecptr ? (*vecptr)->Capacity : 0) < obj_count)
 845     {
 846         size_t old_size = (*vecptr ? (*vecptr)->Size : 0);
 847         void *temp;
 848
 849         /* Use the next power-of-2 size if we don't need to allocate the exact
 850          * amount. This is preferred when regularly increasing the vector since
 851          * it means fewer reallocations. Though it means it also wastes some
 852          * memory. */
 853         if(exact == AL_FALSE && obj_count < INT_MAX)
 854             obj_count = NextPowerOf2((ALuint)obj_count);
 855
 856         /* Need to be explicit with the caller type's base size, because it
 857          * could have extra padding before the start of the array (that is,
 858          * sizeof(*vector_) may not equal base_size). */
 859         temp = al_calloc(16, base_size + obj_size*obj_count);
 860         if(temp == NULL) return AL_FALSE;
 861         memcpy(((ALubyte*)temp)+base_size, ((ALubyte*)*vecptr)+base_size,
 862                obj_size*old_size);
 863
 864         al_free(*vecptr);
 865         *vecptr = temp;
 866         (*vecptr)->Capacity = obj_count;
 867         (*vecptr)->Size = old_size;
 868     }
 869     return AL_TRUE;
 870 }
 871
 872 ALboolean vector_resize(char *ptr, size_t base_size, size_t obj_size, size_t obj_count)
 873 {
 874     vector_ *vecptr = (vector_*)ptr;
 875     if(*vecptr || obj_count > 0)
 876     {
 877         if(!vector_reserve((char*)vecptr, base_size, obj_size, obj_count, AL_TRUE))
 878             return AL_FALSE;
 879         (*vecptr)->Size = obj_count;
 880     }
 881     return AL_TRUE;
 882 }
 883
 884
 885 extern inline void al_string_deinit(al_string *str);
 886 extern inline size_t al_string_length(const_al_string str);
 887 extern inline ALboolean al_string_empty(const_al_string str);
 888 extern inline const al_string_char_type *al_string_get_cstr(const_al_string str);
 889
 890 void al_string_clear(al_string *str)
 891 {
 892     if(!al_string_empty(*str))
 893     {
 894         /* Reserve one more character than the total size of the string. This
 895          * is to ensure we have space to add a null terminator in the string
 896          * data so it can be used as a C-style string.
 897          */
 898         VECTOR_RESERVE(*str, 1);
 899         VECTOR_RESIZE(*str, 0);
 900         *VECTOR_END(*str) = 0;
 901     }
 902 }
 903
 904 static inline int al_string_compare(const al_string_char_type *str1, size_t str1len,
 905                                     const al_string_char_type *str2, size_t str2len)
 906 {
 907     size_t complen = (str1len < str2len) ? str1len : str2len;
 908     int ret = memcmp(str1, str2, complen);
 909     if(ret == 0)
 910     {
 911         if(str1len > str2len) return  1;
 912         if(str1len < str2len) return -1;
 913     }
 914     return ret;
 915 }
 916 int al_string_cmp(const_al_string str1, const_al_string str2)
 917 {
 918     return al_string_compare(&VECTOR_FRONT(str1), al_string_length(str1),
 919                              &VECTOR_FRONT(str2), al_string_length(str2));
 920 }
 921 int al_string_cmp_cstr(const_al_string str1, const al_string_char_type *str2)
 922 {
 923     return al_string_compare(&VECTOR_FRONT(str1), al_string_length(str1),
 924                              str2, strlen(str2));
 925 }
 926
 927 void al_string_copy(al_string *str, const_al_string from)
 928 {
 929     size_t len = al_string_length(from);
 930     size_t i;
 931
 932     VECTOR_RESERVE(*str, len+1);
 933     VECTOR_RESIZE(*str, len);
 934     for(i = 0;i < len;i++)
 935         VECTOR_ELEM(*str, i) = VECTOR_ELEM(from, i);
 936     *VECTOR_END(*str) = 0;
 937 }
 938
 939 void al_string_copy_cstr(al_string *str, const al_string_char_type *from)
 940 {
 941     size_t len = strlen(from);
 942     size_t i;
 943
 944     VECTOR_RESERVE(*str, len+1);
 945     VECTOR_RESIZE(*str, len);
 946     for(i = 0;i < len;i++)
 947         VECTOR_ELEM(*str, i) = from[i];
 948     *VECTOR_END(*str) = 0;
 949 }
 950
 951 void al_string_copy_range(al_string *str, const al_string_char_type *from, const al_string_char_type *to)
 952 {
 953     size_t len = to - from;
 954     size_t i;
 955
 956     VECTOR_RESERVE(*str, len+1);
 957     VECTOR_RESIZE(*str, len);
 958     for(i = 0;i < len;i++)
 959         VECTOR_ELEM(*str, i) = from[i];
 960     *VECTOR_END(*str) = 0;
 961 }
 962
 963 void al_string_append_char(al_string *str, const al_string_char_type c)
 964 {
 965     VECTOR_RESERVE(*str, al_string_length(*str)+2);
 966     VECTOR_PUSH_BACK(*str, c);
 967     *VECTOR_END(*str) = 0;
 968 }
 969
 970 void al_string_append_cstr(al_string *str, const al_string_char_type *from)
 971 {
 972     size_t len = strlen(from);
 973     if(len != 0)
 974     {
 975         size_t base = al_string_length(*str);
 976         size_t i;
 977
 978         VECTOR_RESERVE(*str, base+len+1);
 979         VECTOR_RESIZE(*str, base+len);
 980         for(i = 0;i < len;i++)
 981             VECTOR_ELEM(*str, base+i) = from[i];
 982         *VECTOR_END(*str) = 0;
 983     }
 984 }
 985
 986 void al_string_append_range(al_string *str, const al_string_char_type *from, const al_string_char_type *to)
 987 {
 988     if(to != from)
 989     {
 990         size_t base = al_string_length(*str);
 991         size_t len = to - from;
 992         size_t i;
 993
 994         VECTOR_RESERVE(*str, base+len+1);
 995         VECTOR_RESIZE(*str, base+len);
 996         for(i = 0;i < len;i++)
 997             VECTOR_ELEM(*str, base+i) = from[i];
 998         *VECTOR_END(*str) = 0;
 999     }
1000 }
1001
1002 #ifdef _WIN32
1003 void al_string_copy_wcstr(al_string *str, const wchar_t *from)
1004 {
1005     int len;
1006     if((len=WideCharToMultiByte(CP_UTF8, 0, from, -1, NULL, 0, NULL, NULL)) > 0)
1007     {
1008         VECTOR_RESERVE(*str, len);
1009         VECTOR_RESIZE(*str, len-1);
1010         WideCharToMultiByte(CP_UTF8, 0, from, -1, &VECTOR_FRONT(*str), len, NULL, NULL);
1011         *VECTOR_END(*str) = 0;
1012     }
1013 }
1014
1015 void al_string_append_wcstr(al_string *str, const wchar_t *from)
1016 {
1017     int len;
1018     if((len=WideCharToMultiByte(CP_UTF8, 0, from, -1, NULL, 0, NULL, NULL)) > 0)
1019     {
1020         size_t strlen = al_string_length(*str);
1021         VECTOR_RESERVE(*str, strlen+len);
1022         VECTOR_RESIZE(*str, strlen+len-1);
1023         WideCharToMultiByte(CP_UTF8, 0, from, -1, &VECTOR_FRONT(*str) + strlen, len, NULL, NULL);
1024         *VECTOR_END(*str) = 0;
1025     }
1026 }
1027
1028 void al_string_append_wrange(al_string *str, const wchar_t *from, const wchar_t *to)
1029 {
1030     int len;
1031     if((len=WideCharToMultiByte(CP_UTF8, 0, from, (int)(to-from), NULL, 0, NULL, NULL)) > 0)
1032     {
1033         size_t strlen = al_string_length(*str);
1034         VECTOR_RESERVE(*str, strlen+len+1);
1035         VECTOR_RESIZE(*str, strlen+len);
1036         WideCharToMultiByte(CP_UTF8, 0, from, (int)(to-from), &VECTOR_FRONT(*str) + strlen, len+1, NULL, NULL);
1037         *VECTOR_END(*str) = 0;
1038     }
1039 }
1040 #endif