1  /*
  2   * Copyright (c) 2011-2012 Apple Inc. All Rights Reserved.
  3   * 
  4   * @APPLE_LICENSE_HEADER_START@
  5   * 
  6   * This file contains Original Code and/or Modifications of Original Code
  7   * as defined in and that are subject to the Apple Public Source License
  8   * Version 2.0 (the 'License'). You may not use this file except in
  9   * compliance with the License. Please obtain a copy of the License at
 10   * http://www.opensource.apple.com/apsl/ and read it before using this
 11   * file.
 12   * 
 13   * The Original Code and all software distributed under the License are
 14   * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 15   * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 16   * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 17   * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 18   * Please see the License for the specific language governing rights and
 19   * limitations under the License.
 20   * 
 21   * @APPLE_LICENSE_HEADER_END@
 22   */
 23  
 24  //
 25  // xar++ - interface to XAR-format archive files
 26  //
 27  #include "quarantine++.h"
 28  
 29  
 30  namespace Security {
 31  namespace CodeSigning {
 32  
 33  
 34  //
 35  // Check the int result of a qtn API call.
 36  // If the error is "not quarantined," note in the object (no error).
 37  // Other qtn-specific errors are arbitrarily mapped to ENOSYS (this isn't
 38  // important enough to subclass CommonError).
 39  //
 40  void FileQuarantine::check(int err)
 41  {
 42  	switch (err) {
 43  	case 0:
 44  		mQuarantined = true;
 45  		break;
 46  	case QTN_NOT_QUARANTINED:
 47  		mQuarantined = false;
 48  		return;
 49  	default:	// some flavor of quarantine-not-available
 50  		UnixError::throwMe(err);
 51  	}
 52  }
 53  
 54  
 55  FileQuarantine::~FileQuarantine()
 56  {
 57  	if (mQtn)
 58  		qtn_file_free(mQtn);
 59  }
 60  
 61  
 62  FileQuarantine::FileQuarantine(const char *path)
 63  {
 64  	if (!(mQtn = qtn_file_alloc()))
 65  		UnixError::throwMe();
 66  	check(qtn_file_init_with_path(mQtn, path));
 67  }
 68  
 69  FileQuarantine::FileQuarantine(int fd)
 70  {
 71  	if (!(mQtn = qtn_file_alloc()))
 72  		UnixError::throwMe();
 73  	check(qtn_file_init_with_fd(mQtn, fd));
 74  }
 75  
 76  
 77  void FileQuarantine::setFlags(uint32_t flags)
 78  {
 79  	if (mQuarantined)
 80  		check(qtn_file_set_flags(mQtn, flags));
 81  }
 82  
 83  void FileQuarantine::setFlag(uint32_t flag)
 84  {
 85  	if (mQuarantined)
 86  		setFlags(flags() | flag);
 87  }
 88  
 89  void FileQuarantine::clearFlag(uint32_t flag)
 90  {
 91  	if (mQuarantined)
 92  		setFlags(flags() & ~flag);
 93  }
 94  
 95  void FileQuarantine::applyTo(const char *path)
 96  {
 97  	check(qtn_file_apply_to_path(mQtn, path));
 98  }
 99  
100  void FileQuarantine::applyTo(int fd)
101  {
102  	check(qtn_file_apply_to_fd(mQtn, fd));
103  }
104  
105  
106  } // end namespace CodeSigning
107  } // end namespace Security