ConnectionPointer ConnectionManager::get_or_connect(std::string const& host, unsigned short port)
{
// Look for an existing connection
ConnectionPointer conn = find(host, port);
if (conn.get() == 0)
{
// No connection found.
// Create a new one.
conn = connect(host, port);
}
return conn;
}
bool ConnectionManager2::applyConfig(QVariant const & config)
{
if (config.type() != QVariant::List)
return false;
QList<Connection *> newConns;
struct cleanup
{
QList<Connection *> & conns;
cleanup(QList<Connection *> & conns) : conns(conns) {}
~cleanup() { for (int i = 0; i < conns.size(); ++i) conns[i]->releaseAll(); }
} cleanupGuard(newConns);
QList<QVariant> const & connConfigs = config.toList();
for (int i = 0; i < connConfigs.size(); ++i)
{
if (connConfigs[i].type() != QVariant::Hash)
return false;
QHash<QString, QVariant> const & connConfig = connConfigs[i].toHash();
QVariant typev = connConfig.value("type");
if (typev.type() != QVariant::String)
return false;
QString const & type = typev.toString();
ConnectionPointer<Connection> conn;
if (type == "serial_port")
conn.reset(new SerialPort());
else if (type == "tcp_client")
conn.reset(new TcpSocket());
else if (type == "udp_socket")
conn.reset(new UdpSocket());
#ifdef HAVE_LIBYB
else if (type == "usb_yb_acm")
conn.reset(new UsbAcmConnection2(m_yb_runner));
#endif
if (!conn)
return false;
QVariant settings = connConfig.value("settings");
if (settings.type() != QVariant::Hash || !conn->applyConfig(settings.toHash()))
return false;
newConns.push_back(conn.data());
conn.take();
}
this->clearUserOwnedConns();
while (!newConns.empty())
{
this->addUserOwnedConn(newConns.back());
newConns.pop_back();
}
return true;
}
void ConnectionManager::handle_write(boost::system::error_code const& e, MessagePointer message, ConnectionPointer conn)
{
#if TIME_WRITES
static virvo::FrameCounter counter;
printf("CM: writes/sec: %.2f\n", counter.registerFrame());
#endif
// Call the connection's slot
conn->signal_(Connection::Write, message, e);
// Remove the message from the queue
write_queue_.pop_front();
if (!e)
{
// Message successfully sent.
// Send the next one -- if any.
if (!write_queue_.empty())
{
do_write();
}
}
else
{
#ifndef NDEBUG
printf("ConnectionManager::handle_write: %s", e.message().c_str());
#endif
remove_connection(conn);
}
}
void ConnectionManager::handle_read_data(boost::system::error_code const& e, MessagePointer message, ConnectionPointer conn)
{
#if TIME_READS
static virvo::FrameCounter counter;
printf("CM: reads/sec: %.2f\n", counter.registerFrame());
#endif
// Call the connection's slot
conn->signal_(Connection::Read, message, e);
if (!e)
{
// Read the next message
do_read(conn);
}
else
{
#ifndef NDEBUG
printf("ConnectionManager::handle_read_data: %s", e.message().c_str());
#endif
remove_connection(conn);
}
}
void Client::OnConnect(ConnectionPointer conn, const boost::system::error_code& error, tcp::resolver::iterator it)
{
if (connectHandler != nullptr)
{
connectHandler(conn);
conn->BeginReceive();
}
}
ConnectionPointer<Connection> ChooseConnectionDlg::choose(PrimaryConnectionTypes allowedConns, ConnectionPointer<Connection> const & preselectedConn)
{
m_allowedConns = allowedConns;
this->selectConn(preselectedConn.data());
if (this->exec() != QDialog::Accepted)
return ConnectionPointer<Connection>();
if (PortConnection * pc = dynamic_cast<PortConnection *>(m_current.data()))
{
if (pc->programmerType() == programmer_shupito && (m_allowedConns & pct_port_data) == 0)
{
ConnectionPointer<ShupitoConnection> sc = sConMgr2.createAutoShupito(pc);
m_current = sc;
}
}
return m_current;
}
void ConnectionManager::handle_read_data(boost::system::error_code const& e, MessagePointer message, ConnectionPointer conn)
{
// Call the connection's slot
conn->signal_(Connection::Read, message, e);
if (!e)
{
// Read the next message
do_read(conn);
}
else
{
#ifndef NDEBUG
printf("ConnectionManager::handle_read_data: %s", e.message().c_str());
#endif
remove_connection(conn);
}
}
void ConnectionManager::close(ConnectionPointer conn)
{
Connections::iterator I = connections_.find(conn);
if (I == connections_.end())
{
return;
}
// Remove the handler!
conn->remove_handler();
// Close the connection
conn->socket_.shutdown(tcp::socket::shutdown_both);
conn->socket_.close();
// Remove from the list.
// Eventually deletes the socket.
connections_.erase(I);
}
void ConnectionManager::handle_read_header(boost::system::error_code const& e, MessagePointer message, ConnectionPointer conn)
{
if (!e)
{
//
// TODO:
// Need to deserialize the message-header!
//
// Allocate memory for the message data
message->data_.resize(message->header_.size_);
assert( message->header_.size_ != 0 );
assert( message->header_.size_ == message->data_.size() );
// Start an asynchronous call to receive the data.
boost::asio::async_read(
conn->socket_,
boost::asio::buffer(&message->data_[0], message->data_.size()),
boost::bind(&ConnectionManager::handle_read_data, this, boost::asio::placeholders::error, message, conn)
);
}
else
{
#ifndef NDEBUG
printf("ConnectionManager::handle_read_header: %s", e.message().c_str());
#endif
#if 1
// Call the connection's slot
conn->signal_(Connection::Read, message, e);
#endif
// Delete the connection
remove_connection(conn);
}
}
void LibybUsbEnumerator::pluginEventReceived()
{
std::vector<yb::usb_plugin_event> events;
m_plugin_channel.receive(events);
for (size_t i = 0; i < events.size(); ++i)
{
yb::usb_plugin_event const & ev = events[i];
if (!ev.dev.empty())
{
if (!GenericUsbConnection::isFlipDevice(ev.dev))
continue;
usb_device_standby st;
st.sn = ev.dev.serial_number();
st.vidpid = ev.dev.vidpid();
switch (ev.action)
{
case yb::usb_plugin_event::a_add:
{
ConnectionPointer<GenericUsbConnection> conn = m_standby_usb_devices.extract(st);
if (!conn)
{
conn.reset(new GenericUsbConnection(m_runner, ev.dev));
sConMgr2.addConnection(conn.data());
}
conn->setRemovable(false);
conn->setDevice(ev.dev);
m_usb_devices.insert(std::make_pair(ev.dev, conn));
}
break;
case yb::usb_plugin_event::a_remove:
{
std::map<yb::usb_device, ConnectionPointer<GenericUsbConnection> >::iterator it = m_usb_devices.find(ev.dev);
it->second->clearDevice();
it->second->setRemovable(true);
m_standby_usb_devices.add(st, it->second.data());
m_usb_devices.erase(it);
}
break;
}
}
else
{
yb::usb_interface_descriptor const & intf = ev.intf.descriptor().altsettings[0];
usb_interface_standby st;
st.dev.sn = ev.intf.device().serial_number();
st.dev.vidpid = ev.intf.device().vidpid();
st.intfname = QString::fromUtf8(ev.intf.name().c_str());
if (st.intfname.isEmpty())
st.intfname = QString("#%1").arg(ev.intf.interface_index());
if (GenericUsbConnection::isShupito23Device(ev.intf.device()) && intf.bInterfaceClass == 0xff
&& intf.in_descriptor_count() > 0 && intf.out_descriptor_count() == 1)
{
switch (ev.action)
{
case yb::usb_plugin_event::a_add:
{
ConnectionPointer<UsbShupito23Connection> conn = m_standby_shupito23_devices.extract(st);
if (!conn)
{
conn.reset(new UsbShupito23Connection(m_runner));
conn->setName(GenericUsbConnection::formatDeviceName(ev.intf.device()), /*isDefault=*/true);
sConMgr2.addConnection(conn.data());
}
conn->setup(ev.intf);
conn->setRemovable(false);
m_shupito23_devices.insert(std::make_pair(ev.intf, conn));
}
break;
case yb::usb_plugin_event::a_remove:
{
std::map<yb::usb_device_interface, ConnectionPointer<UsbShupito23Connection> >::iterator it = m_shupito23_devices.find(ev.intf);
it->second->clear();
it->second->setRemovable(true);
m_standby_shupito23_devices.add(st, it->second.data());
m_shupito23_devices.erase(it);
}
break;
}
}
else if (intf.bInterfaceClass == 0xa && intf.bInterfaceSubClass == 0
&& !intf.endpoints.empty())
{
Q_ASSERT(!st.intfname.isEmpty());
switch (ev.action)
{
case yb::usb_plugin_event::a_add:
m_usb_acm_devices_by_info.insert(std::make_pair(st, ev.intf));
for (std::set<UsbAcmConnection2 *>::const_iterator it = m_user_owned_acm_conns.begin(); it != m_user_owned_acm_conns.end(); ++it)
(*it)->notifyIntfPlugin(ev.intf);
break;
case yb::usb_plugin_event::a_remove:
m_usb_acm_devices_by_info.erase(st);
for (std::set<UsbAcmConnection2 *>::const_iterator it = m_user_owned_acm_conns.begin(); it != m_user_owned_acm_conns.end(); ++it)
(*it)->notifyIntfUnplug(ev.intf);
break;
}
if (GenericUsbConnection::isShupito20Device(ev.intf.device()))
{
switch (ev.action)
{
case yb::usb_plugin_event::a_add:
{
ConnectionPointer<UsbShupito22Connection> conn = m_standby_shupito22_devices.extract(st);
if (!conn)
{
conn.reset(new UsbShupito22Connection(m_runner));
conn->setName(GenericUsbConnection::formatDeviceName(ev.intf.device()), /*isDefault=*/true);
sConMgr2.addConnection(conn.data());
}
conn->setup(ev.intf);
conn->setRemovable(false);
m_shupito22_devices.insert(std::make_pair(ev.intf, conn));
}
break;
case yb::usb_plugin_event::a_remove:
{
std::map<yb::usb_device_interface, ConnectionPointer<UsbShupito22Connection> >::iterator it = m_shupito22_devices.find(ev.intf);
it->second->clear();
it->second->setRemovable(true);
m_standby_shupito22_devices.add(st, it->second.data());
m_shupito22_devices.erase(it);
}
break;
}
}
else if (st.intfname[0] != '.' && st.intfname[0] != '#')
{
switch (ev.action)
{
case yb::usb_plugin_event::a_add:
{
ConnectionPointer<UsbAcmConnection2> conn = m_standby_usb_acm_devices.extract(st);
if (!conn)
{
conn.reset(new UsbAcmConnection2(m_runner));
conn->setName(QString("%1 @ %2").arg(st.intfname).arg(GenericUsbConnection::formatDeviceName(ev.intf.device())), /*isDefault=*/true);
sConMgr2.addConnection(conn.data());
}
conn->setEnumeratedIntf(ev.intf);
conn->setRemovable(false);
this->applyConfig(conn.data());
m_usb_acm_devices.insert(std::make_pair(ev.intf, conn));
}
break;
case yb::usb_plugin_event::a_remove:
{
std::map<yb::usb_device_interface, ConnectionPointer<UsbAcmConnection2> >::iterator it = m_usb_acm_devices.find(ev.intf);
this->updateConfig(it->second.data());
it->second->clear();
it->second->setRemovable(true);
m_standby_usb_acm_devices.add(st, it->second.data());
m_usb_acm_devices.erase(it);
}
break;
}
}
}
else if(GenericUsbConnection::isSTLink32LDevice(ev.intf.device()) && intf.bInterfaceClass == 0xff && st.intfname == "ST Link")
{
switch (ev.action)
{
case yb::usb_plugin_event::a_add:
{
for (std::set<STM32Connection *>::const_iterator it = m_user_owned_stm32_conns.begin(); it != m_user_owned_stm32_conns.end(); ++it)
(*it)->notifyIntfPlugin(ev.intf);
ConnectionPointer<STM32Connection> conn = m_standby_stm32_devices.extract(st);
if (!conn)
{
conn.reset(new STM32Connection(m_runner));
sConMgr2.addConnection(conn.data());
}
conn->setEnumeratedIntf(ev.intf);
conn->setRemovable(false);
m_stm32_devices.insert(std::make_pair(ev.intf, conn));
}
break;
case yb::usb_plugin_event::a_remove:
{
for (std::set<STM32Connection *>::const_iterator it = m_user_owned_stm32_conns.begin(); it != m_user_owned_stm32_conns.end(); ++it)
(*it)->notifyIntfUnplug(ev.intf);
std::map<yb::usb_device_interface, ConnectionPointer<STM32Connection> >::iterator it = m_stm32_devices.find(ev.intf);
it->second->clear();
it->second->setRemovable(true);
m_standby_stm32_devices.add(st, it->second.data());
m_stm32_devices.erase(it);
}
break;
}
}
}
}
}