+++ /dev/null
-/*
- * Copyright (c) 2010-2011 Apple Inc. All Rights Reserved.
- *
- * @APPLE_LICENSE_HEADER_START@
- *
- * This file contains Original Code and/or Modifications of Original Code
- * as defined in and that are subject to the Apple Public Source License
- * Version 2.0 (the 'License'). You may not use this file except in
- * compliance with the License. Please obtain a copy of the License at
- * http://www.opensource.apple.com/apsl/ and read it before using this
- * file.
- *
- * The Original Code and all software distributed under the License are
- * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
- * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
- * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
- * Please see the License for the specific language governing rights and
- * limitations under the License.
- *
- * @APPLE_LICENSE_HEADER_END@
- */
-
-#include "Transform.h"
-#include "SecTransform.h"
-#include "SecCollectTransform.h"
-#include "SecCustomTransform.h"
-#include "misc.h"
-#include "c++utils.h"
-#include "Utilities.h"
-
-static CFStringRef kCollectTransformName = CFSTR("com.apple.security.seccollecttransform");
-
-static SecTransformInstanceBlock CollectTransform(CFStringRef name,
- SecTransformRef newTransform,
- SecTransformImplementationRef ref)
-{
- SecTransformInstanceBlock instanceBlock =
- ^{
- __block CFMutableArrayRef allValues =
- CFArrayCreateMutable(kCFAllocatorDefault, 0, &kCFTypeArrayCallBacks);
- __block Boolean isSameType = TRUE;
- CFTypeRef input_ah = SecTranformCustomGetAttribute(ref, kSecTransformInputAttributeName, kSecTransformMetaAttributeRef);
- ah2ta(input_ah)->direct_error_handling = 1;
-
- dispatch_block_t no_more_output = ^
- {
- SecTransformSetAttributeAction(ref, kSecTransformActionAttributeNotification, input_ah, ^(SecTransformStringOrAttributeRef a, CFTypeRef v) { return v; });
- };
-
- // Create a block to deal with out of memory errors
- dispatch_block_t oom = ^
- {
- CFTypeRefHolder localErr(GetNoMemoryError());
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, localErr.Get());
- no_more_output();
- };
-
- SecTransformSetTransformAction(ref, kSecTransformActionFinalize,
- ^()
- {
- if (NULL != allValues)
- {
- CFRelease(allValues);
- }
-
- return (CFTypeRef) NULL;
- });
-
- SecTransformSetAttributeAction(ref, kSecTransformActionAttributeNotification,
- input_ah,
- ^(SecTransformStringOrAttributeRef attribute, CFTypeRef value)
- {
- CFIndex len = CFArrayGetCount(allValues);
-
-#if 0
- if (NULL == value && 0 == len)
- {
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, NULL);
- no_more_output();
- return value;
- }
-#endif
-
- if (value && isSameType && len > 0)
- {
- isSameType = CFGetTypeID(CFArrayGetValueAtIndex(allValues, 0)) == CFGetTypeID(value);
- }
-
- if (value)
- {
-
- if (CFGetTypeID(value) == CFErrorGetTypeID())
- {
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, value);
- no_more_output();
- return value;
- }
-
- // For mutable types, we want an immutable copy.
- /// XXX: write a more general CFImutableCopy and use it here.
- if (CFGetTypeID(value) == CFDataGetTypeID())
- {
- CFDataRef copy = CFDataCreateCopy(NULL, (CFDataRef)value);
-
- CFArrayAppendValue(allValues, copy);
- CFRelease(copy);
- }
- else
- {
- CFArrayAppendValue(allValues, value);
- }
-
- if (CFArrayGetCount(allValues) != len +1)
- {
- oom();
- return value;
- }
- }
- else
- {
- if (isSameType)
- {
- // Deal with data or no items at all
- CFTypeID type = CFArrayGetCount(allValues) ?
- CFGetTypeID(CFArrayGetValueAtIndex(allValues, 0)) : CFDataGetTypeID();
- if (CFDataGetTypeID() == type)
- {
- CFIndex total_len = 0;
- CFIndex prev_total_len = 0;
- CFIndex i;
- const CFIndex n_datas = CFArrayGetCount(allValues);
-
- for(i = 0; i < n_datas; i++)
- {
- total_len +=
- CFDataGetLength((CFDataRef)CFArrayGetValueAtIndex(allValues, i));
- if (total_len < prev_total_len)
- {
- oom();
- return value;
- }
- prev_total_len = total_len;
- }
-
- CFMutableDataRef result = CFDataCreateMutable(NULL, total_len);
- if (!result)
- {
- oom();
- return value;
- }
-
- for(i = 0; i < n_datas; i++)
- {
- CFDataRef d = (CFDataRef)CFArrayGetValueAtIndex(allValues, i);
- CFDataAppendBytes(result, CFDataGetBytePtr(d), CFDataGetLength(d));
- }
-
- if (CFDataGetLength(result) != total_len)
- {
- oom();
- return value;
- }
-
- CFDataRef resultData = CFDataCreateCopy(NULL, result);
-
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)resultData);
-
- CFRelease(resultData);
-
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)value);
- no_more_output();
-
- CFRelease(result);
- return value;
- }
- else if (CFStringGetTypeID() == type)
- {
- // deal with strings
- CFStringRef resultStr = CFStringCreateByCombiningStrings(NULL, allValues, CFSTR(""));
-
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)resultStr);
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)value);
- no_more_output();
-
- return value;
- }
- else
- {
- // special case the singleton
- if (1 == CFArrayGetCount(allValues))
- {
- CFTypeRef result = (CFTypeRef)CFRetain(CFArrayGetValueAtIndex(allValues, 0));
-
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)result);
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)value);
- no_more_output();
-
- return value;
- }
- }
- }
- // Fall through for non-homogenous or un-mergable type
- CFArrayRef resultArray = CFArrayCreateCopy(kCFAllocatorDefault, allValues);
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)resultArray);
- SecTransformCustomSetAttribute(ref, kSecTransformOutputAttributeName,
- kSecTransformMetaAttributeValue, (CFTypeRef)value);
- no_more_output();
-
- return value;
- }
-
- return value;
-
- });
-
- return (CFErrorRef)NULL;
- };
-
- return Block_copy(instanceBlock);
-}
-
-SecTransformRef SecCreateCollectTransform(CFErrorRef* error)
-{
- static dispatch_once_t once;
- __block Boolean ok = TRUE;
-
- dispatch_block_t aBlock = ^
- {
- ok = SecTransformRegister(kCollectTransformName, &CollectTransform, error);
- };
-
- dispatch_once(&once, aBlock);
-
- if (!ok)
- {
- return NULL;
- }
-
- SecTransformRef yatz = SecTransformCreate(kCollectTransformName, error);
- return yatz;
-}
projects / apple / security.git / blobdiff