Properly clean up scheduler state.

Author: robertnishihara

src/photon/photon_algorithm.c

@@ -37,6 +37,11 @@ scheduler_state *make_scheduler_state(void) {
 }
 
 void free_scheduler_state(scheduler_state *s) {
+  for (int i = 0; i < utarray_len(s->task_queue); ++i) {
+    task_instance **instance =
+        (task_instance **) utarray_eltptr(s->task_queue, i);
+    free(*instance);
+  }
   utarray_free(s->task_queue);
   utarray_free(s->available_workers);
   free(s);

src/photon/photon_scheduler.c

@@ -73,6 +73,13 @@ local_scheduler_state *init_local_scheduler(event_loop *loop,
 
 void free_local_scheduler(local_scheduler_state *s) {
   db_disconnect(s->scheduler_info->db);
+  free(s->plasma_conn);
+  worker_index *current_worker_index, *temp_worker_index;
+  HASH_ITER(hh, s->worker_index, current_worker_index, temp_worker_index) {
+    HASH_DEL(s->worker_index, current_worker_index);
+    free(current_worker_index);
+  }
+  utarray_free(s->scheduler_info->workers);
   free(s->scheduler_info);
   free_scheduler_state(s->scheduler_state);
   event_loop_destroy(s->loop);
@@ -97,6 +104,7 @@ void process_plasma_notification(event_loop *loop,
   recv(client_sock, message, sizeof(object_id), 0);
   object_id *obj_id = (object_id *) message;
   handle_object_available(s->scheduler_info, s->scheduler_state, *obj_id);
+  free(message);
 }
 
 void process_message(event_loop *loop, int client_sock, void *context,