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          * Generate a random UID for this Attachment.
  38          * NOTE: This probably should be a CRC32 of file data.
  39          *        In place of being completely random.
  40          */
  41         file_uid = random();
  42
  43         if (w->attachments == NULL) {
  44                 /* Attachments */
  45                 if ((w->attachments = mk_createContext(w, w->root, MATROSKA_ID_ATTACHMENTS)) == NULL)
  46                         return -1;
  47         }
  48         /* AttachedFile */
  49         if ((attach = mk_createContext(w, w->attachments, MATROSKA_ID_ATTACHEDFILE)) == NULL)
  50                 return -1;
  51
  52         /* FileName */
  53         CHECK(mk_writeStr(attach, MATROSKA_ID_FILENAME, name));
  54         if ((description != NULL) && (strlen(description) > 0)) {
  55                 /* FileDescription */
  56                 CHECK(mk_writeStr(attach, MATROSKA_ID_FILEDESCRIPTION, description));
  57         }
  58         /* FileMimeType */
  59         CHECK(mk_writeStr(attach, MATROSKA_ID_FILEMIMETYPE, mime));
  60         /* FileUID */
  61         CHECK(mk_writeUInt(attach, MATROSKA_ID_FILEUID, file_uid));
  62         /* FileData */
  63         CHECK(mk_writeBin(attach, MATROSKA_ID_FILEDATA, data, size));
  64         CHECK(mk_closeContext(attach, 0));
  65
  66         return 0;
  67 }
  68
  69 int mk_writeAttachments(mk_Writer * w)
  70 {
  71         if (w->attachments == NULL)
  72                 return -1;
  73         CHECK(mk_closeContext(w->attachments, 0));
  74
  75         return 0;
  76 }