forked from raphui/rdb
/
cli.c
129 lines (98 loc) · 3.14 KB
/
cli.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/* rdb - In-memory database optimized for embedded world
* Copyright (C) 2013 Raphaël POGGI
*
* 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 3 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, see <http://www.gnu.org/licenses/>.
*/
#include "cli.h"
static char *help( struct environment *env );
static cliCommand_t cliCmd[] =
{
{"memory_count" , &getMemoryCount },
{"thread_count" , &getThreadCount },
{"dump_trace_level" , &dumpTrace },
{"get_level" , &levelInfo },
{"help" , &help },
{"sort" , &sort },
{"print" , &print },
{"set_trace_level" , &setTraceLevel },
{"set" , &setPair },
{"get" , &getPair },
{"remove" , &removePair },
{"flush" , &flush },
};
static char *help( struct environment *env )
{
TRACE_2( CLI , "help().");
int i = 0;
char *buff = ( char * )zmalloc( 1024 * sizeof( char ) );
memset( buff , 0 , 1024 );
for( i = 0 ; i < CLI_COUNT_COMMAND ; i++ )
sprintf( buff + strlen( buff ) , "%s\n" , cliCmd[i].command );
return buff;
}
static int searchCliCmd( const char *cmd )
{
TRACE_2( CLI , "searchCliCmd( %s )." , cmd );
int i = 0;
for( i = 0 ; i < CLI_COUNT_COMMAND ; i++ )
{
if( strstr( cmd , cliCmd[i].command ) != NULL )
{
TRACE_1( CLI , "Command found, id: %d" , i );
return i;
}
}
return -EINVAL;
}
void *doCommand( char *cmd )
{
TRACE_2( CLI , "doCommand( %s )." , cmd );
int i = 0;
int idFuncptr;
void *ret = NULL;
char *ptr = NULL;
/* Dynamic allocation, because after send the response throught the socket, free() is called. (If it's declare like -char errorMsg[]="zedze"- this will crash for sure.) */
char *errorMsg = NULL;
ptr = strtok( cmd , " ");
env->arg_count = 0;
for( i = 0 ; i < MAX_GENERIC_VAL ; i++ )
{
ptr = strtok( NULL , " " );
if( !ptr )
break;
env->genericVal[i] = ptr;
env->arg_count++;
}
if( env->arg_count > 0 )
{
/* Remove '\n' character from the last argument */
ptr = env->genericVal[env->arg_count - 1];
*( ptr + strlen( ptr ) - 1 ) = 0;
}
if( ( idFuncptr = searchCliCmd( cmd ) ) >= 0 )
{
TRACE_3( CLI , "Execute function.");
ret = ( void * )cliCmd[idFuncptr].func( env );
TRACE_1( CLI , "Function return: %d" , ret );
}
else
{
TRACE_ERROR( CLI , "Command is not supported.");
errorMsg = ( char * )zmalloc( 28 * sizeof( char ) );
memset( errorMsg , 0 , 28 );
sprintf( errorMsg , "Command is not supported.\n");
ret = ( void * )errorMsg;
}
return ret;
}