/* return true if address range is valid */
static bool check_addr_range( const int n, const int m, const int addr_cnt )
{
if( addr_cnt == 0 )
{
first_addr = n;
second_addr = m;
}
if( first_addr < 1 || first_addr > second_addr || second_addr > last_addr() )
{ invalid_address(); return false; }
return true;
}
/* get a valid address from the command buffer */
static bool get_third_addr( const char ** const ibufpp, int * const addr )
{
const int old1 = first_addr;
const int old2 = second_addr;
int addr_cnt = extract_addr_range( ibufpp );
if( addr_cnt < 0 ) return false;
if( traditional() && addr_cnt == 0 )
{ set_error_msg( "Destination expected" ); return false; }
if( second_addr < 0 || second_addr > last_addr() )
{ invalid_address(); return false; }
*addr = second_addr;
first_addr = old1; second_addr = old2;
return true;
}
int cmd_poke(struct cli_state *state, int argc, char **argv)
{
/* Valid commands:
poke dac <address> <value>
poke lms <address> <value>
poke si <address> <value>
*/
int rv = CLI_RET_OK;
int status;
bool ok;
int (*f)(struct bladerf *, uint8_t, uint8_t);
unsigned int address, value;
if (!cli_device_is_opened(state)) {
return CLI_RET_NODEV;
}
if( argc == 4 ) {
/* Parse the value */
value = str2uint( argv[3], 0, MAX_VALUE, &ok );
if( !ok ) {
cli_err(state, argv[0],
"Invalid value provided (%s)", argv[3]);
return CLI_RET_INVPARAM;
}
/* Are we reading from the DAC? */
if( strcasecmp( argv[1], "dac" ) == 0 ) {
/* Parse address */
address = str2uint( argv[2], 0, DAC_MAX_ADDRESS, &ok );
if( !ok ) {
invalid_address(state, argv[0], argv[2]);
rv = CLI_RET_INVPARAM;
} else {
/* TODO: Point function pointer */
/* f = vctcxo_dac_write */
f = NULL;
}
}
/* Are we reading from the LMS6002D */
else if( strcasecmp( argv[1], "lms" ) == 0 ) {
/* Parse address */
address = str2uint( argv[2], 0, LMS_MAX_ADDRESS, &ok );
if( !ok ) {
invalid_address(state, argv[0], argv[2]);
rv = CLI_RET_INVPARAM;
} else {
f = bladerf_lms_write;
}
}
/* Are we reading from the Si5338? */
else if( strcasecmp( argv[1], "si" ) == 0 ) {
/* Parse address */
address = str2uint( argv[2], 0, SI_MAX_ADDRESS, &ok );
if( !ok ) {
invalid_address(state, argv[0], argv[2]);
rv = CLI_RET_INVPARAM;
} else {
f = bladerf_si5338_write;
}
}
/* I guess we aren't reading from anything :( */
else {
cli_err(state, argv[0], "%s is not a pokeable device\n", argv[1] );
rv = CLI_RET_INVPARAM;
}
/* Write the value to the address */
if( rv == CLI_RET_OK && f ) {
status = f( state->dev, (uint8_t)address, (uint8_t)value );
if (status < 0) {
state->last_lib_error = status;
rv = CLI_RET_LIBBLADERF;
} else {
printf( " 0x%2.2x: 0x%2.2x\n", address, value );
if (f == bladerf_lms_write) {
uint8_t readback;
int status = bladerf_lms_read(state->dev,
(uint8_t)address,
&readback);
if (status == 0) {
lms_reg_info(address, readback);
putchar('\n'); /* To be consistent with peek output */
}
}
}
}
} else {
cli_err(state, argv[0], "Invalid number of arguments (%d)\n", argc);
rv = CLI_RET_INVPARAM;
}
return rv;
}
/* execute the next command in command buffer; return error status */
static int exec_command( const char ** const ibufpp, const int prev_status,
const bool isglobal )
{
const char * fnp;
int gflags = 0;
int addr, c, n;
const int addr_cnt = extract_addr_range( ibufpp );
if( addr_cnt < 0 ) return ERR;
*ibufpp = skip_blanks( *ibufpp );
c = *(*ibufpp)++;
switch( c )
{
case 'a': if( !get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !append_lines( ibufpp, second_addr, isglobal ) ) return ERR;
break;
case 'c': if( first_addr == 0 ) first_addr = 1;
if( second_addr == 0 ) second_addr = 1;
if( !check_current_addr( addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !delete_lines( first_addr, second_addr, isglobal ) ||
!append_lines( ibufpp, current_addr(), isglobal ) ) return ERR;
break;
case 'd': if( !check_current_addr( addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !delete_lines( first_addr, second_addr, isglobal ) ) return ERR;
inc_current_addr();
break;
case 'e': if( modified() && !scripted() && prev_status != EMOD )
return EMOD; /* fall through */
case 'E': if( unexpected_address( addr_cnt ) ||
unexpected_command_suffix( **ibufpp ) ) return ERR;
fnp = get_filename( ibufpp );
if( !fnp || !delete_lines( 1, last_addr(), isglobal ) ||
!close_sbuf() ) return ERR;
if( !open_sbuf() ) return FATAL;
if( fnp[0] && fnp[0] != '!' ) set_def_filename( fnp );
if( traditional() && !fnp[0] && !def_filename[0] )
{ set_error_msg( "No current filename" ); return ERR; }
if( read_file( fnp[0] ? fnp : def_filename, 0 ) < 0 )
return ERR;
reset_undo_state(); set_modified( false );
break;
case 'f': if( unexpected_address( addr_cnt ) ||
unexpected_command_suffix( **ibufpp ) ) return ERR;
fnp = get_filename( ibufpp );
if( !fnp ) return ERR;
if( fnp[0] == '!' )
{ set_error_msg( "Invalid redirection" ); return ERR; }
if( fnp[0] ) set_def_filename( fnp );
printf( "%s\n", strip_escapes( def_filename ) );
break;
case 'g':
case 'v':
case 'G':
case 'V': if( isglobal )
{ set_error_msg( "Cannot nest global commands" ); return ERR; }
n = ( c == 'g' || c == 'G' ); /* mark matching lines */
if( !check_addr_range( 1, last_addr(), addr_cnt ) ||
!build_active_list( ibufpp, first_addr, second_addr, n ) )
return ERR;
n = ( c == 'G' || c == 'V' ); /* interactive */
if( ( n && !get_command_suffix( ibufpp, &gflags ) ) ||
!exec_global( ibufpp, gflags, n ) )
return ERR;
break;
case 'h':
case 'H': if( unexpected_address( addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( c == 'H' ) verbose = !verbose;
if( ( c == 'h' || verbose ) && errmsg[0] )
fprintf( stderr, "%s\n", errmsg );
break;
case 'i': if( second_addr == 0 ) second_addr = 1;
if( !get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !append_lines( ibufpp, second_addr - 1, isglobal ) )
return ERR;
break;
case 'j': if( !check_addr_range( current_addr(), current_addr() + 1, addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( first_addr != second_addr &&
!join_lines( first_addr, second_addr, isglobal ) ) return ERR;
break;
case 'k': n = *(*ibufpp)++;
if( second_addr == 0 ) { invalid_address(); return ERR; }
if( !get_command_suffix( ibufpp, &gflags ) ||
!mark_line_node( search_line_node( second_addr ), n ) )
return ERR;
break;
case 'l':
case 'n':
case 'p': if( c == 'l' ) n = GLS; else if( c == 'n' ) n = GNP; else n = GPR;
if( !check_current_addr( addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ||
!display_lines( first_addr, second_addr, gflags | n ) )
return ERR;
gflags = 0;
break;
case 'm': if( !check_current_addr( addr_cnt ) ||
!get_third_addr( ibufpp, &addr ) ) return ERR;
if( addr >= first_addr && addr < second_addr )
{ set_error_msg( "Invalid destination" ); return ERR; }
if( !get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !move_lines( first_addr, second_addr, addr, isglobal ) )
return ERR;
break;
case 'P':
case 'q':
case 'Q': if( unexpected_address( addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( c == 'P' ) prompt_on = !prompt_on;
else if( modified() && !scripted() && c == 'q' &&
prev_status != EMOD ) return EMOD;
else return QUIT;
break;
case 'r': if( unexpected_command_suffix( **ibufpp ) ) return ERR;
if( addr_cnt == 0 ) second_addr = last_addr();
fnp = get_filename( ibufpp );
if( !fnp ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !def_filename[0] && fnp[0] != '!' ) set_def_filename( fnp );
if( traditional() && !fnp[0] && !def_filename[0] )
{ set_error_msg( "No current filename" ); return ERR; }
addr = read_file( fnp[0] ? fnp : def_filename, second_addr );
if( addr < 0 ) return ERR;
if( addr ) set_modified( true );
break;
case 's': if( !command_s( ibufpp, &gflags, addr_cnt, isglobal ) )
return ERR;
break;
case 't': if( !check_current_addr( addr_cnt ) ||
!get_third_addr( ibufpp, &addr ) ||
!get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !copy_lines( first_addr, second_addr, addr ) ) return ERR;
break;
case 'u': if( unexpected_address( addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ||
!undo( isglobal ) ) return ERR;
break;
case 'w':
case 'W': n = **ibufpp;
if( n == 'q' || n == 'Q' ) ++*ibufpp;
if( unexpected_command_suffix( **ibufpp ) ) return ERR;
fnp = get_filename( ibufpp );
if( !fnp ) return ERR;
if( addr_cnt == 0 && last_addr() == 0 )
first_addr = second_addr = 0;
else if( !check_addr_range( 1, last_addr(), addr_cnt ) )
return ERR;
if( !def_filename[0] && fnp[0] != '!' ) set_def_filename( fnp );
if( traditional() && !fnp[0] && !def_filename[0] )
{ set_error_msg( "No current filename" ); return ERR; }
addr = write_file( fnp[0] ? fnp : def_filename,
( c == 'W' ) ? "a" : "w", first_addr, second_addr );
if( addr < 0 ) return ERR;
if( addr == last_addr() ) set_modified( false );
else if( modified() && !scripted() && n == 'q' &&
prev_status != EMOD ) return EMOD;
if( n == 'q' || n == 'Q' ) return QUIT;
break;
case 'x': if( second_addr < 0 || last_addr() < second_addr )
{ invalid_address(); return ERR; }
if( !get_command_suffix( ibufpp, &gflags ) ) return ERR;
if( !isglobal ) clear_undo_stack();
if( !put_lines( second_addr ) ) return ERR;
break;
case 'y': if( !check_current_addr( addr_cnt ) ||
!get_command_suffix( ibufpp, &gflags ) ||
!yank_lines( first_addr, second_addr ) ) return ERR;
break;
case 'z': first_addr = 1;
if( !check_addr_range( first_addr, current_addr() +
( traditional() || !isglobal ), addr_cnt ) )
return ERR;
if( **ibufpp > '0' && **ibufpp <= '9' )
{ if( parse_int( &n, *ibufpp, ibufpp ) ) set_window_lines( n );
else return ERR; }
if( !get_command_suffix( ibufpp, &gflags ) ||
!display_lines( second_addr, min( last_addr(), second_addr + window_lines() ),
gflags ) )
return ERR;
gflags = 0;
break;
case '=': if( !get_command_suffix( ibufpp, &gflags ) ) return ERR;
printf( "%d\n", addr_cnt ? second_addr : last_addr() );
break;
case '!': if( unexpected_address( addr_cnt ) ) return ERR;
fnp = get_shell_command( ibufpp );
if( !fnp ) return ERR;
if( system( fnp + 1 ) < 0 )
{ set_error_msg( "Can't create shell process" ); return ERR; }
if( !scripted() ) printf( "!\n" );
break;
case '\n': first_addr = 1;
if( !check_addr_range( first_addr, current_addr() +
( traditional() || !isglobal ), addr_cnt ) ||
!display_lines( second_addr, second_addr, 0 ) )
return ERR;
break;
case '#': while( *(*ibufpp)++ != '\n' ) ;
break;
default : set_error_msg( "Unknown command" ); return ERR;
}
if( gflags && !display_lines( current_addr(), current_addr(), gflags ) )
return ERR;
return 0;
}
/* return the next line address in the command buffer */
static int next_addr( const char ** const ibufpp, int * const addr_cnt )
{
const char * const s = *ibufpp = skip_blanks( *ibufpp );
int addr = current_addr();
bool first = true; /* true == addr, false == offset */
while( true )
{
int n;
const unsigned char ch = **ibufpp;
if( isdigit( ch ) )
{
if( !first ) { invalid_address(); return -2; };
if( !parse_int( &addr, *ibufpp, ibufpp ) ) return -2;
}
else switch( ch )
{
case '+':
case '\t':
case ' ':
case '-': *ibufpp = skip_blanks( ++*ibufpp );
if( isdigit( (unsigned char)**ibufpp ) )
{
if( !parse_int( &n, *ibufpp, ibufpp ) ) return -2;
addr += ( ( ch == '-' ) ? -n : n );
}
else if( ch == '+' ) ++addr;
else if( ch == '-' ) --addr;
break;
case '.':
case '$': if( !first ) { invalid_address(); return -2; };
++*ibufpp;
addr = ( ( ch == '.' ) ? current_addr() : last_addr() );
break;
case '/':
case '?': if( !first ) { invalid_address(); return -2; };
addr = next_matching_node_addr( ibufpp, ch == '/' );
if( addr < 0 ) return -2;
if( ch == **ibufpp ) ++*ibufpp;
break;
case '\'':if( !first ) { invalid_address(); return -2; };
++*ibufpp;
addr = get_marked_node_addr( *(*ibufpp)++ );
if( addr < 0 ) return -2;
break;
case '%':
case ',':
case ';': if( first )
{
++*ibufpp; ++*addr_cnt;
second_addr = ( ( ch == ';' ) ? current_addr() : 1 );
addr = last_addr();
break;
} /* FALL THROUGH */
default : if( *ibufpp == s ) return -1; /* EOF */
if( addr < 0 || addr > last_addr() )
{ invalid_address(); return -2; }
++*addr_cnt; return addr;
}
first = false;
}
}
