//based on libRTMP function
void rtmp_supplement::dumpAMF(AMFObject *obj)
{
int i;
const char opt[] = "NBSO Z";
for (i=0; i < obj->o_num; i++) {
AMFObjectProperty *p = &obj->o_props[i];
printf(" -C ");
if (p->p_name.av_val)
printf("N");
printf("%c:", opt[p->p_type]);
if (p->p_name.av_val)
printf("%.*s:", p->p_name.av_len, p->p_name.av_val);
switch(p->p_type) {
case AMF_BOOLEAN:
printf("%d", p->p_vu.p_number != 0);
break;
case AMF_STRING:
printf("%.*s", p->p_vu.p_aval.av_len, p->p_vu.p_aval.av_val);
break;
case AMF_NUMBER:
printf("%f", p->p_vu.p_number);
break;
case AMF_OBJECT:
printf("1");
dumpAMF(&p->p_vu.p_object);
printf(" -C O:0");
break;
case AMF_NULL:
break;
default:
printf("<type %d>", p->p_type);
}
}
}
static char *
dumpAMF(AMFObject *obj, char *ptr, AVal *argv, int *argc)
{
int i, len, ac = *argc;
const char opt[] = "NBSO Z";
for (i=0, len=0; i < obj->o_num; i++)
{
AMFObjectProperty *p = &obj->o_props[i];
argv[ac].av_val = ptr+1;
argv[ac++].av_len = 2;
ptr += sprintf(ptr, " -C ");
argv[ac].av_val = ptr;
if (p->p_name.av_val)
*ptr++ = 'N';
*ptr++ = opt[p->p_type];
*ptr++ = ':';
if (p->p_name.av_val)
ptr += sprintf(ptr, "%.*s:", p->p_name.av_len, p->p_name.av_val);
switch(p->p_type)
{
case AMF_BOOLEAN:
*ptr++ = p->p_vu.p_number != 0 ? '1' : '0';
argv[ac].av_len = ptr - argv[ac].av_val;
break;
case AMF_STRING:
memcpy(ptr, p->p_vu.p_aval.av_val, p->p_vu.p_aval.av_len);
ptr += p->p_vu.p_aval.av_len;
argv[ac].av_len = ptr - argv[ac].av_val;
break;
case AMF_NUMBER:
ptr += sprintf(ptr, "%f", p->p_vu.p_number);
argv[ac].av_len = ptr - argv[ac].av_val;
break;
case AMF_OBJECT:
*ptr++ = '1';
argv[ac].av_len = ptr - argv[ac].av_val;
ac++;
*argc = ac;
ptr = dumpAMF(&p->p_vu.p_object, ptr, argv, argc);
ac = *argc;
argv[ac].av_val = ptr+1;
argv[ac++].av_len = 2;
argv[ac].av_val = ptr+4;
argv[ac].av_len = 3;
ptr += sprintf(ptr, " -C O:0");
break;
case AMF_NULL:
default:
argv[ac].av_len = ptr - argv[ac].av_val;
break;
}
ac++;
}
*argc = ac;
return ptr;
}
// Returns 0 for OK/Failed/error, 1 for 'Stop or Complete'
int
ServeInvoke(STREAMING_SERVER *server, RTMP * r, RTMPPacket *packet, unsigned int offset)
{
const char *body;
unsigned int nBodySize;
int ret = 0, nRes;
body = packet->m_body + offset;
nBodySize = packet->m_nBodySize - offset;
if (body[0] != 0x02) // make sure it is a string method name we start with
{
RTMP_Log(RTMP_LOGWARNING, "%s, Sanity failed. no string method in invoke packet",
__FUNCTION__);
return 0;
}
AMFObject obj;
nRes = AMF_Decode(&obj, body, nBodySize, FALSE);
if (nRes < 0)
{
RTMP_Log(RTMP_LOGERROR, "%s, error decoding invoke packet", __FUNCTION__);
return 0;
}
AMF_Dump(&obj);
AVal method;
AMFProp_GetString(AMF_GetProp(&obj, NULL, 0), &method);
double txn = AMFProp_GetNumber(AMF_GetProp(&obj, NULL, 1));
RTMP_Log(RTMP_LOGDEBUG, "%s, client invoking <%s>", __FUNCTION__, method.av_val);
if (AVMATCH(&method, &av_connect))
{
AMFObject cobj;
AVal pname, pval;
int i;
server->connect = packet->m_body;
packet->m_body = NULL;
AMFProp_GetObject(AMF_GetProp(&obj, NULL, 2), &cobj);
for (i=0; i<cobj.o_num; i++)
{
pname = cobj.o_props[i].p_name;
pval.av_val = NULL;
pval.av_len = 0;
if (cobj.o_props[i].p_type == AMF_STRING)
pval = cobj.o_props[i].p_vu.p_aval;
if (AVMATCH(&pname, &av_app))
{
r->Link.app = pval;
pval.av_val = NULL;
if (!r->Link.app.av_val)
r->Link.app.av_val = "";
server->arglen += 6 + pval.av_len;
server->argc += 2;
}
else if (AVMATCH(&pname, &av_flashVer))
{
r->Link.flashVer = pval;
pval.av_val = NULL;
server->arglen += 6 + pval.av_len;
server->argc += 2;
}
else if (AVMATCH(&pname, &av_swfUrl))
{
r->Link.swfUrl = pval;
pval.av_val = NULL;
server->arglen += 6 + pval.av_len;
server->argc += 2;
}
else if (AVMATCH(&pname, &av_tcUrl))
{
r->Link.tcUrl = pval;
pval.av_val = NULL;
server->arglen += 6 + pval.av_len;
server->argc += 2;
}
else if (AVMATCH(&pname, &av_pageUrl))
{
r->Link.pageUrl = pval;
pval.av_val = NULL;
server->arglen += 6 + pval.av_len;
server->argc += 2;
}
else if (AVMATCH(&pname, &av_audioCodecs))
{
r->m_fAudioCodecs = cobj.o_props[i].p_vu.p_number;
}
else if (AVMATCH(&pname, &av_videoCodecs))
{
r->m_fVideoCodecs = cobj.o_props[i].p_vu.p_number;
}
else if (AVMATCH(&pname, &av_objectEncoding))
{
r->m_fEncoding = cobj.o_props[i].p_vu.p_number;
}
}
/* Still have more parameters? Copy them */
if (obj.o_num > 3)
{
int i = obj.o_num - 3;
r->Link.extras.o_num = i;
r->Link.extras.o_props = malloc(i*sizeof(AMFObjectProperty));
memcpy(r->Link.extras.o_props, obj.o_props+3, i*sizeof(AMFObjectProperty));
obj.o_num = 3;
server->arglen += countAMF(&r->Link.extras, &server->argc);
}
SendConnectResult(r, txn);
}
else if (AVMATCH(&method, &av_createStream))
{
SendResultNumber(r, txn, ++server->streamID);
}
else if (AVMATCH(&method, &av_getStreamLength))
{
SendResultNumber(r, txn, 10.0);
}
else if (AVMATCH(&method, &av_play))
{
char *file, *p, *q, *cmd, *ptr;
AVal *argv, av;
int len, argc;
uint32_t now;
RTMPPacket pc = {0};
AMFProp_GetString(AMF_GetProp(&obj, NULL, 3), &r->Link.playpath);
/*
r->Link.seekTime = AMFProp_GetNumber(AMF_GetProp(&obj, NULL, 4));
if (obj.o_num > 5)
r->Link.length = AMFProp_GetNumber(AMF_GetProp(&obj, NULL, 5));
*/
if (r->Link.tcUrl.av_len)
{
len = server->arglen + r->Link.playpath.av_len + 4 +
sizeof("rtmpdump") + r->Link.playpath.av_len + 12;
server->argc += 5;
cmd = malloc(len + server->argc * sizeof(AVal));
ptr = cmd;
argv = (AVal *)(cmd + len);
argv[0].av_val = cmd;
argv[0].av_len = sizeof("rtmpdump")-1;
ptr += sprintf(ptr, "rtmpdump");
argc = 1;
argv[argc].av_val = ptr + 1;
argv[argc++].av_len = 2;
argv[argc].av_val = ptr + 5;
ptr += sprintf(ptr," -r \"%s\"", r->Link.tcUrl.av_val);
argv[argc++].av_len = r->Link.tcUrl.av_len;
if (r->Link.app.av_val)
{
argv[argc].av_val = ptr + 1;
argv[argc++].av_len = 2;
argv[argc].av_val = ptr + 5;
ptr += sprintf(ptr, " -a \"%s\"", r->Link.app.av_val);
argv[argc++].av_len = r->Link.app.av_len;
}
if (r->Link.flashVer.av_val)
{
argv[argc].av_val = ptr + 1;
argv[argc++].av_len = 2;
argv[argc].av_val = ptr + 5;
ptr += sprintf(ptr, " -f \"%s\"", r->Link.flashVer.av_val);
argv[argc++].av_len = r->Link.flashVer.av_len;
}
if (r->Link.swfUrl.av_val)
{
argv[argc].av_val = ptr + 1;
argv[argc++].av_len = 2;
argv[argc].av_val = ptr + 5;
ptr += sprintf(ptr, " -W \"%s\"", r->Link.swfUrl.av_val);
argv[argc++].av_len = r->Link.swfUrl.av_len;
}
if (r->Link.pageUrl.av_val)
{
argv[argc].av_val = ptr + 1;
argv[argc++].av_len = 2;
argv[argc].av_val = ptr + 5;
ptr += sprintf(ptr, " -p \"%s\"", r->Link.pageUrl.av_val);
argv[argc++].av_len = r->Link.pageUrl.av_len;
}
if (r->Link.extras.o_num) {
ptr = dumpAMF(&r->Link.extras, ptr, argv, &argc);
AMF_Reset(&r->Link.extras);
}
argv[argc].av_val = ptr + 1;
argv[argc++].av_len = 2;
argv[argc].av_val = ptr + 5;
ptr += sprintf(ptr, " -y \"%.*s\"",
r->Link.playpath.av_len, r->Link.playpath.av_val);
argv[argc++].av_len = r->Link.playpath.av_len;
av = r->Link.playpath;
/* strip trailing URL parameters */
q = memchr(av.av_val, '?', av.av_len);
if (q)
{
if (q == av.av_val)
{
av.av_val++;
av.av_len--;
}
else
{
av.av_len = q - av.av_val;
}
}
/* strip leading slash components */
for (p=av.av_val+av.av_len-1; p>=av.av_val; p--)
if (*p == '/')
{
p++;
av.av_len -= p - av.av_val;
av.av_val = p;
break;
}
/* skip leading dot */
if (av.av_val[0] == '.')
{
av.av_val++;
av.av_len--;
}
file = malloc(av.av_len+5);
memcpy(file, av.av_val, av.av_len);
file[av.av_len] = '\0';
for (p=file; *p; p++)
if (*p == ':')
*p = '_';
/* Add extension if none present */
if (file[av.av_len - 4] != '.')
{
av.av_len += 4;
}
/* Always use flv extension, regardless of original */
if (strcmp(file+av.av_len-4, ".flv"))
{
strcpy(file+av.av_len-4, ".flv");
}
argv[argc].av_val = ptr + 1;
argv[argc++].av_len = 2;
argv[argc].av_val = file;
argv[argc].av_len = av.av_len;
ptr += sprintf(ptr, " -o %s", file);
now = RTMP_GetTime();
if (now - server->filetime < DUPTIME && AVMATCH(&argv[argc], &server->filename))
{
printf("Duplicate request, skipping.\n");
free(file);
}
else
{
printf("\n%s\n\n", cmd);
fflush(stdout);
server->filetime = now;
free(server->filename.av_val);
server->filename = argv[argc++];
spawn_dumper(argc, argv, cmd);
}
free(cmd);
}
pc.m_body = server->connect;
server->connect = NULL;
RTMPPacket_Free(&pc);
ret = 1;
RTMP_SendCtrl(r, 0, 1, 0);
SendPlayStart(r);
RTMP_SendCtrl(r, 1, 1, 0);
SendPlayStop(r);
}
AMF_Reset(&obj);
return ret;
}
