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) || len < 0) {
  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 static int load_module(void *mod)
 119 {
 120         return ast_image_register(&jpeg_format);
 121 }
 122
 123 static int unload_module(void *mod)
 124 {
 125         ast_image_unregister(&jpeg_format);
 126         return 0;
 127 }
 128
 129 static const char *description(void)
 130 {
 131         return desc;
 132 }
 133
 134 static const char *key(void)
 135 {
 136         return ASTERISK_GPL_KEY;
 137 }
 138
 139 STD_MOD(MOD_1 | NO_USECOUNT, NULL, NULL, NULL);