1 // RUN: %clang_cc1 -fsyntax-only -Wthread-safety -verify %s
2 
3 typedef int __attribute__((capability("role"))) ThreadRole;
4 struct __attribute__((shared_capability("mutex"))) Mutex {};
5 struct NotACapability {};
6 
7 // Put capability attributes on unions
8 union __attribute__((capability("mutex"))) MutexUnion { int a; char* b; };
9 typedef union { int a; char* b; } __attribute__((capability("mutex"))) MutexUnion2;
10 
11 // Test a different capability name
12 struct __attribute__((capability("custom"))) CustomName {};
13 
14 int Test1 __attribute__((capability("test1")));  // expected-error {{'capability' attribute only applies to structs, unions, classes, and typedefs}}
15 int Test2 __attribute__((shared_capability("test2"))); // expected-error {{'shared_capability' attribute only applies to structs, unions, classes, and typedefs}}
16 int Test3 __attribute__((acquire_capability("test3")));  // expected-warning {{'acquire_capability' attribute only applies to functions}}
17 int Test4 __attribute__((try_acquire_capability("test4"))); // expected-error {{'try_acquire_capability' attribute only applies to functions}}
18 int Test5 __attribute__((release_capability("test5"))); // expected-warning {{'release_capability' attribute only applies to functions}}
19 
20 struct __attribute__((capability(12))) Test3 {}; // expected-error {{'capability' attribute requires a string}}
21 struct __attribute__((shared_capability(Test2))) Test4 {}; // expected-error {{'shared_capability' attribute requires a string}}
22 
23 struct __attribute__((capability)) Test5 {}; // expected-error {{'capability' attribute takes one argument}}
24 struct __attribute__((shared_capability("test1", 12))) Test6 {}; // expected-error {{'shared_capability' attribute takes one argument}}
25 
26 struct NotACapability BadCapability;
27 ThreadRole GUI, Worker;
Func1(void)28 void Func1(void) __attribute__((requires_capability(GUI))) {}
Func2(void)29 void Func2(void) __attribute__((requires_shared_capability(Worker))) {}
30 
Func3(void)31 void Func3(void) __attribute__((requires_capability)) {}  // expected-error {{'requires_capability' attribute takes at least 1 argument}}
Func4(void)32 void Func4(void) __attribute__((requires_shared_capability)) {}  // expected-error {{'requires_shared_capability' attribute takes at least 1 argument}}
33 
Func5(void)34 void Func5(void) __attribute__((requires_capability(1))) {}  // expected-warning {{'requires_capability' attribute requires arguments whose type is annotated with 'capability' attribute; type here is 'int'}}
Func6(void)35 void Func6(void) __attribute__((requires_shared_capability(BadCapability))) {}  // expected-warning {{'requires_shared_capability' attribute requires arguments whose type is annotated with 'capability' attribute; type here is 'struct NotACapability'}}
36 
Func7(void)37 void Func7(void) __attribute__((assert_capability(GUI))) {}
Func8(void)38 void Func8(void) __attribute__((assert_shared_capability(GUI))) {}
39 
Func9(void)40 void Func9(void) __attribute__((assert_capability())) {} // expected-warning {{'assert_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
Func10(void)41 void Func10(void) __attribute__((assert_shared_capability())) {} // expected-warning {{'assert_shared_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
42 
Func11(void)43 void Func11(void) __attribute__((acquire_capability(GUI))) {}
Func12(void)44 void Func12(void) __attribute__((acquire_shared_capability(GUI))) {}
45 
Func13(void)46 void Func13(void) __attribute__((acquire_capability())) {} // expected-warning {{'acquire_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
Func14(void)47 void Func14(void) __attribute__((acquire_shared_capability())) {} // expected-warning {{'acquire_shared_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
48 
Func15(void)49 void Func15(void) __attribute__((release_capability(GUI))) {}
Func16(void)50 void Func16(void) __attribute__((release_shared_capability(GUI))) {}
Func17(void)51 void Func17(void) __attribute__((release_generic_capability(GUI))) {}
52 
Func18(void)53 void Func18(void) __attribute__((release_capability())) {} // expected-warning {{'release_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
Func19(void)54 void Func19(void) __attribute__((release_shared_capability())) {} // expected-warning {{'release_shared_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
Func20(void)55 void Func20(void) __attribute__((release_generic_capability())) {} // expected-warning {{'release_generic_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
56 
Func21(void)57 void Func21(void) __attribute__((try_acquire_capability(1))) {} // expected-warning {{'try_acquire_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
Func22(void)58 void Func22(void) __attribute__((try_acquire_shared_capability(1))) {} // expected-warning {{'try_acquire_shared_capability' attribute without capability arguments can only be applied to non-static methods of a class}}
59 
Func23(void)60 void Func23(void) __attribute__((try_acquire_capability(1, GUI))) {}
Func24(void)61 void Func24(void) __attribute__((try_acquire_shared_capability(1, GUI))) {}
62 
Func25(void)63 void Func25(void) __attribute__((try_acquire_capability())) {} // expected-error {{'try_acquire_capability' attribute takes at least 1 argument}}
Func26(void)64 void Func26(void) __attribute__((try_acquire_shared_capability())) {} // expected-error {{'try_acquire_shared_capability' attribute takes at least 1 argument}}
65 
66 // Test that boolean logic works with capability attributes
67 void Func27(void) __attribute__((requires_capability(!GUI)));
68 void Func28(void) __attribute__((requires_capability(GUI && Worker)));
69 void Func29(void) __attribute__((requires_capability(GUI || Worker)));
70 void Func30(void) __attribute__((requires_capability((Worker || Worker) && !GUI)));
71 
72 int AlsoNotACapability;
73 void Func31(void) __attribute__((requires_capability(GUI && AlsoNotACapability))); // expected-warning {{'requires_capability' attribute requires arguments whose type is annotated with 'capability' attribute; type here is 'int'}}
74