1 files changed, 21 insertions, 5 deletions

diff --git a/core/friend_requests.h b/core/friend_requests.h
index 3ce0df8c..2ebd557b 100644
--- a/core/friend_requests.h
+++ b/core/friend_requests.h
@@ -31,21 +31,37 @@
 extern "C" {
 #endif
 
+typedef struct {
+    uint32_t nospam;
+    void (*handle_friendrequest)(uint8_t *, uint8_t *, uint16_t, void *);
+    uint8_t handle_friendrequest_isset;
+    void *handle_friendrequest_userdata;
+
+    /*NOTE: the following is just a temporary fix for the multiple friend requests received at the same time problem
+    TODO: Make this better (This will most likely tie in with the way we will handle spam.)*/
+
+#define MAX_RECEIVED_STORED 32
+
+    uint8_t received_requests[MAX_RECEIVED_STORED][crypto_box_PUBLICKEYBYTES];
+    uint16_t received_requests_index;
+} Friend_Requests;
+
 /* Try to send a friendrequest to peer with public_key
     data is the data in the request and length is the length. */
-int send_friendrequest(uint8_t *public_key, uint32_t nospam_num, uint8_t *data, uint32_t length);
+int send_friendrequest(DHT *dht, uint8_t *public_key, uint32_t nospam_num, uint8_t *data, uint32_t length);
 /*
  * Set and get the nospam variable used to prevent one type of friend request spam
  */
-void set_nospam(uint32_t num);
-uint32_t get_nospam();
+void set_nospam(Friend_Requests *fr, uint32_t num);
+uint32_t get_nospam(Friend_Requests *fr);
 
 /* set the function that will be executed when a friend request for us is received.
     function format is function(uint8_t * public_key, uint8_t * data, uint16_t length) */
-void callback_friendrequest(void (*function)(uint8_t *, uint8_t *, uint16_t, void *), void *userdata);
+void callback_friendrequest(Friend_Requests *fr, void (*function)(uint8_t *, uint8_t *, uint16_t, void *),
+                            void *userdata);
 
 /* sets up friendreq packet handlers */
-void friendreq_init(void);
+void friendreq_init(Friend_Requests *fr, Net_Crypto *c);
 
 #ifdef __cplusplus
 }