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 #include <sys/ioctl.h> 38 #include <sys/time.h> 39 #include <sys/types.h> 40 #include <sys/uio.h> 41 42 #if __FreeBSD_version >= 500000 43 #include <arpa/inet.h> 44 #endif 45 46 #include <err.h> 47 #include <errno.h> 48 #include <unistd.h> 49 #include <fcntl.h> 50 #include <stdio.h> 51 #include <stdlib.h> 52 #include <string.h> 53 #include <sysexits.h> 54 55 #if defined(__FreeBSD__) 56 #include <dev/firewire/firewire.h> 57 #include <dev/firewire/iec68113.h> 58 #elif defined(__NetBSD__) 59 #include <dev/ieee1394/firewire.h> 60 #include <dev/ieee1394/iec68113.h> 61 #endif 62 63 #include "fwmethods.h" 64 65 #define DEBUG 0 66 67 /***************************************************************************** 68 69 MPEG-2 Transport Stream (MPEG TS) packet format according to IEC 61883: 70 71 31 15 0 72 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 73 | len |tag| channel | tcode | sy | 74 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ 1394 75 | header_CRC | 76 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 77 |0|0| sid | dbs |fn | qpc |S|RSV| dbc | 78 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ CIP 79 |1|0| fmt | fdf | fdf/syt | 80 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 81 | reserved | cycle_count | cycle_offset | 82 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ 83 | | N x 84 . . MPEG 85 . MPEG TS payload 188 bytes . 86 . . 87 | | 88 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ -------- 89 | data_CRC | 90 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ 91 92 N.b. that CRCs are removed by firewire layer! 93 94 The following fiels are fixed for IEEE-1394: 95 tag = 01b 96 tcode = 1010b 97 The length is payload length, i.e. includes CIP header and data size. 98 99 The following fields are constant for MPEG TS: 100 sph = 1 (denoted as S in CIP header above) 101 dbs = 6 102 fmt = (1<<5) 103 fdf = reserved 104 In the supported streams we also require 105 qpc = 0 106 fn = 3 107 and thus the payload is divided in 8 blocks as follows: 108 109 +-----+-----+-----+-----+-----+-----+-----+-----+ 110 | db0 | db1 | db2 | db3 | db4 | db5 | db6 | db7 | 111 +-----+-----+-----+-----+-----+-----+-----+-----+ 112 113 We have several cases of payload distribution based on stream 114 bandwidth (R): 115 1) R < 1.5 Mbps: any of db0..db7 may be payload, 116 2) 1.5 < R < 3 Mbps: db0/db1 or db2/db3 or db4/db5 or db6/db7 is payload, 117 3) 3 < R < 6 Mbps: db0/db1/db2/db3 or db4/db5/db6/db7 is payload, 118 4) R > 6 Mbps: all db0..db7 contain the payload. 119 Curently, only case (4) is supported in fwmpegts.c 120 121 Each packet may contain N MPEG TS data blocks with timestamp header, 122 which are (4+188)B long. Experimentally, the N ranges from 0 through 3. 123 124 *****************************************************************************/ 125 126 127 typedef uint8_t mpeg_ts_pld[188]; 128 129 struct mpeg_pldt { 130 #if BYTE_ORDER == BIG_ENDIAN 131 uint32_t :7, 132 c_count:13, 133 c_offset:12; 134 #else /* BYTE_ORDER != BIG_ENDIAN */ 135 uint32_t c_offset:12, 136 c_count:13, 137 :7; 138 #endif /* BYTE_ORDER == BIG_ENDIAN */ 139 mpeg_ts_pld payload; 140 }; 141 142 143 #define NCHUNK 8 144 #define PSIZE 596 145 #define NPACKET_R 4096 146 #define RBUFSIZE (PSIZE * NPACKET_R) 147 148 void 149 mpegtsrecv(int d, const char *filename, char ich, int count) 150 { 151 struct ciphdr *ciph; 152 struct fw_isochreq isoreq; 153 struct fw_isobufreq bufreq; 154 struct fw_pkt *pkt; 155 struct mpeg_pldt *pld; 156 uint32_t *ptr; 157 int fd, k, len, m, pkt_size, startwr, tlen; 158 char *buf; 159 160 startwr = 0; 161 162 if (strcmp(filename, "-") == 0) 163 fd = STDOUT_FILENO; 164 else { 165 fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0660); 166 if (fd == -1) 167 err(EX_NOINPUT, filename); 168 } 169 buf = malloc(RBUFSIZE); 170 171 bufreq.rx.nchunk = NCHUNK; 172 bufreq.rx.npacket = NPACKET_R; 173 bufreq.rx.psize = PSIZE; 174 bufreq.tx.nchunk = 0; 175 bufreq.tx.npacket = 0; 176 bufreq.tx.psize = 0; 177 if (ioctl(d, FW_SSTBUF, &bufreq) < 0) 178 err(1, "ioctl"); 179 180 isoreq.ch = ich & 0x3f; 181 isoreq.tag = (ich >> 6) & 3; 182 183 if (ioctl(d, FW_SRSTREAM, &isoreq) < 0) 184 err(1, "ioctl"); 185 186 k = m = 0; 187 while (count <= 0 || k <= count) { 188 len = tlen = read(d, buf, RBUFSIZE); 189 #if DEBUG 190 fprintf(stderr, "Read %d bytes.\n", len); 191 #endif /* DEBUG */ 192 if (len < 0) { 193 if (errno == EAGAIN) { 194 fprintf(stderr, "(EAGAIN) - push 'Play'?\n"); 195 if (len <= 0) 196 continue; 197 } else 198 err(1, "read failed"); 199 } 200 ptr = (uint32_t *) buf; 201 202 do { 203 pkt = (struct fw_pkt *) ptr; 204 #if DEBUG 205 fprintf(stderr, "\nReading new packet.\n"); 206 fprintf(stderr, "%08x %08x %08x %08x\n", 207 htonl(ptr[0]), htonl(ptr[1]), 208 htonl(ptr[2]), htonl(ptr[3])); 209 #endif /* DEBUG */ 210 /* there is no CRC in the 1394 header */ 211 ciph = (struct ciphdr *)(ptr + 1); /* skip iso header */ 212 if (ciph->fmt != CIP_FMT_MPEG) 213 errx(1, "unknown format 0x%x", ciph->fmt); 214 if (ciph->fn != 3) { 215 errx(1, 216 "unsupported MPEG TS stream, fn=%d (only fn=3 is supported)", 217 ciph->fn); 218 } 219 ptr = (uint32_t *) (ciph + 1); /* skip cip header */ 220 221 if (pkt->mode.stream.len <= sizeof(struct ciphdr)) { 222 /* no payload */ 223 /* tlen needs to be decremented before end of the loop */ 224 goto next; 225 } 226 #if DEBUG 227 else { 228 fprintf(stderr, 229 "Packet net payload length (IEEE1394 header): %d\n", 230 pkt->mode.stream.len - sizeof(struct ciphdr)); 231 fprintf(stderr, "Data block size (CIP header): %d [q], %d [B]\n", 232 ciph->len, ciph->len * 4); 233 fprintf(stderr, 234 "Data fraction number (CIP header): %d => DBC increments with %d\n", 235 ciph->fn, (1<<ciph->fn) ); 236 fprintf(stderr, "QCP (CIP header): %d\n", ciph->qpc ); 237 fprintf(stderr, "DBC counter (CIP header): %d\n", ciph->dbc ); 238 fprintf(stderr, "MPEG payload type size: %d\n", 239 sizeof(struct mpeg_pldt)); 240 } 241 #endif /* DEBUG */ 242 243 /* This is a condition that needs to be satisfied to start 244 writing the data */ 245 if (ciph->dbc % (1<<ciph->fn) == 0) 246 startwr = 1; 247 /* Read out all the MPEG TS data blocks from current packet */ 248 for (pld = (struct mpeg_pldt *)ptr; 249 (intptr_t)pld < (intptr_t)((char *)ptr + 250 pkt->mode.stream.len - sizeof(struct ciphdr)); 251 pld++) { 252 if (startwr == 1) 253 write(fd, pld->payload, 254 sizeof(pld->payload)); 255 } 256 257 next: 258 /* CRCs are removed from both header and trailer 259 so that only 4 bytes of 1394 header remains */ 260 pkt_size = pkt->mode.stream.len + 4; 261 ptr = (uint32_t *)((intptr_t)pkt + pkt_size); 262 tlen -= pkt_size; 263 } while (tlen > 0); 264 #if DEBUG 265 fprintf(stderr, "\nReading a data from firewire.\n"); 266 #endif /* DEBUG */ 267 268 } 269 if (fd != STDOUT_FILENO) 270 close(fd); 271 fprintf(stderr, "\n"); 272 } 273