/* for each line in a range, change text matching a pattern according to
a substitution template; return false if error */
char search_and_replace( const int first_addr, const int second_addr,
const int gflags, const int snum, const char isglobal )
{
int lc;
char match_found = 0;
set_current_addr( first_addr - 1 );
for( lc = 0; lc <= second_addr - first_addr; ++lc )
{
line_t *lp = search_line_node( inc_current_addr() );
int len = replace_matching_text( lp, gflags, snum );
if( len < 0 ) return 0;
if( len )
{
const char *txt = rbuf;
const char *eot = rbuf + len;
undo_t *up = 0;
disable_interrupts();
if( !delete_lines( current_addr(), current_addr(), isglobal ) ) return 0;
do {
txt = put_sbuf_line( txt, current_addr() );
if( !txt ) { enable_interrupts(); return 0; }
if( up ) up->tail = search_line_node( current_addr() );
else if( !( up = push_undo_atom( UADD, -1, -1 ) ) )
{ enable_interrupts(); return 0; }
}
while( txt != eot );
enable_interrupts();
match_found = 1;
}
}
if( !match_found && !( gflags & GLB ) )
{ set_error_msg( "No match" ); return 0; }
return 1;
}
static bool command_s( const char ** const ibufpp, int * const gflagsp,
const int addr_cnt, const bool isglobal )
{
static int gflags = 0;
static int snum = 0;
enum Sflags {
SGG = 0x01, /* complement previous global substitute suffix */
SGP = 0x02, /* complement previous print suffix */
SGR = 0x04, /* use last regex instead of last pat */
SGF = 0x08 /* repeat last substitution */
} sflags = 0;
do {
if( isdigit( (unsigned char)**ibufpp ) )
{
if( !parse_int( &snum, *ibufpp, ibufpp ) ) return false;
sflags |= SGF; gflags &= ~GSG; /* override GSG */
}
else switch( **ibufpp )
{
case '\n':sflags |= SGF; break;
case 'g': sflags |= SGG; ++*ibufpp; break;
case 'p': sflags |= SGP; ++*ibufpp; break;
case 'r': sflags |= SGR; ++*ibufpp; break;
default : if( sflags )
{ set_error_msg( "Invalid command suffix" ); return false; }
}
}
while( sflags && **ibufpp != '\n' );
if( sflags && !prev_pattern() )
{ set_error_msg( "No previous substitution" ); return false; }
if( sflags & SGG ) snum = 0; /* override numeric arg */
if( **ibufpp != '\n' && (*ibufpp)[1] == '\n' )
{ set_error_msg( "Invalid pattern delimiter" ); return false; }
if( ( !sflags || ( sflags & SGR ) ) && !new_compiled_pattern( ibufpp ) )
return false;
if( !sflags && !extract_subst_tail( ibufpp, &gflags, &snum, isglobal ) )
return false;
if( isglobal ) gflags |= GLB;
else gflags &= ~GLB;
if( sflags & SGG ) gflags ^= GSG;
if( sflags & SGP ) { gflags ^= GPR; gflags &= ~( GLS | GNP ); }
switch( **ibufpp )
{
case 'l': gflags |= GLS; ++*ibufpp; break;
case 'n': gflags |= GNP; ++*ibufpp; break;
case 'p': gflags |= GPR; ++*ibufpp; break;
}
if( !check_current_addr( addr_cnt ) ||
!get_command_suffix( ibufpp, gflagsp ) ) return false;
if( !isglobal ) clear_undo_stack();
if( !search_and_replace( first_addr, second_addr, gflags, snum, isglobal ) )
return false;
if( ( gflags & ( GPR | GLS | GNP ) ) &&
!display_lines( current_addr(), current_addr(), gflags ) )
return false;
return true;
}
/* apply command list in the command buffer to the active lines in a
range; return false if error */
static bool exec_global( const char ** const ibufpp, const int gflags,
const bool interactive )
{
static char * buf = 0;
static int bufsz = 0;
const char * cmd = 0;
if( !interactive )
{
if( traditional() && !strcmp( *ibufpp, "\n" ) )
cmd = "p\n"; /* null cmd_list == 'p' */
else
{
if( !get_extended_line( ibufpp, 0, false ) ) return false;
cmd = *ibufpp;
}
}
clear_undo_stack();
while( true )
{
const line_t * const lp = next_active_node();
if( !lp ) break;
set_current_addr( get_line_node_addr( lp ) );
if( current_addr() < 0 ) return false;
if( interactive )
{
/* print current_addr; get a command in global syntax */
int len;
if( !display_lines( current_addr(), current_addr(), gflags ) )
return false;
do { *ibufpp = get_tty_line( &len ); }
while( *ibufpp && len > 0 && (*ibufpp)[len-1] != '\n' );
if( !*ibufpp ) return false;
if( len == 0 )
{ set_error_msg( "Unexpected end-of-file" ); return false; }
if( len == 1 && !strcmp( *ibufpp, "\n" ) ) continue;
if( len == 2 && !strcmp( *ibufpp, "&\n" ) )
{ if( !cmd ) { set_error_msg( "No previous command" ); return false; } }
else
{
if( !get_extended_line( ibufpp, &len, false ) ||
!resize_buffer( &buf, &bufsz, len + 1 ) ) return false;
memcpy( buf, *ibufpp, len + 1 );
cmd = buf;
}
}
*ibufpp = cmd;
while( **ibufpp ) if( exec_command( ibufpp, 0, true ) < 0 ) return false;
}
return true;
}
/* return the address of the next line matching a pattern in a given
direction. wrap around begin/end of editor buffer if necessary */
int get_matching_node_addr( const char **ibufpp, const char forward )
{
regex_t *pat = get_compiled_pattern( ibufpp );
int addr = current_addr();
if( !pat ) return -1;
do {
addr = ( forward ? inc_addr( addr ) : dec_addr( addr ) );
if( addr )
{
line_t *lp = search_line_node( addr );
char *s = get_sbuf_line( lp );
if( !s ) return -1;
if( isbinary() ) nul_to_newline( s, lp->len );
if( !regexec( pat, s, 0, 0, 0 ) ) return addr;
}
}
while( addr != current_addr() );
set_error_msg( "No match" );
return -1;
}
/* get line addresses from the command buffer until an invalid address
is seen. Return number of addresses read */
static int extract_addr_range( const char ** const ibufpp )
{
int addr;
int addr_cnt = 0;
first_addr = second_addr = current_addr();
while( true )
{
addr = next_addr( ibufpp, &addr_cnt );
if( addr < 0 ) break;
first_addr = second_addr; second_addr = addr;
if( **ibufpp != ',' && **ibufpp != ';' ) break;
if( **ibufpp == ';' ) set_current_addr( addr );
++*ibufpp;
}
if( addr_cnt == 1 || second_addr != addr ) first_addr = second_addr;
return ( ( addr != -2 ) ? addr_cnt : -1 );
}
/* 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 true if current address is valid */
static bool check_current_addr( const int addr_cnt )
{
return check_addr_range( current_addr(), current_addr(), addr_cnt );
}
/* 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;
}
}
