/** * Main program code for the sambaget utility. Write the SAM-BA boot * monitor program to a given file name. * * Copyright 2006 David Anderson * * 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 #include #include #include #include #include #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 \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; }