1// RUN: rm -rf %t
2// RUN: %clang_cc1 -objcmt-migrate-literals -objcmt-migrate-subscripting -mt-migrate-directory %t %s -x objective-c -triple x86_64-apple-darwin11 
3// RUN: c-arcmt-test -mt-migrate-directory %t | arcmt-test -verify-transformed-files %s.result
4// RUN: %clang_cc1 -triple x86_64-apple-darwin10 -fsyntax-only -x objective-c %s.result
5
6typedef signed char BOOL;
7#define nil ((void*) 0)
8
9typedef const struct __CFString * CFStringRef;
10
11@interface NSObject
12+ (id)alloc;
13@end
14
15@protocol NSCopying
16@end
17
18@interface NSString : NSObject
19+ (id)stringWithString:(NSString *)string;
20- (id)initWithString:(NSString *)aString;
21@end
22
23@interface NSArray : NSObject
24- (id)objectAtIndex:(unsigned long)index;
25@end
26
27@interface NSArray (NSExtendedArray)
28- (id)objectAtIndexedSubscript:(unsigned)idx;
29@end
30
31@interface NSArray (NSArrayCreation)
32+ (id)array;
33+ (id)arrayWithObject:(id)anObject;
34+ (id)arrayWithObjects:(const id [])objects count:(unsigned long)cnt;
35+ (id)arrayWithObjects:(id)firstObj, ...;
36+ (id)arrayWithArray:(NSArray *)array;
37
38- (id)initWithObjects:(const id [])objects count:(unsigned long)cnt;
39- (id)initWithObjects:(id)firstObj, ...;
40- (id)initWithArray:(NSArray *)array;
41@end
42
43@interface NSMutableArray : NSArray
44- (void)replaceObjectAtIndex:(unsigned long)index withObject:(id)anObject;
45@end
46
47@interface NSMutableArray (NSExtendedMutableArray)
48- (void)setObject:(id)obj atIndexedSubscript:(unsigned)idx;
49@end
50
51@interface NSDictionary : NSObject
52- (id)objectForKey:(id)aKey;
53@end
54
55@interface NSDictionary (NSExtendedDictionary)
56- (id)objectForKeyedSubscript:(id)key;
57@end
58
59@interface NSDictionary (NSDictionaryCreation)
60+ (id)dictionary;
61+ (id)dictionaryWithObject:(id)object forKey:(id)key;
62+ (id)dictionaryWithObjects:(const id [])objects forKeys:(const id [])keys count:(unsigned long)cnt;
63+ (id)dictionaryWithObjectsAndKeys:(id)firstObject, ...;
64+ (id)dictionaryWithDictionary:(NSDictionary *)dict;
65+ (id)dictionaryWithObjects:(NSArray *)objects forKeys:(NSArray *)keys;
66
67- (id)initWithObjects:(const id [])objects forKeys:(const id [])keys count:(unsigned long)cnt;
68- (id)initWithObjectsAndKeys:(id)firstObject, ...;
69- (id)initWithDictionary:(NSDictionary *)otherDictionary;
70- (id)initWithObjects:(NSArray *)objects forKeys:(NSArray *)keys;
71@end
72
73@interface NSMutableDictionary : NSDictionary
74- (void)setObject:(id)anObject forKey:(id)aKey;
75@end
76
77@interface NSMutableDictionary (NSExtendedMutableDictionary)
78- (void)setObject:(id)obj forKeyedSubscript:(id <NSCopying>)key;
79@end
80
81@interface NSNumber : NSObject
82@end
83
84@interface NSNumber (NSNumberCreation)
85+ (NSNumber *)numberWithInt:(int)value;
86@end
87
88#define M(x) (x)
89#define PAIR(x) @#x, [NSNumber numberWithInt:(x)]
90#define TWO(x) ((x), (x))
91
92@interface I {
93  NSArray *ivarArr;
94}
95@end
96@implementation I
97-(void) foo {
98  NSString *str;
99  NSArray *arr;
100  NSDictionary *dict;
101
102  arr = [NSArray array];
103  arr = [NSArray arrayWithObject:str];
104  arr = [NSArray arrayWithObjects:str, str, nil];
105  dict = [NSDictionary dictionary];
106  dict = [NSDictionary dictionaryWithObject:arr forKey:str];
107  dict = [NSDictionary dictionaryWithObjectsAndKeys: @"value1", @"key1", @"value2", @"key2", nil];
108  dict = [NSDictionary dictionaryWithObjectsAndKeys: PAIR(1), PAIR(2), nil];
109  dict = [NSDictionary dictionaryWithObjectsAndKeys:
110                                               @"value1", @"key1",
111#ifdef BLAH
112                                               @"value2", @"key2",
113#else
114                                               @"value3", @"key3",
115#endif
116                                               nil ];
117
118  id o = [arr objectAtIndex:2];
119  o = [dict objectForKey:@"key"];
120  o = TWO([dict objectForKey:@"key"]);
121  o = [NSDictionary dictionaryWithObject:[NSDictionary dictionary] forKey:@"key"];
122  NSMutableArray *marr = 0;
123  NSMutableDictionary *mdict = 0;
124  [marr replaceObjectAtIndex:2 withObject:@"val"];
125  [mdict setObject:@"value" forKey:@"key"];
126  [marr replaceObjectAtIndex:2 withObject:[arr objectAtIndex:4]];
127  [mdict setObject:[dict objectForKey:@"key2"] forKey:@"key"];
128  [mdict setObject:[dict objectForKey:@"key2"] forKey:
129#if 1
130                     @"key1"
131#else
132                     @"key2"
133#endif
134                    ];
135  [mdict setObject:[dict objectForKey:
136#if 2
137                     @"key3"
138#else
139                     @"key4"
140#endif
141                   ] forKey:@"key"];
142  [mdict setObject:@"value" forKey:[dict objectForKey:
143#if 3
144                     @"key5"
145#else
146                     @"key6"
147#endif
148                   ] ];
149  [mdict setObject:@"val" forKey:[dict objectForKey:@"key2"]];
150  [mdict setObject:[dict objectForKey:@"key1"] forKey:[dict objectForKey:[NSArray arrayWithObject:@"arrkey"]]];
151  __strong NSArray **parr = 0;
152  o = [*parr objectAtIndex:2];
153  void *hd;
154  o = [(NSArray*)hd objectAtIndex:2];
155  o = [ivarArr objectAtIndex:2];
156
157  dict = [NSDictionary dictionaryWithObjects:[NSArray arrayWithObjects:@"1", [NSArray array], nil] forKeys:[NSArray arrayWithObjects:@"A", [arr objectAtIndex:2], nil]];
158  dict = [NSDictionary dictionaryWithObjects:[NSArray arrayWithObjects:@"1", @"2", nil] forKeys:arr];
159  dict = [NSDictionary dictionaryWithObjects:[NSArray arrayWithObjects:@"1", @"2", nil] forKeys:@[@"A", @"B"]];
160  dict = [NSDictionary dictionaryWithObjectsAndKeys:[NSArray array], @"A", [NSArray array], @"B", nil];
161}
162@end
163
164extern const CFStringRef globStr;
165
166void test1(NSString *str) {
167  NSDictionary *dict = [NSDictionary dictionaryWithObjectsAndKeys: str, globStr, nil];
168  dict = [NSDictionary dictionaryWithObjectsAndKeys: globStr, str, nil];
169  dict = [NSDictionary dictionaryWithObject:str forKey:globStr];
170  dict = [NSDictionary dictionaryWithObject:globStr forKey:str];
171
172  NSArray *arr = [NSArray arrayWithObjects: globStr, globStr, nil];
173  arr = [NSArray arrayWithObjects: str, globStr, nil];
174  arr = [NSArray arrayWithObjects: globStr, str, nil];
175  arr = [NSArray arrayWithObject:globStr];
176}
177
178@interface Custom : NSObject
179- (id)objectAtIndex:(unsigned long)index;
180@end
181
182@interface Custom (Extended)
183- (id)objectAtIndexedSubscript:(unsigned)idx;
184@end
185
186@interface MutableCustom : Custom
187- (void)replaceObjectAtIndex:(unsigned long)index withObject:(id)anObject;
188@end
189
190@interface MutableCustom (Extended)
191- (void)setObject:(id)obj atIndexedSubscript:(unsigned)idx;
192@end
193
194@interface CustomUnavail : NSObject
195- (id)objectAtIndex:(unsigned long)index;
196@end
197
198@interface CustomUnavail (Extended)
199- (id)objectAtIndexedSubscript:(unsigned)idx __attribute__((unavailable));
200@end
201
202@interface MutableCustomUnavail : CustomUnavail
203- (void)replaceObjectAtIndex:(unsigned long)index withObject:(id)anObject;
204@end
205
206@interface MutableCustomUnavail (Extended)
207- (void)setObject:(id)obj atIndexedSubscript:(unsigned)idx __attribute__((unavailable));
208@end
209
210void test2() {
211  MutableCustom *mutc;
212  id o = [mutc objectAtIndex:4];
213  [mutc replaceObjectAtIndex:2 withObject:@"val"];
214
215  MutableCustomUnavail *mutcunaval;
216  o = [mutcunaval objectAtIndex:4];
217  [mutcunaval replaceObjectAtIndex:2 withObject:@"val"];
218}
219
220@interface NSLocale : NSObject
221+ (id)systemLocale;
222+ (id)currentLocale;
223- (id)objectForKey:(id)key;
224@end
225
226void test3(id key) {
227  id o = [[NSLocale currentLocale] objectForKey:key];
228}
229