#include <Security/SecureObjectSync/SOSPeer.h>
#include <Security/SecureObjectSync/SOSPeerCoder.h>
#include <Security/SecureObjectSync/SOSTransportMessage.h>
#include <Security/SecureObjectSync/SOSCoder.h>
#include <Security/SecureObjectSync/SOSAccount.h>
#include <Security/SecureObjectSync/SOSEngine.h>
#include <utilities/debugging.h>
#include <utilities/SecCFWrappers.h>
#include <AssertMacros.h>
#include "SOSInternal.h"
void SOSPeerCoderConsume(SOSEnginePeerMessageSentBlock *sent, bool ok) {
if (*sent)
(*sent)(ok);
}
enum SOSCoderUnwrapStatus SOSPeerHandleCoderMessage(SOSPeerRef peer, SOSCoderRef coder, CFStringRef peer_id, CFDataRef codedMessage, CFDataRef *decodedMessage, bool *forceSave, CFErrorRef *error) {
enum SOSCoderUnwrapStatus result = SOSCoderUnwrapError;
CFMutableDataRef localDecodedMessage = NULL;
SOSCoderStatus coderStatus = kSOSCoderDataReturned;
require_action_quiet(coder, xit, secerror("%@ getCoder: %@", peer_id, error ? *error : NULL));
CFErrorRef localError = NULL;
if (coder) {
coderStatus = SOSCoderUnwrap(coder, codedMessage, &localDecodedMessage, peer_id, error);
switch(coderStatus) {
case kSOSCoderDataReturned: {
logRawMessage(localDecodedMessage, false, 0);
result = SOSCoderUnwrapDecoded;
break;
}
case kSOSCoderNegotiating: result = SOSCoderUnwrapHandled;
secnotice("engine", "%@ engine negotiating", peer_id);
break;
case kSOSCoderNegotiationCompleted:
SOSPeerDidConnect(peer);
result = SOSCoderUnwrapHandled;
*forceSave = true;
secnotice("engine", "%@ engine negotiation complete", peer_id);
break;
case kSOSCoderFailure: secnotice("engine", "%@ engine failed handling message %@", peer_id, error ? *error : NULL);
SOSCoderReset(coder);
if(SOSCoderStart(coder, &localError) == kSOSCoderFailure){
secerror("Attempt to recover coder failed to restart: %@", localError);
}
break;
case kSOSCoderStaleEvent: secinfo("engine", "%@ engine stale event ignored", peer_id);
result = SOSCoderUnwrapHandled;
break;
case kSOSCoderForceMessage:
SOSPeerSetMustSendMessage(peer, true);
result = SOSCoderUnwrapHandled;
break;
case kSOSCoderTooNew: secnotice("engine", "%@ engine received a message too soon, time to restart", peer_id);
SOSCoderReset(coder);
if(SOSCoderStart(coder, &localError) == kSOSCoderFailure){
secerror("Attempt to recover coder failed to restart: %@", localError);
}
break;
default:
assert(false);
break;
}
if(decodedMessage)
*decodedMessage = CFRetainSafe(localDecodedMessage);
CFReleaseNull(localDecodedMessage);
}
CFReleaseNull(localError);
xit:
return result;
}
bool SOSPeerCoderSendMessageIfNeeded(SOSEngineRef engine, SOSTransactionRef txn, SOSPeerRef peer, SOSCoderRef coder, CFDataRef *message_to_send, CFStringRef circle_id, CFStringRef peer_id, SOSEnginePeerMessageSentBlock *sent, CFErrorRef *error) {
bool ok = false;
require_action_quiet(coder, xit, secerror("%@ getCoder: %@", peer_id, error ? *error : NULL));
if (SOSCoderCanWrap(coder)) {
secinfo("transport", "%@ Coder can wrap, getting message from engine", peer_id);
CFMutableDataRef codedMessage = NULL;
CFDataRef message = SOSEngineCreateMessage_locked(engine, txn, peer, error, sent);
if (!message) {
secnotice("transport", "%@ SOSEngineCreateMessageToSyncToPeer failed: %@", peer_id, *error);
} else if (CFDataGetLength(message) || SOSPeerMustSendMessage(peer)) {
ok = message && (SOSCoderWrap(coder, message, &codedMessage, peer_id, error) == kSOSCoderDataReturned);
if (!ok) {
secnotice("transport", "%@ SOSCoderWrap failed: %@", peer_id, *error);
} else {
CFRetainAssign(*message_to_send, codedMessage);
}
CFReleaseNull(codedMessage);
} else {
ok = true;
}
CFReleaseNull(message);
} else {
*message_to_send = SOSCoderCopyPendingResponse(coder);
secinfo("transport", "%@ negotiating, %@", peer_id, message_to_send ? CFSTR("sending negotiation message.") : CFSTR("waiting for negotiation message."));
*sent = Block_copy(^(bool wasSent){
if (wasSent)
SOSCoderConsumeResponse(coder);
});
ok = true;
}
xit:
return ok;
}