cmds/buttonmon.c

/*
 * Copyright 2012-2014 Google Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
// GFLT110 the "reset" button is connected to MPP[18]
//
// This will periodically scan MPP[18].
// If held < 1s &&  sysvar PRODUCTION_UNIT is NOT set
//     start dropbear.
// If held > 2s
//   generate a reset.
// if head > 10s
//   remove sysvar PRODUCTION_UNIT AND
//   generate a reset.
//


#include <fcntl.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <time.h>
#include <unistd.h>


// TODO(jnewlin): Export this LED register via the gpio sysfs.
#define GPIO_INPUT_REG_ADDR 0xf1018110
#define RESET_BIT 18
#define RESET_BIT_MASK (1 << RESET_BIT)
#define TRUE 1
#define FALSE 0


// Only run on gflt110s.
int IsGflt110() {
  int bytes_read;
  char buf[64];
  memset(buf, 0, sizeof(buf));
  FILE* f = fopen("/proc/board_type", "r");
  if (!f) {
    printf("Failed to open /proc/board_type\n");
    return FALSE;
  }
  bytes_read = fread(&buf[0], 1, sizeof(buf)-1, f);
  fclose(f);
  if (bytes_read <= 0) {
    printf("fread of /proc/board_type returned 0 data.\n");
  }
  if (strncmp(buf, "GFLT110", strlen("GFLT110")))
    return FALSE;
  return TRUE;
}


int64_t GetTick() {
  struct timespec ts;
  if (clock_gettime(CLOCK_MONOTONIC, &ts)) {
    printf("clock_gettime failed... exiting.\n");
    exit(1);
  }
  return ((int64_t)(ts.tv_sec) * 1000LL) + (ts.tv_nsec / 1000000);
}


void HandleButtonClick(uint64_t ms_down) {
  if (ms_down > 50 && ms_down < 2000) {
    // Short click will reset the box.
    printf("click0\n");
  } else if (ms_down > 2000 && ms_down < 10000) {
    printf("click2\n");
  } else {
    printf("click10\n");
  }
}


void MonitorReset() {
  uint32_t page_size = sysconf(_SC_PAGESIZE);
  uint32_t page_mask = page_size - 1;
  int fd = open("/dev/mem", O_RDWR | O_SYNC);
  if (fd == -1) {
    printf("Failed to open /dev/mem\n");
    exit(1);
  }

  uint32_t* base = (uint32_t*)mmap(
      0, page_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd,
      GPIO_INPUT_REG_ADDR & ~page_mask);
  if (base == (uint32_t*)-1) {
    printf("Failed to mmap /dev/mem\n");
    exit(1);
  }

  volatile uint32_t* reg_addr = base + ((GPIO_INPUT_REG_ADDR & page_mask) / sizeof(*base));
  int button_down = FALSE;
  int button_down_sent = FALSE;
  uint64_t button_down_start_tick = 0;
  for(;;) {
    int button_down_now = (*reg_addr & RESET_BIT_MASK) == 0;
    if (!button_down && button_down_now) {
      // Handle button down toggle.
      button_down_start_tick = GetTick();
      button_down = TRUE;
      button_down_sent = FALSE;
    }

    // Debounces the button, if it's down for more than 50ms
    // send a message that can be used to do something like flicker
    // the led.
    if (button_down && button_down_now && !button_down_sent) {
      if ((GetTick() - button_down_start_tick) > 50) {
        printf("buttondown\n");
        button_down_sent = TRUE;
      }
    }

    if (button_down && !button_down_now) {
      // Handle button up toggle.
      HandleButtonClick(GetTick() - button_down_start_tick);
      button_down_start_tick = 0;
      button_down = FALSE;
    }
    button_down = button_down_now;
    usleep(1000*100);  // sleep 100ms
  }
}


int main() {
  if (!IsGflt110()) {
    printf("resetmonitor only works on gflt110.\n");
    return 1;
  }
  setlinebuf(stdout);
  MonitorReset();
  return 0;
}