xnu-6153.81.5.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..e09f849cce42cc9eb38abb4201a1860fc8ac7795 100644 (file)
--- a/libsyscall/wrappers/getiopolicy_np.c
+++ b/libsyscall/wrappers/getiopolicy_np.c
@@ -2,14 +2,14 @@
  * Copyright (c) 2006 Apple Computer, Inc. All rights reserved.
  *
  * @APPLE_LICENSE_HEADER_START@
- * 
+ *
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
  * compliance with the License. Please obtain a copy of the License at
  * http://www.opensource.apple.com/apsl/ and read it before using this
  * file.
- * 
+ *
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
@@ -17,13 +17,15 @@
  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  * Please see the License for the specific language governing rights and
  * limitations under the License.
- * 
+ *
  * @APPLE_LICENSE_HEADER_END@
  */
 #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)
@@ -31,8 +33,8 @@ getiopolicy_np(int iotype, int scope)
        int policy, error;
        struct _iopol_param_t iop_param;
 
-       if (iotype != IOPOL_TYPE_DISK ||
-               (scope != IOPOL_SCOPE_PROCESS && scope != IOPOL_SCOPE_THREAD)) {
+       if ((iotype != IOPOL_TYPE_DISK && iotype != IOPOL_TYPE_VFS_ATIME_UPDATES && iotype != IOPOL_TYPE_VFS_MATERIALIZE_DATALESS_FILES) ||
+           (scope != IOPOL_SCOPE_PROCESS && scope != IOPOL_SCOPE_THREAD)) {
                errno = EINVAL;
                policy = -1;
                goto exit;
@@ -49,7 +51,7 @@ getiopolicy_np(int iotype, int scope)
 
        policy = iop_param.iop_policy;
 
-  exit:
+exit:
        return policy;
 }
 
@@ -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;
 }