Merge branch 'tr/submodule-relative-scp-url'
[git] / thread-utils.c
1 #include "cache.h"
2 #include <pthread.h>
3
4 #if defined(hpux) || defined(__hpux) || defined(_hpux)
5 #  include <sys/pstat.h>
6 #endif
7
8 /*
9  * By doing this in two steps we can at least get
10  * the function to be somewhat coherent, even
11  * with this disgusting nest of #ifdefs.
12  */
13 #ifndef _SC_NPROCESSORS_ONLN
14 #  ifdef _SC_NPROC_ONLN
15 #    define _SC_NPROCESSORS_ONLN _SC_NPROC_ONLN
16 #  elif defined _SC_CRAY_NCPU
17 #    define _SC_NPROCESSORS_ONLN _SC_CRAY_NCPU
18 #  endif
19 #endif
20
21 int online_cpus(void)
22 {
23 #ifdef _SC_NPROCESSORS_ONLN
24         long ncpus;
25 #endif
26
27 #ifdef _WIN32
28         SYSTEM_INFO info;
29         GetSystemInfo(&info);
30
31         if ((int)info.dwNumberOfProcessors > 0)
32                 return (int)info.dwNumberOfProcessors;
33 #elif defined(hpux) || defined(__hpux) || defined(_hpux)
34         struct pst_dynamic psd;
35
36         if (!pstat_getdynamic(&psd, sizeof(psd), (size_t)1, 0))
37                 return (int)psd.psd_proc_cnt;
38 #endif
39
40 #ifdef _SC_NPROCESSORS_ONLN
41         if ((ncpus = (long)sysconf(_SC_NPROCESSORS_ONLN)) > 0)
42                 return (int)ncpus;
43 #endif
44
45         return 1;
46 }
47
48 int init_recursive_mutex(pthread_mutex_t *m)
49 {
50         pthread_mutexattr_t a;
51         int ret;
52
53         ret = pthread_mutexattr_init(&a);
54         if (!ret) {
55                 ret = pthread_mutexattr_settype(&a, PTHREAD_MUTEX_RECURSIVE);
56                 if (!ret)
57                         ret = pthread_mutex_init(m, &a);
58                 pthread_mutexattr_destroy(&a);
59         }
60         return ret;
61 }