jmv/gtkballs
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

child.c: consistent indentation

Browse Source
This commit is contained in:
wdlkmpx 2021-01-14 20:58:50 +08:00
parent 9a53cd2c48
commit d4b2abb5f9
1 changed files with 105 additions and 99 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

204
src/child.c
View File

@ -22,120 +22,126 @@
volatile sig_atomic_t _child_writer_alive = 0;
gboolean _child_writer_dead_warned = 0;
int child_writer_alive(void) {
return _child_writer_alive;
int child_writer_alive(void){
return _child_writer_alive;
}
int child_writer_dead_handler(void) {
if(_child_writer_alive == 0 && _child_writer_dead_warned == 0) {
_child_writer_dead_warned = 1;
return 1;
}
return 0;
if (_child_writer_alive == 0 && _child_writer_dead_warned == 0) {
_child_writer_dead_warned = 1;
return 1;
}
return 0;
}
void child_process_score_writer(int chfd, gchar *score_file) {
fd_set rfds;
int i;
int fd=-1;
gchar *buf;
size_t sz;
sigset_t sset;
struct flock lockinfo;
gchar *score_file_full = g_strconcat(LOCALSTATEDIR, score_file, NULL);
void child_process_score_writer(int chfd, gchar *score_file)
{
fd_set rfds;
int i;
int fd=-1;
gchar *buf;
size_t sz;
sigset_t sset;
struct flock lockinfo;
gchar *score_file_full = g_strconcat(LOCALSTATEDIR, score_file, NULL);
while(1) {
FD_ZERO(&rfds);
FD_SET(chfd, &rfds);
if(select(chfd + 1, &rfds, NULL, NULL, NULL) > 0) {
if(read(chfd, &sz, sizeof(sz)) <= 0) {
exit(0);
}
/* block signals before writing, to prevent possible file corruption */
sigemptyset(&sset);
sigaddset(&sset, SIGHUP);
sigaddset(&sset, SIGINT);
sigaddset(&sset, SIGQUIT);
sigaddset(&sset, SIGTERM);
sigprocmask(SIG_BLOCK, &sset, NULL);
fd = open(score_file_full, O_WRONLY | O_TRUNC);
if(fd != -1) {
/* get write lock before writing scores */
lockinfo.l_whence = SEEK_SET;
lockinfo.l_start = 0;
lockinfo.l_len = 0;
while (1)
{
FD_ZERO(&rfds);
FD_SET(chfd, &rfds);
if (select(chfd + 1, &rfds, NULL, NULL, NULL) > 0)
{
if (read(chfd, &sz, sizeof(sz)) <= 0) {
exit(0);
}
/* block signals before writing, to prevent possible file corruption */
sigemptyset(&sset);
sigaddset(&sset, SIGHUP);
sigaddset(&sset, SIGINT);
sigaddset(&sset, SIGQUIT);
sigaddset(&sset, SIGTERM);
sigprocmask(SIG_BLOCK, &sset, NULL);
fd = open(score_file_full, O_WRONLY | O_TRUNC);
if (fd != -1) {
/* get write lock before writing scores */
lockinfo.l_whence = SEEK_SET;
lockinfo.l_start = 0;
lockinfo.l_len = 0;
i = 0;
while(1) {
lockinfo.l_type=F_WRLCK;
if(!fcntl(fd, F_SETLK, &lockinfo)) break;
if(i >= 3) {
close(fd);
fd = -1;
}
i++;
}
}
while(sz > 0) {
buf=g_malloc(sz);
read(chfd, buf, sz);
if(fd != -1) {
write(fd, buf, sz);
}
g_free(buf);
read(chfd, &sz, sizeof(sz));
}
i = 0;
while (1) {
lockinfo.l_type=F_WRLCK;
if (!fcntl(fd, F_SETLK, &lockinfo)) {
break;
}
if (i >= 3) {
close(fd);
fd = -1;
}
i++;
}
}
while (sz > 0) {
buf=g_malloc(sz);
read(chfd, buf, sz);
if (fd != -1) {
write(fd, buf, sz);
}
g_free(buf);
read(chfd, &sz, sizeof(sz));
}
if(fd != -1) {
close(fd);
} else {
/* FIXME: here should be some sort of error
reporting to parent */
}
sigprocmask(SIG_UNBLOCK, &sset, NULL);
}
}
if (fd != -1) {
close(fd);
} else {
/* FIXME: here should be some sort of error reporting to parent */
}
sigprocmask(SIG_UNBLOCK, &sset, NULL);
}
}
}
static void sigchld_handler(int param) {
pid_t ret = waitpid(0, NULL, WNOHANG);
static void sigchld_handler(int param)
{
pid_t ret = waitpid(0, NULL, WNOHANG);
if(ret > 0) { /* score writer process killed by bastards! */
_child_writer_alive = 0;
}
if (ret > 0) { /* score writer process killed by bastards! */
_child_writer_alive = 0;
}
}
int child_setup(gchar *score_file) {
pid_t pid;
struct sigaction sact;
int sfds[2];
int child_setup(gchar *score_file)
{
pid_t pid;
struct sigaction sact;
int sfds[2];
/* set up SIGCHLD handler, so we can know if our score writer process
terminated while we run... */
sact.sa_handler = sigchld_handler;
sigemptyset(&sact.sa_mask);
sact.sa_flags = 0;
/* set up SIGCHLD handler, so we can know if our score writer process
terminated while we run... */
sact.sa_handler = sigchld_handler;
sigemptyset(&sact.sa_mask);
sact.sa_flags = 0;
#ifdef SA_RESTART
sact.sa_flags |= SA_RESTART;
sact.sa_flags |= SA_RESTART;
#endif
if(sigaction(SIGCHLD, &sact, NULL) < 0) {
printf("cannot setup SIGCHLD handler.\n");
return -1;
}
if(pipe(sfds) == -1) {
printf("pipe() failed: %s\n", strerror(errno));
return -1;
}
pid = fork();
if(pid == -1) {
printf("cannot fork: %s\n", strerror(errno));
return -1;
} else if(pid == 0) {
close(sfds[1]);
child_process_score_writer(sfds[0], score_file);
}
close(sfds[0]);
_child_writer_alive = 1;
if (sigaction(SIGCHLD, &sact, NULL) < 0) {
printf("cannot setup SIGCHLD handler.\n");
return -1;
}
if (pipe(sfds) == -1) {
printf("pipe() failed: %s\n", strerror(errno));
return -1;
}
pid = fork();
if (pid == -1) {
printf("cannot fork: %s\n", strerror(errno));
return -1;
} else if (pid == 0) {
close(sfds[1]);
child_process_score_writer(sfds[0], score_file);
}
close(sfds[0]);
_child_writer_alive = 1;
return sfds[1];
return sfds[1];
}