0e81d890fec592199f603d47263cf6186a7e062e
[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 "asterisk/lock.h"
22 #include "asterisk/endian.h"
23 #include <netinet/in.h>
24 #include <arpa/inet.h>
25 #include <stdlib.h>
26 #include <sys/time.h>
27 #include <stdio.h>
28 #include <unistd.h>
29 #include <errno.h>
30 #include <string.h>
31
32
33 static char *desc = "JPEG (Joint Picture Experts Group) Image Format";
34
35
36 static struct ast_frame *jpeg_read_image(int fd, int len)
37 {
38         struct ast_frame fr;
39         int res;
40         char buf[65536];
41         if (len > sizeof(buf)) {
42                 ast_log(LOG_WARNING, "JPEG image too large to read\n");
43                 return NULL;
44         }
45         res = read(fd, buf, len);
46         if (res < len) {
47                 ast_log(LOG_WARNING, "Only read %d of %d bytes: %s\n", res, len, strerror(errno));
48         }
49         memset(&fr, 0, sizeof(fr));
50         fr.frametype = AST_FRAME_IMAGE;
51         fr.subclass = AST_FORMAT_JPEG;
52         fr.data = buf;
53         fr.src = "JPEG Read";
54         fr.datalen = len;
55         return ast_frisolate(&fr);
56 }
57
58 static int jpeg_identify(int fd)
59 {
60         char buf[10];
61         int res;
62         res = read(fd, buf, sizeof(buf));
63         if (res < sizeof(buf))
64                 return 0;
65         if (memcmp(buf + 6, "JFIF", 4))
66                 return 0;
67         return 1;
68 }
69
70 static int jpeg_write_image(int fd, struct ast_frame *fr)
71 {
72         int res=0;
73         if (fr->frametype != AST_FRAME_IMAGE) {
74                 ast_log(LOG_WARNING, "Not an image\n");
75                 return -1;
76         }
77         if (fr->subclass != AST_FORMAT_JPEG) {
78                 ast_log(LOG_WARNING, "Not a jpeg image\n");
79                 return -1;
80         }
81         if (fr->datalen) {
82                 res = write(fd, fr->data, fr->datalen);
83                 if (res != fr->datalen) {
84                         ast_log(LOG_WARNING, "Only wrote %d of %d bytes: %s\n", res, fr->datalen, strerror(errno));
85                         return -1;
86                 }
87         }
88         return res;
89 }
90
91 static struct ast_imager jpeg_format = {
92         "jpg",
93         "JPEG (Joint Picture Experts Group)",
94         "jpg|jpeg",
95         AST_FORMAT_JPEG,
96         jpeg_read_image,
97         jpeg_identify,
98         jpeg_write_image,
99 };
100
101 int load_module()
102 {
103         return ast_image_register(&jpeg_format);
104 }
105
106 int unload_module()
107 {
108         ast_image_unregister(&jpeg_format);
109         return 0;
110 }       
111
112 int usecount()
113 {
114         /* We never really have any users */
115         return 0;
116 }
117
118 char *description()
119 {
120         return desc;
121 }
122
123
124 char *key()
125 {
126         return ASTERISK_GPL_KEY;
127 }