1 /* 2 * Copyright (C) 2005 3 * Petr Holub, Hidetoshi Shimokawa. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. All advertising materials mentioning features or use of this software 14 * must display the following acknowledgement: 15 * 16 * This product includes software developed by Hidetoshi Shimokawa. 17 * 18 * 4. Neither the name of the author nor the names of its contributors 19 * may be used to endorse or promote products derived from this software 20 * without specific prior written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 25 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 32 * SUCH DAMAGE. 33 * 34 * $FreeBSD: src/usr.sbin/fwcontrol/fwmpegts.c,v 1.1 2006/10/26 22:33:38 imp Exp $ 35 */ 36 #include <sys/param.h> 37 #ifndef __HAIKU__ 38 #include <sys/ioctl.h> 39 #endif 40 #include <sys/time.h> 41 #include <sys/types.h> 42 #include <sys/uio.h> 43 44 #if __FreeBSD_version >= 500000 || defined(__HAIKU__) 45 #include <arpa/inet.h> 46 #endif 47 48 #include <err.h> 49 #include <errno.h> 50 #include <unistd.h> 51 #include <fcntl.h> 52 #include <stdio.h> 53 #include <stdlib.h> 54 #include <string.h> 55 #ifdef __HAIKU__ 56 #include <stdint.h> 57 #include <endian.h> 58 #include "firewire.h" 59 #include "iec68113.h" 60 #else 61 #include <sysexits.h> 62 63 #include <dev/firewire/firewire.h> 64 #include <dev/firewire/iec68113.h> 65 #endif 66 67 #include "fwmethods.h" 68 69 #define DEBUG 0 70 71 /***************************************************************************** 72 73 MPEG-2 Transport Stream (MPEG TS) packet format according to IEC 61883: 74 75 31 15 0 76 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 77 | len |tag| channel | tcode | sy | 78 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ 1394 79 | header_CRC | 80 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 81 |0|0| sid | dbs |fn | qpc |S|RSV| dbc | 82 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ CIP 83 |1|0| fmt | fdf | fdf/syt | 84 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 85 | reserved | cycle_count | cycle_offset | 86 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ 87 | | N x 88 . . MPEG 89 . MPEG TS payload 188 bytes . 90 . . 91 | | 92 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 93 | data_CRC | 94 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ 95 96 N.b. that CRCs are removed by firewire layer! 97 98 The following fiels are fixed for IEEE-1394: 99 tag = 01b 100 tcode = 1010b 101 The length is payload length, i.e. includes CIP header and data size. 102 103 The following fields are constant for MPEG TS: 104 sph = 1 (denoted as S in CIP header above) 105 dbs = 6 106 fmt = (1<<5) 107 fdf = reserved 108 In the supported streams we also require 109 qpc = 0 110 fn = 3 111 and thus the payload is divided in 8 blocks as follows: 112 113 +-----+-----+-----+-----+-----+-----+-----+-----+ 114 | db0 | db1 | db2 | db3 | db4 | db5 | db6 | db7 | 115 +-----+-----+-----+-----+-----+-----+-----+-----+ 116 117 We have several cases of payload distribution based on stream 118 bandwidth (R): 119 1) R < 1.5 Mbps: any of db0..db7 may be payload, 120 2) 1.5 < R < 3 Mbps: db0/db1 or db2/db3 or db4/db5 or db6/db7 is payload, 121 3) 3 < R < 6 Mbps: db0/db1/db2/db3 or db4/db5/db6/db7 is payload, 122 4) R > 6 Mbps: all db0..db7 contain the payload. 123 Curently, only case (4) is supported in fwmpegts.c 124 125 Each packet may contain N MPEG TS data blocks with timestamp header, 126 which are (4+188)B long. Experimentally, the N ranges from 0 through 3. 127 128 *****************************************************************************/ 129 130 131 typedef uint8_t mpeg_ts_pld[188]; 132 133 struct mpeg_pldt { 134 #if BYTE_ORDER == BIG_ENDIAN 135 uint32_t :7, 136 c_count:13, 137 c_offset:12; 138 #else /* BYTE_ORDER != BIG_ENDIAN */ 139 uint32_t c_offset:12, 140 c_count:13, 141 :7; 142 #endif /* BYTE_ORDER == BIG_ENDIAN */ 143 mpeg_ts_pld payload; 144 }; 145 146 147 #define NCHUNK 8 148 #define PSIZE 596 149 #define NPACKET_R 4096 150 #define RBUFSIZE (PSIZE * NPACKET_R) 151 152 void 153 mpegtsrecv(int d, const char *filename, char ich, int count) 154 { 155 struct ciphdr *ciph; 156 struct fw_isochreq isoreq; 157 struct fw_isobufreq bufreq; 158 struct fw_pkt *pkt; 159 struct mpeg_pldt *pld; 160 uint32_t *ptr; 161 int fd, k, len, m, pkt_size, startwr, tlen; 162 char *buf; 163 164 startwr = 0; 165 166 if (strcmp(filename, "-") == 0) 167 fd = STDOUT_FILENO; 168 else { 169 fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0660); 170 if (fd == -1) 171 err(EX_NOINPUT, filename); 172 } 173 buf = malloc(RBUFSIZE); 174 175 bufreq.rx.nchunk = NCHUNK; 176 bufreq.rx.npacket = NPACKET_R; 177 bufreq.rx.psize = PSIZE; 178 bufreq.tx.nchunk = 0; 179 bufreq.tx.npacket = 0; 180 bufreq.tx.psize = 0; 181 if (ioctl(d, FW_SSTBUF, &bufreq) < 0) 182 err(1, "ioctl"); 183 184 isoreq.ch = ich & 0x3f; 185 isoreq.tag = (ich >> 6) & 3; 186 187 if (ioctl(d, FW_SRSTREAM, &isoreq) < 0) 188 err(1, "ioctl"); 189 190 k = m = 0; 191 while (count <= 0 || k <= count) { 192 len = tlen = read(d, buf, RBUFSIZE); 193 #if DEBUG 194 fprintf(stderr, "Read %d bytes.\n", len); 195 #endif /* DEBUG */ 196 if (len < 0) { 197 if (errno == EAGAIN) { 198 fprintf(stderr, "(EAGAIN) - push 'Play'?\n"); 199 if (len <= 0) 200 continue; 201 } else 202 err(1, "read failed"); 203 } 204 ptr = (uint32_t *) buf; 205 206 do { 207 pkt = (struct fw_pkt *) ptr; 208 #if DEBUG 209 fprintf(stderr, "\nReading new packet.\n"); 210 fprintf(stderr, "%08x %08x %08x %08x\n", 211 htonl(ptr[0]), htonl(ptr[1]), 212 htonl(ptr[2]), htonl(ptr[3])); 213 #endif /* DEBUG */ 214 /* there is no CRC in the 1394 header */ 215 ciph = (struct ciphdr *)(ptr + 1); /* skip iso header */ 216 if (ciph->fmt != CIP_FMT_MPEG) 217 errx(1, "unknown format 0x%x", ciph->fmt); 218 if (ciph->fn != 3) { 219 errx(1, 220 "unsupported MPEG TS stream, fn=%d (only fn=3 is supported)", 221 ciph->fn); 222 } 223 ptr = (uint32_t *) (ciph + 1); /* skip cip header */ 224 225 if (pkt->mode.stream.len <= sizeof(struct ciphdr)) { 226 /* no payload */ 227 /* tlen needs to be decremented before end of the loop */ 228 goto next; 229 } 230 #if DEBUG 231 else { 232 fprintf(stderr, 233 "Packet net payload length (IEEE1394 header): %d\n", 234 pkt->mode.stream.len - sizeof(struct ciphdr)); 235 fprintf(stderr, "Data block size (CIP header): %d [q], %d [B]\n", 236 ciph->len, ciph->len * 4); 237 fprintf(stderr, 238 "Data fraction number (CIP header): %d => DBC increments with %d\n", 239 ciph->fn, (1<<ciph->fn) ); 240 fprintf(stderr, "QCP (CIP header): %d\n", ciph->qpc ); 241 fprintf(stderr, "DBC counter (CIP header): %d\n", ciph->dbc ); 242 fprintf(stderr, "MPEG payload type size: %d\n", 243 sizeof(struct mpeg_pldt)); 244 } 245 #endif /* DEBUG */ 246 247 /* This is a condition that needs to be satisfied to start 248 writing the data */ 249 if (ciph->dbc % (1<<ciph->fn) == 0) 250 startwr = 1; 251 /* Read out all the MPEG TS data blocks from current packet */ 252 for (pld = (struct mpeg_pldt *)ptr; 253 (intptr_t)pld < (intptr_t)((char *)ptr + 254 pkt->mode.stream.len - sizeof(struct ciphdr)); 255 pld++) { 256 if (startwr == 1) 257 write(fd, pld->payload, 258 sizeof(pld->payload)); 259 } 260 261 next: 262 /* CRCs are removed from both header and trailer 263 so that only 4 bytes of 1394 header remains */ 264 pkt_size = pkt->mode.stream.len + 4; 265 ptr = (uint32_t *)((intptr_t)pkt + pkt_size); 266 tlen -= pkt_size; 267 } while (tlen > 0); 268 #if DEBUG 269 fprintf(stderr, "\nReading a data from firewire.\n"); 270 #endif /* DEBUG */ 271 272 } 273 if (fd != STDOUT_FILENO) 274 close(fd); 275 fprintf(stderr, "\n"); 276 } 277