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
30namespace Security {
31namespace 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//
40void 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
55FileQuarantine::~FileQuarantine()
56{
57	if (mQtn)
58		qtn_file_free(mQtn);
59}
60
61
62FileQuarantine::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
69FileQuarantine::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
77void FileQuarantine::setFlags(uint32_t flags)
78{
79	if (mQuarantined)
80		check(qtn_file_set_flags(mQtn, flags));
81}
82
83void FileQuarantine::setFlag(uint32_t flag)
84{
85	if (mQuarantined)
86		setFlags(flags() | flag);
87}
88
89void FileQuarantine::clearFlag(uint32_t flag)
90{
91	if (mQuarantined)
92		setFlags(flags() & ~flag);
93}
94
95void FileQuarantine::applyTo(const char *path)
96{
97	check(qtn_file_apply_to_path(mQtn, path));
98}
99
100void 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
108