<!DOCTYPE HTML> <html> <head> <meta http-equiv="Content-Type" content="text/html;charset=US-ASCII"> <meta name="generator" content="ZealOS V0.15"> <style type="text/css"> body {background-color:#fef1f0;} .cF0{color:#000000;background-color:#fef1f0;} .cF1{color:#0148a4;background-color:#fef1f0;} .cF2{color:#3b7901;background-color:#fef1f0;} .cF3{color:#057c7e;background-color:#fef1f0;} .cF4{color:#bb2020;background-color:#fef1f0;} .cF5{color:#9e42ae;background-color:#fef1f0;} .cF6{color:#b57901;background-color:#fef1f0;} .cF7{color:#b2b6af;background-color:#fef1f0;} .cF8{color:#555753;background-color:#fef1f0;} .cF9{color:#678fbb;background-color:#fef1f0;} .cFA{color:#82bc49;background-color:#fef1f0;} .cFB{color:#0097a2;background-color:#fef1f0;} .cFC{color:#e26a6a;background-color:#fef1f0;} .cFD{color:#c671bc;background-color:#fef1f0;} .cFE{color:#c7ab00;background-color:#fef1f0;} .cFF{color:#fef1f0;background-color:#fef1f0;} </style> </head> <body> <pre style="font-family:monospace;font-size:12pt"> <a name="l1"></a><span class=cF2>/***************************************************</span><span class=cF0> <a name="l2"></a> <a name="l3"></a></span><span class=cF2>TCP Bound/Connected Socket Tree Functions</span><span class=cF0> <a name="l4"></a> <a name="l5"></a></span><span class=cF2>***************************************************/</span><span class=cF0> <a name="l6"></a> <a name="l7"></a>CTCPTreeNode *TCPTreeNodeInit() <a name="l8"></a>{ </span><span class=cF2>// init new empty tree/node. Init socket queue head links.</span><span class=cF0> <a name="l9"></a> CTCPTreeNode *tree_node = </span><span class=cF5>CAlloc</span><span class=cF0>(</span><span class=cF1>sizeof</span><span class=cF7>(</span><span class=cF0>CTCPTreeNode</span><span class=cF7>)</span><span class=cF0>); <a name="l10"></a> <a name="l11"></a> tree_node->queue = </span><span class=cF5>CAlloc</span><span class=cF0>(</span><span class=cF1>sizeof</span><span class=cF7>(</span><span class=cF0>CTCPTreeQueue</span><span class=cF7>)</span><span class=cF0>); </span><span class=cF2>// CQueue vs CTCPTreeQueue ?...</span><span class=cF0> <a name="l12"></a> </span><span class=cF5>QueueInit</span><span class=cF0>(tree_node->queue); <a name="l13"></a> <a name="l14"></a> </span><span class=cF1>return</span><span class=cF0> tree_node; <a name="l15"></a>} <a name="l16"></a> <a name="l17"></a></span><span class=cF1>U0</span><span class=cF0> TCPTreeNodeAdd(CTCPTreeNode *node, CTCPTreeNode *tree) <a name="l18"></a>{ </span><span class=cF2>// using temp and last allows avoiding recursion and non-growing stack issues.</span><span class=cF0> <a name="l19"></a> BSTAdd(node, tree); <a name="l20"></a>} <a name="l21"></a> <a name="l22"></a>CTCPTreeNode *TCPTreeNodeParamAdd(</span><span class=cF9>I64</span><span class=cF0> port, CTCPTreeNode *tree) <a name="l23"></a>{ </span><span class=cF2>// add a node using params, return pointer to the node</span><span class=cF0> <a name="l24"></a> CTCPTreeNode *result = TCPTreeNodeInit; <a name="l25"></a> <a name="l26"></a> result->value = port; <a name="l27"></a> <a name="l28"></a> TCPTreeNodeAdd(result, tree); <a name="l29"></a> <a name="l30"></a> </span><span class=cF1>return</span><span class=cF0> result; <a name="l31"></a>} <a name="l32"></a> <a name="l33"></a>CTCPTreeNode *TCPTreeNodeParamInit(</span><span class=cF9>I64</span><span class=cF0> port) <a name="l34"></a>{ <a name="l35"></a> CTCPTreeNode *result = TCPTreeNodeInit; <a name="l36"></a> <a name="l37"></a> result->value = port; <a name="l38"></a> <a name="l39"></a> </span><span class=cF1>return</span><span class=cF0> result; <a name="l40"></a>} <a name="l41"></a> <a name="l42"></a>CTCPTreeNode *TCPTreeNodeFind(</span><span class=cF9>I64</span><span class=cF0> port, CTCPTreeNode *tree) <a name="l43"></a>{ <a name="l44"></a> </span><span class=cF1>return</span><span class=cF0> BSTFind(port, tree); <a name="l45"></a>} <a name="l46"></a> <a name="l47"></a>CTCPTreeNode *TCPTreeNodePop(</span><span class=cF9>I64</span><span class=cF0> port, CTCPTreeNode *tree) <a name="l48"></a>{ </span><span class=cF2>// Pops whole sub-tree, original tree loses whole branch.</span><span class=cF0> <a name="l49"></a> </span><span class=cF1>return</span><span class=cF0> BSTPop(port, tree); <a name="l50"></a>} <a name="l51"></a> <a name="l52"></a>CTCPTreeNode *TCPTreeNodeSinglePop(</span><span class=cF9>I64</span><span class=cF0> port, CTCPTreeNode *tree) <a name="l53"></a>{ </span><span class=cF2>// Pop a tree off, then add back in its sub-trees to main tree.</span><span class=cF0> <a name="l54"></a> </span><span class=cF2>// Original node sub-tree links are cleared.</span><span class=cF0> <a name="l55"></a> <a name="l56"></a> </span><span class=cF1>return</span><span class=cF0> BSTSinglePop(port, tree); <a name="l57"></a>} <a name="l58"></a> <a name="l59"></a></span><span class=cF1>U0</span><span class=cF0> TCPTreeNodeQueueAdd(CTCPSocket *socket, CTCPTreeNode *node) <a name="l60"></a>{ <a name="l61"></a> CTCPTreeQueue *new_entry = </span><span class=cF5>CAlloc</span><span class=cF0>(</span><span class=cF1>sizeof</span><span class=cF7>(</span><span class=cF0>CTCPTreeQueue</span><span class=cF7>)</span><span class=cF0>); <a name="l62"></a> <a name="l63"></a> new_entry->socket = socket; <a name="l64"></a> <a name="l65"></a> </span><span class=cF5>QueueInsert</span><span class=cF0>(new_entry, node->queue->last); <a name="l66"></a>} <a name="l67"></a> <a name="l68"></a>CTCPTreeQueue *TCPTreeNodeQueueSocketFind(CTCPSocket *socket, CTCPTreeNode *node) <a name="l69"></a>{ <a name="l70"></a> CTCPTreeQueue *temp_queue; <a name="l71"></a> <a name="l72"></a> temp_queue = node->queue->next; <a name="l73"></a> </span><span class=cF1>while</span><span class=cF0> (temp_queue != node->queue) <a name="l74"></a> </span><span class=cF7>{</span><span class=cF0> <a name="l75"></a> </span><span class=cF1>if</span><span class=cF0> (temp_queue->socket == socket) <a name="l76"></a> </span><span class=cF1>return</span><span class=cF0> temp_queue; <a name="l77"></a> temp_queue = temp_queue->next; <a name="l78"></a> </span><span class=cF7>}</span><span class=cF0> <a name="l79"></a> <a name="l80"></a> </span><span class=cF1>return</span><span class=cF0> </span><span class=cF3>NULL</span><span class=cF0>; <a name="l81"></a>} <a name="l82"></a> <a name="l83"></a>CTCPTreeQueue *TCPTreeNodeQueueSocketSinglePop(CTCPSocket *socket, CTCPTreeNode *node) <a name="l84"></a>{ </span><span class=cF2>// search by socket, pop a single TCPTreeQueue off the node, return popped queue.</span><span class=cF0> <a name="l85"></a> CTCPTreeQueue *temp_queue = TCPTreeNodeQueueSocketFind(socket, node); <a name="l86"></a> <a name="l87"></a> </span><span class=cF1>if</span><span class=cF0> (temp_queue) <a name="l88"></a> </span><span class=cF7>{</span><span class=cF0> <a name="l89"></a> </span><span class=cF5>QueueRemove</span><span class=cF0>(temp_queue); <a name="l90"></a> <a name="l91"></a> </span><span class=cF7>}</span><span class=cF0> <a name="l92"></a> <a name="l93"></a> </span><span class=cF1>return</span><span class=cF0> temp_queue; </span><span class=cF2>// if not found, NULL.</span><span class=cF0> <a name="l94"></a>} <a name="l95"></a> <a name="l96"></a>CTCPTreeQueue *TCPTreeNodeQueueIPV4Find(</span><span class=cF9>U32</span><span class=cF0> address, CTCPTreeNode *node, </span><span class=cF1>Bool</span><span class=cF0> specific=</span><span class=cF3>FALSE</span><span class=cF0>) <a name="l97"></a>{ </span><span class=cF2>// address should be pulled from an instance of CIPV4Address (TODO... double check what bit order we're in ?)</span><span class=cF0> <a name="l98"></a> </span><span class=cF2>// use TRUE or FALSE in specific arg to dictate how to handle INADDR_ANY.</span><span class=cF0> <a name="l99"></a> <a name="l100"></a> CTCPTreeQueue *temp_queue = node->queue->next; <a name="l101"></a> CSocketAddressIPV4 *temp_ip; <a name="l102"></a> <a name="l103"></a> </span><span class=cF1>while</span><span class=cF0> (temp_queue != node->queue) <a name="l104"></a> </span><span class=cF7>{</span><span class=cF0> <a name="l105"></a> </span><span class=cF1>if</span><span class=cF0> (temp_queue->socket->destination_address.family == AF_INET) <a name="l106"></a> { <a name="l107"></a> temp_ip = &temp_queue->socket->destination_address; <a name="l108"></a> NetLog(</span><span class=cF6>"TCPTreeNodeQueueIPV4Find: Comparing: addr, nodequeue addr: %08X, %08X"</span><span class=cF0>, <a name="l109"></a> address, temp_ip->address.address); <a name="l110"></a> <a name="l111"></a> </span><span class=cF1>if</span><span class=cF0> (temp_ip->address.address == address) <a name="l112"></a> </span><span class=cF7>{</span><span class=cF0> <a name="l113"></a> NetLog(</span><span class=cF6>"TCPTreeNodeQueueIPV4Find: Address match: addr, nodequeue addr: %08X, %08X "</span><span class=cF0>, <a name="l114"></a> address, temp_ip->address.address); <a name="l115"></a> <a name="l116"></a> </span><span class=cF1>return</span><span class=cF0> temp_queue; <a name="l117"></a> </span><span class=cF7>}</span><span class=cF0> <a name="l118"></a> } <a name="l119"></a> </span><span class=cF1>else</span><span class=cF0> <a name="l120"></a> NetErr(</span><span class=cF6>"TCPTreeNodeQueueIPV4Find: Skipped iteration of a non AF_INET family: %0X"</span><span class=cF0>, <a name="l121"></a> temp_queue->socket->destination_address.family); <a name="l122"></a> <a name="l123"></a> temp_queue = temp_queue->next; <a name="l124"></a> </span><span class=cF7>}</span><span class=cF0> <a name="l125"></a> <a name="l126"></a> </span><span class=cF1>if</span><span class=cF0> (!specific) <a name="l127"></a> </span><span class=cF7>{</span><span class=cF0> <a name="l128"></a> temp_queue = node->queue->next; <a name="l129"></a> NetDebug(</span><span class=cF6>"TCPTreeNodeQueueIPV4Find: Exact match not found, looking for an INADDR_ANY address."</span><span class=cF0>); <a name="l130"></a> <a name="l131"></a> </span><span class=cF1>while</span><span class=cF0> (temp_queue != node->queue) <a name="l132"></a> { <a name="l133"></a> </span><span class=cF1>if</span><span class=cF0> (temp_queue->socket->destination_address.family == AF_INET) <a name="l134"></a> </span><span class=cF7>{</span><span class=cF0> <a name="l135"></a> temp_ip = &temp_queue->socket->destination_address; <a name="l136"></a> NetLog(</span><span class=cF6>"TCPTreeNodeQueueIPV4Find: Comparing: addr, nodequeue addr: %08X, %08X"</span><span class=cF0>, <a name="l137"></a> address, temp_ip->address.address); <a name="l138"></a> <a name="l139"></a> </span><span class=cF1>if</span><span class=cF0> (temp_ip->address.address == INADDR_ANY) <a name="l140"></a> { <a name="l141"></a> NetLog(</span><span class=cF6>"TCPTreeNodeQueueIPV4Find: Address match: addr, nodequeue addr: %08X, %08X "</span><span class=cF0>, <a name="l142"></a> address, temp_ip->address.address); <a name="l143"></a> <a name="l144"></a> </span><span class=cF1>return</span><span class=cF0> temp_queue; <a name="l145"></a> } <a name="l146"></a> </span><span class=cF7>}</span><span class=cF0> <a name="l147"></a> </span><span class=cF1>else</span><span class=cF0> <a name="l148"></a> NetErr(</span><span class=cF6>"TCPTreeNodeQueueIPV4Find: Skipped iteration of a non AF_INET family: %0X"</span><span class=cF0>, <a name="l149"></a> temp_queue->socket->destination_address.family); <a name="l150"></a> <a name="l151"></a> temp_queue = temp_queue->next; <a name="l152"></a> } <a name="l153"></a> </span><span class=cF7>}</span><span class=cF0> <a name="l154"></a> <a name="l155"></a> <a name="l156"></a> </span><span class=cF1>return</span><span class=cF0> </span><span class=cF3>NULL</span><span class=cF0>; <a name="l157"></a>} <a name="l158"></a> </span></pre></body> </html>