rc4: Round up output buffer size.
[paraslash.git] / dbadm.c
1 /*
2 * Copyright (C) 2004-2006 Andre Noll <maan@systemlinux.org>
3 *
4 * Licensed under the GPL v2. For licencing details see COPYING.
5 */
6
7 /** \file dbadm.c simple attribute setting utility for the mysql selector */
8
9 #include "para.h"
10 #include <menu.h>
11 #include "string.h"
12
13 //#define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0]))
14 #define MAXLINE 255
15
16 #define OFFSET 4
17 static int att_win_lines, att_win_cols;
18 static int att_format_lines, att_format_cols;
19 static int atts_modified, refresh_file = 1;
20 static int n_choices, choice_len;
21 static char *atts;
22 ITEM **my_items;
23 WINDOW *att_win;
24
25 static char **choices;
26
27 /* no looging */
28 void para_log(__a_unused int ll, __a_unused const char *fmt,...)
29 {
30 }
31
32 static int client_cmd(const char *cmd)
33 {
34 pid_t pid;
35 int ret, fds[3] = {0, 1, 0};
36 char *cmdline = make_message(BINDIR "/para_client %s", cmd);
37 ret = para_exec_cmdline_pid(&pid, cmdline, fds);
38 free(cmdline);
39 if (ret < 0)
40 return -1;
41 return fds[1];
42 }
43
44 static char **get_all_atts(int *num_atts)
45 {
46 int fd = client_cmd("laa");
47 FILE *f;
48 char **ret = NULL, *buf;
49
50 if (fd < 0)
51 return NULL;
52 f = fdopen(fd, "r");
53 if (!f) {
54 close(fd);
55 return NULL;
56 }
57 *num_atts = 0;
58 buf = para_malloc(MAXLINE * sizeof(char));
59 while (fgets(buf, MAXLINE - 1, f) && *buf) {
60 size_t n = strlen(buf);
61 buf[n - 1] = '\0';
62 if (choice_len < n - 1)
63 choice_len = n - 1;
64 ret = para_realloc(ret, (*num_atts + 1) * sizeof(char*));
65 ret[*num_atts] = para_strdup(buf);
66 *num_atts += 1;
67 }
68 free(buf);
69 return ret;
70 }
71
72 static char *get_atts(char *filename)
73 {
74 int n, fd, bufsize = (n_choices * (choice_len + 1) + 10) * sizeof(char);
75 char *cmd = make_message("la %s", filename), *buf;
76
77 fd = client_cmd(cmd);
78 free(cmd);
79 if (fd < 0)
80 return NULL;
81 buf = para_malloc(bufsize * sizeof(char));
82 n = read(fd, buf, bufsize - 1);
83 if (n <= 0 ||strstr(buf, "Not contained in database")) {
84 free(buf);
85 return NULL;
86 };
87 return buf;
88 }
89
90 static void _item_init(__a_unused MENU* menu)
91 {
92 // static int subsequent_run;
93 int i, n;
94 char *p = atts;
95 char att[MAXLINE];
96
97 if (!refresh_file)
98 return;
99 refresh_file = 0;
100 for (i = 0; i < n_choices; i++)
101 set_item_value(my_items[i], FALSE);
102 while (sscanf(p, "%s%n", att, &n) > 0) {
103 //mvprintw(LINES - 4, 0, "aaaitem.");
104 p += n + 1;
105 for (i = 0; i < n_choices; i++) {
106 if (!strcmp(item_name(my_items[i]), att)) {
107 set_item_value(my_items[i], TRUE);
108 break;
109 }
110 }
111 }
112 }
113
114 struct color_pair {
115 int bg;
116 int fg;
117 };
118
119 enum {
120 COLOR_DUMMY,
121 COLOR_FRAME,
122 COLOR_FILENAME,
123 COLOR_ACTIVE_ITEM,
124 COLOR_INACTIVE_ITEM
125 };
126
127 static void init_colors(void)
128 {
129 start_color();
130 init_pair(COLOR_FRAME, COLOR_BLUE, COLOR_BLACK);
131 init_pair(COLOR_FILENAME, COLOR_MAGENTA, COLOR_BLACK);
132 init_pair(COLOR_ACTIVE_ITEM, COLOR_RED, COLOR_WHITE);
133 init_pair(COLOR_INACTIVE_ITEM, COLOR_WHITE, COLOR_BLACK);
134 }
135
136 static int commit_changes(char *filename)
137 {
138 // ITEM **items;
139 int i;
140 char buf[MAXLINE] = "para_client sa ";
141 int fds[3] = {0, 0, 0};
142 pid_t pid;
143
144 for (i = 0; i < n_choices; ++i) {
145 strcat(buf, item_name(my_items[i]));
146 if (item_value(my_items[i])) {
147 // printf("%s\n", item_name(my_items[i]));
148 strcat(buf, "+ ");
149 } else
150 strcat(buf, "- ");
151 }
152 strcat(buf, filename);
153 //printf("old atts: %s\n", atts);
154 //printf("%s\n", buf);
155 return para_exec_cmdline_pid(&pid, buf, fds);
156 }
157
158 static char *get_current_filename(void)
159 {
160 char *bn = NULL, *buf = para_malloc(MAXLINE * sizeof(char));
161 int ret, fd;
162
163 fd = client_cmd("sc 1");
164 if (fd < 0)
165 return NULL;
166 ret = read(fd, buf, MAXLINE - 1);
167 if (ret <= 0)
168 goto out;
169 buf[ret] = '\0';
170 bn = para_basename(buf);
171 free(buf);
172 out:
173 close(fd);
174 return bn;
175 }
176
177 static void print_filename(char *filename)
178 {
179 char *tmp = strdup(filename);
180 int maxlen = att_win_cols - 2;
181
182 wattron(att_win, COLOR_PAIR(COLOR_FILENAME));
183 if (strlen(filename) > maxlen)
184 tmp[maxlen] = '\0';
185 wmove(att_win, 1, 1);
186 clrtoeol();
187 mvwprintw(att_win, 1, 1, "%s", tmp);
188 wattron(att_win, COLOR_PAIR(COLOR_FRAME));
189 mvwaddch(att_win, 2, att_win_cols - 1, ACS_RTEE);
190 free(tmp);
191 }
192
193 static int com_refresh_file(char *filename) {
194
195
196 filename = get_current_filename();
197 if (!filename)
198 return -1;
199 atts = get_atts(filename);
200 if (!atts)
201 return -1;
202 print_filename(filename);
203 return 1;
204 }
205
206 static int init_curses(void)
207 {
208 /* Initialize curses */
209 initscr();
210
211 if (LINES <= OFFSET + 4)
212 return -1;
213
214 att_format_cols = (COLS - 2 * OFFSET - 3) / (choice_len + 1);
215 if (att_format_cols < 1)
216 return -1;
217 att_format_lines = (n_choices - 1) / att_format_cols + 1;
218 if (att_format_lines + OFFSET + 4 > LINES)
219 att_format_lines = LINES - OFFSET - 4;
220
221 att_win_lines = att_format_lines + 4;
222 att_win_cols = (choice_len + 1) * att_format_cols + 1;
223 if (att_win_lines + OFFSET > LINES)
224 att_win_lines = LINES - OFFSET - 1;
225 if (att_win_cols + 2 * OFFSET > COLS)
226 att_win_cols = COLS - 2 * OFFSET + 1;
227 //printf ("%i:%i, %i:%i\n", att_format_lines, att_format_cols, att_win_lines, att_win_cols); fflush(stdout); sleep(2);
228
229 cbreak();
230 noecho();
231 keypad(stdscr, TRUE);
232 init_colors();
233 return 1;
234 }
235
236 int main(int argc, char *argv[])
237 {
238 int c, i, ret = EXIT_FAILURE;
239 MENU *my_menu = NULL;
240 char *filename;
241
242 choices = get_all_atts(&n_choices);
243 if (!choices || n_choices <= 0)
244 exit(EXIT_FAILURE);
245 if (argc < 2) {
246 filename = get_current_filename();
247 if (!filename)
248 exit(EXIT_FAILURE);
249 } else
250 filename = strdup(argv[1]);
251 atts = get_atts(filename);
252 if (!atts)
253 goto out;
254 if (init_curses() < 0)
255 goto out;
256
257 /* Initialize items */
258 my_items = (ITEM **)calloc(n_choices + 1, sizeof(ITEM *));
259 for (i = 0; i < n_choices; ++i)
260 my_items[i] = new_item(choices[i], "");
261
262 my_menu = new_menu(my_items);
263 set_item_init(my_menu, _item_init);
264 /* Make the menu multi valued */
265 menu_opts_off(my_menu, O_ONEVALUE);
266 /* Set menu option not to show the description */
267 menu_opts_off(my_menu, O_SHOWDESC);
268
269 /* Create the window to be associated with the menu */
270 att_win = newwin(att_win_lines, att_win_cols, OFFSET, OFFSET);
271 keypad(att_win, TRUE);
272 /* Set main window and sub window */
273 set_menu_win(my_menu, att_win);
274 set_menu_sub(my_menu, derwin(att_win, att_win_lines - 4,
275 att_win_cols - 2, 3, 1));
276 set_menu_format(my_menu, att_format_lines, att_format_cols);
277 //set_menu_format(my_menu, 5, 1);
278 set_menu_mark(my_menu, "");
279
280 /* Print a border around the main window and print a title */
281 wattron(att_win, COLOR_PAIR(COLOR_FRAME));
282 box(att_win, 0, 0);
283 mvwhline(att_win, 2, 1, ACS_HLINE , att_win_cols - 2);
284 mvwaddch(att_win, 2, 0, ACS_LTEE);
285 print_filename(filename);
286 set_menu_fore(my_menu, COLOR_PAIR(COLOR_ACTIVE_ITEM) | A_REVERSE);
287 set_menu_back(my_menu, COLOR_PAIR(COLOR_INACTIVE_ITEM));
288 refresh();
289 post_menu(my_menu);
290
291 repeat:
292 wrefresh(att_win);
293 c = getch();
294 switch(c) {
295 case KEY_DOWN:
296 menu_driver(my_menu, REQ_DOWN_ITEM);
297 goto repeat;
298 case KEY_UP:
299 menu_driver(my_menu, REQ_UP_ITEM);
300 goto repeat;
301 case KEY_LEFT:
302 menu_driver(my_menu, REQ_LEFT_ITEM);
303 goto repeat;
304 case KEY_RIGHT:
305 menu_driver(my_menu, REQ_RIGHT_ITEM);
306 goto repeat;
307
308 case 'q':
309 ret = EXIT_SUCCESS;
310 goto out;
311 case 'r':
312 com_refresh_file(filename);
313 refresh_file = 1;
314 _item_init(NULL);
315 goto repeat;
316
317 case ' ':
318 menu_driver(my_menu, REQ_TOGGLE_ITEM);
319 atts_modified = 1;
320 goto repeat;
321 /* Enter */
322 case 10:
323 if (atts_modified)
324 commit_changes(filename);
325 goto out;
326 default:
327 goto repeat;
328 }
329 out:
330 if (my_items) {
331 free_item(my_items[0]);
332 free_item(my_items[1]);
333 }
334 for (i = 0; i < n_choices; i++)
335 free(choices[i]);
336 free(choices);
337 if (my_menu)
338 free_menu(my_menu);
339 if (atts)
340 free(atts);
341 if (filename)
342 free(filename);
343 endwin();
344 exit(ret);
345 }
346