Embedded/elua
1
0
Fork 0
mirror of https://github.com/elua/elua.git synced 2025-01-25 01:02:54 +08:00
Code Issues Projects Releases Wiki Activity

Corrected a memory leak, and checking writing error on xmodem recv.

Browse Source
This commit is contained in:
Marcelo Politzer Couto 2011-06-12 17:51:32 -03:00
parent d64459f8f5
commit 1ae6f3f865
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/shell.c
View File

@ -145,19 +145,26 @@ static void shell_recv( char* args )
p ++;
printf( "done, got %u bytes\n", ( unsigned )( p - shell_prog ) );
/* if we have an argument, write to it, if there is no argument, run it. */
if (strcmp( args, "") != 0)
// we've received an argument, try saving it to a file
if( strcmp( args, "") != 0 )
{
FILE *foutput = fopen( args, "w" );
size_t file_sz = p - shell_prog;
if( foutput == NULL )
{
printf( "unable to open file %s\n", args);
free( shell_prog );
shell_prog = NULL;
return;
}
fwrite( shell_prog, sizeof (char), p - shell_prog, foutput );
printf ("received and saved as %s\n", args);
if( fwrite( shell_prog, sizeof( char ), file_sz, foutput ) == file_sz )
printf( "received and saved as %s\n", args );
else
printf( "disk full, unable to save file %s\n", args );
fclose( foutput );
} else {
}
else // no arg, running the file with lua.
{
if( ( L = lua_open() ) == NULL )
{
printf( "Unable to create Lua state\n" );
@ -173,8 +180,8 @@ static void shell_recv( char* args )
printf( "Error: %s\n", lua_tostring( L, -1 ) );
lua_close( L );
}
free( shell_prog );
shell_prog = NULL;
free( shell_prog );
shell_prog = NULL;
#endif // #ifndef BUILD_XMODEM
}