From d910ed68a37e206fc84445d0c93870a0452b69fc Mon Sep 17 00:00:00 2001
From: Tyera Eulberg <tyera@solana.com>
Date: Wed, 22 Aug 2018 12:25:21 -0600
Subject: [PATCH] Use balance to verify requestAirdrop success

---
 src/rpc.rs | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/src/rpc.rs b/src/rpc.rs
index 347b7cd8be..2382acca3e 100644
--- a/src/rpc.rs
+++ b/src/rpc.rs
@@ -11,7 +11,9 @@ use std::mem;
 use std::net::{SocketAddr, UdpSocket};
 use std::sync::atomic::{AtomicBool, Ordering};
 use std::sync::Arc;
-use std::thread::{self, Builder, JoinHandle};
+use std::thread::{self, sleep, Builder, JoinHandle};
+use std::time::Duration;
+use std::time::Instant;
 use transaction::Transaction;
 use wallet::request_airdrop;
 
@@ -150,9 +152,25 @@ impl RpcSol for RpcSolImpl {
             return Err(Error::invalid_request());
         }
         let pubkey = Pubkey::new(&pubkey_vec);
-        request_airdrop(&meta.drone_addr, &pubkey, tokens)
+        let previous_balance = meta
+            .request_processor
+            .get_balance(pubkey)
             .map_err(|_| Error::internal_error())?;
-        Ok(true)
+        request_airdrop(&meta.drone_addr, &pubkey, tokens).map_err(|_| Error::internal_error())?;
+        let now = Instant::now();
+        let mut balance;
+        loop {
+            balance = meta
+                .request_processor
+                .get_balance(pubkey)
+                .map_err(|_| Error::internal_error())?;
+            if balance > previous_balance {
+                return Ok(true);
+            } else if now.elapsed().as_secs() > 5 {
+                return Err(Error::internal_error());
+            }
+            sleep(Duration::from_millis(100));
+        }
     }
     fn send_transaction(&self, meta: Self::Metadata, data: Vec<u8>) -> Result<String> {
         let tx: Transaction = deserialize(&data).map_err(|_| Error::invalid_request())?;