1  /*
  2   * Copyright (c) 2013-2014 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  #import "KDSecItems.h"
 26  #include <Security/Security.h>
 27  #include <Security/SecItemPriv.h>
 28  
 29  
 30  NSString *kKDSecItemsUpdated = @"KDSecItemsUpdated";
 31  
 32  @interface KDSecItems ()
 33  @property NSMutableArray *items;
 34  @end
 35  
 36  @implementation KDSecItems
 37  
 38  -(NSInteger)numberOfRowsInTableView:(NSTableView*)t
 39  {
 40      return [self.items count];
 41  }
 42  
 43  +(NSString*)nameOfItem:(NSDictionary*)item
 44  {
 45      id name = item[(id)kSecAttrService];
 46      if (name) {
 47          return name;
 48      }
 49      
 50      NSString *path = item[(id)kSecAttrPath];
 51      if (!path) {
 52          path = @"/";
 53      }
 54      NSString *port = item[(id)kSecAttrPort];
 55      if ([@"0" isEqualToString:port] || [@0 isEqual:port]) {
 56          port = @"";
 57      } else {
 58          port = [NSString stringWithFormat:@":%@", port];
 59      }
 60      
 61      return [NSString stringWithFormat:@"%@://%@%@%@", item[(id)kSecAttrProtocol], item[(id)kSecAttrServer], port, path];
 62  }
 63  
 64  - (id)tableView:(NSTableView *)aTableView objectValueForTableColumn:(NSTableColumn *)aTableColumn row:(NSInteger)rowIndex
 65  {
 66      NSString *identifier = [aTableColumn identifier];
 67      
 68      if ([@"account" isEqualToString:identifier]) {
 69          return self.items[rowIndex][(id)kSecAttrAccount];
 70      }
 71      if ([@"name" isEqualToString:identifier]) {
 72          return [KDSecItems nameOfItem:self.items[rowIndex]];
 73      }
 74      
 75      return [NSString stringWithFormat:@"*** c=%@ r%ld", [aTableColumn identifier], (long)rowIndex];
 76  }
 77  
 78  -(NSArray*)fetchItemsMatching:(NSDictionary *)query
 79  {
 80      CFTypeRef raw_items = NULL;
 81      OSStatus result = SecItemCopyMatching((__bridge CFDictionaryRef)(query), &raw_items);
 82      if (result) {
 83          // XXX: UI
 84          NSLog(@"Error result %d - query: %@", result, query);
 85          return nil;
 86      }
 87      if (CFArrayGetTypeID() == CFGetTypeID(raw_items)) {
 88          return (__bridge NSArray*)raw_items;
 89      }
 90      
 91      NSLog(@"Unexpected result type from copyMatching: %@ (query=%@)", raw_items, query);
 92      CFRelease(raw_items);
 93  
 94      return nil;
 95  }
 96  
 97  -(void)loadItems
 98  {
 99      NSDictionary *query_genp = @{(id)kSecClass: (id)kSecClassGenericPassword,
100                                   (__bridge id)kSecAttrSynchronizable: @1,
101                                   (id)kSecMatchLimit: (id)kSecMatchLimitAll,
102                                   (id)kSecReturnAttributes: (id)kCFBooleanTrue};
103      NSDictionary *query_inet = @{(id)kSecClass: (id)kSecClassInternetPassword,
104                                   (__bridge id)kSecAttrSynchronizable: @1,
105                                   (id)kSecMatchLimit: (id)kSecMatchLimitAll,
106                                   (id)kSecReturnAttributes: (id)kCFBooleanTrue};
107      NSArray *nextItems = [[self fetchItemsMatching:query_genp] arrayByAddingObjectsFromArray:[self fetchItemsMatching:query_inet]];
108      self.items = [[nextItems sortedArrayUsingComparator:^NSComparisonResult(id a, id b) {
109          NSDictionary *da = a, *db = b;
110          return [da[(id)kSecAttrService] caseInsensitiveCompare:db[(id)kSecAttrService]];
111      }] mutableCopy];
112          
113      dispatch_async(dispatch_get_main_queue(), ^{
114          [[NSNotificationCenter defaultCenter] postNotificationName:kKDSecItemsUpdated object:self];
115      });
116  }
117  
118  -(id)init
119  {
120      [self loadItems];
121      return self;
122  }
123  
124  @end