static gboolean initialize_device(struct upnp_device_descriptor *device_def,
struct upnp_device *result_device,
const char *ip_address,
unsigned short port)
{
int rc;
char *buf;
rc = UpnpInit(ip_address, port);
if (UPNP_E_SUCCESS != rc) {
Log_error("upnp", "UpnpInit(ip=%s, port=%d) Error: %s (%d)",
ip_address, port, UpnpGetErrorMessage(rc), rc);
return FALSE;
}
Log_info("upnp", "Registered IP=%s port=%d\n",
UpnpGetServerIpAddress(), UpnpGetServerPort());
rc = UpnpEnableWebserver(TRUE);
if (UPNP_E_SUCCESS != rc) {
Log_error("upnp", "UpnpEnableWebServer() Error: %s (%d)",
UpnpGetErrorMessage(rc), rc);
return FALSE;
}
if (!webserver_register_callbacks())
return FALSE;
rc = UpnpAddVirtualDir("/upnp");
if (UPNP_E_SUCCESS != rc) {
Log_error("upnp", "UpnpAddVirtualDir() Error: %s (%d)",
UpnpGetErrorMessage(rc), rc);
return FALSE;
}
buf = upnp_create_device_desc(device_def);
rc = UpnpRegisterRootDevice2(UPNPREG_BUF_DESC,
buf, strlen(buf), 1,
&event_handler, result_device,
&(result_device->device_handle));
free(buf);
if (UPNP_E_SUCCESS != rc) {
Log_error("upnp", "UpnpRegisterRootDevice2() Error: %s (%d)",
UpnpGetErrorMessage(rc), rc);
return FALSE;
}
rc = UpnpSendAdvertisement(result_device->device_handle, 100);
if (UPNP_E_SUCCESS != rc) {
Log_error("unpp", "Error sending advertisements: %s (%d)",
UpnpGetErrorMessage(rc), rc);
return FALSE;
}
return TRUE;
}
void WebServer::addVirtualDirectory(const std::string& virtualDirName, FileInfoCb fileinfoCb, RequestCb requestCb)
{
std::lock_guard<std::mutex> lock(g_mutex);
if (UPNP_E_SUCCESS != UpnpAddVirtualDir(virtualDirName.c_str()))
{
throw std::logic_error("Failed to add virtual directory to webserver");
}
g_virtualDirs.emplace("/" + virtualDirName, std::make_pair(fileinfoCb, requestCb));
}
void WebServer::addVirtualDirectory(const std::string& virtualDirName)
{
std::lock_guard<std::mutex> lock(g_mutex);
if (UPNP_E_SUCCESS != UpnpAddVirtualDir(virtualDirName.c_str()))
{
throw std::logic_error("Failed to add virtual directory to webserver");
}
g_servedFiles.emplace("/" + virtualDirName, std::vector<std::unique_ptr<HostedFile>>());
}
struct device_private *upnp_device_init(struct device *device_def,
const char *ip_address)
{
int rc;
#ifdef HAVE_LIBUPNP
short int port = 0;
#endif
struct service *srv;
struct icon *icon_entry;
char *buf;
struct device_private *priv = NULL;
int i;
assert(device_def != NULL);
if (device_def->init_function)
{
rc = device_def->init_function();
if (rc != 0) {
goto out;
}
}
priv = malloc(sizeof(*priv));
priv->upnp_device = device_def;
#ifdef HAVE_LIBUPNP
ithread_mutex_init(&(priv->device_mutex), NULL);
#endif
//upnp_device = device_def;
/* register icons in web server */
for (i=0; (icon_entry = device_def->icons[i]); i++) {
webserver_register_file(icon_entry->url, "image/png");
}
/* generate and register service schemas in web server */
for (i=0; (srv = device_def->services[i]); i++) {
buf = upnp_get_scpd(srv);
assert(buf != NULL);
printf("registering '%s'\n", srv->scpd_url);
webserver_register_buf(srv->scpd_url,buf,"text/xml");
}
#ifdef HAVE_LIBUPNP
rc = UpnpInit(ip_address, port);
if (UPNP_E_SUCCESS != rc) {
printf("UpnpInit() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpEnableWebserver(TRUE);
if (UPNP_E_SUCCESS != rc) {
printf("UpnpEnableWebServer() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpSetVirtualDirCallbacks(&virtual_dir_callbacks);
if (UPNP_E_SUCCESS != rc) {
printf("UpnpSetVirtualDirCallbacks() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpAddVirtualDir("/upnp");
if (UPNP_E_SUCCESS != rc) {
printf("UpnpAddVirtualDir() Error: %d\n", rc);
goto upnp_err_out;
}
buf = upnp_get_device_desc(device_def);
rc = UpnpRegisterRootDevice2(UPNPREG_BUF_DESC,
buf, strlen(buf), 1,
&event_handler, priv,
&(priv->device_handle));
if (UPNP_E_SUCCESS != rc) {
printf("UpnpRegisterRootDevice2() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpSendAdvertisement(priv->device_handle, 100);
if (UPNP_E_SUCCESS != rc) {
fprintf(stderr, "Error sending advertisements: %d\n", rc);
goto upnp_err_out;
}
#endif
goto out;
#ifdef HAVE_LIBUPNP
upnp_err_out:
UpnpFinish();
#endif
out:
return priv;
}
ERROR_TYPE IUpnp::AddVirtualDir(const char *dirName)
{
return (ERROR_TYPE)UpnpAddVirtualDir(dirName);
}
int upnp_device_init(struct device *device_def, char *ip_address)
{
int rc;
int result = -1;
short int port = 0;
struct service *srv;
struct icon *icon_entry;
char *buf;
int i;
if (device_def->init_function)
{
rc = device_def->init_function();
if (rc != 0) {
goto out;
}
}
upnp_device = device_def;
/* register icons in web server */
for (i=0; (icon_entry = upnp_device->icons[i]); i++) {
webserver_register_file(icon_entry->url, "image/png");
}
/* generate and register service schemas in web server */
for (i=0; (srv = upnp_device->services[i]); i++) {
buf = upnp_get_scpd(srv);
debug_printf(MSG_INFO, "registering '%s'\n", srv->scpd_url);
webserver_register_buf(srv->scpd_url,buf,"text/xml");
}
rc = UpnpInit(ip_address, port);
if (UPNP_E_SUCCESS != rc) {
debug_printf(MSG_ERROR, "UpnpInit() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpEnableWebserver(TRUE);
if (UPNP_E_SUCCESS != rc) {
debug_printf(MSG_ERROR, "UpnpEnableWebServer() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpSetVirtualDirCallbacks(&virtual_dir_callbacks);
if (UPNP_E_SUCCESS != rc) {
debug_printf(MSG_ERROR, "UpnpSetVirtualDirCallbacks() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpAddVirtualDir("/upnp");
if (UPNP_E_SUCCESS != rc) {
debug_printf(MSG_ERROR, "UpnpAddVirtualDir() Error: %d\n", rc);
goto upnp_err_out;
}
buf = upnp_get_device_desc(device_def);
rc = UpnpRegisterRootDevice2(UPNPREG_BUF_DESC,
buf, strlen(buf), 1,
&event_handler, &device_def,
&device_handle);
if (UPNP_E_SUCCESS != rc) {
debug_printf(MSG_ERROR, "UpnpRegisterRootDevice2() Error: %d\n", rc);
goto upnp_err_out;
}
rc = UpnpSendAdvertisement(device_handle, 100);
if (UPNP_E_SUCCESS != rc) {
debug_printf(MSG_ERROR, "Error sending advertisements: %d\n", rc);
goto upnp_err_out;
}
result = 0;
goto out;
upnp_err_out:
UpnpFinish();
out:
return result;
}
