Index: third_party/psutil/psutil/_psutil_osx.c |
diff --git a/third_party/psutil/psutil/_psutil_osx.c b/third_party/psutil/psutil/_psutil_osx.c |
deleted file mode 100644 |
index 29162f20b6f757eef9e692a2b8dfa83cc4c3a7a7..0000000000000000000000000000000000000000 |
--- a/third_party/psutil/psutil/_psutil_osx.c |
+++ /dev/null |
@@ -1,1430 +0,0 @@ |
-/* |
- * $Id: _psutil_osx.c 1142 2011-10-05 18:45:49Z g.rodola $ |
- * |
- * Copyright (c) 2009, Jay Loden, Giampaolo Rodola'. All rights reserved. |
- * Use of this source code is governed by a BSD-style license that can be |
- * found in the LICENSE file. |
- * |
- * OS X platform-specific module methods for _psutil_osx |
- */ |
- |
-#include <Python.h> |
-#include <assert.h> |
-#include <errno.h> |
-#include <stdbool.h> |
-#include <stdlib.h> |
-#include <stdio.h> |
-#include <sys/sysctl.h> |
-#include <sys/vmmeter.h> |
-#include <libproc.h> |
-#include <sys/proc_info.h> |
-#include <netinet/tcp_fsm.h> |
-#include <arpa/inet.h> |
-#include <net/if_dl.h> |
- |
-#include <mach/mach.h> |
-#include <mach/task.h> |
-#include <mach/mach_init.h> |
-#include <mach/host_info.h> |
-#include <mach/mach_host.h> |
-#include <mach/mach_traps.h> |
-#include <mach/shared_memory_server.h> |
- |
-#include <CoreFoundation/CoreFoundation.h> |
-#include <IOKit/IOKitLib.h> |
-#include <IOKit/storage/IOBlockStorageDriver.h> |
-#include <IOKit/storage/IOMedia.h> |
-#include <IOKit/IOBSD.h> |
- |
-#include "_psutil_osx.h" |
-#include "_psutil_common.h" |
-#include "arch/osx/process_info.h" |
- |
- |
-/* |
- * Return a Python list of all the PIDs running on the system. |
- */ |
-static PyObject* |
-get_pid_list(PyObject* self, PyObject* args) |
-{ |
- kinfo_proc *proclist = NULL; |
- kinfo_proc *orig_address = NULL; |
- size_t num_processes; |
- size_t idx; |
- PyObject *pid; |
- PyObject *retlist = PyList_New(0); |
- |
- if (get_proc_list(&proclist, &num_processes) != 0) { |
- Py_DECREF(retlist); |
- PyErr_SetString(PyExc_RuntimeError, "failed to retrieve process list."); |
- return NULL; |
- } |
- |
- if (num_processes > 0) { |
- // save the address of proclist so we can free it later |
- orig_address = proclist; |
- for (idx=0; idx < num_processes; idx++) { |
- pid = Py_BuildValue("i", proclist->kp_proc.p_pid); |
- PyList_Append(retlist, pid); |
- Py_XDECREF(pid); |
- proclist++; |
- } |
- } |
- free(orig_address); |
- return retlist; |
-} |
- |
- |
-/* |
- * Return process name from kinfo_proc as a Python string. |
- */ |
-static PyObject* |
-get_process_name(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- struct kinfo_proc kp; |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- if (get_kinfo_proc(pid, &kp) == -1) { |
- return NULL; |
- } |
- return Py_BuildValue("s", kp.kp_proc.p_comm); |
-} |
- |
- |
-/* |
- * Return process cmdline as a Python list of cmdline arguments. |
- */ |
-static PyObject* |
-get_process_cmdline(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- PyObject* arglist = NULL; |
- |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- |
- // get the commandline, defined in arch/osx/process_info.c |
- arglist = get_arg_list(pid); |
- return arglist; |
-} |
- |
- |
-/* |
- * Return process parent pid from kinfo_proc as a Python integer. |
- */ |
-static PyObject* |
-get_process_ppid(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- struct kinfo_proc kp; |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- if (get_kinfo_proc(pid, &kp) == -1) { |
- return NULL; |
- } |
- return Py_BuildValue("l", (long)kp.kp_eproc.e_ppid); |
-} |
- |
- |
-/* |
- * Return process real uid from kinfo_proc as a Python integer. |
- */ |
-static PyObject* |
-get_process_uids(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- struct kinfo_proc kp; |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- if (get_kinfo_proc(pid, &kp) == -1) { |
- return NULL; |
- } |
- return Py_BuildValue("lll", (long)kp.kp_eproc.e_pcred.p_ruid, |
- (long)kp.kp_eproc.e_ucred.cr_uid, |
- (long)kp.kp_eproc.e_pcred.p_svuid); |
-} |
- |
- |
-/* |
- * Return process real group id from ki_comm as a Python integer. |
- */ |
-static PyObject* |
-get_process_gids(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- struct kinfo_proc kp; |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- if (get_kinfo_proc(pid, &kp) == -1) { |
- return NULL; |
- } |
- return Py_BuildValue("lll", (long)kp.kp_eproc.e_pcred.p_rgid, |
- (long)kp.kp_eproc.e_ucred.cr_groups[0], |
- (long)kp.kp_eproc.e_pcred.p_svgid); |
-} |
- |
- |
-/* |
- * Return process controlling terminal number as an integer. |
- */ |
-static PyObject* |
-get_process_tty_nr(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- struct kinfo_proc kp; |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- if (get_kinfo_proc(pid, &kp) == -1) { |
- return NULL; |
- } |
- return Py_BuildValue("i", kp.kp_eproc.e_tdev); |
-} |
- |
- |
-/* |
- * Return a Python integer indicating the number of CPUs on the system. |
- */ |
-static PyObject* |
-get_num_cpus(PyObject* self, PyObject* args) |
-{ |
- |
- int mib[2]; |
- int ncpu; |
- size_t len; |
- |
- mib[0] = CTL_HW; |
- mib[1] = HW_NCPU; |
- len = sizeof(ncpu); |
- |
- if (sysctl(mib, 2, &ncpu, &len, NULL, 0) == -1) { |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- |
- return Py_BuildValue("i", ncpu); |
-} |
- |
- |
-#define TV2DOUBLE(t) ((t).tv_sec + (t).tv_usec / 1000000.0) |
- |
-/* |
- * Return a Python tuple (user_time, kernel_time) |
- */ |
-static PyObject* |
-get_cpu_times(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- int err; |
- unsigned int info_count = TASK_BASIC_INFO_COUNT; |
- task_port_t task; // = (task_port_t)NULL; |
- time_value_t user_time, system_time; |
- struct task_basic_info tasks_info; |
- struct task_thread_times_info task_times; |
- |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- |
- /* task_for_pid() requires special privileges |
- * "This function can be called only if the process is owned by the |
- * procmod group or if the caller is root." |
- * - http://developer.apple.com/documentation/MacOSX/Conceptual/universal_binary/universal_binary_tips/chapter_5_section_19.html */ |
- err = task_for_pid(mach_task_self(), pid, &task); |
- if ( err == KERN_SUCCESS) { |
- info_count = TASK_BASIC_INFO_COUNT; |
- err = task_info(task, TASK_BASIC_INFO, (task_info_t)&tasks_info, &info_count); |
- if (err != KERN_SUCCESS) { |
- // errcode 4 is "invalid argument" (access denied) |
- if (err == 4) { |
- return AccessDenied(); |
- } |
- |
- // otherwise throw a runtime error with appropriate error code |
- return PyErr_Format(PyExc_RuntimeError, |
- "task_info(TASK_BASIC_INFO) failed"); |
- } |
- |
- info_count = TASK_THREAD_TIMES_INFO_COUNT; |
- err = task_info(task, TASK_THREAD_TIMES_INFO, |
- (task_info_t)&task_times, &info_count); |
- if (err != KERN_SUCCESS) { |
- // errcode 4 is "invalid argument" (access denied) |
- if (err == 4) { |
- return AccessDenied(); |
- } |
- return PyErr_Format(PyExc_RuntimeError, |
- "task_info(TASK_BASIC_INFO) failed"); |
- } |
- } |
- |
- else { // task_for_pid failed |
- if (! pid_exists(pid) ) { |
- return NoSuchProcess(); |
- } |
- // pid exists, so return AccessDenied error since task_for_pid() failed |
- return AccessDenied(); |
- } |
- |
- float user_t = -1.0; |
- float sys_t = -1.0; |
- user_time = tasks_info.user_time; |
- system_time = tasks_info.system_time; |
- |
- time_value_add(&user_time, &task_times.user_time); |
- time_value_add(&system_time, &task_times.system_time); |
- |
- user_t = (float)user_time.seconds + ((float)user_time.microseconds / 1000000.0); |
- sys_t = (float)system_time.seconds + ((float)system_time.microseconds / 1000000.0); |
- return Py_BuildValue("(dd)", user_t, sys_t); |
-} |
- |
- |
-/* |
- * Return a Python float indicating the process create time expressed in |
- * seconds since the epoch. |
- */ |
-static PyObject* |
-get_process_create_time(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- struct kinfo_proc kp; |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- if (get_kinfo_proc(pid, &kp) == -1) { |
- return NULL; |
- } |
- return Py_BuildValue("d", TV2DOUBLE(kp.kp_proc.p_starttime)); |
-} |
- |
- |
-/* |
- * Return a tuple of RSS and VMS memory usage. |
- */ |
-static PyObject* |
-get_memory_info(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- int err; |
- unsigned int info_count = TASK_BASIC_INFO_COUNT; |
- mach_port_t task; |
- struct task_basic_info tasks_info; |
- vm_region_basic_info_data_64_t b_info; |
- vm_address_t address = GLOBAL_SHARED_TEXT_SEGMENT; |
- vm_size_t size; |
- mach_port_t object_name; |
- |
- // the argument passed should be a process id |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- |
- /* task_for_pid() requires special privileges |
- * "This function can be called only if the process is owned by the |
- * procmod group or if the caller is root." |
- * - http://developer.apple.com/documentation/MacOSX/Conceptual/universal_binary/universal_binary_tips/chapter_5_section_19.html */ |
- err = task_for_pid(mach_task_self(), pid, &task); |
- if ( err == KERN_SUCCESS) { |
- info_count = TASK_BASIC_INFO_COUNT; |
- err = task_info(task, TASK_BASIC_INFO, (task_info_t)&tasks_info, &info_count); |
- if (err != KERN_SUCCESS) { |
- if (err == 4) { |
- // errcode 4 is "invalid argument" (access denied) |
- return AccessDenied(); |
- } |
- // otherwise throw a runtime error with appropriate error code |
- return PyErr_Format(PyExc_RuntimeError, |
- "task_info(TASK_BASIC_INFO) failed"); |
- } |
- |
- /* Issue #73 http://code.google.com/p/psutil/issues/detail?id=73 |
- * adjust the virtual memory size down to account for |
- * shared memory that task_info.virtual_size includes w/every process |
- */ |
- info_count = VM_REGION_BASIC_INFO_COUNT_64; |
- err = vm_region_64(task, &address, &size, VM_REGION_BASIC_INFO, |
- (vm_region_info_t)&b_info, &info_count, &object_name); |
- if (err == KERN_SUCCESS) { |
- if (b_info.reserved && size == (SHARED_TEXT_REGION_SIZE) && |
- tasks_info.virtual_size > (SHARED_TEXT_REGION_SIZE + SHARED_DATA_REGION_SIZE)) |
- { |
- tasks_info.virtual_size -= (SHARED_TEXT_REGION_SIZE + SHARED_DATA_REGION_SIZE); |
- } |
- } |
- } |
- |
- else { |
- if (! pid_exists(pid) ) { |
- return NoSuchProcess(); |
- } |
- |
- // pid exists, so return AccessDenied error since task_for_pid() failed |
- return AccessDenied(); |
- } |
- |
- return Py_BuildValue("(ll)", tasks_info.resident_size, tasks_info.virtual_size); |
-} |
- |
- |
-/* |
- * Return number of threads used by process as a Python integer. |
- */ |
-static PyObject* |
-get_process_num_threads(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- int err, ret; |
- unsigned int info_count = TASK_BASIC_INFO_COUNT; |
- mach_port_t task; |
- struct task_basic_info tasks_info; |
- thread_act_port_array_t thread_list; |
- mach_msg_type_number_t thread_count; |
- |
- // the argument passed should be a process id |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- |
- /* task_for_pid() requires special privileges |
- * "This function can be called only if the process is owned by the |
- * procmod group or if the caller is root." |
- * - http://developer.apple.com/documentation/MacOSX/Conceptual/universal_binary/universal_binary_tips/chapter_5_section_19.html |
- */ |
- err = task_for_pid(mach_task_self(), pid, &task); |
- if ( err == KERN_SUCCESS) { |
- info_count = TASK_BASIC_INFO_COUNT; |
- err = task_info(task, TASK_BASIC_INFO, (task_info_t)&tasks_info, &info_count); |
- if (err != KERN_SUCCESS) { |
- // errcode 4 is "invalid argument" (access denied) |
- if (err == 4) { |
- return AccessDenied(); |
- } |
- |
- // otherwise throw a runtime error with appropriate error code |
- return PyErr_Format(PyExc_RuntimeError, |
- "task_info(TASK_BASIC_INFO) failed"); |
- } |
- err = task_threads(task, &thread_list, &thread_count); |
- if (err == KERN_SUCCESS) { |
- ret = vm_deallocate(task, (vm_address_t)thread_list, |
- thread_count * sizeof(int)); |
- if (ret != KERN_SUCCESS) { |
- printf("vm_deallocate() failed\n"); |
- } |
- return Py_BuildValue("l", (long)thread_count); |
- } |
- else { |
- return PyErr_Format(PyExc_RuntimeError, "task_thread() failed"); |
- } |
- } |
- else { |
- if (! pid_exists(pid) ) { |
- return NoSuchProcess(); |
- } |
- |
- // pid exists, so return AccessDenied error since task_for_pid() failed |
- return AccessDenied(); |
- } |
- return NULL; |
-} |
- |
- |
-/* |
- * Return a Python integer indicating the total amount of physical memory |
- * in bytes. |
- */ |
-static PyObject* |
-get_total_phymem(PyObject* self, PyObject* args) |
-{ |
- int mib[2]; |
- uint64_t total_phymem; |
- size_t len; |
- |
- mib[0] = CTL_HW; |
- mib[1] = HW_MEMSIZE; |
- len = sizeof(total_phymem); |
- |
- if (sysctl(mib, 2, &total_phymem, &len, NULL, 0) == -1) { |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- return Py_BuildValue("L", total_phymem); |
-} |
- |
- |
-/* |
- * Return a Python long indicating the amount of available physical memory in |
- * bytes. |
- */ |
-static PyObject* |
-get_avail_phymem(PyObject* self, PyObject* args) |
-{ |
- vm_statistics_data_t vm_stat; |
- mach_msg_type_number_t count; |
- kern_return_t error; |
- unsigned long long mem_free; |
- int pagesize = getpagesize(); |
- mach_port_t mport = mach_host_self(); |
- |
- count = sizeof(vm_stat) / sizeof(natural_t); |
- error = host_statistics(mport, HOST_VM_INFO, (host_info_t)&vm_stat, &count); |
- |
- if (error != KERN_SUCCESS) { |
- return PyErr_Format(PyExc_RuntimeError, |
- "Error in host_statistics(): %s", mach_error_string(error)); |
- } |
- |
- mem_free = (unsigned long long) vm_stat.free_count * pagesize; |
- return Py_BuildValue("L", mem_free); |
-} |
- |
- |
-/* |
- * Return a Python integer indicating the total amount of virtual memory |
- * in bytes. |
- */ |
-static PyObject* |
-get_total_virtmem(PyObject* self, PyObject* args) |
-{ |
- int mib[2]; |
- size_t size; |
- struct xsw_usage totals; |
- |
- mib[0] = CTL_VM; |
- mib[1] = VM_SWAPUSAGE; |
- size = sizeof(totals); |
- |
- if (sysctl(mib, 2, &totals, &size, NULL, 0) == -1) { |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- |
- return Py_BuildValue("L", totals.xsu_total); |
-} |
- |
-/* |
- * Return a Python integer indicating the avail amount of virtual memory |
- * in bytes. |
- */ |
-static PyObject* |
-get_avail_virtmem(PyObject* self, PyObject* args) |
-{ |
- int mib[2]; |
- size_t size; |
- struct xsw_usage totals; |
- |
- mib[0] = CTL_VM; |
- mib[1] = VM_SWAPUSAGE; |
- size = sizeof(totals); |
- |
- if (sysctl(mib, 2, &totals, &size, NULL, 0) == -1) { |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- |
- return Py_BuildValue("L", totals.xsu_avail); |
-} |
- |
-/* |
- * Return a Python tuple representing user, kernel and idle CPU times |
- */ |
-static PyObject* |
-get_system_cpu_times(PyObject* self, PyObject* args) |
-{ |
- mach_msg_type_number_t count = HOST_CPU_LOAD_INFO_COUNT; |
- kern_return_t error; |
- host_cpu_load_info_data_t r_load; |
- |
- error = host_statistics(mach_host_self(), HOST_CPU_LOAD_INFO, (host_info_t)&r_load, &count); |
- if (error != KERN_SUCCESS) { |
- return PyErr_Format(PyExc_RuntimeError, |
- "Error in host_statistics(): %s", mach_error_string(error)); |
- } |
- |
- return Py_BuildValue("(dddd)", |
- (double)r_load.cpu_ticks[CPU_STATE_USER] / CLK_TCK, |
- (double)r_load.cpu_ticks[CPU_STATE_NICE] / CLK_TCK, |
- (double)r_load.cpu_ticks[CPU_STATE_SYSTEM] / CLK_TCK, |
- (double)r_load.cpu_ticks[CPU_STATE_IDLE] / CLK_TCK |
- ); |
-} |
- |
- |
-/* |
- * Return a Python list of tuple representing per-cpu times |
- */ |
-static PyObject* |
-get_system_per_cpu_times(PyObject* self, PyObject* args) |
-{ |
- natural_t cpu_count; |
- processor_info_array_t info_array; |
- mach_msg_type_number_t info_count; |
- kern_return_t error; |
- processor_cpu_load_info_data_t* cpu_load_info; |
- PyObject* py_retlist = PyList_New(0); |
- PyObject* py_cputime; |
- int i, ret; |
- |
- error = host_processor_info(mach_host_self(), PROCESSOR_CPU_LOAD_INFO, |
- &cpu_count, &info_array, &info_count); |
- if (error != KERN_SUCCESS) { |
- return PyErr_Format(PyExc_RuntimeError, |
- "Error in host_processor_info(): %s", mach_error_string(error)); |
- } |
- |
- cpu_load_info = (processor_cpu_load_info_data_t*) info_array; |
- |
- for (i = 0; i < cpu_count; i++) { |
- py_cputime = Py_BuildValue("(dddd)", |
- (double)cpu_load_info[i].cpu_ticks[CPU_STATE_USER] / CLK_TCK, |
- (double)cpu_load_info[i].cpu_ticks[CPU_STATE_NICE] / CLK_TCK, |
- (double)cpu_load_info[i].cpu_ticks[CPU_STATE_SYSTEM] / CLK_TCK, |
- (double)cpu_load_info[i].cpu_ticks[CPU_STATE_IDLE] / CLK_TCK |
- ); |
- PyList_Append(py_retlist, py_cputime); |
- Py_XDECREF(py_cputime); |
- } |
- |
- ret = vm_deallocate(mach_task_self(), (vm_address_t)info_array, |
- info_count * sizeof(int)); |
- if (ret != KERN_SUCCESS) { |
- printf("vm_deallocate() failed\n"); |
- } |
- return py_retlist; |
-} |
- |
- |
-/* |
- * Return a Python float indicating the system boot time expressed in |
- * seconds since the epoch. |
- */ |
-static PyObject* |
-get_system_boot_time(PyObject* self, PyObject* args) |
-{ |
- /* fetch sysctl "kern.boottime" */ |
- static int request[2] = { CTL_KERN, KERN_BOOTTIME }; |
- struct timeval result; |
- size_t result_len = sizeof result; |
- time_t boot_time = 0; |
- |
- if (sysctl(request, 2, &result, &result_len, NULL, 0) == -1) { |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- boot_time = result.tv_sec; |
- return Py_BuildValue("f", (float)boot_time); |
-} |
- |
- |
-/* |
- * Return a list of tuples including device, mount point and fs type |
- * for all partitions mounted on the system. |
- */ |
-static PyObject* |
-get_disk_partitions(PyObject* self, PyObject* args) |
-{ |
- int num; |
- int i; |
- long len; |
- struct statfs *fs; |
- PyObject* py_retlist = PyList_New(0); |
- PyObject* py_tuple; |
- |
- // get the number of mount points |
- Py_BEGIN_ALLOW_THREADS |
- num = getfsstat(NULL, 0, MNT_NOWAIT); |
- Py_END_ALLOW_THREADS |
- if (num == -1) { |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- |
- len = sizeof(*fs) * num; |
- fs = malloc(len); |
- |
- Py_BEGIN_ALLOW_THREADS |
- num = getfsstat(fs, len, MNT_NOWAIT); |
- Py_END_ALLOW_THREADS |
- if (num == -1) { |
- free(fs); |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- |
- for (i = 0; i < num; i++) { |
- py_tuple = Py_BuildValue("(sss)", fs[i].f_mntfromname, // device |
- fs[i].f_mntonname, // mount point |
- fs[i].f_fstypename); // fs type |
- PyList_Append(py_retlist, py_tuple); |
- Py_XDECREF(py_tuple); |
- } |
- |
- free(fs); |
- return py_retlist; |
-} |
- |
- |
-/* |
- * Return process status as a Python integer. |
- */ |
-static PyObject* |
-get_process_status(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- struct kinfo_proc kp; |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- if (get_kinfo_proc(pid, &kp) == -1) { |
- return NULL; |
- } |
- return Py_BuildValue("i", (int)kp.kp_proc.p_stat); |
-} |
- |
- |
-/* |
- * Return process threads |
- */ |
-static PyObject* |
-get_process_threads(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- int err, j, ret; |
- kern_return_t kr; |
- unsigned int info_count = TASK_BASIC_INFO_COUNT; |
- mach_port_t task; |
- struct task_basic_info tasks_info; |
- thread_act_port_array_t thread_list; |
- thread_info_data_t thinfo; |
- thread_basic_info_t basic_info_th; |
- mach_msg_type_number_t thread_count, thread_info_count; |
- |
- PyObject* retList = PyList_New(0); |
- PyObject* pyTuple = NULL; |
- |
- // the argument passed should be a process id |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- |
- // task_for_pid() requires special privileges |
- err = task_for_pid(mach_task_self(), pid, &task); |
- if (err != KERN_SUCCESS) { |
- if (! pid_exists(pid) ) { |
- return NoSuchProcess(); |
- } |
- return AccessDenied(); |
- } |
- |
- info_count = TASK_BASIC_INFO_COUNT; |
- err = task_info(task, TASK_BASIC_INFO, (task_info_t)&tasks_info, &info_count); |
- if (err != KERN_SUCCESS) { |
- // errcode 4 is "invalid argument" (access denied) |
- if (err == 4) { |
- return AccessDenied(); |
- } |
- // otherwise throw a runtime error with appropriate error code |
- return PyErr_Format(PyExc_RuntimeError, |
- "task_info(TASK_BASIC_INFO) failed"); |
- } |
- |
- err = task_threads(task, &thread_list, &thread_count); |
- if (err != KERN_SUCCESS) { |
- return PyErr_Format(PyExc_RuntimeError, "task_threads() failed"); |
- } |
- |
- for (j = 0; j < thread_count; j++) { |
- thread_info_count = THREAD_INFO_MAX; |
- kr = thread_info(thread_list[j], THREAD_BASIC_INFO, |
- (thread_info_t)thinfo, &thread_info_count); |
- if (kr != KERN_SUCCESS) { |
- return PyErr_Format(PyExc_RuntimeError, "thread_info() failed"); |
- } |
- basic_info_th = (thread_basic_info_t)thinfo; |
- // XXX - thread_info structure does not provide any process id; |
- // the best we can do is assigning an incremental bogus value |
- pyTuple = Py_BuildValue("Iff", j + 1, |
- (float)basic_info_th->user_time.microseconds / 1000000.0, |
- (float)basic_info_th->system_time.microseconds / 1000000.0 |
- ); |
- PyList_Append(retList, pyTuple); |
- Py_XDECREF(pyTuple); |
- } |
- |
- ret = vm_deallocate(task, (vm_address_t)thread_list, |
- thread_count * sizeof(int)); |
- if (ret != KERN_SUCCESS) { |
- printf("vm_deallocate() failed\n"); |
- } |
- |
- return retList; |
-} |
- |
- |
-/* |
- * Return process open files as a Python tuple. |
- * References: |
- * - lsof source code: http://goo.gl/SYW79 and http://goo.gl/m78fd |
- * - /usr/include/sys/proc_info.h |
- */ |
-static PyObject* |
-get_process_open_files(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- int pidinfo_result; |
- int iterations; |
- int i; |
- int nb; |
- |
- struct proc_fdinfo *fds_pointer; |
- struct proc_fdinfo *fdp_pointer; |
- struct vnode_fdinfowithpath vi; |
- |
- PyObject *retList = PyList_New(0); |
- PyObject *tuple = NULL; |
- |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- |
- pidinfo_result = proc_pidinfo(pid, PROC_PIDLISTFDS, 0, NULL, 0); |
- if (pidinfo_result <= 0) { |
- goto error; |
- } |
- |
- fds_pointer = malloc(pidinfo_result); |
- pidinfo_result = proc_pidinfo(pid, PROC_PIDLISTFDS, 0, fds_pointer, |
- pidinfo_result); |
- free(fds_pointer); |
- |
- if (pidinfo_result <= 0) { |
- goto error; |
- } |
- |
- iterations = (pidinfo_result / PROC_PIDLISTFD_SIZE); |
- |
- for (i = 0; i < iterations; i++) { |
- fdp_pointer = &fds_pointer[i]; |
- |
- // |
- if (fdp_pointer->proc_fdtype == PROX_FDTYPE_VNODE) |
- { |
- nb = proc_pidfdinfo(pid, |
- fdp_pointer->proc_fd, |
- PROC_PIDFDVNODEPATHINFO, |
- &vi, |
- sizeof(vi)); |
- |
- // --- errors checking |
- if (nb <= 0) { |
- if ((errno == ENOENT) || (errno == EBADF)) { |
- // no such file or directory or bad file descriptor; |
- // let's assume the file has been closed or removed |
- continue; |
- } |
- if (errno != 0) { |
- return PyErr_SetFromErrno(PyExc_OSError); |
- } |
- else |
- return PyErr_Format(PyExc_RuntimeError, |
- "proc_pidinfo(PROC_PIDFDVNODEPATHINFO) failed"); |
- } |
- if (nb < sizeof(vi)) { |
- return PyErr_Format(PyExc_RuntimeError, |
- "proc_pidinfo(PROC_PIDFDVNODEPATHINFO) failed (buffer mismatch)"); |
- } |
- // --- /errors checking |
- |
- // --- construct python list |
- tuple = Py_BuildValue("(si)", vi.pvip.vip_path, |
- (int)fdp_pointer->proc_fd); |
- PyList_Append(retList, tuple); |
- Py_DECREF(tuple); |
- // --- /construct python list |
- } |
- } |
- |
- return retList; |
- |
-error: |
- if (errno != 0) |
- return PyErr_SetFromErrno(PyExc_OSError); |
- else if (! pid_exists(pid) ) |
- return NoSuchProcess(); |
- else |
- return PyErr_Format(PyExc_RuntimeError, |
- "proc_pidinfo(PROC_PIDLISTFDS) failed"); |
-} |
- |
- |
-/* |
- * mathes Linux net/tcp_states.h: |
- * http://students.mimuw.edu.pl/lxr/source/include/net/tcp_states.h |
- */ |
-static char * |
-get_connection_status(int st) { |
- switch (st) { |
- case TCPS_CLOSED: |
- return "CLOSE"; |
- case TCPS_CLOSING: |
- return "CLOSING"; |
- case TCPS_CLOSE_WAIT: |
- return "CLOSE_WAIT"; |
- case TCPS_LISTEN: |
- return "LISTEN"; |
- case TCPS_ESTABLISHED: |
- return "ESTABLISHED"; |
- case TCPS_SYN_SENT: |
- return "SYN_SENT"; |
- case TCPS_SYN_RECEIVED: |
- return "SYN_RECV"; |
- case TCPS_FIN_WAIT_1: |
- return "FIN_WAIT_1"; |
- case TCPS_FIN_WAIT_2: |
- return "FIN_WAIT_2"; |
- case TCPS_LAST_ACK: |
- return "LAST_ACK"; |
- case TCPS_TIME_WAIT: |
- return "TIME_WAIT"; |
- default: |
- return ""; |
- } |
-} |
- |
- |
-/* |
- * Return process TCP and UDP connections as a list of tuples. |
- * References: |
- * - lsof source code: http://goo.gl/SYW79 and http://goo.gl/wNrC0 |
- * - /usr/include/sys/proc_info.h |
- */ |
-static PyObject* |
-get_process_connections(PyObject* self, PyObject* args) |
-{ |
- long pid; |
- int pidinfo_result; |
- int iterations; |
- int i; |
- int nb; |
- |
- struct proc_fdinfo *fds_pointer; |
- struct proc_fdinfo *fdp_pointer; |
- struct socket_fdinfo si; |
- |
- |
- PyObject *retList = PyList_New(0); |
- PyObject *tuple = NULL; |
- PyObject *laddr = NULL; |
- PyObject *raddr = NULL; |
- |
- if (! PyArg_ParseTuple(args, "l", &pid)) { |
- return NULL; |
- } |
- |
- if (pid == 0) { |
- return retList; |
- } |
- |
- pidinfo_result = proc_pidinfo(pid, PROC_PIDLISTFDS, 0, NULL, 0); |
- if (pidinfo_result <= 0) { |
- goto error; |
- } |
- |
- fds_pointer = malloc(pidinfo_result); |
- pidinfo_result = proc_pidinfo(pid, PROC_PIDLISTFDS, 0, fds_pointer, |
- pidinfo_result); |
- free(fds_pointer); |
- |
- if (pidinfo_result <= 0) { |
- goto error; |
- } |
- |
- iterations = (pidinfo_result / PROC_PIDLISTFD_SIZE); |
- |
- for (i = 0; i < iterations; i++) { |
- errno = 0; |
- fdp_pointer = &fds_pointer[i]; |
- |
- // |
- if (fdp_pointer->proc_fdtype == PROX_FDTYPE_SOCKET) |
- { |
- nb = proc_pidfdinfo(pid, fdp_pointer->proc_fd, PROC_PIDFDSOCKETINFO, |
- &si, sizeof(si)); |
- |
- // --- errors checking |
- if (nb <= 0) { |
- if (errno == EBADF) { |
- // let's assume socket has been closed |
- continue; |
- } |
- if (errno != 0) { |
- return PyErr_SetFromErrno(PyExc_OSError); |
- } |
- else { |
- return PyErr_Format(PyExc_RuntimeError, |
- "proc_pidinfo(PROC_PIDFDVNODEPATHINFO) failed"); |
- } |
- } |
- if (nb < sizeof(si)) { |
- return PyErr_Format(PyExc_RuntimeError, |
- "proc_pidinfo(PROC_PIDFDVNODEPATHINFO) failed (buffer mismatch)"); |
- } |
- // --- /errors checking |
- |
- // |
- int fd, family, type, lport, rport; |
- char lip[200], rip[200]; |
- char *state; |
- |
- fd = (int)fdp_pointer->proc_fd; |
- family = si.psi.soi_family; |
- type = si.psi.soi_kind; |
- |
- if ((family != AF_INET) && (family != AF_INET6)) { |
- continue; |
- } |
- |
- if (type == 2) |
- type = SOCK_STREAM; |
- else if (type == 1) |
- type = SOCK_DGRAM; |
- else |
- continue; |
- |
- if (errno != 0) { |
- printf("errno 1 = %i\n", errno); |
- return PyErr_SetFromErrno(PyExc_OSError); |
- } |
- |
- |
- if (family == AF_INET) { |
- inet_ntop(AF_INET, |
- &si.psi.soi_proto.pri_tcp.tcpsi_ini.insi_laddr.ina_46.i46a_addr4, |
- lip, |
- sizeof(lip)); |
- inet_ntop(AF_INET, |
- &si.psi.soi_proto.pri_tcp.tcpsi_ini.insi_faddr.ina_46.i46a_addr4, |
- rip, |
- sizeof(lip)); |
- } |
- else { |
- inet_ntop(AF_INET6, |
- &si.psi.soi_proto.pri_tcp.tcpsi_ini.insi_laddr.ina_6, |
- lip, sizeof(lip)); |
- inet_ntop(AF_INET6, |
- &si.psi.soi_proto.pri_tcp.tcpsi_ini.insi_faddr.ina_6, |
- lip, sizeof(rip)); |
- } |
- |
- // check for inet_ntop failures |
- if (errno != 0) { |
- return PyErr_SetFromErrno(PyExc_OSError); |
- } |
- |
- lport = ntohs(si.psi.soi_proto.pri_tcp.tcpsi_ini.insi_lport); |
- rport = ntohs(si.psi.soi_proto.pri_tcp.tcpsi_ini.insi_fport); |
- if (type == SOCK_STREAM) |
- state = get_connection_status((int)si.psi.soi_proto.pri_tcp.tcpsi_state); |
- else |
- state = ""; |
- |
- laddr = Py_BuildValue("(si)", lip, lport); |
- if (rport != 0) |
- raddr = Py_BuildValue("(si)", rip, rport); |
- else |
- raddr = PyTuple_New(0); |
- |
- // --- construct python list |
- tuple = Py_BuildValue("(iiiNNs)", fd, family, type, laddr, raddr, |
- state); |
- PyList_Append(retList, tuple); |
- Py_DECREF(tuple); |
- // --- /construct python list |
- } |
- } |
- |
- return retList; |
- |
-error: |
- if (errno != 0) |
- return PyErr_SetFromErrno(PyExc_OSError); |
- else if (! pid_exists(pid) ) |
- return NoSuchProcess(); |
- else |
- return PyErr_Format(PyExc_RuntimeError, |
- "proc_pidinfo(PROC_PIDLISTFDS) failed"); |
-} |
- |
- |
-/* |
- * Return a Python list of named tuples with overall network I/O information |
- */ |
-static PyObject* |
-get_network_io_counters(PyObject* self, PyObject* args) |
-{ |
- PyObject* py_retdict = PyDict_New(); |
- PyObject* py_ifc_info; |
- |
- char *buf = NULL, *lim, *next; |
- struct if_msghdr *ifm; |
- int mib[6]; |
- size_t len; |
- |
- mib[0] = CTL_NET; // networking subsystem |
- mib[1] = PF_ROUTE; // type of information |
- mib[2] = 0; // protocol (IPPROTO_xxx) |
- mib[3] = 0; // address family |
- mib[4] = NET_RT_IFLIST2; // operation |
- mib[5] = 0; |
- |
- if (sysctl(mib, 6, NULL, &len, NULL, 0) < 0) { |
- Py_DECREF(py_retdict); |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- |
- buf = malloc(len); |
- |
- if (sysctl(mib, 6, buf, &len, NULL, 0) < 0) { |
- if (buf) { |
- free(buf); |
- } |
- Py_DECREF(py_retdict); |
- PyErr_SetFromErrno(0); |
- return NULL; |
- } |
- |
- lim = buf + len; |
- |
- for (next = buf; next < lim; ) { |
- ifm = (struct if_msghdr *)next; |
- next += ifm->ifm_msglen; |
- |
- if (ifm->ifm_type == RTM_IFINFO2) { |
- struct if_msghdr2 *if2m = (struct if_msghdr2 *)ifm; |
- struct sockaddr_dl *sdl = (struct sockaddr_dl *)(if2m + 1); |
- char ifc_name[32]; |
- |
- strncpy(ifc_name, sdl->sdl_data, sdl->sdl_nlen); |
- ifc_name[sdl->sdl_nlen] = 0; |
- |
- py_ifc_info = Py_BuildValue("(KKKK)", |
- if2m->ifm_data.ifi_obytes, |
- if2m->ifm_data.ifi_ibytes, |
- if2m->ifm_data.ifi_opackets, |
- if2m->ifm_data.ifi_ipackets); |
- PyDict_SetItemString(py_retdict, ifc_name, py_ifc_info); |
- Py_XDECREF(py_ifc_info); |
- } |
- else { |
- continue; |
- } |
- } |
- |
- free(buf); |
- |
- return py_retdict; |
-} |
- |
- |
-/* |
- * Return a Python dict of tuples for disk I/O information |
- */ |
-static PyObject* |
-get_disk_io_counters(PyObject* self, PyObject* args) |
-{ |
- PyObject* py_retdict = PyDict_New(); |
- PyObject* py_disk_info; |
- |
- CFDictionaryRef parent_dict; |
- CFDictionaryRef props_dict; |
- CFDictionaryRef stats_dict; |
- io_registry_entry_t parent; |
- io_registry_entry_t disk; |
- io_iterator_t disk_list; |
- |
- /* Get list of disks */ |
- if (IOServiceGetMatchingServices(kIOMasterPortDefault, |
- IOServiceMatching(kIOMediaClass), |
- &disk_list) != kIOReturnSuccess) { |
- Py_DECREF(py_retdict); |
- PyErr_SetString(PyExc_RuntimeError, "Unable to get the list of disks."); |
- return NULL; |
- } |
- |
- /* Iterate over disks */ |
- while ((disk = IOIteratorNext(disk_list)) != 0) { |
- parent_dict = NULL; |
- props_dict = NULL; |
- stats_dict = NULL; |
- |
- if (IORegistryEntryGetParentEntry(disk, kIOServicePlane, &parent) != kIOReturnSuccess) { |
- PyErr_SetString(PyExc_RuntimeError, "Unable to get the disk's parent."); |
- Py_DECREF(py_retdict); |
- IOObjectRelease(disk); |
- return NULL; |
- } |
- |
- if (IOObjectConformsTo(parent, "IOBlockStorageDriver")) { |
- if(IORegistryEntryCreateCFProperties( |
- disk, |
- (CFMutableDictionaryRef *) &parent_dict, |
- kCFAllocatorDefault, |
- kNilOptions) != kIOReturnSuccess) |
- { |
- PyErr_SetString(PyExc_RuntimeError, |
- "Unable to get the parent's properties."); |
- Py_DECREF(py_retdict); |
- IOObjectRelease(disk); |
- IOObjectRelease(parent); |
- return NULL; |
- } |
- |
- if (IORegistryEntryCreateCFProperties(parent, |
- (CFMutableDictionaryRef *) &props_dict, |
- kCFAllocatorDefault, |
- kNilOptions) != kIOReturnSuccess) |
- { |
- PyErr_SetString(PyExc_RuntimeError, |
- "Unable to get the disk properties."); |
- Py_DECREF(py_retdict); |
- IOObjectRelease(disk); |
- return NULL; |
- } |
- |
- const int kMaxDiskNameSize = 64; |
- CFStringRef disk_name_ref = (CFStringRef)CFDictionaryGetValue( |
- parent_dict, |
- CFSTR(kIOBSDNameKey)); |
- char disk_name[kMaxDiskNameSize]; |
- |
- CFStringGetCString(disk_name_ref, |
- disk_name, |
- kMaxDiskNameSize, |
- CFStringGetSystemEncoding()); |
- |
- stats_dict = (CFDictionaryRef)CFDictionaryGetValue( |
- props_dict, |
- CFSTR(kIOBlockStorageDriverStatisticsKey)); |
- |
- if (stats_dict == NULL) { |
- PyErr_SetString(PyExc_RuntimeError, "Unable to get disk stats."); |
- Py_DECREF(py_retdict); |
- CFRelease(props_dict); |
- IOObjectRelease(disk); |
- IOObjectRelease(parent); |
- return NULL; |
- } |
- |
- CFNumberRef number; |
- int64_t reads, writes, read_bytes, write_bytes, read_time, write_time = 0; |
- |
- /* Get disk reads/writes */ |
- if ((number = (CFNumberRef)CFDictionaryGetValue( |
- stats_dict, |
- CFSTR(kIOBlockStorageDriverStatisticsReadsKey)))) |
- { |
- CFNumberGetValue(number, kCFNumberSInt64Type, &reads); |
- } |
- if ((number = (CFNumberRef)CFDictionaryGetValue( |
- stats_dict, |
- CFSTR(kIOBlockStorageDriverStatisticsWritesKey)))) |
- { |
- CFNumberGetValue(number, kCFNumberSInt64Type, &writes); |
- } |
- |
- /* Get disk bytes read/written */ |
- if ((number = (CFNumberRef)CFDictionaryGetValue( |
- stats_dict, |
- CFSTR(kIOBlockStorageDriverStatisticsBytesReadKey)))) |
- { |
- CFNumberGetValue(number, kCFNumberSInt64Type, &read_bytes); |
- } |
- if ((number = (CFNumberRef)CFDictionaryGetValue( |
- stats_dict, |
- CFSTR(kIOBlockStorageDriverStatisticsBytesWrittenKey)))) |
- { |
- CFNumberGetValue(number, kCFNumberSInt64Type, &write_bytes); |
- } |
- |
- /* Get disk time spent reading/writing (nanoseconds) */ |
- if ((number = (CFNumberRef)CFDictionaryGetValue( |
- stats_dict, |
- CFSTR(kIOBlockStorageDriverStatisticsTotalReadTimeKey)))) |
- { |
- CFNumberGetValue(number, kCFNumberSInt64Type, &read_time); |
- } |
- if ((number = (CFNumberRef)CFDictionaryGetValue( |
- stats_dict, |
- CFSTR(kIOBlockStorageDriverStatisticsTotalWriteTimeKey)))) { |
- CFNumberGetValue(number, kCFNumberSInt64Type, &write_time); |
- } |
- |
- py_disk_info = Py_BuildValue("(KKKKKK)", |
- reads, writes, |
- read_bytes, write_bytes, |
- read_time, write_time); |
- PyDict_SetItemString(py_retdict, disk_name, py_disk_info); |
- Py_XDECREF(py_disk_info); |
- |
- CFRelease(parent_dict); |
- IOObjectRelease(parent); |
- CFRelease(props_dict); |
- IOObjectRelease(disk); |
- } |
- } |
- |
- IOObjectRelease (disk_list); |
- |
- return py_retdict; |
-} |
- |
- |
-/* |
- * define the psutil C module methods and initialize the module. |
- */ |
-static PyMethodDef |
-PsutilMethods[] = |
-{ |
- // --- per-process functions |
- |
- {"get_process_name", get_process_name, METH_VARARGS, |
- "Return process name"}, |
- {"get_process_cmdline", get_process_cmdline, METH_VARARGS, |
- "Return process cmdline as a list of cmdline arguments"}, |
- {"get_process_ppid", get_process_ppid, METH_VARARGS, |
- "Return process ppid as an integer"}, |
- {"get_process_uids", get_process_uids, METH_VARARGS, |
- "Return process real user id as an integer"}, |
- {"get_process_gids", get_process_gids, METH_VARARGS, |
- "Return process real group id as an integer"}, |
- {"get_cpu_times", get_cpu_times, METH_VARARGS, |
- "Return tuple of user/kern time for the given PID"}, |
- {"get_process_create_time", get_process_create_time, METH_VARARGS, |
- "Return a float indicating the process create time expressed in " |
- "seconds since the epoch"}, |
- {"get_memory_info", get_memory_info, METH_VARARGS, |
- "Return a tuple of RSS/VMS memory information"}, |
- {"get_process_num_threads", get_process_num_threads, METH_VARARGS, |
- "Return number of threads used by process"}, |
- {"get_process_status", get_process_status, METH_VARARGS, |
- "Return process status as an integer"}, |
- {"get_process_threads", get_process_threads, METH_VARARGS, |
- "Return process threads as a list of tuples"}, |
- {"get_process_open_files", get_process_open_files, METH_VARARGS, |
- "Return files opened by process as a list of tuples"}, |
- {"get_process_connections", get_process_connections, METH_VARARGS, |
- "Get process TCP and UDP connections as a list of tuples"}, |
- {"get_process_tty_nr", get_process_tty_nr, METH_VARARGS, |
- "Return process tty number as an integer"}, |
- |
- // --- system-related functions |
- |
- {"get_pid_list", get_pid_list, METH_VARARGS, |
- "Returns a list of PIDs currently running on the system"}, |
- {"get_num_cpus", get_num_cpus, METH_VARARGS, |
- "Return number of CPUs on the system"}, |
- {"get_total_phymem", get_total_phymem, METH_VARARGS, |
- "Return the total amount of physical memory, in bytes"}, |
- {"get_avail_phymem", get_avail_phymem, METH_VARARGS, |
- "Return the amount of available physical memory, in bytes"}, |
- {"get_total_virtmem", get_total_virtmem, METH_VARARGS, |
- "Return the total amount of virtual memory, in bytes"}, |
- {"get_avail_virtmem", get_avail_virtmem, METH_VARARGS, |
- "Return the amount of available virtual memory, in bytes"}, |
- {"get_system_cpu_times", get_system_cpu_times, METH_VARARGS, |
- "Return system cpu times as a tuple (user, system, nice, idle, irc)"}, |
- {"get_system_per_cpu_times", get_system_per_cpu_times, METH_VARARGS, |
- "Return system per-cpu times as a list of tuples"}, |
- {"get_system_boot_time", get_system_boot_time, METH_VARARGS, |
- "Return a float indicating the system boot time expressed in " |
- "seconds since the epoch"}, |
- {"get_disk_partitions", get_disk_partitions, METH_VARARGS, |
- "Return a list of tuples including device, mount point and " |
- "fs type for all partitions mounted on the system."}, |
- {"get_network_io_counters", get_network_io_counters, METH_VARARGS, |
- "Return dict of tuples of networks I/O information."}, |
- {"get_disk_io_counters", get_disk_io_counters, METH_VARARGS, |
- "Return dict of tuples of disks I/O information."}, |
- |
- {NULL, NULL, 0, NULL} |
-}; |
- |
- |
-struct module_state { |
- PyObject *error; |
-}; |
- |
-#if PY_MAJOR_VERSION >= 3 |
-#define GETSTATE(m) ((struct module_state*)PyModule_GetState(m)) |
-#else |
-#define GETSTATE(m) (&_state) |
-#endif |
- |
-#if PY_MAJOR_VERSION >= 3 |
- |
-static int |
-psutil_osx_traverse(PyObject *m, visitproc visit, void *arg) { |
- Py_VISIT(GETSTATE(m)->error); |
- return 0; |
-} |
- |
-static int |
-psutil_osx_clear(PyObject *m) { |
- Py_CLEAR(GETSTATE(m)->error); |
- return 0; |
-} |
- |
- |
-static struct PyModuleDef |
-moduledef = { |
- PyModuleDef_HEAD_INIT, |
- "psutil_osx", |
- NULL, |
- sizeof(struct module_state), |
- PsutilMethods, |
- NULL, |
- psutil_osx_traverse, |
- psutil_osx_clear, |
- NULL |
-}; |
- |
-#define INITERROR return NULL |
- |
-PyObject * |
-PyInit__psutil_osx(void) |
- |
-#else |
-#define INITERROR return |
- |
-void |
-init_psutil_osx(void) |
-#endif |
-{ |
-#if PY_MAJOR_VERSION >= 3 |
- PyObject *module = PyModule_Create(&moduledef); |
-#else |
- PyObject *module = Py_InitModule("_psutil_osx", PsutilMethods); |
-#endif |
- // process status constants, defined in: |
- // http://fxr.watson.org/fxr/source/bsd/sys/proc.h?v=xnu-792.6.70#L149 |
- PyModule_AddIntConstant(module, "SIDL", SIDL); |
- PyModule_AddIntConstant(module, "SRUN", SRUN); |
- PyModule_AddIntConstant(module, "SSLEEP", SSLEEP); |
- PyModule_AddIntConstant(module, "SSTOP", SSTOP); |
- PyModule_AddIntConstant(module, "SZOMB", SZOMB); |
- |
- if (module == NULL) { |
- INITERROR; |
- } |
-#if PY_MAJOR_VERSION >= 3 |
- return module; |
-#endif |
-} |