b34ea7bd5278e0c27dc720c4d59da196ab747416
[uwplayer.git] / application / player.c
1 /*
2  * Copyright 2022 Olaf Wintermann
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a 
5  * copy of this software and associated documentation files (the "Software"), 
6  * to deal in the Software without restriction, including without limitation 
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense, 
8  * and/or sell copies of the Software, and to permit persons to whom the 
9  * Software is furnished to do so, subject to the following conditions:
10  * 
11  * The above copyright notice and this permission notice shall be included in 
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 
17  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 
18  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 
19  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER 
20  * DEALINGS IN THE SOFTWARE.
21  */
22
23 #include "player.h"
24
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <sys/fcntl.h>
29 #include <spawn.h>
30 #include <sys/wait.h>
31 #include <signal.h>
32 #include <poll.h>
33 #include <fcntl.h>
34 #include <sys/stat.h>
35 #include <errno.h>
36 #include <sys/socket.h>
37 #include <sys/un.h>
38 #include <pthread.h>
39
40 #include "json.h"
41
42 extern char **environ;
43
44 #define STR_BUFSIZE 512
45 #define WID_ARG_BUFSIZE 24
46
47 #define PLAYER_POLL_TIMEOUT 500
48 #define PLAYER_IN_BUFSIZE   8192
49
50 static void json_print(JSONValue *value, char *name, int indent);
51
52 static void* start_player(void *data);
53
54 static void player_io(Player *p);
55
56 static void handle_json_rpc_msg(Player *player, JSONValue *v);
57 static void handle_json_rpc_reqid(Player *player, JSONValue *v, int reqid);
58 static void handle_json_rpc_event(Player *player, JSONValue *v, JSONValue *event);
59
60 void PlayerOpenFile(MainWindow *win) {
61     pthread_t tid;
62     if(pthread_create(&tid, NULL, start_player, win)) {
63         perror("pthread_create");
64     }
65 }
66
67 static int prepare_player(Player *player, char *log_arg, char *ipc_arg) {
68     // create tmp directory for IPC
69     char *player_tmp = NULL;
70     char buf[STR_BUFSIZE];
71     snprintf(buf, STR_BUFSIZE, "/tmp/uwplayer-%x", rand());
72     int mkdir_success = 0;
73     for(int t=0;t<5;t++) {
74         if(!mkdir(buf, S_IRWXU)) {
75             mkdir_success = 1;
76             break;
77         } else if(errno != EEXIST) {
78             break;
79         }
80     }
81     if(!mkdir_success) return 1;
82     player_tmp = strdup(buf);
83     player->tmp = player_tmp;
84     
85     // prepare log/ipc args and create log fifo
86     int err = 0;
87     
88     if(snprintf(log_arg, STR_BUFSIZE, "--log-file=%s/%s", player_tmp, "log.fifo") >= STR_BUFSIZE) {
89         err = 1;
90     }
91     if(snprintf(ipc_arg, STR_BUFSIZE, "--input-ipc-server=%s/%s", player_tmp, "ipc.socket") >= STR_BUFSIZE) {
92         err = 1;
93     }
94     
95     snprintf(buf, STR_BUFSIZE, "%s/log.fifo", player_tmp);
96     if(err || mkfifo(buf, S_IRUSR|S_IWUSR)) {
97         rmdir(player_tmp);
98         return 1;
99     }
100     
101     return 0;
102 }
103
104 static int start_player_process(Player *player, MainWindow *win) {
105     char log_arg[STR_BUFSIZE];
106     char ipc_arg[STR_BUFSIZE];
107     
108     if(prepare_player(player, log_arg, ipc_arg)) {
109         return 1;
110     }
111     
112     char *player_bin = "/usr/local/bin/mpv"; // TODO: get bin from settings
113     
114     // -wid parameter value for embedding the player in the player_widget
115     Window wid = XtWindow(win->player_widget);
116     char wid_arg[WID_ARG_BUFSIZE];
117     if(snprintf(wid_arg, WID_ARG_BUFSIZE, "%lu", wid) >= WID_ARG_BUFSIZE) {
118         return 1;
119     }
120     
121     // create player arg list
122     char *args[32];
123     args[0] = player_bin;
124     args[1] = "-wid";
125     args[2] = wid_arg;
126     args[3] = "--no-terminal";
127     args[4] = log_arg;
128     args[5] = ipc_arg;
129     args[6] = win->file;
130     args[7] = NULL;
131     
132     posix_spawn_file_actions_t actions;
133     posix_spawn_file_actions_init(&actions);
134     
135     //posix_spawn_file_actions_adddup2(&actions, pin[0], STDIN_FILENO);
136     //posix_spawn_file_actions_adddup2(&actions, pout[1], STDOUT_FILENO);
137     
138     // start player
139     pid_t player_pid;
140     if(posix_spawn(&player_pid, player_bin, &actions, NULL, args, environ)) {
141         perror("posix_spawn");
142         return 1;
143     }
144     posix_spawn_file_actions_destroy(&actions);
145     
146     player->process = player_pid;
147     player->isactive = TRUE;
148     
149     return 0;
150 }
151
152 static int wait_for_ipc(Player *player) {
153     char buf[STR_BUFSIZE];
154     snprintf(buf, STR_BUFSIZE, "%s/log.fifo", player->tmp); // cannot fail
155     
156     // open log
157     int fd_log = open(buf, O_RDONLY);
158     if(fd_log < 0) {
159         perror("Cannot open log");
160         return 1;
161     }
162     player->log = fd_log;
163     
164     // read log until IPC socket is created
165     char *scan_str = "Listening to IPC";
166     int scan_pos = 0;
167     int scan_len = strlen(scan_str);
168     int ipc_listen = 0;
169     ssize_t r;
170     while((r = read(fd_log, buf, STR_BUFSIZE)) > 0) {
171         for(int i=0;i<r;i++) {
172             char c = buf[i];
173             char *s_str = buf+i;
174             
175             if(scan_pos == scan_len) {
176                 ipc_listen = 1;
177                 break;
178             }
179             
180             if(scan_str[scan_pos] == c) {
181                 scan_pos++;
182             } else {
183                 scan_pos = 0;
184             }
185         }
186         if(ipc_listen) break;
187     }
188     
189     return 0;
190 }
191
192 static int connect_to_ipc(Player *player) {
193     // connect to IPC socket
194     int fd_ipc = socket(AF_UNIX, SOCK_STREAM, 0);
195     if(fd_ipc < 0) {
196         perror("Cannot create IPC socket");
197         return 1;
198     }
199     player->ipc = fd_ipc;
200     
201     char buf[STR_BUFSIZE];
202     snprintf(buf, STR_BUFSIZE, "%s/%s", player->tmp, "ipc.socket"); // cannot fail
203     
204     struct sockaddr_un ipc_addr;
205     memset(&ipc_addr, 0, sizeof(struct sockaddr_un));
206     ipc_addr.sun_family = AF_UNIX;
207     memcpy(ipc_addr.sun_path, buf, strlen(buf));
208     if(connect(fd_ipc, (struct sockaddr *)&ipc_addr, sizeof(ipc_addr)) == -1) {
209         perror("Cannot connect to IPC socket");
210         return 1;
211     }
212     
213     return 0;
214 }
215
216 static void* start_player(void *data) {
217     MainWindow *win = data;
218     
219     Player *player = malloc(sizeof(Player));
220     memset(player, 0, sizeof(Player));
221     
222     // start mpv
223     if(start_player_process(player, win)) {
224         PlayerDestroy(player);
225         return NULL;
226     } 
227     
228     // wait until IPC socket is ready
229     if(wait_for_ipc(player)) {
230         PlayerDestroy(player);
231         return NULL;
232     }
233      
234     if(connect_to_ipc(player)) {
235         PlayerDestroy(player);
236         return NULL;
237     }
238     
239     // set player in main window
240     if(win->player) {
241         PlayerDestroy(win->player);
242     }
243     win->player = player;
244     
245     // IO
246     player_io(player);
247     
248     return NULL;
249 }
250
251 static void player_io(Player *p) {
252     int flags = fcntl(p->log, F_GETFL, 0);
253     fcntl(p->log, F_SETFL, flags | O_NONBLOCK);
254     
255     struct pollfd fds[2];
256     fds[0].fd = p->log;
257     fds[0].events = POLLIN;
258     fds[0].revents = 0;
259     fds[1].fd = p->ipc;
260     fds[1].events = POLLIN;
261     fds[1].revents = 0;
262     
263     JSONParser *js = json_parser_new();
264     
265     char buf[PLAYER_IN_BUFSIZE];
266     while(poll(fds, 2, PLAYER_POLL_TIMEOUT)) {
267         if(fds[0].revents == POLLIN) {
268             // clean up fifo
269             read(fds[0].fd, buf, PLAYER_IN_BUFSIZE);
270         }
271         
272         if(fds[1].revents == POLLIN) {
273             ssize_t r;
274             if((r = read(fds[1].fd, buf, PLAYER_IN_BUFSIZE)) <= 0) {
275                 break;
276             }
277             //fwrite(buf, 1, r, stdout);
278             fflush(stdout);
279             json_parser_fill(js, buf, r);
280             
281             JSONValue *value;
282             int ret;
283             while((ret = json_read_value(js, &value)) == 1) {
284                 handle_json_rpc_msg(p, value);
285                 json_value_free(value);
286             }
287             
288             if(ret == -1) {
289                 fprintf(stderr, "JSON-RPC error\n");
290                 break;
291             }
292         }
293         
294         char *cmd = "{ \"command\": [\"get_property\", \"playback-time\"], request_id=\"" REQ_ID_PLAYBACK_TIME "\" }\n";
295         //write(p->ipc, cmd, strlen(cmd));
296     }
297 }
298
299
300 static void handle_json_rpc_msg(Player *player, JSONValue *v) {
301     if(v->type != JSON_OBJECT) return;
302     
303     JSONValue *request_id_v = json_obj_get(&v->value.object, "request_id");
304     JSONValue *event = NULL;
305     if(request_id_v && request_id_v->type == JSON_STRING) {
306         int request_id = 0;
307         if(request_id_v->value.string.length == 2) {
308             request_id = 10 * (request_id_v->value.string.string[0] - '0') + (request_id_v->value.string.string[1] - '0');
309             handle_json_rpc_reqid(player, v, request_id);
310             return;
311         }
312     } else if ((event = json_obj_get(&v->value.object, "event")) != NULL) {
313         handle_json_rpc_event(player, v, event);
314     }
315     
316     json_print(v, NULL, 0);
317     fflush(stdout);
318 }
319
320 static void player_set_size(Player *player, int width, int height) {
321     if(width >= 0) {
322         player->width = width;
323     }
324     if(height >= 0) {
325         player->height = height;
326     }
327     if(player->width > 0 && player->height > 0) {
328         printf("TODO: set player size\n");
329     }
330 }
331
332 static void handle_json_rpc_reqid(Player *player, JSONValue *v, int reqid) {
333     JSONValue *data = json_obj_get(&v->value.object, "data");
334     if(!data) return;
335     
336     switch(reqid) {
337         case REQ_ID_PLAYBACK_TIME_INT: {
338             if(data->type == JSON_NUMBER) {
339                 player->playback_time = data->value.number.value;
340             }
341             break;
342         }
343         case REQ_ID_WIDTH_INT: {
344             if(data->type == JSON_INTEGER) {
345                 player_set_size(player, data->value.integer.value, -1);
346             }
347             break;
348         }
349         case REQ_ID_HEIGHT_INT: {
350             if(data->type == JSON_INTEGER) {
351                 player_set_size(player, -1, data->value.integer.value);
352             }
353             break;
354         }
355     }
356 }
357
358 static void handle_json_rpc_event(Player *p, JSONValue *v, JSONValue *event) {
359     if(!json_strcmp(event, "property-change")) {
360         printf("property change\n");
361     } else if(!json_strcmp(event, "playback-restart")) {
362         char *cmd = "{ \"command\": [\"observe_property\", 1, \"playback-time\"] }\n"
363                     "{ \"command\": [\"get_property\", \"width\"], request_id=\"" REQ_ID_WIDTH "\" }\n"
364                     "{ \"command\": [\"get_property\", \"height\"], request_id=\"" REQ_ID_HEIGHT "\" }\n";
365         write(p->ipc, cmd, strlen(cmd));
366     }
367 }
368
369 void PlayerDestroy(Player *p) {
370     if(p->log >= 0) {
371         close(p->log);
372     }
373     if(p->ipc >= 0) {
374         close(p->ipc);
375     }
376     
377     if(p->tmp) {
378         free(p->tmp);
379     }
380     
381     if(p->isactive) {
382         kill(p->process, SIGTERM);
383     }
384     
385     free(p);
386 }
387
388
389 static void json_print(JSONValue *value, char *name, int indent) {
390     if(name) {
391         printf("%*s%s: ", indent*4, "", name);
392     } else {
393         printf("%*s", indent*4, "");
394     }
395     
396     
397     switch(value->type) {
398         case JSON_OBJECT: {
399             printf("{\n");
400             
401             for(int i=0;i<value->value.object.size;i++) {
402                 JSONObjValue val = value->value.object.values[i];
403                 json_print(val.value, val.name, indent+1);
404                 if(i+1 < value->value.object.size) {
405                     printf(",\n");
406                 } else {
407                     printf("\n");
408                 }
409             }
410             
411             printf("%*s}", indent*4, "");
412             break;
413         }
414         case JSON_ARRAY: {
415             printf("[\n");
416             
417             for(int i=0;i<value->value.object.size;i++) {
418                 JSONValue *v = value->value.array.array[i];
419                 json_print(v, NULL, indent+1);
420                 if(i+1 < value->value.array.size) {
421                     printf(",\n");
422                 } else {
423                     printf("\n");
424                 }
425             }
426             
427             printf("%*s]", indent*4, "");
428             break;
429         }
430         case JSON_STRING: {
431             printf("\"%s\"", value->value.string.string);
432             break;
433         }
434         case JSON_INTEGER: {
435             printf("%i", (int)value->value.integer.value);
436             break;
437         }
438         case JSON_NUMBER: {
439             printf("%f", value->value.number.value);
440             break;
441         }
442         case JSON_LITERAL: {
443             printf("%s\n", "literal\n");
444             break;
445         }
446     }
447     
448     if(indent == 0) {
449         putchar('\n');
450     }
451 }
452