Home | History | Annotate | Line # | Download | only in fwctl
fwmpegts.c revision 1.1
      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