src/attachments.c

   1 /*****************************************************************************
   2  * attachments.c:
   3  *****************************************************************************
   4  * Copyright (C) 2007 libmkv
   5  *
   6  * Authors: John A. Stebbins
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  21  *****************************************************************************/
  22 #include "config.h"
  23 #include "libmkv.h"
  24 #include "matroska.h"
  25
  26 int mk_createAttachment(
  27         mk_Writer * w,
  28         char *name,
  29         char *description,
  30         char *mime,
  31         const void *data,
  32         unsigned size)
  33 {
  34         mk_Context *attach;
  35         unsigned long file_uid;
  36
  37         if ((data == NULL) || (size == 0))
  38                 return -1;
  39
  40         /*
  41          * Generate a random UID for this Attachment.
  42          * NOTE: This probably should be a CRC32 of file data.
  43          *        In place of being completely random.
  44          */
  45         file_uid = random();
  46
  47         if (w->attachments == NULL) {
  48                 /* Attachments */
  49                 if ((w->attachments = mk_createContext(w, w->root, MATROSKA_ID_ATTACHMENTS)) == NULL)
  50                         return -1;
  51         }
  52         /* AttachedFile */
  53         if ((attach = mk_createContext(w, w->attachments, MATROSKA_ID_ATTACHEDFILE)) == NULL)
  54                 return -1;
  55
  56         /* FileName */
  57         CHECK(mk_writeStr(attach, MATROSKA_ID_FILENAME, name));
  58         if ((description != NULL) && (strlen(description) > 0)) {
  59                 /* FileDescription */
  60                 CHECK(mk_writeStr(attach, MATROSKA_ID_FILEDESCRIPTION, description));
  61         }
  62         /* FileMimeType */
  63         CHECK(mk_writeStr(attach, MATROSKA_ID_FILEMIMETYPE, mime));
  64         /* FileUID */
  65         CHECK(mk_writeUInt(attach, MATROSKA_ID_FILEUID, file_uid));
  66         /* FileData */
  67         CHECK(mk_writeBin(attach, MATROSKA_ID_FILEDATA, data, size));
  68         CHECK(mk_closeContext(attach, 0));
  69
  70         return 0;
  71 }
  72
  73 int mk_writeAttachments(mk_Writer * w)
  74 {
  75         if (w->attachments == NULL)
  76                 return -1;
  77         CHECK(mk_closeContext(w->attachments, 0));
  78
  79         return 0;
  80 }