+++ /dev/null
-/*
- * Copyright (c) 2012-2014 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 <SecureObjectSync/SOSEngine.h>
-#include <SecureObjectSync/SOSPeer.h>
-
-#include "SOSCircle_regressions.h"
-
-#include <corecrypto/ccsha2.h>
-#include <Security/SecBase64.h>
-
-#include <utilities/SecCFWrappers.h>
-
-#include <stdint.h>
-#include "SOSTestDataSource.h"
-
-#if 1
-static int kTestTestCount = 0;
-#else
-static int kTestTestCount = 61;
-
-static void tests(void)
-{
- CFErrorRef error = NULL;
-
- /* Transport. */
- __block unsigned msg_count = 0;
- uint8_t msg_digest_buffer[CCSHA256_OUTPUT_SIZE];
- uint8_t *msg_digest = msg_digest_buffer;
-
- SOSPeerSendBlock sendBlock = ^bool (CFDataRef message, CFErrorRef *error) {
- size_t msglen = CFDataGetLength(message);
- const uint8_t *msg = CFDataGetBytePtr(message);
- const struct ccdigest_info *sha256 = ccsha256_di();
- if (msg_count++ == 0) {
- /* message n=0 */
- ccdigest(sha256, msglen, msg, msg_digest);
- } else {
- /* message n=n+1 */
- ccdigest_di_decl(sha256, sha256_ctx);
- ccdigest_init(sha256, sha256_ctx);
- ccdigest_update(sha256, sha256_ctx, sizeof(msg_digest_buffer), msg_digest);
- ccdigest_update(sha256, sha256_ctx, msglen, msg);
- ccdigest_final(sha256, sha256_ctx, msg_digest);
- }
- size_t encmaxlen = SecBase64Encode(msg, msglen, NULL, 0);
- CFMutableDataRef encoded = CFDataCreateMutable(NULL, encmaxlen);
- CFDataSetLength(encoded, encmaxlen);
- SecBase64Result rc;
- char *enc = (char *)CFDataGetMutableBytePtr(encoded);
-#ifndef NDEBUG
- size_t enclen =
-#endif
- SecBase64Encode2(msg, msglen,
- enc,
- encmaxlen, kSecB64_F_LINE_LEN_USE_PARAM,
- 64, &rc);
- assert(enclen < INT32_MAX);
-// printf("=== BEGIN SOSMESSAGE ===\n%.*s\n=== END SOSMESSAGE ===\n", (int)enclen, enc);
- CFRelease(encoded);
- return true;
- };
-
- /* Create peer test. */
- CFStringRef peer_id = CFSTR("peer 70");
- SOSPeerRef peer;
- ok(peer = SOSPeerCreateSimple(peer_id, kSOSPeerVersion, &error),
- "create peer: %@", error);
- CFReleaseNull(error);
-
- /* DataSource */
- SOSDataSourceRef ds = SOSTestDataSourceCreate();
-
- /* Create engine test. */
- SOSEngineRef engine;
- ok(engine = SOSEngineCreate(ds, &error), "create engine: %@", error);
- CFReleaseNull(error);
-
- /* Make sure the engine did not yet send a any messages to the peer. */
- unsigned expected_msg_count = 0;
- is(msg_count, expected_msg_count, "Engine sent %d/%d messages",
- msg_count, expected_msg_count);
-
- /* Test passing peer messages to the engine. */
- CFDataRef message;
-
- /* Hand an empty message to the engine for handeling. */
- message = CFDataCreate(NULL, NULL, 0);
- ok(false == SOSEngineHandleMessage(engine, peer, message, &error),
- "handle empty message: %@", error);
- ok(false == SOSPeerSendMessageIfNeeded(engine, peer, &error, sendBlock),
- "send response: %@", error);
- CFReleaseNull(error);
- CFReleaseNull(message);
- /* Make sure the engine did not yet send a response to the peer. */
- is(msg_count, expected_msg_count, "Engine sent %d/%d messages",
- msg_count, expected_msg_count);
-
-
- /* Create manifest digest message. */
- SKIP: {
- skip("Create manifest digest message failed", 2,
- ok(message = SOSEngineCreateManifestDigestMessage(engine, peer,
- &error),
- "create manifest digest message: %@", error));
- CFReleaseNull(error);
- /* Pass manifest digest message to the engine for handeling. */
- skip("", 1,
- ok(SOSEngineHandleMessage(engine, peer, message, &error),
- "handle manifest digest message: %@", error));
- /* Make sure the engine sent a response to the peer. */
- expected_msg_count++;
- is(msg_count, expected_msg_count, "Engine sent %d/%d messages",
- msg_count, expected_msg_count);
- }
- CFReleaseNull(message);
- CFReleaseNull(error);
-
- /* Create manifest message. */
- SKIP: {
- skip("Create manifest message failed", 2,
- ok(message = SOSEngineCreateManifestMessage(engine, peer, &error),
- "create manifest message: %@", error));
- CFReleaseNull(error);
- /* Pass manifest message to the engine for handeling. */
- skip("", 1,
- ok(SOSEngineHandleMessage(engine, peer, message, &error),
- "handle manifest message: %@", error));
- /* Make sure the engine sent a response to the peer. */
- expected_msg_count++;
- is(msg_count, expected_msg_count, "Engine sent %d/%d messages",
- msg_count, expected_msg_count);
- }
- CFReleaseNull(message);
- CFReleaseNull(error);
-
- /* Create manifest and objects message. */
- SKIP: {
- skip("Create manifest and objects message failed", 2,
- ok(message = SOSEngineCreateManifestAndObjectsMessage(engine, peer,
- &error),
- "create manifest and objects message: %@", error));
- CFReleaseNull(error);
- /* Pass manifest and objects message to the engine for handeling. */
- skip("", 1,
- ok(SOSEngineHandleMessage(engine, peer, message, &error),
- "handle manifest and objects message: %@", error));
- /* Make sure the engine sent a response to the peer. */
- expected_msg_count++;
- is(msg_count, expected_msg_count, "Engine sent %d/%d messages",
- msg_count, expected_msg_count);
- }
- CFReleaseNull(message);
- CFReleaseNull(error);
-
- /* Clean up. */
- CFReleaseSafe(peer);
- SOSEngineDispose(engine);
-}
-
-static CFStringRef SOSMessageCopyDigestHex(CFDataRef message) {
- uint8_t digest[CCSHA1_OUTPUT_SIZE];
- ccdigest(ccsha1_di(), CFDataGetLength(message), CFDataGetBytePtr(message), digest);
- CFMutableStringRef hex = CFStringCreateMutable(0, 2 * sizeof(digest));
- for (unsigned int ix = 0; ix < sizeof(digest); ++ix) {
- CFStringAppendFormat(hex, 0, CFSTR("%02X"), digest[ix]);
- }
- return hex;
-}
-
-static void testsync(const char *name, void (^aliceInit)(SOSDataSourceRef ds), void (^bobInit)(SOSDataSourceRef ds), CFStringRef msg, ...) {
- CFErrorRef error = NULL;
-
- /* Setup Alice engine, dataSource and peer for Alice to talk to Bob */
- SOSDataSourceRef aliceDataSource = SOSTestDataSourceCreate();
- SOSEngineRef aliceEngine;
- ok(aliceEngine = SOSEngineCreate(aliceDataSource, &error), "create alice engine: %@", error);
- CFReleaseNull(error);
- CFStringRef bobID = CFStringCreateWithFormat(kCFAllocatorDefault, 0, CFSTR("Bob-%s"), name);
-
- __block CFDataRef queued_message = NULL;
-
- SOSPeerSendBlock enqueueMessage = ^bool (CFDataRef message, CFErrorRef *error) {
- if (queued_message)
- fail("We already had an unproccessed message");
-
- queued_message = (CFDataRef) CFRetain(message);
- return true;
- };
-
- CFDataRef (^dequeueMessage)() = ^CFDataRef () {
- CFDataRef result = queued_message;
- queued_message = NULL;
-
- return result;
- };
-
- SOSPeerRef bobPeer;
- ok(bobPeer = SOSPeerCreateSimple(bobID, kSOSPeerVersion, &error, enqueueMessage),
- "create peer: %@", error);
-
- /* Setup Bob engine, dataSource and peer for Bob to talk to Alice */
- SOSDataSourceRef bobDataSource = SOSTestDataSourceCreate();
- SOSEngineRef bobEngine;
- ok(bobEngine = SOSEngineCreate(bobDataSource, &error), "create bob engine: %@", error);
- CFReleaseNull(error);
- CFStringRef aliceID = CFStringCreateWithFormat(kCFAllocatorDefault, 0, CFSTR("Alice-%s"), name);
-
- SOSPeerRef alicePeer;
- ok(alicePeer = SOSPeerCreateSimple(aliceID, kSOSPeerVersion, &error, enqueueMessage),
- "create peer: %@", error);
- CFReleaseNull(error);
-
- /* Now call provided setup blocks to populate the dataSources with
- interesting stuff. */
- aliceInit(aliceDataSource);
- bobInit(bobDataSource);
-
- CFDataRef message;
-
- va_list msgs;
- va_start(msgs, msg);
-
- int msg_index = 0;
- bool alice = false;
- for (;;) {
- message = dequeueMessage();
- msg_index++;
- /* We are expecting a message and msg is it's digest. */
- if (message) {
- CFStringRef messageDigestStr = SOSMessageCopyDigestHex(message);
- if (msg) {
- bool handeled = SOSEngineHandleMessage(alice ? aliceEngine : bobEngine, alice ? bobPeer : alicePeer, message, &error);
- if (!CFEqual(messageDigestStr, msg)) {
- if (handeled) {
- fail("%s %s received message [%d] digest %@ != %@ %@", name, alice ? "Alice" : "Bob", msg_index, messageDigestStr, msg, message);
- } else {
- fail("%s %s failed to handle message [%d] digest %@ != %@ %@: %@", name, alice ? "Alice" : "Bob", msg_index, messageDigestStr, msg, message, error);
- CFReleaseNull(error);
- }
- } else if (handeled) {
- pass("%s %s handled message [%d] %@", name, alice ? "Alice" : "Bob", msg_index, message);
- } else {
- fail("%s %s failed to handle message [%d] %@: %@", name, alice ? "Alice" : "Bob", msg_index, message, error);
- CFReleaseNull(error);
- }
- } else {
- fail("%s %s sent extra message [%d] with digest %@: %@", name, alice ? "Bob" : "Alice", msg_index, messageDigestStr, message);
- }
- CFRelease(messageDigestStr);
- CFRelease(message);
- } else {
- if (msg) {
- fail("%s %s expected message [%d] with digest %@, none received", name, alice ? "Alice" : "Bob", msg_index, msg);
- } else {
- /* Compare alice and bobs dataSources databases. */
- ok(CFEqual(SOSTestDataSourceGetDatabase(aliceDataSource), SOSTestDataSourceGetDatabase(bobDataSource)), "%s Alice and Bob are in sync", name);
- }
- }
-
- if (msg) {
- alice = !alice;
- msg = va_arg(msgs, CFStringRef);
- } else
- break;
- }
-
- va_end(msgs);
-
-
-
-
- SOSEngineDispose(aliceEngine); // Also disposes aliceDataSource
- CFReleaseSafe(alicePeer);
- CFReleaseSafe(aliceID);
-
- SOSEngineDispose(bobEngine); // Also disposes bobDataSource
- CFReleaseSafe(bobPeer);
- CFReleaseSafe(bobID);
-}
-
-#if 0
-static void synctests(void) {
- // Sync between 2 empty dataSources
- testsync("empty",
- ^ (SOSDataSourceRef dataSource) {},
- ^ (SOSDataSourceRef dataSource) {},
- CFSTR("2AF312E092D67308A0083DFFBF2B6B754B967864"),
- CFSTR("2AF312E092D67308A0083DFFBF2B6B754B967864"),
- CFSTR("2AF312E092D67308A0083DFFBF2B6B754B967864"),
- NULL);
-
- // Sync a dataSource with one object to an empty dataSource
- testsync("alice1",
- ^ (SOSDataSourceRef dataSource) {
- CFErrorRef error = NULL;
- SOSObjectRef object = SOSDataSourceCreateGenericItem(dataSource, CFSTR("test_account"), CFSTR("test service"));
- ok(dataSource->add(dataSource, object, &error), "dataSource added object %@", error);
- CFReleaseSafe(object);
- CFReleaseNull(error);
- },
- ^ (SOSDataSourceRef dataSource) {},
- CFSTR("ADAA3ACE75ED516CB91893413EE9CC9ED04CA47B"),
- CFSTR("147B6C509908CC4A9FC4263973A842104A64CE01"),
- CFSTR("019B494F3C06B48BB02C280AF1E19AD861A7003C"),
- CFSTR("ADAA3ACE75ED516CB91893413EE9CC9ED04CA47B"),
- NULL);
-
- // Sync a dataSource with one object to another dataSource with the same object
- testsync("alice1bob1",
- ^ (SOSDataSourceRef dataSource) {
- CFErrorRef error = NULL;
- SOSObjectRef object = SOSDataSourceCreateGenericItem(dataSource, CFSTR("test_account"), CFSTR("test service"));
- ok(dataSource->add(dataSource, object, &error), "dataSource added object %@", error);
- CFReleaseSafe(object);
- CFReleaseNull(error);
- },
- ^ (SOSDataSourceRef dataSource) {
- CFErrorRef error = NULL;
- SOSObjectRef object = SOSDataSourceCreateGenericItem(dataSource, CFSTR("test_account"), CFSTR("test service"));
- ok(dataSource->add(dataSource, object, &error), "dataSource added object %@", error);
- CFReleaseSafe(object);
- CFReleaseNull(error);
- },
- CFSTR("ADAA3ACE75ED516CB91893413EE9CC9ED04CA47B"),
- CFSTR("ADAA3ACE75ED516CB91893413EE9CC9ED04CA47B"),
- CFSTR("ADAA3ACE75ED516CB91893413EE9CC9ED04CA47B"),
- NULL);
-
- // Sync a dataSource with one object to another dataSource with the same object
- testsync("alice1bob2",
- ^ (SOSDataSourceRef dataSource) {
- CFErrorRef error = NULL;
- SOSObjectRef object = SOSDataSourceCreateGenericItem(dataSource, CFSTR("test_account"), CFSTR("test service"));
- ok(dataSource->add(dataSource, object, &error), "dataSource added object %@", error);
- CFReleaseSafe(object);
- CFReleaseNull(error);
- },
- ^ (SOSDataSourceRef dataSource) {
- CFErrorRef error = NULL;
- SOSObjectRef object = SOSDataSourceCreateGenericItem(dataSource, CFSTR("test_account"), CFSTR("test service"));
- ok(dataSource->add(dataSource, object, &error), "dataSource added object %@", error);
- CFReleaseSafe(object);
- object = SOSDataSourceCreateGenericItem(dataSource, CFSTR("account1"), CFSTR("service1"));
- ok(dataSource->add(dataSource, object, &error), "dataSource added object %@", error);
- CFReleaseSafe(object);
- CFReleaseNull(error);
- },
- CFSTR("ADAA3ACE75ED516CB91893413EE9CC9ED04CA47B"),
- CFSTR("D4049A1063CFBF7CAF8424E13DE3CE926FF5856C"),
- CFSTR("9624EA855BBED6B668868BB723443E804D04F6A1"),
- CFSTR("063E097CCD4FEB7F3610ED12B3DA828467314846"),
- CFSTR("D1B3944E3084425F41B2C2EA0BE82170E10AA37D"),
- NULL);
-
-}
-#endif
-
-#endif
-
-int sc_70_engine(int argc, char *const *argv)
-{
- plan_tests(kTestTestCount);
-
- //tests();
- //synctests();
- //testsync(NULL, NULL, NULL, NULL);
-
- return 0;
-}
projects / apple / security.git / blobdiff