d6733020ec61ba3a5da6b112ee5faa8cbe4f00b3
[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 /*** MODULEINFO
28         <support_level>extended</support_level>
29  ***/
30  
31 #include "asterisk.h"
32
33 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
34
35 #include "asterisk/mod_format.h"
36 #include "asterisk/module.h"
37 #include "asterisk/image.h"
38 #include "asterisk/endian.h"
39
40 static struct ast_frame *jpeg_read_image(int fd, int len)
41 {
42         struct ast_frame fr;
43         int res;
44         char buf[65536];
45         if (len > sizeof(buf) || len < 0) {
46                 ast_log(LOG_WARNING, "JPEG image too large to read\n");
47                 return NULL;
48         }
49         res = read(fd, buf, len);
50         if (res < len) {
51                 ast_log(LOG_WARNING, "Only read %d of %d bytes: %s\n", res, len, strerror(errno));
52         }
53         memset(&fr, 0, sizeof(fr));
54         fr.frametype = AST_FRAME_IMAGE;
55         ast_format_set(&fr.subclass.format, AST_FORMAT_JPEG, 0);
56         fr.data.ptr = buf;
57         fr.src = "JPEG Read";
58         fr.datalen = len;
59         return ast_frisolate(&fr);
60 }
61
62 static int jpeg_identify(int fd)
63 {
64         char buf[10];
65         int res;
66         res = read(fd, buf, sizeof(buf));
67         if (res < sizeof(buf))
68                 return 0;
69         if (memcmp(buf + 6, "JFIF", 4))
70                 return 0;
71         return 1;
72 }
73
74 static int jpeg_write_image(int fd, struct ast_frame *fr)
75 {
76         int res=0;
77         if (fr->frametype != AST_FRAME_IMAGE) {
78                 ast_log(LOG_WARNING, "Not an image\n");
79                 return -1;
80         }
81         if (fr->subclass.format.id != AST_FORMAT_JPEG) {
82                 ast_log(LOG_WARNING, "Not a jpeg image\n");
83                 return -1;
84         }
85         if (fr->datalen) {
86                 res = write(fd, fr->data.ptr, fr->datalen);
87                 if (res != fr->datalen) {
88                         ast_log(LOG_WARNING, "Only wrote %d of %d bytes: %s\n", res, fr->datalen, strerror(errno));
89                         return -1;
90                 }
91         }
92         return res;
93 }
94
95 static struct ast_imager jpeg_format = {
96         .name = "jpg",
97         .desc = "JPEG (Joint Picture Experts Group)",
98         .exts = "jpg|jpeg",
99         .read_image = jpeg_read_image,
100         .identify = jpeg_identify,
101         .write_image = jpeg_write_image,
102 };
103
104 static int load_module(void)
105 {
106         ast_format_set(&jpeg_format.format, AST_FORMAT_JPEG, 0);
107         if (ast_image_register(&jpeg_format))
108                 return AST_MODULE_LOAD_FAILURE;
109         return AST_MODULE_LOAD_SUCCESS;
110 }
111
112 static int unload_module(void)
113 {
114         ast_image_unregister(&jpeg_format);
115
116         return 0;
117 }
118
119 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "jpeg (joint picture experts group) image format",
120         .load = load_module,
121         .unload = unload_module,
122         .load_pri = AST_MODPRI_APP_DEPEND
123 );