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