text: /**
   * Main program code for the sambaget utility. Write the SAM-BA boot
   * monitor program to a given file name.
   *
   * Copyright 2006 David Anderson <david.anderson@calixo.net>
   *
   * This program is free software; you can redistribute it and/or
   * modify it under the terms of the GNU General Public License as
   * published by the Free Software Foundation; either version 2 of the
   * License, or (at your option) any later version.
   *
   * This program is distributed in the hope that it will be useful, but
   * WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   * General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License
   * along with this program; if not, write to the Free Software
   * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
   * USA
   */
  
  #include <stdio.h>
  #include <stdlib.h>
  #include <unistd.h>
  #include <sys/types.h>
  #include <sys/stat.h>
  #include <fcntl.h>
  
  #include "error.h"
  #include "lowlevel.h"
  #include "samba.h"
  #include "firmware.h"
  
  #define NXT_HANDLE_ERR(expr, nxt, msg)     \
    do {                                     \
      nxt_error_t nxt__err_temp = (expr);    \
      if (nxt__err_temp)                     \
        return handle_error(nxt, msg, nxt__err_temp);  \
    } while(0)
  
  static int handle_error(nxt_t *nxt, char *msg, nxt_error_t err)
  {
    printf("%s: %s\n", msg, nxt_str_error(err));
    if (nxt != NULL)
      nxt_close(nxt);
    exit(err);
  }
  
  nxt_error_t get_samba(nxt_t *nxt, char *filename)
  {
    FILE *fd;
    int page = 0;
  
    fd = fopen(filename, "w");
    if (fd == NULL)
      return NXT_FILE_ERROR;
  
    for (page = 0; page < 32*1024; page+=4)
      {
        nxt_word_t w;
  
        NXT_ERR(nxt_read_word(nxt, 0x100000+page, &w));
  
        if(fwrite((char *)&w, 4, 1, fd) != 1)
          return NXT_FILE_ERROR;
        fprintf(stderr,".");
      }
    fprintf(stderr, "\nDONE\n");
    fclose(fd);
  
    return NXT_OK;
  }
  
  int main(int argc, char *argv[])
  {
    nxt_t *nxt;
    nxt_error_t err;
    char *fw_file;
  
    if (argc != 2)
      {
        printf("Syntax: %s <filename>\n"
               "\n"
               "Example: %s samba.bin\n", argv[0], argv[0]);
        exit(1);
      }
  
    fw_file = argv[1];
  
    NXT_HANDLE_ERR(nxt_init(&nxt), NULL,
                   "Error during library initialization");
  
    err = nxt_find(nxt);
    if (err)
      {
        if (err == NXT_NOT_PRESENT)
          printf("NXT not found. Is it properly plugged in via USB?\n");
        else
          NXT_HANDLE_ERR(0, NULL, "Error while scanning for NXT");
        exit(1);
      }
  
    if (!nxt_in_reset_mode(nxt))
      {
        printf("NXT found, but not running in reset mode.\n");
        printf("Please reset your NXT manually and restart this program.\n");
        exit(2);
      }
  
    NXT_HANDLE_ERR(nxt_open(nxt), NULL, "Error while connecting to NXT");
  
    printf("NXT device in reset mode located and opened.\n"
           "Retrieving SAM-BA monitor code...\n");
  
    /* Retrieve the sam-ba boot code */
    NXT_HANDLE_ERR(get_samba(nxt, argv[1]), nxt,
                   "Error trying to get SAM-BA code");
  
    NXT_HANDLE_ERR(nxt_close(nxt), NULL,
                   "Error while closing connection to NXT");
    return 0;
  }
  