526b5580da
git-svn-id: svn://svn.rockbox.org/rockbox/trunk@21070 a1c6a512-1295-4272-9138-f99709370657
312 lines
6.6 KiB
C
312 lines
6.6 KiB
C
/* (C) Guenter Geiger <geiger@epy.co.at> */
|
|
|
|
#include "m_pd.h"
|
|
#ifdef NT
|
|
#pragma warning( disable : 4244 )
|
|
#pragma warning( disable : 4305 )
|
|
#endif
|
|
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <signal.h>
|
|
#include <sched.h>
|
|
|
|
void sys_rmpollfn(int fd);
|
|
void sys_addpollfn(int fd, void* fn, void *ptr);
|
|
|
|
/* ------------------------ shell ----------------------------- */
|
|
|
|
#define INBUFSIZE 1024
|
|
|
|
static t_class *shell_class;
|
|
|
|
|
|
static void drop_priority(void)
|
|
{
|
|
#ifdef _POSIX_PRIORITY_SCHEDULING
|
|
struct sched_param par;
|
|
int p1 ,p2, p3;
|
|
par.sched_priority = 0;
|
|
sched_setscheduler(0,SCHED_OTHER,&par);
|
|
#endif
|
|
}
|
|
|
|
|
|
typedef struct _shell
|
|
{
|
|
t_object x_obj;
|
|
int x_echo;
|
|
char *sr_inbuf;
|
|
int sr_inhead;
|
|
int sr_intail;
|
|
void* x_binbuf;
|
|
int fdpipe[2];
|
|
int fdinpipe[2];
|
|
int pid;
|
|
int x_del;
|
|
t_outlet* x_done;
|
|
t_clock* x_clock;
|
|
} t_shell;
|
|
|
|
static int shell_pid;
|
|
|
|
|
|
void shell_cleanup(t_shell* x)
|
|
{
|
|
sys_rmpollfn(x->fdpipe[0]);
|
|
|
|
if (x->fdpipe[0]>0) close(x->fdpipe[0]);
|
|
if (x->fdpipe[1]>0) close(x->fdpipe[1]);
|
|
if (x->fdinpipe[0]>0) close(x->fdinpipe[0]);
|
|
if (x->fdinpipe[1]>0) close(x->fdinpipe[1]);
|
|
|
|
x->fdpipe[0] = -1;
|
|
x->fdpipe[1] = -1;
|
|
x->fdinpipe[0] = -1;
|
|
x->fdinpipe[1] = -1;
|
|
clock_unset(x->x_clock);
|
|
}
|
|
|
|
void shell_check(t_shell* x)
|
|
{
|
|
int ret;
|
|
int status;
|
|
ret = waitpid(x->pid,&status,WNOHANG);
|
|
if (ret == x->pid) {
|
|
shell_cleanup(x);
|
|
if (WIFEXITED(status)) {
|
|
outlet_float(x->x_done,WEXITSTATUS(status));
|
|
}
|
|
else outlet_float(x->x_done,0);
|
|
}
|
|
else {
|
|
if (x->x_del < 100) x->x_del+=2; /* increment poll times */
|
|
clock_delay(x->x_clock,x->x_del);
|
|
}
|
|
}
|
|
|
|
|
|
void shell_bang(t_shell *x)
|
|
{
|
|
post("bang");
|
|
}
|
|
|
|
/* snippet from pd's code */
|
|
static void shell_doit(void *z, t_binbuf *b)
|
|
{
|
|
t_shell *x = (t_shell *)z;
|
|
int msg, natom = binbuf_getnatom(b);
|
|
t_atom *at = binbuf_getvec(b);
|
|
|
|
for (msg = 0; msg < natom;)
|
|
{
|
|
int emsg;
|
|
for (emsg = msg; emsg < natom && at[emsg].a_type != A_COMMA
|
|
&& at[emsg].a_type != A_SEMI; emsg++)
|
|
;
|
|
if (emsg > msg)
|
|
{
|
|
int i;
|
|
for (i = msg; i < emsg; i++)
|
|
if (at[i].a_type == A_DOLLAR || at[i].a_type == A_DOLLSYM)
|
|
{
|
|
pd_error(x, "netreceive: got dollar sign in message");
|
|
goto nodice;
|
|
}
|
|
if (at[msg].a_type == A_FLOAT)
|
|
{
|
|
if (emsg > msg + 1)
|
|
outlet_list(x->x_obj.ob_outlet, 0, emsg-msg, at + msg);
|
|
else outlet_float(x->x_obj.ob_outlet, at[msg].a_w.w_float);
|
|
}
|
|
else if (at[msg].a_type == A_SYMBOL)
|
|
outlet_anything(x->x_obj.ob_outlet, at[msg].a_w.w_symbol,
|
|
emsg-msg-1, at + msg + 1);
|
|
}
|
|
nodice:
|
|
msg = emsg + 1;
|
|
}
|
|
}
|
|
|
|
|
|
void shell_read(t_shell *x, int fd)
|
|
{
|
|
char buf[INBUFSIZE];
|
|
t_binbuf* bbuf = binbuf_new();
|
|
int i;
|
|
int readto =
|
|
(x->sr_inhead >= x->sr_intail ? INBUFSIZE : x->sr_intail-1);
|
|
int ret;
|
|
|
|
ret = read(fd, buf,INBUFSIZE-1);
|
|
buf[ret] = '\0';
|
|
|
|
for (i=0;i<ret;i++)
|
|
if (buf[i] == '\n') buf[i] = ';';
|
|
if (ret < 0)
|
|
{
|
|
error("shell: pipe read error");
|
|
sys_rmpollfn(fd);
|
|
x->fdpipe[0] = -1;
|
|
close(fd);
|
|
return;
|
|
}
|
|
else if (ret == 0)
|
|
{
|
|
post("EOF on socket %d\n", fd);
|
|
sys_rmpollfn(fd);
|
|
x->fdpipe[0] = -1;
|
|
close(fd);
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
int natom;
|
|
t_atom *at;
|
|
binbuf_text(bbuf, buf, strlen(buf));
|
|
|
|
natom = binbuf_getnatom(bbuf);
|
|
at = binbuf_getvec(bbuf);
|
|
shell_doit(x,bbuf);
|
|
}
|
|
binbuf_free(bbuf);
|
|
}
|
|
|
|
|
|
static void shell_send(t_shell *x, t_symbol *s,int ac, t_atom *at)
|
|
{
|
|
int i;
|
|
char tmp[MAXPDSTRING];
|
|
int size = 0;
|
|
|
|
if (x->fdinpipe[0] == -1) return; /* nothing to send to */
|
|
|
|
for (i=0;i<ac;i++) {
|
|
atom_string(at,tmp+size,MAXPDSTRING - size);
|
|
at++;
|
|
size=strlen(tmp);
|
|
tmp[size++] = ' ';
|
|
}
|
|
tmp[size-1] = '\0';
|
|
post("sending %s",tmp);
|
|
write(x->fdinpipe[0],tmp,strlen(tmp));
|
|
}
|
|
|
|
static void shell_anything(t_shell *x, t_symbol *s, int ac, t_atom *at)
|
|
{
|
|
int i;
|
|
char* argv[20];
|
|
t_symbol* sym;
|
|
|
|
if (!strcmp(s->s_name,"send")) {
|
|
post("send");
|
|
shell_send(x,s,ac,at);
|
|
return;
|
|
}
|
|
|
|
argv[0] = s->s_name;
|
|
|
|
if (x->fdpipe[0] != -1) {
|
|
post("shell: old process still running");
|
|
kill(x->pid,SIGKILL);
|
|
shell_cleanup(x);
|
|
}
|
|
|
|
|
|
if (pipe(x->fdpipe) < 0) {
|
|
error("unable to create pipe");
|
|
return;
|
|
}
|
|
|
|
if (pipe(x->fdinpipe) < 0) {
|
|
error("unable to create input pipe");
|
|
return;
|
|
}
|
|
|
|
|
|
sys_addpollfn(x->fdpipe[0],shell_read,x);
|
|
|
|
if (!(x->pid = fork())) {
|
|
int status;
|
|
char* cmd = getbytes(1024);
|
|
char* tcmd = getbytes(1024);
|
|
strcpy(cmd,s->s_name);
|
|
|
|
#if 0
|
|
for (i=1;i<=ac;i++) {
|
|
argv[i] = getbytes(255);
|
|
atom_string(at,argv[i],255);
|
|
/* post("argument %s",argv[i]); */
|
|
at++;
|
|
}
|
|
argv[i] = 0;
|
|
#endif
|
|
for (i=1;i<=ac;i++) {
|
|
atom_string(at,tcmd,255);
|
|
strcat(cmd," ");
|
|
strcat(cmd,tcmd);
|
|
at++;
|
|
}
|
|
|
|
|
|
/* reassign stdout */
|
|
dup2(x->fdpipe[1],1);
|
|
dup2(x->fdinpipe[1],0);
|
|
|
|
/* drop privileges */
|
|
drop_priority();
|
|
seteuid(getuid()); /* lose setuid priveliges */
|
|
|
|
post("executing %s",cmd);
|
|
system(cmd);
|
|
// execvp(s->s_name,argv);
|
|
exit(0);
|
|
}
|
|
x->x_del = 4;
|
|
clock_delay(x->x_clock,x->x_del);
|
|
|
|
if (x->x_echo)
|
|
outlet_anything(x->x_obj.ob_outlet, s, ac, at);
|
|
}
|
|
|
|
|
|
|
|
void shell_free(t_shell* x)
|
|
{
|
|
binbuf_free(x->x_binbuf);
|
|
}
|
|
|
|
static void *shell_new(void)
|
|
{
|
|
t_shell *x = (t_shell *)pd_new(shell_class);
|
|
|
|
x->x_echo = 0;
|
|
x->fdpipe[0] = -1;
|
|
x->fdpipe[1] = -1;
|
|
x->fdinpipe[0] = -1;
|
|
x->fdinpipe[1] = -1;
|
|
|
|
x->sr_inhead = x->sr_intail = 0;
|
|
if (!(x->sr_inbuf = (char*) malloc(INBUFSIZE))) bug("t_shell");;
|
|
|
|
x->x_binbuf = binbuf_new();
|
|
|
|
outlet_new(&x->x_obj, &s_list);
|
|
x->x_done = outlet_new(&x->x_obj, &s_bang);
|
|
x->x_clock = clock_new(x, (t_method) shell_check);
|
|
return (x);
|
|
}
|
|
|
|
void shell_setup(void)
|
|
{
|
|
shell_class = class_new(gensym("shell"), (t_newmethod)shell_new,
|
|
(t_method)shell_free,sizeof(t_shell), 0,0);
|
|
class_addbang(shell_class,shell_bang);
|
|
class_addanything(shell_class, shell_anything);
|
|
}
|
|
|
|
|