basic library tests are passing, now to add the poison pill

Chapter06/src/elevator_magic.c

@@ -1,16 +1,20 @@
 int input_codes[3] = {0, 0, 0};
 int error_code = 0;
+void *active_session = 0;
+void *poll_session = 0;
+int active_light = 0;
+int active_light_toggle = 0;
 
 int is_override() {
-   return 0;
+   return (active_session == ((void*)1));
 }
 
 int is_privileged() {
-   return 0;
+   return (active_session == ((void*)2));
 }
 
 int is_admin() {
-   return 0;
+   return (active_session == ((void*)3));
 }
 
 void issue_override_code(int code) {
@@ -18,6 +22,31 @@ void issue_override_code(int code) {
    input_codes[1] = 0;
    input_codes[2] = 0;
    input_codes[0] = code;
+   if( code == 1 ) {
+      if( active_session == ((void*)0) ) {
+         active_session = (void*)1;
+         poll_session = (void*)1;
+      } else {
+         poll_session = (void*)0;
+         error_code = 1;
+      }
+   } else if( code == 2 ) {
+      if( active_session == ((void*)0) ) {
+         active_session = (void*)2;
+         poll_session = (void*)2;
+      } else {
+         poll_session = (void*)0;
+         error_code = 1;
+      }
+   } else if( code == 3 ) {
+      if( active_session == ((void*)0) ) {
+         active_session = (void*)3;
+         poll_session = (void*)3;
+      } else {
+         poll_session = (void*)0;
+         error_code = 1;
+      }
+   }
 }
 
 int poll_override_code() {
@@ -38,33 +67,52 @@ int poll_override_error() {
 }
 
 void* poll_override_session() {
-   //C Code
-   return 0;
+   return poll_session;
 }
 
 void free_override_session(void* session) {
-   //C Code
+   if( active_session == 0 ) {
+      error_code = 2;
+   }
+   active_session = 0;
 }
 
 void* poll_physical_override_privileged_session() {
-   //C Code
-   return 0;
+   return poll_session;
 }
 
 void* poll_physical_override_admin_session() {
-   //C Code
-   return 0;
+   return poll_session;
 }
 
 void override_input_floor(int floor) {
+   if(active_session) {
+      input_codes[0] = 4;
+      input_codes[1] = floor;
+      input_codes[2] = 0;
+   } else {
+      error_code = 3;
+   }
 }
 
 void override_manual_mode() {
-   //C Code
+   if(active_session) {
+      input_codes[0] = 5;
+      input_codes[1] = 0;
+      input_codes[2] = 0;
+   } else {
+      error_code = 3;
+   }
 }
 
 void override_normal_mode() {
-   //C Code
+   if(active_session) {
+      input_codes[0] = 6;
+      input_codes[1] = 0;
+      input_codes[2] = 0;
+   } else {
+      error_code = 3;
+   }
 }
 
 void override_reset_state() {
@@ -72,16 +120,32 @@ void override_reset_state() {
    input_codes[1] = 0;
    input_codes[2] = 0;
    error_code = 0;
+   active_session = 0;
+   poll_session = 0;
+   active_light = 0;
+   active_light_toggle = 0;
 }
 
 void elevator_display_flash(int pattern) {
-   //C Code
+   input_codes[0] = 7;
+   input_codes[1] = pattern;
+   input_codes[2] = 0;
 }
 
 void elevator_display_toggle_light(int light_id) {
-   //C Code
+   int last_state = 0;
+   if( active_light == light_id ) {
+      last_state = active_light_toggle;
+   }
+   active_light = light_id;
+   active_light_toggle = last_state ? 0 : 1;
+   input_codes[0] = 8;
+   input_codes[1] = light_id;
+   input_codes[2] = active_light_toggle;
 }
 
 void elevator_display_set_light_color(int light_id, int color) {
-   //C Code
+   input_codes[0] = 9;
+   input_codes[1] = light_id;
+   input_codes[2] = color;
 }

Chapter06/src/tests/magic.rs

@@ -135,9 +135,11 @@ fn double_free_admin_failure() {
 fn input_floor() {
    unsafe {
       magic::override_reset_state();
+      magic::issue_override_code(3);
       magic::override_input_floor(2);
       assert!(magic::poll_override_code() == 4);
       assert!(magic::poll_override_code() == 2);
+      assert!(magic::poll_override_error() == 0);
    }
 }
 
@@ -145,8 +147,10 @@ fn input_floor() {
 fn manual_mode() {
    unsafe {
       magic::override_reset_state();
-      magic::issue_override_code(5);
+      magic::issue_override_code(3);
+      magic::override_manual_mode();
       assert!(magic::poll_override_code() == 5);
+      assert!(magic::poll_override_error() == 0);
    }
 }
 
@@ -154,8 +158,10 @@ fn manual_mode() {
 fn normal_mode() {
    unsafe {
       magic::override_reset_state();
-      magic::issue_override_code(6);
+      magic::issue_override_code(3);
+      magic::override_normal_mode();
       assert!(magic::poll_override_code() == 6);
+      assert!(magic::poll_override_error() == 0);
    }
 }
 
@@ -208,7 +214,7 @@ fn deny_input_floor() {
 fn deny_manual_mode() {
    unsafe {
       magic::override_reset_state();
-      magic::issue_override_code(5);
+      magic::override_manual_mode();
       assert!(magic::poll_override_error() == 3);
    }
 }
@@ -217,7 +223,7 @@ fn deny_manual_mode() {
 fn deny_normal_mode() {
    unsafe {
       magic::override_reset_state();
-      magic::issue_override_code(6);
+      magic::override_manual_mode();
       assert!(magic::poll_override_error() == 3);
    }
 }