mirror of
https://github.com/Netflix/dial-reference.git
synced 2026-06-08 10:59:59 +00:00
dial spec 1.7
This commit is contained in:
jcli
298
client/main.cpp
Normal file
298
client/main.cpp
Normal file
@@ -0,0 +1,298 @@
|
||||
/*
|
||||
* Copyright (c) 2014 Netflix, Inc.
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* Redistributions of source code must retain the above copyright notice, this
|
||||
* list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY NETFLIX, INC. AND CONTRIBUTORS "AS IS" AND ANY
|
||||
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
* DISCLAIMED. IN NO EVENT SHALL NETFLIX OR CONTRIBUTORS BE LIABLE FOR ANY
|
||||
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
||||
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include <string>
|
||||
#include "DialDiscovery.h"
|
||||
#include "DialConformance.h"
|
||||
#include <assert.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
|
||||
using namespace std;
|
||||
|
||||
static DialDiscovery* gpDiscovery;
|
||||
static bool gUseMenu = true;
|
||||
|
||||
// TODO: Make it possible to pass applications from the command line
|
||||
static vector<string> gAppList;
|
||||
static string gOutputFile;
|
||||
static string gInputFile;
|
||||
|
||||
// IP address of the DIAL server
|
||||
static string gIpAddress;
|
||||
|
||||
static void printServerList( vector<DialServer*> list )
|
||||
{
|
||||
int i;
|
||||
vector<DialServer*>::iterator it;
|
||||
for( i = 0, it = list.begin(); it < list.end(); it++, i++ )
|
||||
{
|
||||
string uuid, name;
|
||||
(*it)->getFriendlyName( name );
|
||||
(*it)->getUuid( uuid );
|
||||
printf("%Zu: Server IP[%s] UUID[%s] FriendlyName[%s] \n",
|
||||
i+1, (*it)->getIpAddress().c_str(),
|
||||
uuid.c_str(), name.c_str() );
|
||||
}
|
||||
}
|
||||
|
||||
static DialServer* getServerFromUser( vector<DialServer*> list )
|
||||
{
|
||||
DialServer* pServer;
|
||||
// show a list to the user
|
||||
if( list.size() > 1 )
|
||||
{
|
||||
char buf[80] = {0,};
|
||||
vector<DialServer*>::iterator it;
|
||||
|
||||
printf("Found Multiple servers\n");
|
||||
printServerList(list);
|
||||
printf("Enter server: ");
|
||||
scanf("%s", buf);
|
||||
unsigned int server = atoi(buf);
|
||||
assert( server > 0 && server <= list.size() );
|
||||
pServer = list[server-1];
|
||||
}
|
||||
else
|
||||
{
|
||||
pServer = list.front();
|
||||
}
|
||||
return pServer;
|
||||
}
|
||||
|
||||
static void runConformance()
|
||||
{
|
||||
vector<DialServer*> list;
|
||||
gpDiscovery->getServerList(list);
|
||||
|
||||
if( list.size() )
|
||||
{
|
||||
DialServer *pServer = NULL;
|
||||
if( !gIpAddress.empty() )
|
||||
{
|
||||
pServer = NULL;
|
||||
vector<DialServer*>::iterator it;
|
||||
for( it = list.begin(); it < list.end(); it ++ )
|
||||
{
|
||||
if( gIpAddress.compare((*it)->getIpAddress()) == 0 )
|
||||
{
|
||||
ATRACE("Found server %s in the list of servers\n",
|
||||
gIpAddress.c_str() );
|
||||
pServer = (*it);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
pServer = getServerFromUser( list );
|
||||
}
|
||||
|
||||
if( pServer )
|
||||
{
|
||||
string name;
|
||||
bool serverExists = pServer->getFriendlyName(name);
|
||||
assert( serverExists );
|
||||
string uuid;
|
||||
pServer->getUuid( uuid );
|
||||
printf("\nRunning conformance against: IP[%s] UUID[%s] FriendlyName[%s] \n",
|
||||
pServer->getIpAddress().c_str(),
|
||||
uuid.c_str(), name.c_str() );
|
||||
DialConformance::instance()->run(
|
||||
pServer,
|
||||
gAppList,
|
||||
gInputFile,
|
||||
gOutputFile );
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("DIAL server not found\n");
|
||||
printf("%Zu available server(s): \n", list.size());
|
||||
printServerList(list);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("No servers available\n");
|
||||
}
|
||||
}
|
||||
|
||||
int handleUser(DialDiscovery *pDial) {
|
||||
int processInput = 1;
|
||||
char buf[80];
|
||||
vector<DialServer*> list;
|
||||
|
||||
pDial->getServerList(list);
|
||||
if( list.size() == 0 )
|
||||
{
|
||||
printf("No servers available\n");
|
||||
return 1;
|
||||
}
|
||||
DialServer* pServer = getServerFromUser( list );
|
||||
|
||||
while(processInput)
|
||||
{
|
||||
string responseHeaders, responseBody, payload;
|
||||
string netflix = "Netflix";
|
||||
string youtube = "YouTube";
|
||||
|
||||
memset(buf, 0, 80);
|
||||
printf("0. List DIAL servers\n");
|
||||
printf("1. Launch Netflix\n");
|
||||
printf("2. Kill Netflix\n");
|
||||
printf("3. Netflix status\n");
|
||||
printf("4. Launch YouTube\n");
|
||||
printf("5. Kill YouTube\n");
|
||||
printf("6. YouTube status\n");
|
||||
printf("7. Run conformance tests\n");
|
||||
printf("8. QUIT\n");
|
||||
printf("Command (0:1:2:3:4:5:6:7:8): ");
|
||||
scanf("%s", buf);
|
||||
switch( atoi(buf) )
|
||||
{
|
||||
case 0:
|
||||
{
|
||||
printf("\n\n******** %Zu servers found ********\n\n", list.size());
|
||||
for( unsigned int i = 0; i < list.size(); i++ )
|
||||
{
|
||||
string name;
|
||||
list[i]->getFriendlyName(name);
|
||||
printf("Server %Zu: %s\n", i+1, name.c_str());
|
||||
}
|
||||
printf("\n*********************************\n\n");
|
||||
}break;
|
||||
case 1:
|
||||
printf("Launch Netflix\n");
|
||||
pServer->launchApplication( netflix, payload, responseHeaders, responseBody );
|
||||
break;
|
||||
case 2:
|
||||
printf("Kill Netflix\n");
|
||||
pServer->stopApplication( netflix, responseHeaders );
|
||||
break;
|
||||
case 3:
|
||||
printf("Netflix Status: \n");
|
||||
pServer->getStatus( netflix, responseHeaders, responseBody );
|
||||
printf("RESPONSE: \n%s\n", responseBody.c_str());
|
||||
break;
|
||||
case 4:
|
||||
printf("Launch YouTube\n");
|
||||
pServer->launchApplication( youtube, payload, responseHeaders, responseBody );
|
||||
break;
|
||||
case 5:
|
||||
printf("Kill YouTube\n");
|
||||
pServer->stopApplication( youtube, responseHeaders );
|
||||
break;
|
||||
case 6:
|
||||
printf("YouTube Status: \n");
|
||||
pServer->getStatus( youtube, responseHeaders, responseBody );
|
||||
break;
|
||||
case 7:
|
||||
runConformance();
|
||||
break;
|
||||
case 8:
|
||||
processInput = 0;
|
||||
break;
|
||||
default:
|
||||
printf("Invalid, try again\n");
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
static const char usage[] = "\n"
|
||||
" If no option is specified, the program will run a conformance test.\n"
|
||||
"\n"
|
||||
"usage: dialclient <option>\n"
|
||||
" Option Parameter Description\n"
|
||||
" -h none Usage menu\n"
|
||||
" -s filename (optional) Run conformance test. Use filename as\n"
|
||||
" the input, if provided\n"
|
||||
" -o filename Reporter output file (./report.html)\n"
|
||||
" -a ip_address IP address of DIAL server (used for conformance\n"
|
||||
" testing)\n"
|
||||
"\n";
|
||||
|
||||
inline void notSupported( string s )
|
||||
{
|
||||
printf( "%s not supported", s.c_str() );
|
||||
printf( "%s\n", usage );
|
||||
exit(0);
|
||||
}
|
||||
|
||||
int parseArgs( int argc, char* argv[] )
|
||||
{
|
||||
for( int i = 1; i < argc; i++ )
|
||||
{
|
||||
string input(argv[i]);
|
||||
if( input[0] != '-' ) notSupported(input);
|
||||
switch(input[1])
|
||||
{
|
||||
case 's':
|
||||
gUseMenu = false;
|
||||
if( argv[i+1] != NULL && argv[i+1][0] != '-' )
|
||||
{
|
||||
//filename provided
|
||||
gInputFile = argv[++i];
|
||||
}
|
||||
break;
|
||||
case 'o':
|
||||
gOutputFile = argv[++i];
|
||||
break;
|
||||
case 'a':
|
||||
gIpAddress = argv[++i];
|
||||
break;
|
||||
case 'h':
|
||||
printf("%s", usage);
|
||||
exit(0);
|
||||
break;
|
||||
default:
|
||||
notSupported(input);
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int main(int argc, char* argv[]) {
|
||||
parseArgs(argc, argv);
|
||||
|
||||
gpDiscovery = DialDiscovery::create();
|
||||
gpDiscovery->init();
|
||||
DialConformance::create();
|
||||
|
||||
// Sleep for 2 seconds to allow DIAL servers to response to MSEARCH.
|
||||
sleep(2);
|
||||
|
||||
if ( gUseMenu )
|
||||
{
|
||||
return handleUser(gpDiscovery);
|
||||
}
|
||||
|
||||
// not using the menu, just run the conformance test.
|
||||
runConformance();
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user