X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=src%2Fmeshlink%2B%2B.h;h=d7f88fd119e8c9c9b51149858142b930d3a85da5;hb=21843b323d8c006fe17e0c2d7fff74fc8dfa3c48;hp=49d7f602bdc88acdd1e502b81fa4ccea00dfa9fc;hpb=ad6a0bf4c6b0fce2cf29e0a7089909e23a577155;p=meshlink

diff --git a/src/meshlink++.h b/src/meshlink++.h
index 49d7f602..d7f88fd1 100644
--- a/src/meshlink++.h
+++ b/src/meshlink++.h
@@ -172,6 +172,10 @@ namespace meshlink {
 
 		/// This functions is called whenever another node attemps to open a channel to the local node.
 		/** 
+		 *  If the channel is accepted, the poll_callback will be set to channel_poll and can be
+		 *  changed using set_channel_poll_cb(). Likewise, the receive callback is set to
+		 *  channel_receive().
+		 *
 		 *  The function is run in MeshLink's own thread.
 		 *  It is therefore important that the callback uses apprioriate methods (queues, pipes, locking, etc.)
 		 *  to pass data to or from the application's thread.
@@ -187,8 +191,8 @@ namespace meshlink {
 		 */
 		virtual bool channel_accept(channel *channel, uint16_t port, const void *data, size_t len)
 		{
-		        /* by default reject all channels */
-		        return false;
+			/* by default reject all channels */
+			return false;
 		}
 
 		/// This function is called by Meshlink for receiving data from a channel.
@@ -403,6 +407,9 @@ namespace meshlink {
 		/** This function is called whenever a remote node wants to open a channel to the local node.
 		 *  The application then has to decide whether to accept or reject this channel.
 		 *
+		 *  This function sets the channel poll callback to channel_poll_trampoline, which in turn
+		 *  calls channel_poll. To set a differnt, channel-specific poll callback, use set_channel_poll_cb.
+		 *
 		 *  @param node         The node to which this channel is being initiated.
 		 *  @param port         The port number the peer wishes to connect to.
 		 *  @param cb           A pointer to the function which will be called when the remote node sends data to the local node.
@@ -412,7 +419,9 @@ namespace meshlink {
 		 *  @return             A handle for the channel, or NULL in case of an error.
 		 */
 		channel *channel_open(node *node, uint16_t port, channel_receive_cb_t cb, const void *data, size_t len) {
-			return (channel *)meshlink_channel_open(handle, node, port, (meshlink_channel_receive_cb_t)cb, data, len);
+			channel *ch = (channel *)meshlink_channel_open(handle, node, port, (meshlink_channel_receive_cb_t)cb, data, len);
+			meshlink_set_channel_poll_cb(handle, ch, &channel_poll_trampoline);
+			return ch;
 		}
 
 		/**
@@ -420,7 +429,9 @@ namespace meshlink {
 		 * Sets channel_receive_trampoline as cb, which in turn calls this->channel_receive( ... ).
 		 */
 		channel *channel_open(node *node, uint16_t port, const void *data, size_t len) {
-			return (channel *)meshlink_channel_open(handle, node, port, &channel_receive_trampoline, data, len);
+			channel *ch = (channel *)meshlink_channel_open(handle, node, port, &channel_receive_trampoline, data, len);
+			meshlink_set_channel_poll_cb(handle, ch, &channel_poll_trampoline);
+			return ch;
 		}
 
 		/// Partially close a reliable stream channel.
@@ -489,6 +500,7 @@ namespace meshlink {
 			bool accepted = that->channel_accept(static_cast<meshlink::channel*>(channel), port, data, len);
 			if (accepted)
 			{
+				meshlink_set_channel_receive_cb(handle, channel, &channel_receive_trampoline);
 				meshlink_set_channel_poll_cb(handle, channel, &channel_poll_trampoline);
 			}
 			return accepted;