Converted all m_free to xfree etc.
[gnupg.git] / tools / shmtest.c
1 /* shmtest.c
2  * Copyright (C) 1998, 1999, 2000, 2001 Free Software Foundation, Inc.
3  *
4  * This file is free software; as a special exception the author gives
5  * unlimited permission to copy and/or distribute it, with or without
6  * modifications, as long as this notice is preserved.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY, to the extent permitted by law; without even the
10  * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
11  */
12
13
14 #include <config.h>
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <string.h>
18 #include <errno.h>
19 #include <signal.h>
20 #include <unistd.h>
21 #ifdef HAVE_SYS_IPC_H
22 #include <sys/types.h>
23 #include <sys/ipc.h>
24 #endif
25 #ifdef HAVE_SYS_SHM_H
26 #include <sys/shm.h>
27 #endif
28 #include "util.h"
29 #include "ttyio.h"
30 #include "i18n.h"
31
32 #ifdef HAVE_DOSISH_SYSTEM
33 int main( int argc, char **argv )
34 {
35     fprintf(stderr, "Sorry, not yet available for DOSish systems\n");
36     exit(1);
37 }
38 #else
39
40 static int serverpid = -1;
41
42 static void
43 my_usage(void)
44 {
45     fprintf(stderr, "usage: shmtest gpg-command-line\n");
46     exit(1);
47 }
48
49 const char *
50 strusage( int level )
51 {
52     return default_strusage(level);
53 }
54
55 static void
56 i18n_init(void)
57 {
58 #ifdef ENABLE_NLS
59     setlocale( LC_ALL, "" );
60     bindtextdomain( PACKAGE, G10_LOCALEDIR );
61     textdomain( PACKAGE );
62 #endif
63 }
64
65
66 static void
67 do_get_string( int mode, const char *keyword, byte *area, size_t areasize )
68 {
69     size_t n, len;
70     char *p=NULL;
71     int yes=0;
72
73     n = area[0] << 8 | area[1];
74     /* fixme: do some sanity checks here */
75     if( mode == 1 )
76         p = tty_get( keyword );
77     else if( mode == 3 )
78         p = tty_get_hidden( keyword );
79     else
80         yes = tty_get_answer_is_yes( keyword );
81     if( p ) {
82         len = strlen(p);
83         memcpy( area+n+2, p, len );
84         area[n] = len >> 8;
85         area[n+1] = len;
86         xfree(p);
87     }
88     else { /* bool */
89         area[n] = 0;
90         area[n+1] = 1;
91         area[n+2] = yes;
92     }
93     area[3] = 1; /* we should better use a semaphore */
94     kill( serverpid, SIGUSR1 );
95 }
96
97
98
99 int
100 main(int argc, char **argv)
101 {
102     void  *area = NULL;
103     size_t areasize = 4096;
104     int shm_id = -1;
105     FILE *fp;
106     char buf[200];
107     char *p, *p2;
108     size_t n;
109     int i;
110
111     log_set_name("shmtest");
112     i18n_init();
113 #ifndef USE_SHM_COPROCESSING
114     log_info("SHM_COPRPOCESSING is not available\n");
115 #else
116     if( argc < 1 )
117         my_usage();
118
119     for(n=0,i=1; i < argc; i++ )
120         n += strlen(argv[i]) + 1;
121     p = xmalloc( 100 + n );
122     strcpy( p, "../g10/gpg --status-fd 1 --run-as-shm-coprocess 0");
123     for(i=1; i < argc; i++ ) {
124         strcat(p, " " );
125         strcat(p, argv[i] );
126     }
127
128     fp = popen( p, "r" );
129     xfree( p );
130     if( !fp )
131         log_error("popen failed: %s\n", strerror(errno));
132
133     while ( fgets (buf, sizeof (buf) - 1, fp ) != NULL ) {
134         size_t len = strlen(buf);
135         if( len >= 9 && !memcmp( buf, "[GNUPG:] ", 9 ) ) {
136             int word=0;
137             int is_info = 0, is_get = 0;
138
139             for( p = strtok(buf+9, " \n"); p ; p = strtok(NULL, " \n")) {
140                 word++;
141                 if( word==1 && !strcmp(p,"SHM_INFO") ) {
142                     if( !area )
143                         is_info=1;
144                     else
145                         log_error("duplicate SHM_INFO ignored\n" );
146                 }
147                 else if( is_info && (p2 = strchr(p, '=' )) ) {
148                     int val;
149                     *p2++ = 0;
150                     val = atoi(p2); /* should be atou() for some values */
151                     if( !strcmp(p, "pv" ) ) {
152                         if( atoi(p2) != 1 )
153                             log_fatal("invalid protocol version %d\n", val );
154                         is_info = 2;
155                     }
156                     else if( !strcmp(p, "pid" ) )
157                         serverpid = val;
158                     else if( !strcmp(p, "shmid" ) )
159                         shm_id = val;
160                 }
161                 else if( word == 1 && !strcmp(p,"SHM_GET") )
162                     is_get = 1;
163                 else if( word == 1 && !strcmp(p,"SHM_GET_BOOL") )
164                     is_get = 2;
165                 else if( word == 1 && !strcmp(p,"SHM_GET_HIDDEN") )
166                     is_get = 3;
167                 else if( word == 2 && is_get )  {
168                     do_get_string( is_get, p, area, areasize );
169                     break;
170                 }
171                 else if( word == 1 )
172                     log_info("Status: %s\n", p);
173             }
174             if( is_info ) {
175                 if( is_info < 2 )
176                     log_fatal("SHM info without protocol version\n");
177                 if( serverpid == -1 )
178                     log_fatal("SHM info without server's pid\n");
179                 if( shm_id == -1 )
180                     log_fatal("SHM info without id\n");
181                 log_info("Shared memory info: server=%d shm_id=%d\n",
182                                                             serverpid, shm_id);
183                 area = shmat( shm_id, 0, 0 );
184                 if( area == (void*)-1 )
185                     log_fatal("attach to shared memory failed: %s\n",
186                                                             strerror(errno));
187             }
188         }
189         else
190             fputs (buf, stdout);
191     }
192
193
194     if( pclose(fp) )
195         log_error("pclose failed\n");
196
197     return 0;
198 #endif
199 }
200
201 #endif