/*
 * \brief  Downlink interface in form of a NIC session component
 * \author Martin Stein
 * \date   2017-03-08
 */

/*
 * Copyright (C) 2016-2017 Genode Labs GmbH
 *
 * This file is part of the Genode OS framework, which is distributed
 * under the terms of the GNU Affero General Public License version 3.
 */

#ifndef _COMPONENT_H_
#define _COMPONENT_H_

/* Genode includes */
#include <root/component.h>
#include <nic/packet_allocator.h>
#include <nic_session/rpc_object.h>
#include <base/heap.h>

/* local includes */
#include <interface.h>
#include <uplink.h>

namespace Net {

	class Communication_buffer;
	class Session_component_base;
	class Session_component;
	class Root;
}


class Net::Communication_buffer : public Genode::Ram_dataspace_capability
{
	private:

		Genode::Ram_allocator &_ram;

	public:

		Communication_buffer(Genode::Ram_allocator &ram,
		                     Genode::size_t const size);

		~Communication_buffer() { _ram.free(*this); }
};


class Net::Session_component_base
{
	protected:

		Genode::Ram_quota_guard           _ram_quota_guard;
		Genode::Cap_quota_guard           _cap_quota_guard;
		Genode::Constrained_ram_allocator _ram;
		Genode::Sliced_heap               _alloc;
		Nic::Packet_allocator             _range_alloc;
		Communication_buffer              _tx_buf;
		Communication_buffer              _rx_buf;

	public:

		Session_component_base(Genode::Ram_allocator &ram,
		                       Genode::Region_map    &local_rm,
		                       Genode::Ram_quota      ram_quota,
		                       Genode::Cap_quota      cap_quota,
		                       Genode::size_t   const tx_buf_size,
		                       Genode::size_t   const rx_buf_size);
};


class Net::Session_component : private Session_component_base,
                               public  ::Nic::Session_rpc_object,
                               private Interface
{
	private:

		Uplink                                    _uplink;
		Genode::Signal_context_capability         _link_state_sigh { };
		Genode::Signal_handler<Session_component> _link_state_handler;

		void _handle_link_state();
		void _print_state();


		/********************
		 ** Net::Interface **
		 ********************/

		Packet_stream_sink   &_sink()   override { return *_tx.sink(); }
		Packet_stream_source &_source() override { return *_rx.source(); }

	public:

		Session_component(Genode::Ram_quota  ram_quota,
		                  Genode::Cap_quota  cap_quota,
		                  Genode::size_t     tx_buf_size,
		                  Genode::size_t     rx_buf_size,
		                  Genode::Xml_node   config,
		                  Timer::Connection &timer,
		                  Genode::Duration  &curr_time,
		                  Genode::Env       &env);


		/******************
		 ** Nic::Session **
		 ******************/

		Mac_address mac_address() override { return _uplink.mac_address(); }
		bool link_state() override { return _uplink.link_state(); }
		void link_state_sigh(Genode::Signal_context_capability sigh) override {
			_link_state_sigh = sigh; }
};


class Net::Root : public Genode::Root_component<Session_component,
                                                Genode::Single_client>
{
	private:

		Genode::Env       &_env;
		Genode::Xml_node   _config;
		Timer::Connection &_timer;
		Genode::Duration  &_curr_time;


		/********************
		 ** Root_component **
		 ********************/

		Session_component *_create_session(char const *args) override;

	public:

		Root(Genode::Env       &env,
		     Genode::Allocator &alloc,
		     Genode::Xml_node   config,
		     Timer::Connection &timer,
		     Genode::Duration  &curr_time);
};

#endif /* _COMPONENT_H_ */