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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
/*
* media.c:
* Extract various media types from files.
*
* Copyright (c) 2002 Chris Lightfoot. All rights reserved.
* Email: chris@ex-parrot.com; WWW: http://www.ex-parrot.com/~chris/
*
*/
static const char rcsid[] = "$Id: media.c,v 1.6 2002/07/08 23:32:33 chris Exp $";
#include <assert.h>
#include <dirent.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
#include "driftnet.h"
extern char *tmpdir; /* in driftnet.c */
extern int adjunct;
extern int dpychld_fd;
/* image.c */
unsigned char *find_gif_image(const unsigned char *data, const size_t len, unsigned char **gifdata, size_t *giflen);
unsigned char *find_jpeg_image(const unsigned char *data, const size_t len, unsigned char **jpegdata, size_t *jpeglen);
/* audio.c */
unsigned char *find_mpeg_stream(const unsigned char *data, const size_t len, unsigned char **mpegdata, size_t *mpeglen);
/* playaudio.c */
void mpeg_submit_chunk(const unsigned char *data, const size_t len);
/* count_temporary_files:
* How many of our files remain in the temporary directory? We do this a
* maximum of once every five seconds. */
static int count_temporary_files(void) {
static int num;
static time_t last_counted;
if (last_counted < time(NULL) - 5) {
DIR *d;
struct dirent *de;
num = 0;
d = opendir(tmpdir);
if (d) {
while ((de = readdir(d))) {
char *p;
p = strrchr(de->d_name, '.');
if (p && (strncmp(de->d_name, "driftnet-", 9) == 0 && (strcmp(p, ".jpeg") == 0 || strcmp(p, ".gif") == 0 || strcmp(p, ".mp3") == 0)))
++num;
}
closedir(d);
last_counted = time(NULL);
}
}
return num;
}
/* dispatch_image:
* Throw some image data at the display process. */
void dispatch_image(const char *mname, const unsigned char *data, const size_t len) {
char *buf, name[TMPNAMELEN];
int fd;
buf = malloc(strlen(tmpdir) + 64);
sprintf(name, "driftnet-%08x%08x.%s", (unsigned int)time(NULL), rand(), mname);
sprintf(buf, "%s/%s", tmpdir, name);
fd = open(buf, O_WRONLY | O_CREAT | O_EXCL, 0644);
if (fd == -1) return;
write(fd, data, len);
close(fd);
if (adjunct)
printf("%s\n", buf);
#ifndef NO_DISPLAY_WINDOW
else
write(dpychld_fd, name, sizeof name);
#endif /* !NO_DISPLAY_WINDOW */
free(buf);
}
/* dispatch_mpeg_audio:
* Throw some MPEG audio into the player process or temporary directory as
* appropriate. */
void dispatch_mpeg_audio(const char *mname, const unsigned char *data, const size_t len) {
mpeg_submit_chunk(data, len);
}
/* Media types we handle. */
static struct mediadrv {
char *name;
enum mediatype type;
unsigned char *(*find_data)(const unsigned char *data, const size_t len, unsigned char **found, size_t *foundlen);
void (*dispatch_data)(const char *mname, const unsigned char *data, const size_t len);
} driver[NMEDIATYPES] = {
{ "gif", m_image, find_gif_image, dispatch_image },
{ "jpeg", m_image, find_jpeg_image, dispatch_image },
{ "mpeg", m_audio, find_mpeg_stream, dispatch_mpeg_audio }
};
/* connection_extract_media:
* Attempt to extract media data of given type/s from the named connection. */
void connection_extract_media(connection c, const enum mediatype T) {
struct datablock *b;
extern int max_tmpfiles; /* in driftnet.c */
/* Walk through the list of blocks and try to extract media data from
* those which have changed. */
for (b = c->blocks; b; b = b->next) {
if (b->len > 0 && b->dirty) {
int i;
for (i = 0; i < NMEDIATYPES; ++i)
if (driver[i].type & T) {
unsigned char *ptr, *oldptr, *media;
size_t mlen;
ptr = c->data + b->off + b->moff[i];
oldptr = NULL;
while (ptr != oldptr && ptr < c->data + b->off + b->len) {
oldptr = ptr;
ptr = driver[i].find_data(ptr, (b->off + b->len) - (ptr - c->data), &media, &mlen);
if (media && (!max_tmpfiles || count_temporary_files() < max_tmpfiles))
driver[i].dispatch_data(driver[i].name, media, mlen);
}
b->moff[i] = ptr - c->data - b->off;
}
b->dirty = 0;
}
}
}
|
