diff options
| -rw-r--r-- | movement/watch_faces/complication/totp_face.c | 46 | ||||
| -rw-r--r-- | movement/watch_faces/complication/totp_face.h | 3 | 
2 files changed, 35 insertions, 14 deletions
diff --git a/movement/watch_faces/complication/totp_face.c b/movement/watch_faces/complication/totp_face.c index 468e6a8c..b026803a 100644 --- a/movement/watch_faces/complication/totp_face.c +++ b/movement/watch_faces/complication/totp_face.c @@ -1,7 +1,3 @@ -/** - * TODO: - * - Support for multiple codes - */  #include <stdlib.h>  #include <string.h>  #include "totp_face.h" @@ -9,25 +5,38 @@  #include "watch_utility.h"  #include "TOTP.h" -// test key: JBSWY3DPEHPK3PXP  // Use https://cryptii.com/pipes/base32-to-hex to convert base32 to hex  // Use https://totp.danhersam.com/ to generate test codes for verification -static uint8_t hmacKey[] = {0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x21, 0xde, 0xad, 0xbe, 0xef}; // Secret key - -static const uint32_t TIMESTEP = 30; +static const uint8_t num_keys = 2; +static uint8_t keys[] = { +    0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x21, 0xde, 0xad, 0xbe, 0xef, // 1 - JBSWY3DPEHPK3PXP +    0x5c, 0x0d, 0x27, 0x6b, 0x6d, 0x9a, 0x01, 0x22, 0x20, 0x4f  // 2 - E9M348K0ADIDFBC2 +}; +static const uint8_t key_sizes[] = { +    10, +    10 +}; +static const uint32_t timesteps[] = { +    30, +    30 +}; +static const char labels[][2] = { +    { 'a', 'b' }, +    { 'c', 'd' } +};  void totp_face_setup(movement_settings_t *settings, uint8_t watch_face_index, void ** context_ptr) {      (void) settings;      (void) watch_face_index;      if (*context_ptr == NULL) *context_ptr = malloc(sizeof(totp_state_t)); -    TOTP(hmacKey, sizeof(hmacKey), TIMESTEP);  }  void totp_face_activate(movement_settings_t *settings, void *context) {      (void) settings;      memset(context, 0, sizeof(totp_state_t));      totp_state_t *totp_state = (totp_state_t *)context; +    TOTP(keys, key_sizes[0], timesteps[0]);      totp_state->timestamp = watch_utility_date_time_to_unix_time(watch_rtc_get_date_time(), movement_timezone_offsets[settings->bit.time_zone] * 60);      totp_state->current_code = getCodeFromTimestamp(totp_state->timestamp);  } @@ -39,19 +48,20 @@ bool totp_face_loop(movement_event_t event, movement_settings_t *settings, void      char buf[14];      uint8_t valid_for;      div_t result; +    uint8_t index = totp_state->current_index;      switch (event.event_type) {          case EVENT_TICK:              totp_state->timestamp++;              // fall through          case EVENT_ACTIVATE: -            result = div(totp_state->timestamp, TIMESTEP); +            result = div(totp_state->timestamp, timesteps[index]);              if (result.quot != totp_state->steps) {                  totp_state->current_code = getCodeFromTimestamp(totp_state->timestamp);                  totp_state->steps = result.quot;              } -            valid_for = TIMESTEP - result.rem; -            sprintf(buf, "2f%2d%06lu", valid_for, totp_state->current_code); +            valid_for = timesteps[index] - result.rem; +            sprintf(buf, "%c%c%2d%06lu", labels[index][0], labels[index][1], valid_for, totp_state->current_code);              watch_display_string(buf, 0);              break; @@ -64,8 +74,18 @@ bool totp_face_loop(movement_event_t event, movement_settings_t *settings, void          case EVENT_TIMEOUT:              movement_move_to_face(0);              break; -        case EVENT_ALARM_BUTTON_DOWN:          case EVENT_ALARM_BUTTON_UP: +            if (index + 1 < num_keys) { +                totp_state->current_key_offset += key_sizes[index]; +                totp_state->current_index++; +            } else { +                // wrap around to first key +                totp_state->current_key_offset = 0; +                totp_state->current_index = 0; +            } +            TOTP(keys + totp_state->current_key_offset, key_sizes[totp_state->current_index], timesteps[totp_state->current_index]); +            break; +        case EVENT_ALARM_BUTTON_DOWN:          case EVENT_ALARM_LONG_PRESS:          default:              break; diff --git a/movement/watch_faces/complication/totp_face.h b/movement/watch_faces/complication/totp_face.h index dfa4a6d0..cff51919 100644 --- a/movement/watch_faces/complication/totp_face.h +++ b/movement/watch_faces/complication/totp_face.h @@ -7,7 +7,8 @@ typedef struct {      uint32_t timestamp;      uint8_t steps;      uint32_t current_code; - +    uint8_t current_index; +    uint8_t current_key_offset;  } totp_state_t;  void totp_face_setup(movement_settings_t *settings, uint8_t watch_face_index, void ** context_ptr);  | 
