xnu-4570.71.2.tar.gz

[apple/xnu.git] / libsyscall / wrappers / getiopolicy_np.c

diff --git a/libsyscall/wrappers/getiopolicy_np.c b/libsyscall/wrappers/getiopolicy_np.c
index 146fdac6d93d21cb1559cce0eefacb171add4286..5833216228be4eef6adb9c4a8750d11b84f67537 100644 (file)
--- a/libsyscall/wrappers/getiopolicy_np.c
+++ b/libsyscall/wrappers/getiopolicy_np.c
@@ -22,8 +22,10 @@
  */
 #include <errno.h>
 #include <sys/resource.h>
+#include <mach/port.h>
 
 extern int __iopolicysys(int, struct _iopol_param_t *);
+extern void _pthread_clear_qos_tsd(mach_port_t);
 
 int
 getiopolicy_np(int iotype, int scope)
@@ -58,10 +60,17 @@ setiopolicy_np(int iotype, int scope, int policy)
 {
        /* kernel validates the indiv values, no need to repeat it */
        struct _iopol_param_t iop_param;
-       
+
        iop_param.iop_scope = scope;
        iop_param.iop_iotype = iotype;
        iop_param.iop_policy = policy;
 
-       return( __iopolicysys(IOPOL_CMD_SET, &iop_param));
+       int rv = __iopolicysys(IOPOL_CMD_SET, &iop_param);
+       if (rv == -2) {
+               /* not an actual error but indication that __iopolicysys removed the thread QoS */
+        _pthread_clear_qos_tsd(MACH_PORT_NULL);
+               rv = 0;
+       }
+
+       return rv;
 }