1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#include <sys/socket.h>
#include <pthread.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "noeintr.h"
#include "warnp.h"
#include "pushbits.h"
struct push {
uint8_t buf[BUFSIZ];
int in;
int out;
};
/* Bit-pushing thread. */
static void *
workthread(void * cookie)
{
struct push * P = cookie;
ssize_t readlen;
/* Infinite loop unless we hit EOF or an error. */
do {
/* Read data and die on error. */
if ((readlen = read(P->in, P->buf, BUFSIZ)) == -1) {
if (errno == EINTR)
continue;
warnp("Error reading");
exit(1);
}
/* If we hit EOF, exit the loop. */
if (readlen == 0)
break;
/* Write the data back out. */
if (noeintr_write(P->out, &P->buf, readlen) != readlen) {
warnp("Error writing");
exit(1);
}
} while (1);
/* Close the descriptor we hit EOF on. */
close(P->in);
/*
* Try to shut down the descriptor we're writing to. Ignore ENOTSOCK,
* since it might, indeed, not be a socket.
*/
if (shutdown(P->out, SHUT_WR)) {
if (errno != ENOTSOCK) {
warnp("Error shutting down socket");
exit(1);
}
}
/* Free our parameters. */
free(P);
/* We're done. */
return (NULL);
}
/**
* pushbits(in, out):
* Create a thread which copies data from ${in} to ${out}.
*/
int
pushbits(int in, int out)
{
struct push * P;
pthread_t thr;
int rc;
/* Allocate structure. */
if ((P = malloc(sizeof(struct push))) == NULL)
goto err0;
P->in = in;
P->out = out;
/* Create thread. */
if ((rc = pthread_create(&thr, NULL, workthread, P)) != 0) {
warn0("pthread_create: %s", strerror(rc));
goto err1;
}
/* Success! */
return (0);
err1:
free(P);
err0:
/* Failure! */
return (-1);
}
|
