From 3356bd4cf11bbcbae036d8fb61561833aba7a113 Mon Sep 17 00:00:00 2001
From: dhood <dhood@users.noreply.github.com>
Date: Wed, 1 Aug 2018 11:24:13 -0700
Subject: [PATCH] Copy allocator passed in to clock init (#274)

* Store allocator passed in to clock init

* Linter fix
---
 rcl/include/rcl/time.h |  2 +-
 rcl/src/rcl/time.c     | 16 +++++++++++-----
 2 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/rcl/include/rcl/time.h b/rcl/include/rcl/time.h
index e36991aa3..c6251dbdc 100644
--- a/rcl/include/rcl/time.h
+++ b/rcl/include/rcl/time.h
@@ -63,7 +63,7 @@ typedef struct rcl_clock_t
   rcl_ret_t (* get_now)(void * data, rcl_time_point_value_t * now);
   // void (*set_now) (rcl_time_point_value_t);
   void * data;
-  rcl_allocator_t * allocator;
+  rcl_allocator_t allocator;
 } rcl_clock_t;
 
 /// A single point in time, measured in nanoseconds, the reference point is based on the source.
diff --git a/rcl/src/rcl/time.c b/rcl/src/rcl/time.c
index b3fbe8f76..89ad83b11 100644
--- a/rcl/src/rcl/time.c
+++ b/rcl/src/rcl/time.c
@@ -88,7 +88,7 @@ rcl_clock_init(
   enum rcl_clock_type_t clock_type, rcl_clock_t * clock,
   rcl_allocator_t * allocator)
 {
-  RCL_CHECK_ARGUMENT_FOR_NULL(allocator, RCL_RET_INVALID_ARGUMENT, rcl_get_default_allocator());
+  RCL_CHECK_ALLOCATOR_WITH_MSG(allocator, "invalid allocator", return RCL_RET_INVALID_ARGUMENT);
   switch (clock_type) {
     case RCL_CLOCK_UNINITIALIZED:
       RCL_CHECK_ARGUMENT_FOR_NULL(
@@ -111,6 +111,8 @@ rcl_clock_fini(
   rcl_clock_t * clock)
 {
   RCL_CHECK_ARGUMENT_FOR_NULL(clock, RCL_RET_INVALID_ARGUMENT, rcl_get_default_allocator());
+  RCL_CHECK_ALLOCATOR_WITH_MSG(&clock->allocator, "clock has invalid allocator",
+    return RCL_RET_ERROR);
   switch (clock->type) {
     case RCL_ROS_TIME:
       return rcl_ros_clock_fini(clock);
@@ -138,7 +140,7 @@ rcl_ros_clock_init(
   storage->active = false;
   clock->get_now = rcl_get_ros_time;
   clock->type = RCL_ROS_TIME;
-  clock->allocator = allocator;
+  clock->allocator = *allocator;
   return RCL_RET_OK;
 }
 
@@ -151,7 +153,11 @@ rcl_ros_clock_fini(
     RCL_SET_ERROR_MSG("clock not of type RCL_ROS_TIME", rcl_get_default_allocator());
     return RCL_RET_ERROR;
   }
-  clock->allocator->deallocate((rcl_ros_clock_storage_t *)clock->data, clock->allocator->state);
+  if (!clock->data) {
+    RCL_SET_ERROR_MSG("clock data invalid", rcl_get_default_allocator());
+    return RCL_RET_ERROR;
+  }
+  clock->allocator.deallocate((rcl_ros_clock_storage_t *)clock->data, clock->allocator.state);
   return RCL_RET_OK;
 }
 
@@ -165,7 +171,7 @@ rcl_steady_clock_init(
   rcl_init_generic_clock(clock);
   clock->get_now = rcl_get_steady_time;
   clock->type = RCL_STEADY_TIME;
-  clock->allocator = allocator;
+  clock->allocator = *allocator;
   return RCL_RET_OK;
 }
 
@@ -191,7 +197,7 @@ rcl_system_clock_init(
   rcl_init_generic_clock(clock);
   clock->get_now = rcl_get_system_time;
   clock->type = RCL_SYSTEM_TIME;
-  clock->allocator = allocator;
+  clock->allocator = *allocator;
   return RCL_RET_OK;
 }