List of usage examples for org.bouncycastle.crypto.digests SHA384Digest update
public void update(byte[] in, int inOff, int len)
From source file:dorkbox.network.connection.registration.remote.RegistrationRemoteHandlerClient.java
License:Apache License
@SuppressWarnings("Duplicates") void readClient(final Channel channel, final Registration registration, final String type, final MetaChannel metaChannel) { final InetSocketAddress remoteAddress = (InetSocketAddress) channel.remoteAddress(); // IN: session ID + public key + ecc parameters (which are a nonce. the SERVER defines what these are) // OUT: remote ECDH shared payload if (metaChannel.aesKey == null && registration.publicKey != null) { // whoa! Didn't send valid public key info! if (invalidPublicKey(registration, type)) { shutdown(channel, registration.sessionID); return; }//w ww.ja v a 2s. co m // want to validate the public key used! This is similar to how SSH works, in that once we use a public key, we want to validate // against that ip-address::key pair, so we can better protect against MITM/spoof attacks. if (invalidRemoteAddress(metaChannel, registration, type, remoteAddress)) { // whoa! abort since something messed up! (log and recording if key changed happens inside of validate method) shutdown(channel, registration.sessionID); return; } // save off remote public key. This is ALWAYS the same, where the ECDH changes every time... metaChannel.publicKey = registration.publicKey; // It is OK that we generate a new ECC keypair for ECDHE every time that we connect from the client. // The server rotates keys every XXXX seconds, since this step is expensive (and the server is the 'trusted' endpoint). metaChannel.ecdhKey = CryptoECC.generateKeyPair(eccSpec, registrationWrapper.getSecureRandom()); Registration outboundRegister = new Registration(metaChannel.sessionId); Output output = new Output(1024); EccPublicKeySerializer.write(output, (ECPublicKeyParameters) metaChannel.ecdhKey.getPublic()); outboundRegister.payload = output.toBytes(); channel.writeAndFlush(outboundRegister); return; } // IN: remote ECDH shared payload // OUT: hasMore=true if we have more registrations to do, false otherwise if (metaChannel.aesKey == null) { /* * Diffie-Hellman-Merkle key exchange for the AES key * http://en.wikipedia.org/wiki/Diffie%E2%80%93Hellman_key_exchange */ byte[] ecdhPubKeyBytes = Arrays.copyOfRange(registration.payload, 0, registration.payload.length); ECPublicKeyParameters ecdhPubKey; try { ecdhPubKey = EccPublicKeySerializer.read(new Input(ecdhPubKeyBytes)); } catch (KryoException e) { logger.error("Invalid decode of ECDH public key. Aborting."); shutdown(channel, registration.sessionID); return; } BasicAgreement agreement = new ECDHCBasicAgreement(); agreement.init(metaChannel.ecdhKey.getPrivate()); BigInteger shared = agreement.calculateAgreement(ecdhPubKey); // now we setup our AES key based on our shared secret! (from ECDH) // the shared secret is different each time a connection is made byte[] keySeed = shared.toByteArray(); SHA384Digest sha384 = new SHA384Digest(); byte[] digest = new byte[sha384.getDigestSize()]; sha384.update(keySeed, 0, keySeed.length); sha384.doFinal(digest, 0); metaChannel.aesKey = Arrays.copyOfRange(digest, 0, 32); // 256bit keysize (32 bytes) metaChannel.aesIV = Arrays.copyOfRange(digest, 32, 44); // 96bit blocksize (12 bytes) required by AES-GCM if (invalidAES(metaChannel)) { // abort if something messed up! shutdown(channel, registration.sessionID); return; } Registration outboundRegister = new Registration(metaChannel.sessionId); // do we have any more registrations? outboundRegister.hasMore = registrationWrapper.hasMoreRegistrations(); channel.writeAndFlush(outboundRegister); // wait for ack from the server before registering the next protocol return; } // IN: upgrade=true if we must upgrade this connection if (registration.upgrade) { // this pipeline can now be marked to be upgraded // upgrade the connection to an encrypted connection // this pipeline encoder/decoder can now be upgraded, and the "connection" added upgradeEncoders(channel, metaChannel, remoteAddress); upgradeDecoders(channel, metaChannel); } // IN: hasMore=true if we have more registrations to do, false otherwise if (registration.hasMore) { logger.trace("Starting another protocol registration"); metaChannel.totalProtocols.incrementAndGet(); registrationWrapper.startNextProtocolRegistration(); return; } // // // we only get this when we are 100% done with the registration of all connection types. // // if (!registration.upgraded) { // setup the pipeline with the real connection upgradePipeline(metaChannel, remoteAddress); // we don't verify anything on the CLIENT. We only verify on the server. // we don't support registering NEW classes after the client starts. if (!registrationWrapper.initClassRegistration(channel, registration)) { // abort if something messed up! shutdown(channel, registration.sessionID); } return; } // IN: upgraded=true this means we are ready to connect, and the server is done with it's onConnect calls // we defer the messages until after our own onConnect() is called... // we have to wait for ALL messages to be received, this way we can prevent out-of-order oddities... int protocolsRemaining = metaChannel.totalProtocols.decrementAndGet(); if (protocolsRemaining > 0) { logger.trace("{} done. Waiting for {} more protocols registrations to arrive...", type, protocolsRemaining); return; } // remove the ConnectionWrapper (that was used to upgrade the connection) and cleanup the pipeline // always wait until AFTER the server calls "onConnect", then we do this cleanupPipeline(metaChannel, new Runnable() { @Override public void run() { // this method runs after the "onConnect()" runs and only after all of the channels have be correctly updated // get all of the out of order messages that we missed List<Object> messages = new LinkedList<Object>(); if (metaChannel.tcpChannel != null) { List<Object> list = getOutOfOrderMessagesAndReset(metaChannel.tcpChannel); if (list != null) { logger.trace("Getting deferred TCP messages: {}", list.size()); messages.addAll(list); } } if (metaChannel.udpChannel != null) { List<Object> list = getOutOfOrderMessagesAndReset(metaChannel.udpChannel); if (list != null) { logger.trace("Getting deferred UDP messages: {}", list.size()); messages.addAll(list); } } // now call 'onMessage' in the connection object with our messages! try { ConnectionImpl connection = (ConnectionImpl) metaChannel.connection; for (Object message : messages) { logger.trace(" deferred onMessage({}, {})", connection.id(), message); try { connection.channelRead(null, message); } catch (Exception e) { logger.error("Error running deferred messages!", e); } } } catch (Exception e) { logger.error("Error initialising deferred messages!", e); } } }); }
From source file:dorkbox.network.connection.registration.remote.RegistrationRemoteHandlerServer.java
License:Apache License
@SuppressWarnings("Duplicates") void readServer(final ChannelHandlerContext context, final Channel channel, final Registration registration, final String type, final MetaChannel metaChannel) { final InetSocketAddress remoteAddress = (InetSocketAddress) channel.remoteAddress(); // IN: session ID == 0 (start of new connection) // OUT: session ID + public key + ecc parameters (which are a nonce. the SERVER defines what these are) if (registration.sessionID == 0) { // whoa! Didn't send valid public key info! if (invalidPublicKey(registration, type)) { shutdown(channel, registration.sessionID); return; }/* ww w . j ava 2 s . c o m*/ // want to validate the public key used! This is similar to how SSH works, in that once we use a public key, we want to validate // against that ip-address::key pair, so we can better protect against MITM/spoof attacks. if (invalidRemoteAddress(metaChannel, registration, type, remoteAddress)) { // whoa! abort since something messed up! (log and recording if key changed happens inside of validate method) shutdown(channel, registration.sessionID); return; } // save off remote public key. This is ALWAYS the same, where the ECDH changes every time... metaChannel.publicKey = registration.publicKey; // tell the client to continue it's registration process. Registration outboundRegister = new Registration(metaChannel.sessionId); outboundRegister.publicKey = registrationWrapper.getPublicKey(); outboundRegister.eccParameters = CryptoECC .generateSharedParameters(registrationWrapper.getSecureRandom()); channel.writeAndFlush(outboundRegister); return; } // IN: remote ECDH shared payload // OUT: server ECDH shared payload if (metaChannel.aesKey == null) { /* * Diffie-Hellman-Merkle key exchange for the AES key * http://en.wikipedia.org/wiki/Diffie%E2%80%93Hellman_key_exchange */ // the ECDH key will ROTATE every 10 minutes, since generating it for EVERY connection is expensive // and since we are combining ECDHE+ECC public/private keys for each connection, other // connections cannot break someone else's connection, since they are still protected by their own private keys. metaChannel.ecdhKey = getEchdKeyOnRotate(registrationWrapper.getSecureRandom()); byte[] ecdhPubKeyBytes = java.util.Arrays.copyOfRange(registration.payload, 0, registration.payload.length); ECPublicKeyParameters ecdhPubKey; try { ecdhPubKey = EccPublicKeySerializer.read(new Input(ecdhPubKeyBytes)); } catch (KryoException e) { logger.error("Invalid decode of ECDH public key. Aborting."); shutdown(channel, registration.sessionID); return; } BasicAgreement agreement = new ECDHCBasicAgreement(); agreement.init(metaChannel.ecdhKey.getPrivate()); BigInteger shared = agreement.calculateAgreement(ecdhPubKey); // now we setup our AES key based on our shared secret! (from ECDH) // the shared secret is different each time a connection is made byte[] keySeed = shared.toByteArray(); SHA384Digest sha384 = new SHA384Digest(); byte[] digest = new byte[sha384.getDigestSize()]; sha384.update(keySeed, 0, keySeed.length); sha384.doFinal(digest, 0); metaChannel.aesKey = Arrays.copyOfRange(digest, 0, 32); // 256bit keysize (32 bytes) metaChannel.aesIV = Arrays.copyOfRange(digest, 32, 44); // 96bit blocksize (12 bytes) required by AES-GCM if (invalidAES(metaChannel)) { // abort if something messed up! shutdown(channel, registration.sessionID); return; } Registration outboundRegister = new Registration(metaChannel.sessionId); Output output = new Output(1024); EccPublicKeySerializer.write(output, (ECPublicKeyParameters) metaChannel.ecdhKey.getPublic()); outboundRegister.payload = output.toBytes(); channel.writeAndFlush(outboundRegister); return; } // NOTE: if we have more registrations, we will "bounce back" that status so the client knows what to do. // IN: hasMore=true if we have more registrations to do, false otherwise // ALWAYS upgrade the connection at this point. // IN: upgraded=false if we haven't upgraded to encryption yet (this will always be the case right after encryption is setup) if (!registration.upgraded) { // upgrade the connection to an encrypted connection registration.upgrade = true; upgradeDecoders(channel, metaChannel); // bounce back to the client so it knows we received it channel.write(registration); // this pipeline encoder/decoder can now be upgraded, and the "connection" added upgradeEncoders(channel, metaChannel, remoteAddress); if (!registration.hasMore) { // we only get this when we are 100% done with the registration of all connection types. // setup the pipeline with the real connection upgradePipeline(metaChannel, remoteAddress); } channel.flush(); return; } // the client will send their class registration data. VERIFY IT IS CORRECT! STATE state = registrationWrapper.verifyClassRegistration(metaChannel, registration); if (state == STATE.ERROR) { // abort! There was an error shutdown(channel, registration.sessionID); return; } else if (state == STATE.WAIT) { return; } // else, continue. // // // we only get this when we are 100% done with the registration of all connection types. // the context is the LAST protocol to be registered // // // remove the ConnectionWrapper (that was used to upgrade the connection) and cleanup the pipeline cleanupPipeline(metaChannel, new Runnable() { @Override public void run() { // this method runs after the "onConnect()" runs and only after all of the channels have be correctly updated // this tells the client we are ready to connect (we just bounce back the original message over ALL protocols) if (metaChannel.tcpChannel != null) { logger.trace("Sending TCP upgraded command"); Registration reg = new Registration(registration.sessionID); reg.upgraded = true; metaChannel.tcpChannel.writeAndFlush(reg); } if (metaChannel.udpChannel != null) { logger.trace("Sending UDP upgraded command"); Registration reg = new Registration(registration.sessionID); reg.upgraded = true; metaChannel.udpChannel.writeAndFlush(reg); } } }); }