public int Pull()
        {
            int changes = 0;

            ThrowExceptionIfNotConnected();

            lock (mutex)
            {
                Log("Sending PULL command.");
                BinaryEncoder.SendCommand(stream, ConnectionCommand.PULL);
#if (DEBUG)
                Log("Sending ClientPullCapsula.");
#endif
                ClientPullCapsula clientCapsula;
                using (Context context = new Context(config))
                {
                    clientCapsula = new ClientPullCapsula()
                    {
                        AesKeysUserIds = context.Contacts
                                         .Where(u => u.ReceiveAesKey == null)
                                         .Select(u => u.PublicId)
                                         .ToArray()
                    };
                }
                TextEncoder.SendJson(stream, clientCapsula);

                ServerPullCapsula capsula = TextEncoder.ReadJson <ServerPullCapsula>(stream);
#if (DEBUG)
                Log("Received ServerPullCapsula.");
#endif
                changes += capsula.AesKeysUserIds.Count;
                changes += capsula.Messages.Count;
                using (Context context = new Context(config))
                {
#if (DEBUG)
                    Log("Receiving and saving AES keys.");
#endif
                    foreach (var id in capsula.AesKeysUserIds)
                    {
                        var user = new UContact(context.Contacts.Where(con => con.PublicId == id).SingleOrDefault());
                        try
                        {
                            user.ReceiveAesKey = RSAEncoder.DecryptAndVerify(BinaryEncoder.ReceiveBytes(stream), MyCertificate, X509Certificate2Utils.ImportFromPem(user.PublicCertificate));
                        }
                        catch (Exception ex)
                        {
                            Log($"Loading of Receive AESKey from {user.PublicId} has failed.");
                            logger.LogException(this, ex);
                        }
                        PushOperations.Update(context, user, UserId, UserId);
                    }
                    context.SaveChanges();

#if (DEBUG)
                    Log("Receiving and saving messages.");
#endif
                    foreach (PullMessage metaMessage in capsula.Messages)
                    {
                        BlobMessages metaBlob = new BlobMessages()
                        {
                            PublicId = metaMessage.PublicId,
                            SenderId = metaMessage.SenderId,
                            Failed   = 0,
                            DoDelete = 0
                        };
                        context.BlobMessages.Add(metaBlob);
                        context.SaveChanges();

                        try
                        {
                            PullMessageParser.ParseEncryptedMessage(context, BinaryEncoder.ReceiveBytes(stream), metaBlob.SenderId, metaBlob.Id, UserId);
                        }
                        catch (Exception ex)
                        {
                            Log($"Loading of message {metaMessage.PublicId} has failed.");
                            metaBlob.Failed = 1;
                            logger.LogException(this, ex);
                        }
                        context.SaveChanges();
                    }
                }
                Log("Pull have been done.");
            }
            return(changes);
        }
Exemple #2
0
        public void Pull()
        {
            Log("Sending PULL command.");
            TextEncoder.SendCommand(stream, ConnectionCommand.PULL);

            PullCapsula capsula = TextEncoder.ReadPullCapsula(stream);

#if (DEBUG)
            Log("Received PullCapsula.");
#endif
            using (Context context = new Context(config))
            {
#if (DEBUG)
                Log("Saving new users.");
#endif
                foreach (PullUser user in capsula.Users)
                {
                    context.Contacts.Add(new Contacts
                    {
                        PublicId          = user.UserId,
                        PublicCertificate = user.PublicCertificate,
                        UserName          = user.UserName
                    });
                }
                context.SaveChanges();

#if (DEBUG)
                Log("Saving trusted contacts.");
#endif
                context.Database.ExecuteSqlCommand("update CONTACTS set TRUSTED=0;");
                context.SaveChanges();

                foreach (var user in context.Contacts
                         .Where(users => capsula.TrustedUserIds.Contains(users.PublicId)))
                {
                    user.Trusted = 1;
                }
                context.SaveChanges();
#if (DEBUG)
                Log("Receiving and saving AES keys.");
#endif
                foreach (var id in capsula.AesKeysUserIds)
                {
                    var user = context.Contacts.Where(con => con.PublicId == id).SingleOrDefault();
                    user.ReceiveAesKey = RSAEncoder.Decrypt(BinaryEncoder.ReceiveBytes(stream), ClientCertificate);
                }
                context.SaveChanges();
#if (DEBUG)
                Log("Receiving and saving messages.");
#endif
                foreach (PullMessage metaMessage in capsula.Messages)
                {
                    BlobMessages metaBlob = new BlobMessages()
                    {
                        PublicId = metaMessage.PublicId,
                        SenderId = metaMessage.SenderId,
                        Failed   = 0,
                        DoDelete = 0
                    };
                    context.BlobMessages.Add(metaBlob);
                    context.SaveChanges();

                    try
                    {
                        PullMessageParser.ParseEncryptedMessage(context, BinaryEncoder.ReceiveBytes(stream), metaBlob.SenderId, metaBlob.Id, UserId);
                    }
                    catch (Exception ex)
                    {
                        Log($"Loading of message {metaMessage.PublicId} has failed.");
                        metaBlob.Failed = 1;
                        logger.LogException(this, ex);
                    }
                    context.SaveChanges();
                }
            }
            Log("Pull have been done.");
        }