issue #5605
[asterisk/asterisk.git] / formats / format_jpeg.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Mark Spencer <markster@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief JPEG File format support.
22  * 
23  * \arg File name extension: jpeg, jpg
24  * \ingroup formats
25  */
26  
27 #include <sys/types.h>
28 #include <netinet/in.h>
29 #include <arpa/inet.h>
30 #include <stdlib.h>
31 #include <sys/time.h>
32 #include <stdio.h>
33 #include <unistd.h>
34 #include <errno.h>
35 #include <string.h>
36
37 #include "asterisk.h"
38
39 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
40
41 #include "asterisk/channel.h"
42 #include "asterisk/file.h"
43 #include "asterisk/logger.h"
44 #include "asterisk/sched.h"
45 #include "asterisk/module.h"
46 #include "asterisk/image.h"
47 #include "asterisk/lock.h"
48 #include "asterisk/endian.h"
49
50 static char *desc = "JPEG (Joint Picture Experts Group) Image Format";
51
52
53 static struct ast_frame *jpeg_read_image(int fd, int len)
54 {
55         struct ast_frame fr;
56         int res;
57         char buf[65536];
58         if (len > sizeof(buf)) {
59                 ast_log(LOG_WARNING, "JPEG image too large to read\n");
60                 return NULL;
61         }
62         res = read(fd, buf, len);
63         if (res < len) {
64                 ast_log(LOG_WARNING, "Only read %d of %d bytes: %s\n", res, len, strerror(errno));
65         }
66         memset(&fr, 0, sizeof(fr));
67         fr.frametype = AST_FRAME_IMAGE;
68         fr.subclass = AST_FORMAT_JPEG;
69         fr.data = buf;
70         fr.src = "JPEG Read";
71         fr.datalen = len;
72         return ast_frisolate(&fr);
73 }
74
75 static int jpeg_identify(int fd)
76 {
77         char buf[10];
78         int res;
79         res = read(fd, buf, sizeof(buf));
80         if (res < sizeof(buf))
81                 return 0;
82         if (memcmp(buf + 6, "JFIF", 4))
83                 return 0;
84         return 1;
85 }
86
87 static int jpeg_write_image(int fd, struct ast_frame *fr)
88 {
89         int res=0;
90         if (fr->frametype != AST_FRAME_IMAGE) {
91                 ast_log(LOG_WARNING, "Not an image\n");
92                 return -1;
93         }
94         if (fr->subclass != AST_FORMAT_JPEG) {
95                 ast_log(LOG_WARNING, "Not a jpeg image\n");
96                 return -1;
97         }
98         if (fr->datalen) {
99                 res = write(fd, fr->data, fr->datalen);
100                 if (res != fr->datalen) {
101                         ast_log(LOG_WARNING, "Only wrote %d of %d bytes: %s\n", res, fr->datalen, strerror(errno));
102                         return -1;
103                 }
104         }
105         return res;
106 }
107
108 static struct ast_imager jpeg_format = {
109         "jpg",
110         "JPEG (Joint Picture Experts Group)",
111         "jpg|jpeg",
112         AST_FORMAT_JPEG,
113         jpeg_read_image,
114         jpeg_identify,
115         jpeg_write_image,
116 };
117
118 int load_module()
119 {
120         return ast_image_register(&jpeg_format);
121 }
122
123 int unload_module()
124 {
125         ast_image_unregister(&jpeg_format);
126         return 0;
127 }       
128
129 int usecount()
130 {
131         /* We never really have any users */
132         return 0;
133 }
134
135 char *description()
136 {
137         return desc;
138 }
139
140
141 char *key()
142 {
143         return ASTERISK_GPL_KEY;
144 }