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