More include fixes
[asterisk/asterisk.git] / formats / format_jpeg.c
1 /*
2  * Asterisk -- A telephony toolkit for Linux.
3  *
4  * JPEG File format support.
5  * 
6  * Copyright (C) 1999, Mark Spencer
7  *
8  * Mark Spencer <markster@linux-support.net>
9  *
10  * This program is free software, distributed under the terms of
11  * the GNU General Public License
12  */
13  
14 #include <sys/types.h>
15 #include <asterisk/channel.h>
16 #include <asterisk/file.h>
17 #include <asterisk/logger.h>
18 #include <asterisk/sched.h>
19 #include <asterisk/module.h>
20 #include <asterisk/image.h>
21 #include <arpa/inet.h>
22 #include <stdlib.h>
23 #include <sys/time.h>
24 #include <stdio.h>
25 #include <unistd.h>
26 #include <errno.h>
27 #include <string.h>
28 #include <pthread.h>
29 #ifdef __linux__
30 #include <endian.h>
31 #else
32 #include <machine/endian.h>
33 #endif
34
35
36 static char *desc = "JPEG (Joint Picture Experts Group) Image Format";
37
38
39 static struct ast_frame *jpeg_read_image(int fd, int len)
40 {
41         struct ast_frame fr;
42         int res;
43         char buf[65536];
44         if (len > sizeof(buf)) {
45                 ast_log(LOG_WARNING, "JPEG image too large to read\n");
46                 return NULL;
47         }
48         res = read(fd, buf, len);
49         if (res < len) {
50                 ast_log(LOG_WARNING, "Only read %d of %d bytes: %s\n", res, len, strerror(errno));
51         }
52         memset(&fr, 0, sizeof(fr));
53         fr.frametype = AST_FRAME_IMAGE;
54         fr.subclass = AST_FORMAT_JPEG;
55         fr.data = buf;
56         fr.src = "JPEG Read";
57         fr.datalen = len;
58         return ast_frisolate(&fr);
59 }
60
61 static int jpeg_identify(int fd)
62 {
63         char buf[10];
64         int res;
65         res = read(fd, buf, sizeof(buf));
66         if (res < sizeof(buf))
67                 return 0;
68         if (memcmp(buf + 6, "JFIF", 4))
69                 return 0;
70         return 1;
71 }
72
73 static int jpeg_write_image(int fd, struct ast_frame *fr)
74 {
75         int res=0;
76         if (fr->frametype != AST_FRAME_IMAGE) {
77                 ast_log(LOG_WARNING, "Not an image\n");
78                 return -1;
79         }
80         if (fr->subclass != AST_FORMAT_JPEG) {
81                 ast_log(LOG_WARNING, "Not a jpeg image\n");
82                 return -1;
83         }
84         if (fr->datalen) {
85                 res = write(fd, fr->data, fr->datalen);
86                 if (res != fr->datalen) {
87                         ast_log(LOG_WARNING, "Only wrote %d of %d bytes: %s\n", res, fr->datalen, strerror(errno));
88                         return -1;
89                 }
90         }
91         return res;
92 }
93
94 static struct ast_imager jpeg_format = {
95         "jpg",
96         "JPEG (Joint Picture Experts Group)",
97         "jpg|jpeg",
98         AST_FORMAT_JPEG,
99         jpeg_read_image,
100         jpeg_identify,
101         jpeg_write_image,
102 };
103
104 int load_module()
105 {
106         return ast_image_register(&jpeg_format);
107 }
108
109 int unload_module()
110 {
111         ast_image_unregister(&jpeg_format);
112         return 0;
113 }       
114
115 int usecount()
116 {
117         /* We never really have any users */
118         return 0;
119 }
120
121 char *description()
122 {
123         return desc;
124 }
125
126
127 char *key()
128 {
129         return ASTERISK_GPL_KEY;
130 }