socket.c 86 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374
  1. /*
  2. * net/tipc/socket.c: TIPC socket API
  3. *
  4. * Copyright (c) 2001-2007, 2012-2017, Ericsson AB
  5. * Copyright (c) 2004-2008, 2010-2013, Wind River Systems
  6. * All rights reserved.
  7. *
  8. * Redistribution and use in source and binary forms, with or without
  9. * modification, are permitted provided that the following conditions are met:
  10. *
  11. * 1. Redistributions of source code must retain the above copyright
  12. * notice, this list of conditions and the following disclaimer.
  13. * 2. Redistributions in binary form must reproduce the above copyright
  14. * notice, this list of conditions and the following disclaimer in the
  15. * documentation and/or other materials provided with the distribution.
  16. * 3. Neither the names of the copyright holders nor the names of its
  17. * contributors may be used to endorse or promote products derived from
  18. * this software without specific prior written permission.
  19. *
  20. * Alternatively, this software may be distributed under the terms of the
  21. * GNU General Public License ("GPL") version 2 as published by the Free
  22. * Software Foundation.
  23. *
  24. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  25. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  28. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  29. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  30. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  31. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  32. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  33. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  34. * POSSIBILITY OF SUCH DAMAGE.
  35. */
  36. #include <linux/rhashtable.h>
  37. #include <linux/sched/signal.h>
  38. #include "core.h"
  39. #include "name_table.h"
  40. #include "node.h"
  41. #include "link.h"
  42. #include "name_distr.h"
  43. #include "socket.h"
  44. #include "bcast.h"
  45. #include "netlink.h"
  46. #include "group.h"
  47. #define CONN_TIMEOUT_DEFAULT 8000 /* default connect timeout = 8s */
  48. #define CONN_PROBING_INTV msecs_to_jiffies(3600000) /* [ms] => 1 h */
  49. #define TIPC_FWD_MSG 1
  50. #define TIPC_MAX_PORT 0xffffffff
  51. #define TIPC_MIN_PORT 1
  52. #define TIPC_ACK_RATE 4 /* ACK at 1/4 of of rcv window size */
  53. enum {
  54. TIPC_LISTEN = TCP_LISTEN,
  55. TIPC_ESTABLISHED = TCP_ESTABLISHED,
  56. TIPC_OPEN = TCP_CLOSE,
  57. TIPC_DISCONNECTING = TCP_CLOSE_WAIT,
  58. TIPC_CONNECTING = TCP_SYN_SENT,
  59. };
  60. struct sockaddr_pair {
  61. struct sockaddr_tipc sock;
  62. struct sockaddr_tipc member;
  63. };
  64. /**
  65. * struct tipc_sock - TIPC socket structure
  66. * @sk: socket - interacts with 'port' and with user via the socket API
  67. * @conn_type: TIPC type used when connection was established
  68. * @conn_instance: TIPC instance used when connection was established
  69. * @published: non-zero if port has one or more associated names
  70. * @max_pkt: maximum packet size "hint" used when building messages sent by port
  71. * @portid: unique port identity in TIPC socket hash table
  72. * @phdr: preformatted message header used when sending messages
  73. * #cong_links: list of congested links
  74. * @publications: list of publications for port
  75. * @blocking_link: address of the congested link we are currently sleeping on
  76. * @pub_count: total # of publications port has made during its lifetime
  77. * @probing_state:
  78. * @conn_timeout: the time we can wait for an unresponded setup request
  79. * @dupl_rcvcnt: number of bytes counted twice, in both backlog and rcv queue
  80. * @cong_link_cnt: number of congested links
  81. * @snt_unacked: # messages sent by socket, and not yet acked by peer
  82. * @rcv_unacked: # messages read by user, but not yet acked back to peer
  83. * @peer: 'connected' peer for dgram/rdm
  84. * @node: hash table node
  85. * @mc_method: cookie for use between socket and broadcast layer
  86. * @rcu: rcu struct for tipc_sock
  87. */
  88. struct tipc_sock {
  89. struct sock sk;
  90. u32 conn_type;
  91. u32 conn_instance;
  92. int published;
  93. u32 max_pkt;
  94. u32 portid;
  95. struct tipc_msg phdr;
  96. struct list_head cong_links;
  97. struct list_head publications;
  98. u32 pub_count;
  99. uint conn_timeout;
  100. atomic_t dupl_rcvcnt;
  101. bool probe_unacked;
  102. u16 cong_link_cnt;
  103. u16 snt_unacked;
  104. u16 snd_win;
  105. u16 peer_caps;
  106. u16 rcv_unacked;
  107. u16 rcv_win;
  108. struct sockaddr_tipc peer;
  109. struct rhash_head node;
  110. struct tipc_mc_method mc_method;
  111. struct rcu_head rcu;
  112. struct tipc_group *group;
  113. };
  114. static int tipc_sk_backlog_rcv(struct sock *sk, struct sk_buff *skb);
  115. static void tipc_data_ready(struct sock *sk);
  116. static void tipc_write_space(struct sock *sk);
  117. static void tipc_sock_destruct(struct sock *sk);
  118. static int tipc_release(struct socket *sock);
  119. static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags,
  120. bool kern);
  121. static void tipc_sk_timeout(struct timer_list *t);
  122. static int tipc_sk_publish(struct tipc_sock *tsk, uint scope,
  123. struct tipc_name_seq const *seq);
  124. static int tipc_sk_withdraw(struct tipc_sock *tsk, uint scope,
  125. struct tipc_name_seq const *seq);
  126. static int tipc_sk_leave(struct tipc_sock *tsk);
  127. static struct tipc_sock *tipc_sk_lookup(struct net *net, u32 portid);
  128. static int tipc_sk_insert(struct tipc_sock *tsk);
  129. static void tipc_sk_remove(struct tipc_sock *tsk);
  130. static int __tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dsz);
  131. static int __tipc_sendmsg(struct socket *sock, struct msghdr *m, size_t dsz);
  132. static const struct proto_ops packet_ops;
  133. static const struct proto_ops stream_ops;
  134. static const struct proto_ops msg_ops;
  135. static struct proto tipc_proto;
  136. static const struct rhashtable_params tsk_rht_params;
  137. static u32 tsk_own_node(struct tipc_sock *tsk)
  138. {
  139. return msg_prevnode(&tsk->phdr);
  140. }
  141. static u32 tsk_peer_node(struct tipc_sock *tsk)
  142. {
  143. return msg_destnode(&tsk->phdr);
  144. }
  145. static u32 tsk_peer_port(struct tipc_sock *tsk)
  146. {
  147. return msg_destport(&tsk->phdr);
  148. }
  149. static bool tsk_unreliable(struct tipc_sock *tsk)
  150. {
  151. return msg_src_droppable(&tsk->phdr) != 0;
  152. }
  153. static void tsk_set_unreliable(struct tipc_sock *tsk, bool unreliable)
  154. {
  155. msg_set_src_droppable(&tsk->phdr, unreliable ? 1 : 0);
  156. }
  157. static bool tsk_unreturnable(struct tipc_sock *tsk)
  158. {
  159. return msg_dest_droppable(&tsk->phdr) != 0;
  160. }
  161. static void tsk_set_unreturnable(struct tipc_sock *tsk, bool unreturnable)
  162. {
  163. msg_set_dest_droppable(&tsk->phdr, unreturnable ? 1 : 0);
  164. }
  165. static int tsk_importance(struct tipc_sock *tsk)
  166. {
  167. return msg_importance(&tsk->phdr);
  168. }
  169. static int tsk_set_importance(struct tipc_sock *tsk, int imp)
  170. {
  171. if (imp > TIPC_CRITICAL_IMPORTANCE)
  172. return -EINVAL;
  173. msg_set_importance(&tsk->phdr, (u32)imp);
  174. return 0;
  175. }
  176. static struct tipc_sock *tipc_sk(const struct sock *sk)
  177. {
  178. return container_of(sk, struct tipc_sock, sk);
  179. }
  180. static bool tsk_conn_cong(struct tipc_sock *tsk)
  181. {
  182. return tsk->snt_unacked > tsk->snd_win;
  183. }
  184. static u16 tsk_blocks(int len)
  185. {
  186. return ((len / FLOWCTL_BLK_SZ) + 1);
  187. }
  188. /* tsk_blocks(): translate a buffer size in bytes to number of
  189. * advertisable blocks, taking into account the ratio truesize(len)/len
  190. * We can trust that this ratio is always < 4 for len >= FLOWCTL_BLK_SZ
  191. */
  192. static u16 tsk_adv_blocks(int len)
  193. {
  194. return len / FLOWCTL_BLK_SZ / 4;
  195. }
  196. /* tsk_inc(): increment counter for sent or received data
  197. * - If block based flow control is not supported by peer we
  198. * fall back to message based ditto, incrementing the counter
  199. */
  200. static u16 tsk_inc(struct tipc_sock *tsk, int msglen)
  201. {
  202. if (likely(tsk->peer_caps & TIPC_BLOCK_FLOWCTL))
  203. return ((msglen / FLOWCTL_BLK_SZ) + 1);
  204. return 1;
  205. }
  206. /**
  207. * tsk_advance_rx_queue - discard first buffer in socket receive queue
  208. *
  209. * Caller must hold socket lock
  210. */
  211. static void tsk_advance_rx_queue(struct sock *sk)
  212. {
  213. kfree_skb(__skb_dequeue(&sk->sk_receive_queue));
  214. }
  215. /* tipc_sk_respond() : send response message back to sender
  216. */
  217. static void tipc_sk_respond(struct sock *sk, struct sk_buff *skb, int err)
  218. {
  219. u32 selector;
  220. u32 dnode;
  221. u32 onode = tipc_own_addr(sock_net(sk));
  222. if (!tipc_msg_reverse(onode, &skb, err))
  223. return;
  224. dnode = msg_destnode(buf_msg(skb));
  225. selector = msg_origport(buf_msg(skb));
  226. tipc_node_xmit_skb(sock_net(sk), skb, dnode, selector);
  227. }
  228. /**
  229. * tsk_rej_rx_queue - reject all buffers in socket receive queue
  230. *
  231. * Caller must hold socket lock
  232. */
  233. static void tsk_rej_rx_queue(struct sock *sk)
  234. {
  235. struct sk_buff *skb;
  236. while ((skb = __skb_dequeue(&sk->sk_receive_queue)))
  237. tipc_sk_respond(sk, skb, TIPC_ERR_NO_PORT);
  238. }
  239. static bool tipc_sk_connected(struct sock *sk)
  240. {
  241. return sk->sk_state == TIPC_ESTABLISHED;
  242. }
  243. /* tipc_sk_type_connectionless - check if the socket is datagram socket
  244. * @sk: socket
  245. *
  246. * Returns true if connection less, false otherwise
  247. */
  248. static bool tipc_sk_type_connectionless(struct sock *sk)
  249. {
  250. return sk->sk_type == SOCK_RDM || sk->sk_type == SOCK_DGRAM;
  251. }
  252. /* tsk_peer_msg - verify if message was sent by connected port's peer
  253. *
  254. * Handles cases where the node's network address has changed from
  255. * the default of <0.0.0> to its configured setting.
  256. */
  257. static bool tsk_peer_msg(struct tipc_sock *tsk, struct tipc_msg *msg)
  258. {
  259. struct sock *sk = &tsk->sk;
  260. struct tipc_net *tn = net_generic(sock_net(sk), tipc_net_id);
  261. u32 peer_port = tsk_peer_port(tsk);
  262. u32 orig_node;
  263. u32 peer_node;
  264. if (unlikely(!tipc_sk_connected(sk)))
  265. return false;
  266. if (unlikely(msg_origport(msg) != peer_port))
  267. return false;
  268. orig_node = msg_orignode(msg);
  269. peer_node = tsk_peer_node(tsk);
  270. if (likely(orig_node == peer_node))
  271. return true;
  272. if (!orig_node && (peer_node == tn->own_addr))
  273. return true;
  274. if (!peer_node && (orig_node == tn->own_addr))
  275. return true;
  276. return false;
  277. }
  278. /* tipc_set_sk_state - set the sk_state of the socket
  279. * @sk: socket
  280. *
  281. * Caller must hold socket lock
  282. *
  283. * Returns 0 on success, errno otherwise
  284. */
  285. static int tipc_set_sk_state(struct sock *sk, int state)
  286. {
  287. int oldsk_state = sk->sk_state;
  288. int res = -EINVAL;
  289. switch (state) {
  290. case TIPC_OPEN:
  291. res = 0;
  292. break;
  293. case TIPC_LISTEN:
  294. case TIPC_CONNECTING:
  295. if (oldsk_state == TIPC_OPEN)
  296. res = 0;
  297. break;
  298. case TIPC_ESTABLISHED:
  299. if (oldsk_state == TIPC_CONNECTING ||
  300. oldsk_state == TIPC_OPEN)
  301. res = 0;
  302. break;
  303. case TIPC_DISCONNECTING:
  304. if (oldsk_state == TIPC_CONNECTING ||
  305. oldsk_state == TIPC_ESTABLISHED)
  306. res = 0;
  307. break;
  308. }
  309. if (!res)
  310. sk->sk_state = state;
  311. return res;
  312. }
  313. static int tipc_sk_sock_err(struct socket *sock, long *timeout)
  314. {
  315. struct sock *sk = sock->sk;
  316. int err = sock_error(sk);
  317. int typ = sock->type;
  318. if (err)
  319. return err;
  320. if (typ == SOCK_STREAM || typ == SOCK_SEQPACKET) {
  321. if (sk->sk_state == TIPC_DISCONNECTING)
  322. return -EPIPE;
  323. else if (!tipc_sk_connected(sk))
  324. return -ENOTCONN;
  325. }
  326. if (!*timeout)
  327. return -EAGAIN;
  328. if (signal_pending(current))
  329. return sock_intr_errno(*timeout);
  330. return 0;
  331. }
  332. #define tipc_wait_for_cond(sock_, timeo_, condition_) \
  333. ({ \
  334. struct sock *sk_; \
  335. int rc_; \
  336. \
  337. while ((rc_ = !(condition_))) { \
  338. DEFINE_WAIT_FUNC(wait_, woken_wake_function); \
  339. sk_ = (sock_)->sk; \
  340. rc_ = tipc_sk_sock_err((sock_), timeo_); \
  341. if (rc_) \
  342. break; \
  343. prepare_to_wait(sk_sleep(sk_), &wait_, TASK_INTERRUPTIBLE); \
  344. release_sock(sk_); \
  345. *(timeo_) = wait_woken(&wait_, TASK_INTERRUPTIBLE, *(timeo_)); \
  346. sched_annotate_sleep(); \
  347. lock_sock(sk_); \
  348. remove_wait_queue(sk_sleep(sk_), &wait_); \
  349. } \
  350. rc_; \
  351. })
  352. /**
  353. * tipc_sk_create - create a TIPC socket
  354. * @net: network namespace (must be default network)
  355. * @sock: pre-allocated socket structure
  356. * @protocol: protocol indicator (must be 0)
  357. * @kern: caused by kernel or by userspace?
  358. *
  359. * This routine creates additional data structures used by the TIPC socket,
  360. * initializes them, and links them together.
  361. *
  362. * Returns 0 on success, errno otherwise
  363. */
  364. static int tipc_sk_create(struct net *net, struct socket *sock,
  365. int protocol, int kern)
  366. {
  367. struct tipc_net *tn;
  368. const struct proto_ops *ops;
  369. struct sock *sk;
  370. struct tipc_sock *tsk;
  371. struct tipc_msg *msg;
  372. /* Validate arguments */
  373. if (unlikely(protocol != 0))
  374. return -EPROTONOSUPPORT;
  375. switch (sock->type) {
  376. case SOCK_STREAM:
  377. ops = &stream_ops;
  378. break;
  379. case SOCK_SEQPACKET:
  380. ops = &packet_ops;
  381. break;
  382. case SOCK_DGRAM:
  383. case SOCK_RDM:
  384. ops = &msg_ops;
  385. break;
  386. default:
  387. return -EPROTOTYPE;
  388. }
  389. /* Allocate socket's protocol area */
  390. sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto, kern);
  391. if (sk == NULL)
  392. return -ENOMEM;
  393. tsk = tipc_sk(sk);
  394. tsk->max_pkt = MAX_PKT_DEFAULT;
  395. INIT_LIST_HEAD(&tsk->publications);
  396. INIT_LIST_HEAD(&tsk->cong_links);
  397. msg = &tsk->phdr;
  398. tn = net_generic(sock_net(sk), tipc_net_id);
  399. /* Finish initializing socket data structures */
  400. sock->ops = ops;
  401. sock_init_data(sock, sk);
  402. tipc_set_sk_state(sk, TIPC_OPEN);
  403. if (tipc_sk_insert(tsk)) {
  404. pr_warn("Socket create failed; port number exhausted\n");
  405. return -EINVAL;
  406. }
  407. /* Ensure tsk is visible before we read own_addr. */
  408. smp_mb();
  409. tipc_msg_init(tn->own_addr, msg, TIPC_LOW_IMPORTANCE, TIPC_NAMED_MSG,
  410. NAMED_H_SIZE, 0);
  411. msg_set_origport(msg, tsk->portid);
  412. timer_setup(&sk->sk_timer, tipc_sk_timeout, 0);
  413. sk->sk_shutdown = 0;
  414. sk->sk_backlog_rcv = tipc_sk_backlog_rcv;
  415. sk->sk_rcvbuf = sysctl_tipc_rmem[1];
  416. sk->sk_data_ready = tipc_data_ready;
  417. sk->sk_write_space = tipc_write_space;
  418. sk->sk_destruct = tipc_sock_destruct;
  419. tsk->conn_timeout = CONN_TIMEOUT_DEFAULT;
  420. atomic_set(&tsk->dupl_rcvcnt, 0);
  421. /* Start out with safe limits until we receive an advertised window */
  422. tsk->snd_win = tsk_adv_blocks(RCVBUF_MIN);
  423. tsk->rcv_win = tsk->snd_win;
  424. if (tipc_sk_type_connectionless(sk)) {
  425. tsk_set_unreturnable(tsk, true);
  426. if (sock->type == SOCK_DGRAM)
  427. tsk_set_unreliable(tsk, true);
  428. }
  429. return 0;
  430. }
  431. static void tipc_sk_callback(struct rcu_head *head)
  432. {
  433. struct tipc_sock *tsk = container_of(head, struct tipc_sock, rcu);
  434. sock_put(&tsk->sk);
  435. }
  436. /* Caller should hold socket lock for the socket. */
  437. static void __tipc_shutdown(struct socket *sock, int error)
  438. {
  439. struct sock *sk = sock->sk;
  440. struct tipc_sock *tsk = tipc_sk(sk);
  441. struct net *net = sock_net(sk);
  442. long timeout = CONN_TIMEOUT_DEFAULT;
  443. u32 dnode = tsk_peer_node(tsk);
  444. struct sk_buff *skb;
  445. /* Avoid that hi-prio shutdown msgs bypass msgs in link wakeup queue */
  446. tipc_wait_for_cond(sock, &timeout, (!tsk->cong_link_cnt &&
  447. !tsk_conn_cong(tsk)));
  448. /* Reject all unreceived messages, except on an active connection
  449. * (which disconnects locally & sends a 'FIN+' to peer).
  450. */
  451. while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
  452. if (TIPC_SKB_CB(skb)->bytes_read) {
  453. kfree_skb(skb);
  454. continue;
  455. }
  456. if (!tipc_sk_type_connectionless(sk) &&
  457. sk->sk_state != TIPC_DISCONNECTING) {
  458. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  459. tipc_node_remove_conn(net, dnode, tsk->portid);
  460. }
  461. tipc_sk_respond(sk, skb, error);
  462. }
  463. if (tipc_sk_type_connectionless(sk))
  464. return;
  465. if (sk->sk_state != TIPC_DISCONNECTING) {
  466. skb = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE,
  467. TIPC_CONN_MSG, SHORT_H_SIZE, 0, dnode,
  468. tsk_own_node(tsk), tsk_peer_port(tsk),
  469. tsk->portid, error);
  470. if (skb)
  471. tipc_node_xmit_skb(net, skb, dnode, tsk->portid);
  472. tipc_node_remove_conn(net, dnode, tsk->portid);
  473. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  474. }
  475. }
  476. /**
  477. * tipc_release - destroy a TIPC socket
  478. * @sock: socket to destroy
  479. *
  480. * This routine cleans up any messages that are still queued on the socket.
  481. * For DGRAM and RDM socket types, all queued messages are rejected.
  482. * For SEQPACKET and STREAM socket types, the first message is rejected
  483. * and any others are discarded. (If the first message on a STREAM socket
  484. * is partially-read, it is discarded and the next one is rejected instead.)
  485. *
  486. * NOTE: Rejected messages are not necessarily returned to the sender! They
  487. * are returned or discarded according to the "destination droppable" setting
  488. * specified for the message by the sender.
  489. *
  490. * Returns 0 on success, errno otherwise
  491. */
  492. static int tipc_release(struct socket *sock)
  493. {
  494. struct sock *sk = sock->sk;
  495. struct tipc_sock *tsk;
  496. /*
  497. * Exit if socket isn't fully initialized (occurs when a failed accept()
  498. * releases a pre-allocated child socket that was never used)
  499. */
  500. if (sk == NULL)
  501. return 0;
  502. tsk = tipc_sk(sk);
  503. lock_sock(sk);
  504. __tipc_shutdown(sock, TIPC_ERR_NO_PORT);
  505. sk->sk_shutdown = SHUTDOWN_MASK;
  506. tipc_sk_leave(tsk);
  507. tipc_sk_withdraw(tsk, 0, NULL);
  508. sk_stop_timer(sk, &sk->sk_timer);
  509. tipc_sk_remove(tsk);
  510. /* Reject any messages that accumulated in backlog queue */
  511. release_sock(sk);
  512. tipc_dest_list_purge(&tsk->cong_links);
  513. tsk->cong_link_cnt = 0;
  514. call_rcu(&tsk->rcu, tipc_sk_callback);
  515. sock->sk = NULL;
  516. return 0;
  517. }
  518. /**
  519. * tipc_bind - associate or disassocate TIPC name(s) with a socket
  520. * @sock: socket structure
  521. * @uaddr: socket address describing name(s) and desired operation
  522. * @uaddr_len: size of socket address data structure
  523. *
  524. * Name and name sequence binding is indicated using a positive scope value;
  525. * a negative scope value unbinds the specified name. Specifying no name
  526. * (i.e. a socket address length of 0) unbinds all names from the socket.
  527. *
  528. * Returns 0 on success, errno otherwise
  529. *
  530. * NOTE: This routine doesn't need to take the socket lock since it doesn't
  531. * access any non-constant socket information.
  532. */
  533. static int tipc_bind(struct socket *sock, struct sockaddr *uaddr,
  534. int uaddr_len)
  535. {
  536. struct sock *sk = sock->sk;
  537. struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
  538. struct tipc_sock *tsk = tipc_sk(sk);
  539. int res = -EINVAL;
  540. lock_sock(sk);
  541. if (unlikely(!uaddr_len)) {
  542. res = tipc_sk_withdraw(tsk, 0, NULL);
  543. goto exit;
  544. }
  545. if (tsk->group) {
  546. res = -EACCES;
  547. goto exit;
  548. }
  549. if (uaddr_len < sizeof(struct sockaddr_tipc)) {
  550. res = -EINVAL;
  551. goto exit;
  552. }
  553. if (addr->family != AF_TIPC) {
  554. res = -EAFNOSUPPORT;
  555. goto exit;
  556. }
  557. if (addr->addrtype == TIPC_ADDR_NAME)
  558. addr->addr.nameseq.upper = addr->addr.nameseq.lower;
  559. else if (addr->addrtype != TIPC_ADDR_NAMESEQ) {
  560. res = -EAFNOSUPPORT;
  561. goto exit;
  562. }
  563. if ((addr->addr.nameseq.type < TIPC_RESERVED_TYPES) &&
  564. (addr->addr.nameseq.type != TIPC_TOP_SRV) &&
  565. (addr->addr.nameseq.type != TIPC_CFG_SRV)) {
  566. res = -EACCES;
  567. goto exit;
  568. }
  569. res = (addr->scope > 0) ?
  570. tipc_sk_publish(tsk, addr->scope, &addr->addr.nameseq) :
  571. tipc_sk_withdraw(tsk, -addr->scope, &addr->addr.nameseq);
  572. exit:
  573. release_sock(sk);
  574. return res;
  575. }
  576. /**
  577. * tipc_getname - get port ID of socket or peer socket
  578. * @sock: socket structure
  579. * @uaddr: area for returned socket address
  580. * @uaddr_len: area for returned length of socket address
  581. * @peer: 0 = own ID, 1 = current peer ID, 2 = current/former peer ID
  582. *
  583. * Returns 0 on success, errno otherwise
  584. *
  585. * NOTE: This routine doesn't need to take the socket lock since it only
  586. * accesses socket information that is unchanging (or which changes in
  587. * a completely predictable manner).
  588. */
  589. static int tipc_getname(struct socket *sock, struct sockaddr *uaddr,
  590. int *uaddr_len, int peer)
  591. {
  592. struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
  593. struct sock *sk = sock->sk;
  594. struct tipc_sock *tsk = tipc_sk(sk);
  595. struct tipc_net *tn = net_generic(sock_net(sock->sk), tipc_net_id);
  596. memset(addr, 0, sizeof(*addr));
  597. if (peer) {
  598. if ((!tipc_sk_connected(sk)) &&
  599. ((peer != 2) || (sk->sk_state != TIPC_DISCONNECTING)))
  600. return -ENOTCONN;
  601. addr->addr.id.ref = tsk_peer_port(tsk);
  602. addr->addr.id.node = tsk_peer_node(tsk);
  603. } else {
  604. addr->addr.id.ref = tsk->portid;
  605. addr->addr.id.node = tn->own_addr;
  606. }
  607. *uaddr_len = sizeof(*addr);
  608. addr->addrtype = TIPC_ADDR_ID;
  609. addr->family = AF_TIPC;
  610. addr->scope = 0;
  611. addr->addr.name.domain = 0;
  612. return 0;
  613. }
  614. /**
  615. * tipc_poll - read and possibly block on pollmask
  616. * @file: file structure associated with the socket
  617. * @sock: socket for which to calculate the poll bits
  618. * @wait: ???
  619. *
  620. * Returns pollmask value
  621. *
  622. * COMMENTARY:
  623. * It appears that the usual socket locking mechanisms are not useful here
  624. * since the pollmask info is potentially out-of-date the moment this routine
  625. * exits. TCP and other protocols seem to rely on higher level poll routines
  626. * to handle any preventable race conditions, so TIPC will do the same ...
  627. *
  628. * IMPORTANT: The fact that a read or write operation is indicated does NOT
  629. * imply that the operation will succeed, merely that it should be performed
  630. * and will not block.
  631. */
  632. static unsigned int tipc_poll(struct file *file, struct socket *sock,
  633. poll_table *wait)
  634. {
  635. struct sock *sk = sock->sk;
  636. struct tipc_sock *tsk = tipc_sk(sk);
  637. struct tipc_group *grp;
  638. u32 revents = 0;
  639. sock_poll_wait(file, sk_sleep(sk), wait);
  640. if (sk->sk_shutdown & RCV_SHUTDOWN)
  641. revents |= POLLRDHUP | POLLIN | POLLRDNORM;
  642. if (sk->sk_shutdown == SHUTDOWN_MASK)
  643. revents |= POLLHUP;
  644. switch (sk->sk_state) {
  645. case TIPC_ESTABLISHED:
  646. case TIPC_CONNECTING:
  647. if (!tsk->cong_link_cnt && !tsk_conn_cong(tsk))
  648. revents |= POLLOUT;
  649. /* fall thru' */
  650. case TIPC_LISTEN:
  651. if (!skb_queue_empty(&sk->sk_receive_queue))
  652. revents |= POLLIN | POLLRDNORM;
  653. break;
  654. case TIPC_OPEN:
  655. grp = tsk->group;
  656. if ((!grp || tipc_group_is_open(grp)) && !tsk->cong_link_cnt)
  657. revents |= POLLOUT;
  658. if (!tipc_sk_type_connectionless(sk))
  659. break;
  660. if (skb_queue_empty(&sk->sk_receive_queue))
  661. break;
  662. revents |= POLLIN | POLLRDNORM;
  663. break;
  664. case TIPC_DISCONNECTING:
  665. revents = POLLIN | POLLRDNORM | POLLHUP;
  666. break;
  667. }
  668. return revents;
  669. }
  670. /**
  671. * tipc_sendmcast - send multicast message
  672. * @sock: socket structure
  673. * @seq: destination address
  674. * @msg: message to send
  675. * @dlen: length of data to send
  676. * @timeout: timeout to wait for wakeup
  677. *
  678. * Called from function tipc_sendmsg(), which has done all sanity checks
  679. * Returns the number of bytes sent on success, or errno
  680. */
  681. static int tipc_sendmcast(struct socket *sock, struct tipc_name_seq *seq,
  682. struct msghdr *msg, size_t dlen, long timeout)
  683. {
  684. struct sock *sk = sock->sk;
  685. struct tipc_sock *tsk = tipc_sk(sk);
  686. struct tipc_msg *hdr = &tsk->phdr;
  687. struct net *net = sock_net(sk);
  688. int mtu = tipc_bcast_get_mtu(net);
  689. struct tipc_mc_method *method = &tsk->mc_method;
  690. u32 domain = addr_domain(net, TIPC_CLUSTER_SCOPE);
  691. struct sk_buff_head pkts;
  692. struct tipc_nlist dsts;
  693. int rc;
  694. if (tsk->group)
  695. return -EACCES;
  696. /* Block or return if any destination link is congested */
  697. rc = tipc_wait_for_cond(sock, &timeout, !tsk->cong_link_cnt);
  698. if (unlikely(rc))
  699. return rc;
  700. /* Lookup destination nodes */
  701. tipc_nlist_init(&dsts, tipc_own_addr(net));
  702. tipc_nametbl_lookup_dst_nodes(net, seq->type, seq->lower,
  703. seq->upper, domain, &dsts);
  704. if (!dsts.local && !dsts.remote)
  705. return -EHOSTUNREACH;
  706. /* Build message header */
  707. msg_set_type(hdr, TIPC_MCAST_MSG);
  708. msg_set_hdr_sz(hdr, MCAST_H_SIZE);
  709. msg_set_lookup_scope(hdr, TIPC_CLUSTER_SCOPE);
  710. msg_set_destport(hdr, 0);
  711. msg_set_destnode(hdr, 0);
  712. msg_set_nametype(hdr, seq->type);
  713. msg_set_namelower(hdr, seq->lower);
  714. msg_set_nameupper(hdr, seq->upper);
  715. /* Build message as chain of buffers */
  716. skb_queue_head_init(&pkts);
  717. rc = tipc_msg_build(hdr, msg, 0, dlen, mtu, &pkts);
  718. /* Send message if build was successful */
  719. if (unlikely(rc == dlen))
  720. rc = tipc_mcast_xmit(net, &pkts, method, &dsts,
  721. &tsk->cong_link_cnt);
  722. tipc_nlist_purge(&dsts);
  723. return rc ? rc : dlen;
  724. }
  725. /**
  726. * tipc_send_group_msg - send a message to a member in the group
  727. * @net: network namespace
  728. * @m: message to send
  729. * @mb: group member
  730. * @dnode: destination node
  731. * @dport: destination port
  732. * @dlen: total length of message data
  733. */
  734. static int tipc_send_group_msg(struct net *net, struct tipc_sock *tsk,
  735. struct msghdr *m, struct tipc_member *mb,
  736. u32 dnode, u32 dport, int dlen)
  737. {
  738. u16 bc_snd_nxt = tipc_group_bc_snd_nxt(tsk->group);
  739. struct tipc_mc_method *method = &tsk->mc_method;
  740. int blks = tsk_blocks(GROUP_H_SIZE + dlen);
  741. struct tipc_msg *hdr = &tsk->phdr;
  742. struct sk_buff_head pkts;
  743. int mtu, rc;
  744. /* Complete message header */
  745. msg_set_type(hdr, TIPC_GRP_UCAST_MSG);
  746. msg_set_hdr_sz(hdr, GROUP_H_SIZE);
  747. msg_set_destport(hdr, dport);
  748. msg_set_destnode(hdr, dnode);
  749. msg_set_grp_bc_seqno(hdr, bc_snd_nxt);
  750. /* Build message as chain of buffers */
  751. skb_queue_head_init(&pkts);
  752. mtu = tipc_node_get_mtu(net, dnode, tsk->portid);
  753. rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
  754. if (unlikely(rc != dlen))
  755. return rc;
  756. /* Send message */
  757. rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
  758. if (unlikely(rc == -ELINKCONG)) {
  759. tipc_dest_push(&tsk->cong_links, dnode, 0);
  760. tsk->cong_link_cnt++;
  761. }
  762. /* Update send window */
  763. tipc_group_update_member(mb, blks);
  764. /* A broadcast sent within next EXPIRE period must follow same path */
  765. method->rcast = true;
  766. method->mandatory = true;
  767. return dlen;
  768. }
  769. /**
  770. * tipc_send_group_unicast - send message to a member in the group
  771. * @sock: socket structure
  772. * @m: message to send
  773. * @dlen: total length of message data
  774. * @timeout: timeout to wait for wakeup
  775. *
  776. * Called from function tipc_sendmsg(), which has done all sanity checks
  777. * Returns the number of bytes sent on success, or errno
  778. */
  779. static int tipc_send_group_unicast(struct socket *sock, struct msghdr *m,
  780. int dlen, long timeout)
  781. {
  782. struct sock *sk = sock->sk;
  783. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  784. int blks = tsk_blocks(GROUP_H_SIZE + dlen);
  785. struct tipc_sock *tsk = tipc_sk(sk);
  786. struct tipc_group *grp = tsk->group;
  787. struct net *net = sock_net(sk);
  788. struct tipc_member *mb = NULL;
  789. u32 node, port;
  790. int rc;
  791. node = dest->addr.id.node;
  792. port = dest->addr.id.ref;
  793. if (!port && !node)
  794. return -EHOSTUNREACH;
  795. /* Block or return if destination link or member is congested */
  796. rc = tipc_wait_for_cond(sock, &timeout,
  797. !tipc_dest_find(&tsk->cong_links, node, 0) &&
  798. !tipc_group_cong(grp, node, port, blks, &mb));
  799. if (unlikely(rc))
  800. return rc;
  801. if (unlikely(!mb))
  802. return -EHOSTUNREACH;
  803. rc = tipc_send_group_msg(net, tsk, m, mb, node, port, dlen);
  804. return rc ? rc : dlen;
  805. }
  806. /**
  807. * tipc_send_group_anycast - send message to any member with given identity
  808. * @sock: socket structure
  809. * @m: message to send
  810. * @dlen: total length of message data
  811. * @timeout: timeout to wait for wakeup
  812. *
  813. * Called from function tipc_sendmsg(), which has done all sanity checks
  814. * Returns the number of bytes sent on success, or errno
  815. */
  816. static int tipc_send_group_anycast(struct socket *sock, struct msghdr *m,
  817. int dlen, long timeout)
  818. {
  819. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  820. struct sock *sk = sock->sk;
  821. struct tipc_sock *tsk = tipc_sk(sk);
  822. struct list_head *cong_links = &tsk->cong_links;
  823. int blks = tsk_blocks(GROUP_H_SIZE + dlen);
  824. struct tipc_group *grp = tsk->group;
  825. struct tipc_msg *hdr = &tsk->phdr;
  826. struct tipc_member *first = NULL;
  827. struct tipc_member *mbr = NULL;
  828. struct net *net = sock_net(sk);
  829. u32 node, port, exclude;
  830. struct list_head dsts;
  831. u32 type, inst, scope;
  832. int lookups = 0;
  833. int dstcnt, rc;
  834. bool cong;
  835. INIT_LIST_HEAD(&dsts);
  836. type = msg_nametype(hdr);
  837. inst = dest->addr.name.name.instance;
  838. scope = msg_lookup_scope(hdr);
  839. exclude = tipc_group_exclude(grp);
  840. while (++lookups < 4) {
  841. first = NULL;
  842. /* Look for a non-congested destination member, if any */
  843. while (1) {
  844. if (!tipc_nametbl_lookup(net, type, inst, scope, &dsts,
  845. &dstcnt, exclude, false))
  846. return -EHOSTUNREACH;
  847. tipc_dest_pop(&dsts, &node, &port);
  848. cong = tipc_group_cong(grp, node, port, blks, &mbr);
  849. if (!cong)
  850. break;
  851. if (mbr == first)
  852. break;
  853. if (!first)
  854. first = mbr;
  855. }
  856. /* Start over if destination was not in member list */
  857. if (unlikely(!mbr))
  858. continue;
  859. if (likely(!cong && !tipc_dest_find(cong_links, node, 0)))
  860. break;
  861. /* Block or return if destination link or member is congested */
  862. rc = tipc_wait_for_cond(sock, &timeout,
  863. !tipc_dest_find(cong_links, node, 0) &&
  864. !tipc_group_cong(grp, node, port,
  865. blks, &mbr));
  866. if (unlikely(rc))
  867. return rc;
  868. /* Send, unless destination disappeared while waiting */
  869. if (likely(mbr))
  870. break;
  871. }
  872. if (unlikely(lookups >= 4))
  873. return -EHOSTUNREACH;
  874. rc = tipc_send_group_msg(net, tsk, m, mbr, node, port, dlen);
  875. return rc ? rc : dlen;
  876. }
  877. /**
  878. * tipc_send_group_bcast - send message to all members in communication group
  879. * @sk: socket structure
  880. * @m: message to send
  881. * @dlen: total length of message data
  882. * @timeout: timeout to wait for wakeup
  883. *
  884. * Called from function tipc_sendmsg(), which has done all sanity checks
  885. * Returns the number of bytes sent on success, or errno
  886. */
  887. static int tipc_send_group_bcast(struct socket *sock, struct msghdr *m,
  888. int dlen, long timeout)
  889. {
  890. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  891. struct sock *sk = sock->sk;
  892. struct net *net = sock_net(sk);
  893. struct tipc_sock *tsk = tipc_sk(sk);
  894. struct tipc_group *grp = tsk->group;
  895. struct tipc_nlist *dsts = tipc_group_dests(grp);
  896. struct tipc_mc_method *method = &tsk->mc_method;
  897. bool ack = method->mandatory && method->rcast;
  898. int blks = tsk_blocks(MCAST_H_SIZE + dlen);
  899. struct tipc_msg *hdr = &tsk->phdr;
  900. int mtu = tipc_bcast_get_mtu(net);
  901. struct sk_buff_head pkts;
  902. int rc = -EHOSTUNREACH;
  903. if (!dsts->local && !dsts->remote)
  904. return -EHOSTUNREACH;
  905. /* Block or return if any destination link or member is congested */
  906. rc = tipc_wait_for_cond(sock, &timeout, !tsk->cong_link_cnt &&
  907. !tipc_group_bc_cong(grp, blks));
  908. if (unlikely(rc))
  909. return rc;
  910. /* Complete message header */
  911. if (dest) {
  912. msg_set_type(hdr, TIPC_GRP_MCAST_MSG);
  913. msg_set_nameinst(hdr, dest->addr.name.name.instance);
  914. } else {
  915. msg_set_type(hdr, TIPC_GRP_BCAST_MSG);
  916. msg_set_nameinst(hdr, 0);
  917. }
  918. msg_set_hdr_sz(hdr, GROUP_H_SIZE);
  919. msg_set_destport(hdr, 0);
  920. msg_set_destnode(hdr, 0);
  921. msg_set_grp_bc_seqno(hdr, tipc_group_bc_snd_nxt(grp));
  922. /* Avoid getting stuck with repeated forced replicasts */
  923. msg_set_grp_bc_ack_req(hdr, ack);
  924. /* Build message as chain of buffers */
  925. skb_queue_head_init(&pkts);
  926. rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
  927. if (unlikely(rc != dlen))
  928. return rc;
  929. /* Send message */
  930. rc = tipc_mcast_xmit(net, &pkts, method, dsts, &tsk->cong_link_cnt);
  931. if (unlikely(rc))
  932. return rc;
  933. /* Update broadcast sequence number and send windows */
  934. tipc_group_update_bc_members(tsk->group, blks, ack);
  935. /* Broadcast link is now free to choose method for next broadcast */
  936. method->mandatory = false;
  937. method->expires = jiffies;
  938. return dlen;
  939. }
  940. /**
  941. * tipc_send_group_mcast - send message to all members with given identity
  942. * @sock: socket structure
  943. * @m: message to send
  944. * @dlen: total length of message data
  945. * @timeout: timeout to wait for wakeup
  946. *
  947. * Called from function tipc_sendmsg(), which has done all sanity checks
  948. * Returns the number of bytes sent on success, or errno
  949. */
  950. static int tipc_send_group_mcast(struct socket *sock, struct msghdr *m,
  951. int dlen, long timeout)
  952. {
  953. struct sock *sk = sock->sk;
  954. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  955. struct tipc_sock *tsk = tipc_sk(sk);
  956. struct tipc_group *grp = tsk->group;
  957. struct tipc_msg *hdr = &tsk->phdr;
  958. struct net *net = sock_net(sk);
  959. u32 type, inst, scope, exclude;
  960. struct list_head dsts;
  961. u32 dstcnt;
  962. INIT_LIST_HEAD(&dsts);
  963. type = msg_nametype(hdr);
  964. inst = dest->addr.name.name.instance;
  965. scope = msg_lookup_scope(hdr);
  966. exclude = tipc_group_exclude(grp);
  967. if (!tipc_nametbl_lookup(net, type, inst, scope, &dsts,
  968. &dstcnt, exclude, true))
  969. return -EHOSTUNREACH;
  970. if (dstcnt == 1) {
  971. tipc_dest_pop(&dsts, &dest->addr.id.node, &dest->addr.id.ref);
  972. return tipc_send_group_unicast(sock, m, dlen, timeout);
  973. }
  974. tipc_dest_list_purge(&dsts);
  975. return tipc_send_group_bcast(sock, m, dlen, timeout);
  976. }
  977. /**
  978. * tipc_sk_mcast_rcv - Deliver multicast messages to all destination sockets
  979. * @arrvq: queue with arriving messages, to be cloned after destination lookup
  980. * @inputq: queue with cloned messages, delivered to socket after dest lookup
  981. *
  982. * Multi-threaded: parallel calls with reference to same queues may occur
  983. */
  984. void tipc_sk_mcast_rcv(struct net *net, struct sk_buff_head *arrvq,
  985. struct sk_buff_head *inputq)
  986. {
  987. u32 self = tipc_own_addr(net);
  988. u32 type, lower, upper, scope;
  989. struct sk_buff *skb, *_skb;
  990. u32 portid, oport, onode;
  991. struct sk_buff_head tmpq;
  992. struct list_head dports;
  993. struct tipc_msg *hdr;
  994. int user, mtyp, hlen;
  995. bool exact;
  996. __skb_queue_head_init(&tmpq);
  997. INIT_LIST_HEAD(&dports);
  998. skb = tipc_skb_peek(arrvq, &inputq->lock);
  999. for (; skb; skb = tipc_skb_peek(arrvq, &inputq->lock)) {
  1000. hdr = buf_msg(skb);
  1001. user = msg_user(hdr);
  1002. mtyp = msg_type(hdr);
  1003. hlen = skb_headroom(skb) + msg_hdr_sz(hdr);
  1004. oport = msg_origport(hdr);
  1005. onode = msg_orignode(hdr);
  1006. type = msg_nametype(hdr);
  1007. if (mtyp == TIPC_GRP_UCAST_MSG || user == GROUP_PROTOCOL) {
  1008. spin_lock_bh(&inputq->lock);
  1009. if (skb_peek(arrvq) == skb) {
  1010. __skb_dequeue(arrvq);
  1011. __skb_queue_tail(inputq, skb);
  1012. }
  1013. kfree_skb(skb);
  1014. spin_unlock_bh(&inputq->lock);
  1015. continue;
  1016. }
  1017. /* Group messages require exact scope match */
  1018. if (msg_in_group(hdr)) {
  1019. lower = 0;
  1020. upper = ~0;
  1021. scope = msg_lookup_scope(hdr);
  1022. exact = true;
  1023. } else {
  1024. /* TIPC_NODE_SCOPE means "any scope" in this context */
  1025. if (onode == self)
  1026. scope = TIPC_NODE_SCOPE;
  1027. else
  1028. scope = TIPC_CLUSTER_SCOPE;
  1029. exact = false;
  1030. lower = msg_namelower(hdr);
  1031. upper = msg_nameupper(hdr);
  1032. }
  1033. /* Create destination port list: */
  1034. tipc_nametbl_mc_lookup(net, type, lower, upper,
  1035. scope, exact, &dports);
  1036. /* Clone message per destination */
  1037. while (tipc_dest_pop(&dports, NULL, &portid)) {
  1038. _skb = __pskb_copy(skb, hlen, GFP_ATOMIC);
  1039. if (_skb) {
  1040. msg_set_destport(buf_msg(_skb), portid);
  1041. __skb_queue_tail(&tmpq, _skb);
  1042. continue;
  1043. }
  1044. pr_warn("Failed to clone mcast rcv buffer\n");
  1045. }
  1046. /* Append to inputq if not already done by other thread */
  1047. spin_lock_bh(&inputq->lock);
  1048. if (skb_peek(arrvq) == skb) {
  1049. skb_queue_splice_tail_init(&tmpq, inputq);
  1050. kfree_skb(__skb_dequeue(arrvq));
  1051. }
  1052. spin_unlock_bh(&inputq->lock);
  1053. __skb_queue_purge(&tmpq);
  1054. kfree_skb(skb);
  1055. }
  1056. tipc_sk_rcv(net, inputq);
  1057. }
  1058. /**
  1059. * tipc_sk_conn_proto_rcv - receive a connection mng protocol message
  1060. * @tsk: receiving socket
  1061. * @skb: pointer to message buffer.
  1062. */
  1063. static void tipc_sk_conn_proto_rcv(struct tipc_sock *tsk, struct sk_buff *skb,
  1064. struct sk_buff_head *xmitq)
  1065. {
  1066. struct tipc_msg *hdr = buf_msg(skb);
  1067. u32 onode = tsk_own_node(tsk);
  1068. struct sock *sk = &tsk->sk;
  1069. int mtyp = msg_type(hdr);
  1070. bool conn_cong;
  1071. /* Ignore if connection cannot be validated: */
  1072. if (!tsk_peer_msg(tsk, hdr))
  1073. goto exit;
  1074. if (unlikely(msg_errcode(hdr))) {
  1075. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  1076. tipc_node_remove_conn(sock_net(sk), tsk_peer_node(tsk),
  1077. tsk_peer_port(tsk));
  1078. sk->sk_state_change(sk);
  1079. goto exit;
  1080. }
  1081. tsk->probe_unacked = false;
  1082. if (mtyp == CONN_PROBE) {
  1083. msg_set_type(hdr, CONN_PROBE_REPLY);
  1084. if (tipc_msg_reverse(onode, &skb, TIPC_OK))
  1085. __skb_queue_tail(xmitq, skb);
  1086. return;
  1087. } else if (mtyp == CONN_ACK) {
  1088. conn_cong = tsk_conn_cong(tsk);
  1089. tsk->snt_unacked -= msg_conn_ack(hdr);
  1090. if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL)
  1091. tsk->snd_win = msg_adv_win(hdr);
  1092. if (conn_cong)
  1093. sk->sk_write_space(sk);
  1094. } else if (mtyp != CONN_PROBE_REPLY) {
  1095. pr_warn("Received unknown CONN_PROTO msg\n");
  1096. }
  1097. exit:
  1098. kfree_skb(skb);
  1099. }
  1100. /**
  1101. * tipc_sendmsg - send message in connectionless manner
  1102. * @sock: socket structure
  1103. * @m: message to send
  1104. * @dsz: amount of user data to be sent
  1105. *
  1106. * Message must have an destination specified explicitly.
  1107. * Used for SOCK_RDM and SOCK_DGRAM messages,
  1108. * and for 'SYN' messages on SOCK_SEQPACKET and SOCK_STREAM connections.
  1109. * (Note: 'SYN+' is prohibited on SOCK_STREAM.)
  1110. *
  1111. * Returns the number of bytes sent on success, or errno otherwise
  1112. */
  1113. static int tipc_sendmsg(struct socket *sock,
  1114. struct msghdr *m, size_t dsz)
  1115. {
  1116. struct sock *sk = sock->sk;
  1117. int ret;
  1118. lock_sock(sk);
  1119. ret = __tipc_sendmsg(sock, m, dsz);
  1120. release_sock(sk);
  1121. return ret;
  1122. }
  1123. static int __tipc_sendmsg(struct socket *sock, struct msghdr *m, size_t dlen)
  1124. {
  1125. struct sock *sk = sock->sk;
  1126. struct net *net = sock_net(sk);
  1127. struct tipc_sock *tsk = tipc_sk(sk);
  1128. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  1129. long timeout = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
  1130. struct list_head *clinks = &tsk->cong_links;
  1131. bool syn = !tipc_sk_type_connectionless(sk);
  1132. struct tipc_group *grp = tsk->group;
  1133. struct tipc_msg *hdr = &tsk->phdr;
  1134. struct tipc_name_seq *seq;
  1135. struct sk_buff_head pkts;
  1136. u32 type, inst, domain;
  1137. u32 dnode, dport;
  1138. int mtu, rc;
  1139. if (unlikely(dlen > TIPC_MAX_USER_MSG_SIZE))
  1140. return -EMSGSIZE;
  1141. if (likely(dest)) {
  1142. if (unlikely(m->msg_namelen < sizeof(*dest)))
  1143. return -EINVAL;
  1144. if (unlikely(dest->family != AF_TIPC))
  1145. return -EINVAL;
  1146. }
  1147. if (grp) {
  1148. if (!dest)
  1149. return tipc_send_group_bcast(sock, m, dlen, timeout);
  1150. if (dest->addrtype == TIPC_ADDR_NAME)
  1151. return tipc_send_group_anycast(sock, m, dlen, timeout);
  1152. if (dest->addrtype == TIPC_ADDR_ID)
  1153. return tipc_send_group_unicast(sock, m, dlen, timeout);
  1154. if (dest->addrtype == TIPC_ADDR_MCAST)
  1155. return tipc_send_group_mcast(sock, m, dlen, timeout);
  1156. return -EINVAL;
  1157. }
  1158. if (unlikely(!dest)) {
  1159. dest = &tsk->peer;
  1160. if (!syn || dest->family != AF_TIPC)
  1161. return -EDESTADDRREQ;
  1162. }
  1163. if (unlikely(syn)) {
  1164. if (sk->sk_state == TIPC_LISTEN)
  1165. return -EPIPE;
  1166. if (sk->sk_state != TIPC_OPEN)
  1167. return -EISCONN;
  1168. if (tsk->published)
  1169. return -EOPNOTSUPP;
  1170. if (dest->addrtype == TIPC_ADDR_NAME) {
  1171. tsk->conn_type = dest->addr.name.name.type;
  1172. tsk->conn_instance = dest->addr.name.name.instance;
  1173. }
  1174. }
  1175. seq = &dest->addr.nameseq;
  1176. if (dest->addrtype == TIPC_ADDR_MCAST)
  1177. return tipc_sendmcast(sock, seq, m, dlen, timeout);
  1178. if (dest->addrtype == TIPC_ADDR_NAME) {
  1179. type = dest->addr.name.name.type;
  1180. inst = dest->addr.name.name.instance;
  1181. domain = dest->addr.name.domain;
  1182. dnode = domain;
  1183. msg_set_type(hdr, TIPC_NAMED_MSG);
  1184. msg_set_hdr_sz(hdr, NAMED_H_SIZE);
  1185. msg_set_nametype(hdr, type);
  1186. msg_set_nameinst(hdr, inst);
  1187. msg_set_lookup_scope(hdr, tipc_addr_scope(domain));
  1188. dport = tipc_nametbl_translate(net, type, inst, &dnode);
  1189. msg_set_destnode(hdr, dnode);
  1190. msg_set_destport(hdr, dport);
  1191. if (unlikely(!dport && !dnode))
  1192. return -EHOSTUNREACH;
  1193. } else if (dest->addrtype == TIPC_ADDR_ID) {
  1194. dnode = dest->addr.id.node;
  1195. msg_set_type(hdr, TIPC_DIRECT_MSG);
  1196. msg_set_lookup_scope(hdr, 0);
  1197. msg_set_destnode(hdr, dnode);
  1198. msg_set_destport(hdr, dest->addr.id.ref);
  1199. msg_set_hdr_sz(hdr, BASIC_H_SIZE);
  1200. }
  1201. /* Block or return if destination link is congested */
  1202. rc = tipc_wait_for_cond(sock, &timeout,
  1203. !tipc_dest_find(clinks, dnode, 0));
  1204. if (unlikely(rc))
  1205. return rc;
  1206. skb_queue_head_init(&pkts);
  1207. mtu = tipc_node_get_mtu(net, dnode, tsk->portid);
  1208. rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
  1209. if (unlikely(rc != dlen))
  1210. return rc;
  1211. rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
  1212. if (unlikely(rc == -ELINKCONG)) {
  1213. tipc_dest_push(clinks, dnode, 0);
  1214. tsk->cong_link_cnt++;
  1215. rc = 0;
  1216. }
  1217. if (unlikely(syn && !rc))
  1218. tipc_set_sk_state(sk, TIPC_CONNECTING);
  1219. return rc ? rc : dlen;
  1220. }
  1221. /**
  1222. * tipc_sendstream - send stream-oriented data
  1223. * @sock: socket structure
  1224. * @m: data to send
  1225. * @dsz: total length of data to be transmitted
  1226. *
  1227. * Used for SOCK_STREAM data.
  1228. *
  1229. * Returns the number of bytes sent on success (or partial success),
  1230. * or errno if no data sent
  1231. */
  1232. static int tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dsz)
  1233. {
  1234. struct sock *sk = sock->sk;
  1235. int ret;
  1236. lock_sock(sk);
  1237. ret = __tipc_sendstream(sock, m, dsz);
  1238. release_sock(sk);
  1239. return ret;
  1240. }
  1241. static int __tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dlen)
  1242. {
  1243. struct sock *sk = sock->sk;
  1244. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  1245. long timeout = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
  1246. struct tipc_sock *tsk = tipc_sk(sk);
  1247. struct tipc_msg *hdr = &tsk->phdr;
  1248. struct net *net = sock_net(sk);
  1249. struct sk_buff_head pkts;
  1250. u32 dnode = tsk_peer_node(tsk);
  1251. int send, sent = 0;
  1252. int rc = 0;
  1253. skb_queue_head_init(&pkts);
  1254. if (unlikely(dlen > INT_MAX))
  1255. return -EMSGSIZE;
  1256. /* Handle implicit connection setup */
  1257. if (unlikely(dest)) {
  1258. rc = __tipc_sendmsg(sock, m, dlen);
  1259. if (dlen && (dlen == rc))
  1260. tsk->snt_unacked = tsk_inc(tsk, dlen + msg_hdr_sz(hdr));
  1261. return rc;
  1262. }
  1263. do {
  1264. rc = tipc_wait_for_cond(sock, &timeout,
  1265. (!tsk->cong_link_cnt &&
  1266. !tsk_conn_cong(tsk) &&
  1267. tipc_sk_connected(sk)));
  1268. if (unlikely(rc))
  1269. break;
  1270. send = min_t(size_t, dlen - sent, TIPC_MAX_USER_MSG_SIZE);
  1271. rc = tipc_msg_build(hdr, m, sent, send, tsk->max_pkt, &pkts);
  1272. if (unlikely(rc != send))
  1273. break;
  1274. rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
  1275. if (unlikely(rc == -ELINKCONG)) {
  1276. tsk->cong_link_cnt = 1;
  1277. rc = 0;
  1278. }
  1279. if (likely(!rc)) {
  1280. tsk->snt_unacked += tsk_inc(tsk, send + MIN_H_SIZE);
  1281. sent += send;
  1282. }
  1283. } while (sent < dlen && !rc);
  1284. return sent ? sent : rc;
  1285. }
  1286. /**
  1287. * tipc_send_packet - send a connection-oriented message
  1288. * @sock: socket structure
  1289. * @m: message to send
  1290. * @dsz: length of data to be transmitted
  1291. *
  1292. * Used for SOCK_SEQPACKET messages.
  1293. *
  1294. * Returns the number of bytes sent on success, or errno otherwise
  1295. */
  1296. static int tipc_send_packet(struct socket *sock, struct msghdr *m, size_t dsz)
  1297. {
  1298. if (dsz > TIPC_MAX_USER_MSG_SIZE)
  1299. return -EMSGSIZE;
  1300. return tipc_sendstream(sock, m, dsz);
  1301. }
  1302. /* tipc_sk_finish_conn - complete the setup of a connection
  1303. */
  1304. static void tipc_sk_finish_conn(struct tipc_sock *tsk, u32 peer_port,
  1305. u32 peer_node)
  1306. {
  1307. struct sock *sk = &tsk->sk;
  1308. struct net *net = sock_net(sk);
  1309. struct tipc_msg *msg = &tsk->phdr;
  1310. msg_set_destnode(msg, peer_node);
  1311. msg_set_destport(msg, peer_port);
  1312. msg_set_type(msg, TIPC_CONN_MSG);
  1313. msg_set_lookup_scope(msg, 0);
  1314. msg_set_hdr_sz(msg, SHORT_H_SIZE);
  1315. sk_reset_timer(sk, &sk->sk_timer, jiffies + CONN_PROBING_INTV);
  1316. tipc_set_sk_state(sk, TIPC_ESTABLISHED);
  1317. tipc_node_add_conn(net, peer_node, tsk->portid, peer_port);
  1318. tsk->max_pkt = tipc_node_get_mtu(net, peer_node, tsk->portid);
  1319. tsk->peer_caps = tipc_node_get_capabilities(net, peer_node);
  1320. if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL)
  1321. return;
  1322. /* Fall back to message based flow control */
  1323. tsk->rcv_win = FLOWCTL_MSG_WIN;
  1324. tsk->snd_win = FLOWCTL_MSG_WIN;
  1325. }
  1326. /**
  1327. * tipc_sk_set_orig_addr - capture sender's address for received message
  1328. * @m: descriptor for message info
  1329. * @hdr: received message header
  1330. *
  1331. * Note: Address is not captured if not requested by receiver.
  1332. */
  1333. static void tipc_sk_set_orig_addr(struct msghdr *m, struct sk_buff *skb)
  1334. {
  1335. DECLARE_SOCKADDR(struct sockaddr_pair *, srcaddr, m->msg_name);
  1336. struct tipc_msg *hdr = buf_msg(skb);
  1337. if (!srcaddr)
  1338. return;
  1339. srcaddr->sock.family = AF_TIPC;
  1340. srcaddr->sock.addrtype = TIPC_ADDR_ID;
  1341. srcaddr->sock.addr.id.ref = msg_origport(hdr);
  1342. srcaddr->sock.addr.id.node = msg_orignode(hdr);
  1343. srcaddr->sock.addr.name.domain = 0;
  1344. srcaddr->sock.scope = 0;
  1345. m->msg_namelen = sizeof(struct sockaddr_tipc);
  1346. if (!msg_in_group(hdr))
  1347. return;
  1348. /* Group message users may also want to know sending member's id */
  1349. srcaddr->member.family = AF_TIPC;
  1350. srcaddr->member.addrtype = TIPC_ADDR_NAME;
  1351. srcaddr->member.addr.name.name.type = msg_nametype(hdr);
  1352. srcaddr->member.addr.name.name.instance = TIPC_SKB_CB(skb)->orig_member;
  1353. srcaddr->member.addr.name.domain = 0;
  1354. m->msg_namelen = sizeof(*srcaddr);
  1355. }
  1356. /**
  1357. * tipc_sk_anc_data_recv - optionally capture ancillary data for received message
  1358. * @m: descriptor for message info
  1359. * @msg: received message header
  1360. * @tsk: TIPC port associated with message
  1361. *
  1362. * Note: Ancillary data is not captured if not requested by receiver.
  1363. *
  1364. * Returns 0 if successful, otherwise errno
  1365. */
  1366. static int tipc_sk_anc_data_recv(struct msghdr *m, struct tipc_msg *msg,
  1367. struct tipc_sock *tsk)
  1368. {
  1369. u32 anc_data[3];
  1370. u32 err;
  1371. u32 dest_type;
  1372. int has_name;
  1373. int res;
  1374. if (likely(m->msg_controllen == 0))
  1375. return 0;
  1376. /* Optionally capture errored message object(s) */
  1377. err = msg ? msg_errcode(msg) : 0;
  1378. if (unlikely(err)) {
  1379. anc_data[0] = err;
  1380. anc_data[1] = msg_data_sz(msg);
  1381. res = put_cmsg(m, SOL_TIPC, TIPC_ERRINFO, 8, anc_data);
  1382. if (res)
  1383. return res;
  1384. if (anc_data[1]) {
  1385. res = put_cmsg(m, SOL_TIPC, TIPC_RETDATA, anc_data[1],
  1386. msg_data(msg));
  1387. if (res)
  1388. return res;
  1389. }
  1390. }
  1391. /* Optionally capture message destination object */
  1392. dest_type = msg ? msg_type(msg) : TIPC_DIRECT_MSG;
  1393. switch (dest_type) {
  1394. case TIPC_NAMED_MSG:
  1395. has_name = 1;
  1396. anc_data[0] = msg_nametype(msg);
  1397. anc_data[1] = msg_namelower(msg);
  1398. anc_data[2] = msg_namelower(msg);
  1399. break;
  1400. case TIPC_MCAST_MSG:
  1401. has_name = 1;
  1402. anc_data[0] = msg_nametype(msg);
  1403. anc_data[1] = msg_namelower(msg);
  1404. anc_data[2] = msg_nameupper(msg);
  1405. break;
  1406. case TIPC_CONN_MSG:
  1407. has_name = (tsk->conn_type != 0);
  1408. anc_data[0] = tsk->conn_type;
  1409. anc_data[1] = tsk->conn_instance;
  1410. anc_data[2] = tsk->conn_instance;
  1411. break;
  1412. default:
  1413. has_name = 0;
  1414. }
  1415. if (has_name) {
  1416. res = put_cmsg(m, SOL_TIPC, TIPC_DESTNAME, 12, anc_data);
  1417. if (res)
  1418. return res;
  1419. }
  1420. return 0;
  1421. }
  1422. static void tipc_sk_send_ack(struct tipc_sock *tsk)
  1423. {
  1424. struct sock *sk = &tsk->sk;
  1425. struct net *net = sock_net(sk);
  1426. struct sk_buff *skb = NULL;
  1427. struct tipc_msg *msg;
  1428. u32 peer_port = tsk_peer_port(tsk);
  1429. u32 dnode = tsk_peer_node(tsk);
  1430. if (!tipc_sk_connected(sk))
  1431. return;
  1432. skb = tipc_msg_create(CONN_MANAGER, CONN_ACK, INT_H_SIZE, 0,
  1433. dnode, tsk_own_node(tsk), peer_port,
  1434. tsk->portid, TIPC_OK);
  1435. if (!skb)
  1436. return;
  1437. msg = buf_msg(skb);
  1438. msg_set_conn_ack(msg, tsk->rcv_unacked);
  1439. tsk->rcv_unacked = 0;
  1440. /* Adjust to and advertize the correct window limit */
  1441. if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL) {
  1442. tsk->rcv_win = tsk_adv_blocks(tsk->sk.sk_rcvbuf);
  1443. msg_set_adv_win(msg, tsk->rcv_win);
  1444. }
  1445. tipc_node_xmit_skb(net, skb, dnode, msg_link_selector(msg));
  1446. }
  1447. static int tipc_wait_for_rcvmsg(struct socket *sock, long *timeop)
  1448. {
  1449. struct sock *sk = sock->sk;
  1450. DEFINE_WAIT(wait);
  1451. long timeo = *timeop;
  1452. int err = sock_error(sk);
  1453. if (err)
  1454. return err;
  1455. for (;;) {
  1456. prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
  1457. if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
  1458. if (sk->sk_shutdown & RCV_SHUTDOWN) {
  1459. err = -ENOTCONN;
  1460. break;
  1461. }
  1462. release_sock(sk);
  1463. timeo = schedule_timeout(timeo);
  1464. lock_sock(sk);
  1465. }
  1466. err = 0;
  1467. if (!skb_queue_empty(&sk->sk_receive_queue))
  1468. break;
  1469. err = -EAGAIN;
  1470. if (!timeo)
  1471. break;
  1472. err = sock_intr_errno(timeo);
  1473. if (signal_pending(current))
  1474. break;
  1475. err = sock_error(sk);
  1476. if (err)
  1477. break;
  1478. }
  1479. finish_wait(sk_sleep(sk), &wait);
  1480. *timeop = timeo;
  1481. return err;
  1482. }
  1483. /**
  1484. * tipc_recvmsg - receive packet-oriented message
  1485. * @m: descriptor for message info
  1486. * @buflen: length of user buffer area
  1487. * @flags: receive flags
  1488. *
  1489. * Used for SOCK_DGRAM, SOCK_RDM, and SOCK_SEQPACKET messages.
  1490. * If the complete message doesn't fit in user area, truncate it.
  1491. *
  1492. * Returns size of returned message data, errno otherwise
  1493. */
  1494. static int tipc_recvmsg(struct socket *sock, struct msghdr *m,
  1495. size_t buflen, int flags)
  1496. {
  1497. struct sock *sk = sock->sk;
  1498. bool connected = !tipc_sk_type_connectionless(sk);
  1499. struct tipc_sock *tsk = tipc_sk(sk);
  1500. int rc, err, hlen, dlen, copy;
  1501. struct sk_buff_head xmitq;
  1502. struct tipc_msg *hdr;
  1503. struct sk_buff *skb;
  1504. bool grp_evt;
  1505. long timeout;
  1506. /* Catch invalid receive requests */
  1507. if (unlikely(!buflen))
  1508. return -EINVAL;
  1509. lock_sock(sk);
  1510. if (unlikely(connected && sk->sk_state == TIPC_OPEN)) {
  1511. rc = -ENOTCONN;
  1512. goto exit;
  1513. }
  1514. timeout = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
  1515. /* Step rcv queue to first msg with data or error; wait if necessary */
  1516. do {
  1517. rc = tipc_wait_for_rcvmsg(sock, &timeout);
  1518. if (unlikely(rc))
  1519. goto exit;
  1520. skb = skb_peek(&sk->sk_receive_queue);
  1521. hdr = buf_msg(skb);
  1522. dlen = msg_data_sz(hdr);
  1523. hlen = msg_hdr_sz(hdr);
  1524. err = msg_errcode(hdr);
  1525. grp_evt = msg_is_grp_evt(hdr);
  1526. if (likely(dlen || err))
  1527. break;
  1528. tsk_advance_rx_queue(sk);
  1529. } while (1);
  1530. /* Collect msg meta data, including error code and rejected data */
  1531. tipc_sk_set_orig_addr(m, skb);
  1532. rc = tipc_sk_anc_data_recv(m, hdr, tsk);
  1533. if (unlikely(rc))
  1534. goto exit;
  1535. /* Capture data if non-error msg, otherwise just set return value */
  1536. if (likely(!err)) {
  1537. copy = min_t(int, dlen, buflen);
  1538. if (unlikely(copy != dlen))
  1539. m->msg_flags |= MSG_TRUNC;
  1540. rc = skb_copy_datagram_msg(skb, hlen, m, copy);
  1541. } else {
  1542. copy = 0;
  1543. rc = 0;
  1544. if (err != TIPC_CONN_SHUTDOWN && connected && !m->msg_control)
  1545. rc = -ECONNRESET;
  1546. }
  1547. if (unlikely(rc))
  1548. goto exit;
  1549. /* Mark message as group event if applicable */
  1550. if (unlikely(grp_evt)) {
  1551. if (msg_grp_evt(hdr) == TIPC_WITHDRAWN)
  1552. m->msg_flags |= MSG_EOR;
  1553. m->msg_flags |= MSG_OOB;
  1554. copy = 0;
  1555. }
  1556. /* Caption of data or error code/rejected data was successful */
  1557. if (unlikely(flags & MSG_PEEK))
  1558. goto exit;
  1559. /* Send group flow control advertisement when applicable */
  1560. if (tsk->group && msg_in_group(hdr) && !grp_evt) {
  1561. skb_queue_head_init(&xmitq);
  1562. tipc_group_update_rcv_win(tsk->group, tsk_blocks(hlen + dlen),
  1563. msg_orignode(hdr), msg_origport(hdr),
  1564. &xmitq);
  1565. tipc_node_distr_xmit(sock_net(sk), &xmitq);
  1566. }
  1567. tsk_advance_rx_queue(sk);
  1568. if (likely(!connected))
  1569. goto exit;
  1570. /* Send connection flow control advertisement when applicable */
  1571. tsk->rcv_unacked += tsk_inc(tsk, hlen + dlen);
  1572. if (tsk->rcv_unacked >= tsk->rcv_win / TIPC_ACK_RATE)
  1573. tipc_sk_send_ack(tsk);
  1574. exit:
  1575. release_sock(sk);
  1576. return rc ? rc : copy;
  1577. }
  1578. /**
  1579. * tipc_recvstream - receive stream-oriented data
  1580. * @m: descriptor for message info
  1581. * @buflen: total size of user buffer area
  1582. * @flags: receive flags
  1583. *
  1584. * Used for SOCK_STREAM messages only. If not enough data is available
  1585. * will optionally wait for more; never truncates data.
  1586. *
  1587. * Returns size of returned message data, errno otherwise
  1588. */
  1589. static int tipc_recvstream(struct socket *sock, struct msghdr *m,
  1590. size_t buflen, int flags)
  1591. {
  1592. struct sock *sk = sock->sk;
  1593. struct tipc_sock *tsk = tipc_sk(sk);
  1594. struct sk_buff *skb;
  1595. struct tipc_msg *hdr;
  1596. struct tipc_skb_cb *skb_cb;
  1597. bool peek = flags & MSG_PEEK;
  1598. int offset, required, copy, copied = 0;
  1599. int hlen, dlen, err, rc;
  1600. long timeout;
  1601. /* Catch invalid receive attempts */
  1602. if (unlikely(!buflen))
  1603. return -EINVAL;
  1604. lock_sock(sk);
  1605. if (unlikely(sk->sk_state == TIPC_OPEN)) {
  1606. rc = -ENOTCONN;
  1607. goto exit;
  1608. }
  1609. required = sock_rcvlowat(sk, flags & MSG_WAITALL, buflen);
  1610. timeout = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
  1611. do {
  1612. /* Look at first msg in receive queue; wait if necessary */
  1613. rc = tipc_wait_for_rcvmsg(sock, &timeout);
  1614. if (unlikely(rc))
  1615. break;
  1616. skb = skb_peek(&sk->sk_receive_queue);
  1617. skb_cb = TIPC_SKB_CB(skb);
  1618. hdr = buf_msg(skb);
  1619. dlen = msg_data_sz(hdr);
  1620. hlen = msg_hdr_sz(hdr);
  1621. err = msg_errcode(hdr);
  1622. /* Discard any empty non-errored (SYN-) message */
  1623. if (unlikely(!dlen && !err)) {
  1624. tsk_advance_rx_queue(sk);
  1625. continue;
  1626. }
  1627. /* Collect msg meta data, incl. error code and rejected data */
  1628. if (!copied) {
  1629. tipc_sk_set_orig_addr(m, skb);
  1630. rc = tipc_sk_anc_data_recv(m, hdr, tsk);
  1631. if (rc)
  1632. break;
  1633. }
  1634. /* Copy data if msg ok, otherwise return error/partial data */
  1635. if (likely(!err)) {
  1636. offset = skb_cb->bytes_read;
  1637. copy = min_t(int, dlen - offset, buflen - copied);
  1638. rc = skb_copy_datagram_msg(skb, hlen + offset, m, copy);
  1639. if (unlikely(rc))
  1640. break;
  1641. copied += copy;
  1642. offset += copy;
  1643. if (unlikely(offset < dlen)) {
  1644. if (!peek)
  1645. skb_cb->bytes_read = offset;
  1646. break;
  1647. }
  1648. } else {
  1649. rc = 0;
  1650. if ((err != TIPC_CONN_SHUTDOWN) && !m->msg_control)
  1651. rc = -ECONNRESET;
  1652. if (copied || rc)
  1653. break;
  1654. }
  1655. if (unlikely(peek))
  1656. break;
  1657. tsk_advance_rx_queue(sk);
  1658. /* Send connection flow control advertisement when applicable */
  1659. tsk->rcv_unacked += tsk_inc(tsk, hlen + dlen);
  1660. if (unlikely(tsk->rcv_unacked >= tsk->rcv_win / TIPC_ACK_RATE))
  1661. tipc_sk_send_ack(tsk);
  1662. /* Exit if all requested data or FIN/error received */
  1663. if (copied == buflen || err)
  1664. break;
  1665. } while (!skb_queue_empty(&sk->sk_receive_queue) || copied < required);
  1666. exit:
  1667. release_sock(sk);
  1668. return copied ? copied : rc;
  1669. }
  1670. /**
  1671. * tipc_write_space - wake up thread if port congestion is released
  1672. * @sk: socket
  1673. */
  1674. static void tipc_write_space(struct sock *sk)
  1675. {
  1676. struct socket_wq *wq;
  1677. rcu_read_lock();
  1678. wq = rcu_dereference(sk->sk_wq);
  1679. if (skwq_has_sleeper(wq))
  1680. wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
  1681. POLLWRNORM | POLLWRBAND);
  1682. rcu_read_unlock();
  1683. }
  1684. /**
  1685. * tipc_data_ready - wake up threads to indicate messages have been received
  1686. * @sk: socket
  1687. * @len: the length of messages
  1688. */
  1689. static void tipc_data_ready(struct sock *sk)
  1690. {
  1691. struct socket_wq *wq;
  1692. rcu_read_lock();
  1693. wq = rcu_dereference(sk->sk_wq);
  1694. if (skwq_has_sleeper(wq))
  1695. wake_up_interruptible_sync_poll(&wq->wait, POLLIN |
  1696. POLLRDNORM | POLLRDBAND);
  1697. rcu_read_unlock();
  1698. }
  1699. static void tipc_sock_destruct(struct sock *sk)
  1700. {
  1701. __skb_queue_purge(&sk->sk_receive_queue);
  1702. }
  1703. static void tipc_sk_proto_rcv(struct sock *sk,
  1704. struct sk_buff_head *inputq,
  1705. struct sk_buff_head *xmitq)
  1706. {
  1707. struct sk_buff *skb = __skb_dequeue(inputq);
  1708. struct tipc_sock *tsk = tipc_sk(sk);
  1709. struct tipc_msg *hdr = buf_msg(skb);
  1710. struct tipc_group *grp = tsk->group;
  1711. bool wakeup = false;
  1712. switch (msg_user(hdr)) {
  1713. case CONN_MANAGER:
  1714. tipc_sk_conn_proto_rcv(tsk, skb, xmitq);
  1715. return;
  1716. case SOCK_WAKEUP:
  1717. tipc_dest_del(&tsk->cong_links, msg_orignode(hdr), 0);
  1718. tsk->cong_link_cnt--;
  1719. wakeup = true;
  1720. break;
  1721. case GROUP_PROTOCOL:
  1722. tipc_group_proto_rcv(grp, &wakeup, hdr, inputq, xmitq);
  1723. break;
  1724. case TOP_SRV:
  1725. tipc_group_member_evt(tsk->group, &wakeup, &sk->sk_rcvbuf,
  1726. hdr, inputq, xmitq);
  1727. break;
  1728. default:
  1729. break;
  1730. }
  1731. if (wakeup)
  1732. sk->sk_write_space(sk);
  1733. kfree_skb(skb);
  1734. }
  1735. /**
  1736. * tipc_filter_connect - Handle incoming message for a connection-based socket
  1737. * @tsk: TIPC socket
  1738. * @skb: pointer to message buffer. Set to NULL if buffer is consumed
  1739. *
  1740. * Returns true if everything ok, false otherwise
  1741. */
  1742. static bool tipc_sk_filter_connect(struct tipc_sock *tsk, struct sk_buff *skb)
  1743. {
  1744. struct sock *sk = &tsk->sk;
  1745. struct net *net = sock_net(sk);
  1746. struct tipc_msg *hdr = buf_msg(skb);
  1747. u32 pport = msg_origport(hdr);
  1748. u32 pnode = msg_orignode(hdr);
  1749. if (unlikely(msg_mcast(hdr)))
  1750. return false;
  1751. switch (sk->sk_state) {
  1752. case TIPC_CONNECTING:
  1753. /* Accept only ACK or NACK message */
  1754. if (unlikely(!msg_connected(hdr))) {
  1755. if (pport != tsk_peer_port(tsk) ||
  1756. pnode != tsk_peer_node(tsk))
  1757. return false;
  1758. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  1759. sk->sk_err = ECONNREFUSED;
  1760. sk->sk_state_change(sk);
  1761. return true;
  1762. }
  1763. if (unlikely(msg_errcode(hdr))) {
  1764. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  1765. sk->sk_err = ECONNREFUSED;
  1766. sk->sk_state_change(sk);
  1767. return true;
  1768. }
  1769. if (unlikely(!msg_isdata(hdr))) {
  1770. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  1771. sk->sk_err = EINVAL;
  1772. sk->sk_state_change(sk);
  1773. return true;
  1774. }
  1775. tipc_sk_finish_conn(tsk, msg_origport(hdr), msg_orignode(hdr));
  1776. msg_set_importance(&tsk->phdr, msg_importance(hdr));
  1777. /* If 'ACK+' message, add to socket receive queue */
  1778. if (msg_data_sz(hdr))
  1779. return true;
  1780. /* If empty 'ACK-' message, wake up sleeping connect() */
  1781. sk->sk_data_ready(sk);
  1782. /* 'ACK-' message is neither accepted nor rejected: */
  1783. msg_set_dest_droppable(hdr, 1);
  1784. return false;
  1785. case TIPC_OPEN:
  1786. case TIPC_DISCONNECTING:
  1787. break;
  1788. case TIPC_LISTEN:
  1789. /* Accept only SYN message */
  1790. if (!msg_connected(hdr) && !(msg_errcode(hdr)))
  1791. return true;
  1792. break;
  1793. case TIPC_ESTABLISHED:
  1794. /* Accept only connection-based messages sent by peer */
  1795. if (unlikely(!tsk_peer_msg(tsk, hdr)))
  1796. return false;
  1797. if (unlikely(msg_errcode(hdr))) {
  1798. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  1799. /* Let timer expire on it's own */
  1800. tipc_node_remove_conn(net, tsk_peer_node(tsk),
  1801. tsk->portid);
  1802. sk->sk_state_change(sk);
  1803. }
  1804. return true;
  1805. default:
  1806. pr_err("Unknown sk_state %u\n", sk->sk_state);
  1807. }
  1808. return false;
  1809. }
  1810. /**
  1811. * rcvbuf_limit - get proper overload limit of socket receive queue
  1812. * @sk: socket
  1813. * @skb: message
  1814. *
  1815. * For connection oriented messages, irrespective of importance,
  1816. * default queue limit is 2 MB.
  1817. *
  1818. * For connectionless messages, queue limits are based on message
  1819. * importance as follows:
  1820. *
  1821. * TIPC_LOW_IMPORTANCE (2 MB)
  1822. * TIPC_MEDIUM_IMPORTANCE (4 MB)
  1823. * TIPC_HIGH_IMPORTANCE (8 MB)
  1824. * TIPC_CRITICAL_IMPORTANCE (16 MB)
  1825. *
  1826. * Returns overload limit according to corresponding message importance
  1827. */
  1828. static unsigned int rcvbuf_limit(struct sock *sk, struct sk_buff *skb)
  1829. {
  1830. struct tipc_sock *tsk = tipc_sk(sk);
  1831. struct tipc_msg *hdr = buf_msg(skb);
  1832. if (unlikely(msg_in_group(hdr)))
  1833. return sk->sk_rcvbuf;
  1834. if (unlikely(!msg_connected(hdr)))
  1835. return sk->sk_rcvbuf << msg_importance(hdr);
  1836. if (likely(tsk->peer_caps & TIPC_BLOCK_FLOWCTL))
  1837. return sk->sk_rcvbuf;
  1838. return FLOWCTL_MSG_LIM;
  1839. }
  1840. /**
  1841. * tipc_sk_filter_rcv - validate incoming message
  1842. * @sk: socket
  1843. * @skb: pointer to message.
  1844. *
  1845. * Enqueues message on receive queue if acceptable; optionally handles
  1846. * disconnect indication for a connected socket.
  1847. *
  1848. * Called with socket lock already taken
  1849. *
  1850. */
  1851. static void tipc_sk_filter_rcv(struct sock *sk, struct sk_buff *skb,
  1852. struct sk_buff_head *xmitq)
  1853. {
  1854. bool sk_conn = !tipc_sk_type_connectionless(sk);
  1855. struct tipc_sock *tsk = tipc_sk(sk);
  1856. struct tipc_group *grp = tsk->group;
  1857. struct tipc_msg *hdr = buf_msg(skb);
  1858. struct net *net = sock_net(sk);
  1859. struct sk_buff_head inputq;
  1860. int limit, err = TIPC_OK;
  1861. TIPC_SKB_CB(skb)->bytes_read = 0;
  1862. __skb_queue_head_init(&inputq);
  1863. __skb_queue_tail(&inputq, skb);
  1864. if (unlikely(!msg_isdata(hdr)))
  1865. tipc_sk_proto_rcv(sk, &inputq, xmitq);
  1866. if (unlikely(grp))
  1867. tipc_group_filter_msg(grp, &inputq, xmitq);
  1868. /* Validate and add to receive buffer if there is space */
  1869. while ((skb = __skb_dequeue(&inputq))) {
  1870. hdr = buf_msg(skb);
  1871. limit = rcvbuf_limit(sk, skb);
  1872. if ((sk_conn && !tipc_sk_filter_connect(tsk, skb)) ||
  1873. (!sk_conn && msg_connected(hdr)) ||
  1874. (!grp && msg_in_group(hdr)))
  1875. err = TIPC_ERR_NO_PORT;
  1876. else if (sk_rmem_alloc_get(sk) + skb->truesize >= limit)
  1877. err = TIPC_ERR_OVERLOAD;
  1878. if (unlikely(err)) {
  1879. tipc_skb_reject(net, err, skb, xmitq);
  1880. err = TIPC_OK;
  1881. continue;
  1882. }
  1883. __skb_queue_tail(&sk->sk_receive_queue, skb);
  1884. skb_set_owner_r(skb, sk);
  1885. sk->sk_data_ready(sk);
  1886. }
  1887. }
  1888. /**
  1889. * tipc_sk_backlog_rcv - handle incoming message from backlog queue
  1890. * @sk: socket
  1891. * @skb: message
  1892. *
  1893. * Caller must hold socket lock
  1894. */
  1895. static int tipc_sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
  1896. {
  1897. unsigned int before = sk_rmem_alloc_get(sk);
  1898. struct sk_buff_head xmitq;
  1899. unsigned int added;
  1900. __skb_queue_head_init(&xmitq);
  1901. tipc_sk_filter_rcv(sk, skb, &xmitq);
  1902. added = sk_rmem_alloc_get(sk) - before;
  1903. atomic_add(added, &tipc_sk(sk)->dupl_rcvcnt);
  1904. /* Send pending response/rejected messages, if any */
  1905. tipc_node_distr_xmit(sock_net(sk), &xmitq);
  1906. return 0;
  1907. }
  1908. /**
  1909. * tipc_sk_enqueue - extract all buffers with destination 'dport' from
  1910. * inputq and try adding them to socket or backlog queue
  1911. * @inputq: list of incoming buffers with potentially different destinations
  1912. * @sk: socket where the buffers should be enqueued
  1913. * @dport: port number for the socket
  1914. *
  1915. * Caller must hold socket lock
  1916. */
  1917. static void tipc_sk_enqueue(struct sk_buff_head *inputq, struct sock *sk,
  1918. u32 dport, struct sk_buff_head *xmitq)
  1919. {
  1920. unsigned long time_limit = jiffies + 2;
  1921. struct sk_buff *skb;
  1922. unsigned int lim;
  1923. atomic_t *dcnt;
  1924. u32 onode;
  1925. while (skb_queue_len(inputq)) {
  1926. if (unlikely(time_after_eq(jiffies, time_limit)))
  1927. return;
  1928. skb = tipc_skb_dequeue(inputq, dport);
  1929. if (unlikely(!skb))
  1930. return;
  1931. /* Add message directly to receive queue if possible */
  1932. if (!sock_owned_by_user(sk)) {
  1933. tipc_sk_filter_rcv(sk, skb, xmitq);
  1934. continue;
  1935. }
  1936. /* Try backlog, compensating for double-counted bytes */
  1937. dcnt = &tipc_sk(sk)->dupl_rcvcnt;
  1938. if (!sk->sk_backlog.len)
  1939. atomic_set(dcnt, 0);
  1940. lim = rcvbuf_limit(sk, skb) + atomic_read(dcnt);
  1941. if (likely(!sk_add_backlog(sk, skb, lim)))
  1942. continue;
  1943. /* Overload => reject message back to sender */
  1944. onode = tipc_own_addr(sock_net(sk));
  1945. if (tipc_msg_reverse(onode, &skb, TIPC_ERR_OVERLOAD))
  1946. __skb_queue_tail(xmitq, skb);
  1947. break;
  1948. }
  1949. }
  1950. /**
  1951. * tipc_sk_rcv - handle a chain of incoming buffers
  1952. * @inputq: buffer list containing the buffers
  1953. * Consumes all buffers in list until inputq is empty
  1954. * Note: may be called in multiple threads referring to the same queue
  1955. */
  1956. void tipc_sk_rcv(struct net *net, struct sk_buff_head *inputq)
  1957. {
  1958. struct sk_buff_head xmitq;
  1959. u32 dnode, dport = 0;
  1960. int err;
  1961. struct tipc_sock *tsk;
  1962. struct sock *sk;
  1963. struct sk_buff *skb;
  1964. __skb_queue_head_init(&xmitq);
  1965. while (skb_queue_len(inputq)) {
  1966. dport = tipc_skb_peek_port(inputq, dport);
  1967. tsk = tipc_sk_lookup(net, dport);
  1968. if (likely(tsk)) {
  1969. sk = &tsk->sk;
  1970. if (likely(spin_trylock_bh(&sk->sk_lock.slock))) {
  1971. tipc_sk_enqueue(inputq, sk, dport, &xmitq);
  1972. spin_unlock_bh(&sk->sk_lock.slock);
  1973. }
  1974. /* Send pending response/rejected messages, if any */
  1975. tipc_node_distr_xmit(sock_net(sk), &xmitq);
  1976. sock_put(sk);
  1977. continue;
  1978. }
  1979. /* No destination socket => dequeue skb if still there */
  1980. skb = tipc_skb_dequeue(inputq, dport);
  1981. if (!skb)
  1982. return;
  1983. /* Try secondary lookup if unresolved named message */
  1984. err = TIPC_ERR_NO_PORT;
  1985. if (tipc_msg_lookup_dest(net, skb, &err))
  1986. goto xmit;
  1987. /* Prepare for message rejection */
  1988. if (!tipc_msg_reverse(tipc_own_addr(net), &skb, err))
  1989. continue;
  1990. xmit:
  1991. dnode = msg_destnode(buf_msg(skb));
  1992. tipc_node_xmit_skb(net, skb, dnode, dport);
  1993. }
  1994. }
  1995. static int tipc_wait_for_connect(struct socket *sock, long *timeo_p)
  1996. {
  1997. DEFINE_WAIT_FUNC(wait, woken_wake_function);
  1998. struct sock *sk = sock->sk;
  1999. int done;
  2000. do {
  2001. int err = sock_error(sk);
  2002. if (err)
  2003. return err;
  2004. if (!*timeo_p)
  2005. return -ETIMEDOUT;
  2006. if (signal_pending(current))
  2007. return sock_intr_errno(*timeo_p);
  2008. add_wait_queue(sk_sleep(sk), &wait);
  2009. done = sk_wait_event(sk, timeo_p,
  2010. sk->sk_state != TIPC_CONNECTING, &wait);
  2011. remove_wait_queue(sk_sleep(sk), &wait);
  2012. } while (!done);
  2013. return 0;
  2014. }
  2015. /**
  2016. * tipc_connect - establish a connection to another TIPC port
  2017. * @sock: socket structure
  2018. * @dest: socket address for destination port
  2019. * @destlen: size of socket address data structure
  2020. * @flags: file-related flags associated with socket
  2021. *
  2022. * Returns 0 on success, errno otherwise
  2023. */
  2024. static int tipc_connect(struct socket *sock, struct sockaddr *dest,
  2025. int destlen, int flags)
  2026. {
  2027. struct sock *sk = sock->sk;
  2028. struct tipc_sock *tsk = tipc_sk(sk);
  2029. struct sockaddr_tipc *dst = (struct sockaddr_tipc *)dest;
  2030. struct msghdr m = {NULL,};
  2031. long timeout = (flags & O_NONBLOCK) ? 0 : tsk->conn_timeout;
  2032. int previous;
  2033. int res = 0;
  2034. if (destlen != sizeof(struct sockaddr_tipc))
  2035. return -EINVAL;
  2036. lock_sock(sk);
  2037. if (tsk->group) {
  2038. res = -EINVAL;
  2039. goto exit;
  2040. }
  2041. if (dst->family == AF_UNSPEC) {
  2042. memset(&tsk->peer, 0, sizeof(struct sockaddr_tipc));
  2043. if (!tipc_sk_type_connectionless(sk))
  2044. res = -EINVAL;
  2045. goto exit;
  2046. } else if (dst->family != AF_TIPC) {
  2047. res = -EINVAL;
  2048. }
  2049. if (dst->addrtype != TIPC_ADDR_ID && dst->addrtype != TIPC_ADDR_NAME)
  2050. res = -EINVAL;
  2051. if (res)
  2052. goto exit;
  2053. /* DGRAM/RDM connect(), just save the destaddr */
  2054. if (tipc_sk_type_connectionless(sk)) {
  2055. memcpy(&tsk->peer, dest, destlen);
  2056. goto exit;
  2057. }
  2058. previous = sk->sk_state;
  2059. switch (sk->sk_state) {
  2060. case TIPC_OPEN:
  2061. /* Send a 'SYN-' to destination */
  2062. m.msg_name = dest;
  2063. m.msg_namelen = destlen;
  2064. /* If connect is in non-blocking case, set MSG_DONTWAIT to
  2065. * indicate send_msg() is never blocked.
  2066. */
  2067. if (!timeout)
  2068. m.msg_flags = MSG_DONTWAIT;
  2069. res = __tipc_sendmsg(sock, &m, 0);
  2070. if ((res < 0) && (res != -EWOULDBLOCK))
  2071. goto exit;
  2072. /* Just entered TIPC_CONNECTING state; the only
  2073. * difference is that return value in non-blocking
  2074. * case is EINPROGRESS, rather than EALREADY.
  2075. */
  2076. res = -EINPROGRESS;
  2077. /* fall thru' */
  2078. case TIPC_CONNECTING:
  2079. if (!timeout) {
  2080. if (previous == TIPC_CONNECTING)
  2081. res = -EALREADY;
  2082. goto exit;
  2083. }
  2084. timeout = msecs_to_jiffies(timeout);
  2085. /* Wait until an 'ACK' or 'RST' arrives, or a timeout occurs */
  2086. res = tipc_wait_for_connect(sock, &timeout);
  2087. break;
  2088. case TIPC_ESTABLISHED:
  2089. res = -EISCONN;
  2090. break;
  2091. default:
  2092. res = -EINVAL;
  2093. }
  2094. exit:
  2095. release_sock(sk);
  2096. return res;
  2097. }
  2098. /**
  2099. * tipc_listen - allow socket to listen for incoming connections
  2100. * @sock: socket structure
  2101. * @len: (unused)
  2102. *
  2103. * Returns 0 on success, errno otherwise
  2104. */
  2105. static int tipc_listen(struct socket *sock, int len)
  2106. {
  2107. struct sock *sk = sock->sk;
  2108. int res;
  2109. lock_sock(sk);
  2110. res = tipc_set_sk_state(sk, TIPC_LISTEN);
  2111. release_sock(sk);
  2112. return res;
  2113. }
  2114. static int tipc_wait_for_accept(struct socket *sock, long timeo)
  2115. {
  2116. struct sock *sk = sock->sk;
  2117. DEFINE_WAIT(wait);
  2118. int err;
  2119. /* True wake-one mechanism for incoming connections: only
  2120. * one process gets woken up, not the 'whole herd'.
  2121. * Since we do not 'race & poll' for established sockets
  2122. * anymore, the common case will execute the loop only once.
  2123. */
  2124. for (;;) {
  2125. prepare_to_wait_exclusive(sk_sleep(sk), &wait,
  2126. TASK_INTERRUPTIBLE);
  2127. if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
  2128. release_sock(sk);
  2129. timeo = schedule_timeout(timeo);
  2130. lock_sock(sk);
  2131. }
  2132. err = 0;
  2133. if (!skb_queue_empty(&sk->sk_receive_queue))
  2134. break;
  2135. err = -EAGAIN;
  2136. if (!timeo)
  2137. break;
  2138. err = sock_intr_errno(timeo);
  2139. if (signal_pending(current))
  2140. break;
  2141. }
  2142. finish_wait(sk_sleep(sk), &wait);
  2143. return err;
  2144. }
  2145. /**
  2146. * tipc_accept - wait for connection request
  2147. * @sock: listening socket
  2148. * @newsock: new socket that is to be connected
  2149. * @flags: file-related flags associated with socket
  2150. *
  2151. * Returns 0 on success, errno otherwise
  2152. */
  2153. static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags,
  2154. bool kern)
  2155. {
  2156. struct sock *new_sk, *sk = sock->sk;
  2157. struct sk_buff *buf;
  2158. struct tipc_sock *new_tsock;
  2159. struct tipc_msg *msg;
  2160. long timeo;
  2161. int res;
  2162. lock_sock(sk);
  2163. if (sk->sk_state != TIPC_LISTEN) {
  2164. res = -EINVAL;
  2165. goto exit;
  2166. }
  2167. timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
  2168. res = tipc_wait_for_accept(sock, timeo);
  2169. if (res)
  2170. goto exit;
  2171. buf = skb_peek(&sk->sk_receive_queue);
  2172. res = tipc_sk_create(sock_net(sock->sk), new_sock, 0, kern);
  2173. if (res)
  2174. goto exit;
  2175. security_sk_clone(sock->sk, new_sock->sk);
  2176. new_sk = new_sock->sk;
  2177. new_tsock = tipc_sk(new_sk);
  2178. msg = buf_msg(buf);
  2179. /* we lock on new_sk; but lockdep sees the lock on sk */
  2180. lock_sock_nested(new_sk, SINGLE_DEPTH_NESTING);
  2181. /*
  2182. * Reject any stray messages received by new socket
  2183. * before the socket lock was taken (very, very unlikely)
  2184. */
  2185. tsk_rej_rx_queue(new_sk);
  2186. /* Connect new socket to it's peer */
  2187. tipc_sk_finish_conn(new_tsock, msg_origport(msg), msg_orignode(msg));
  2188. tsk_set_importance(new_tsock, msg_importance(msg));
  2189. if (msg_named(msg)) {
  2190. new_tsock->conn_type = msg_nametype(msg);
  2191. new_tsock->conn_instance = msg_nameinst(msg);
  2192. }
  2193. /*
  2194. * Respond to 'SYN-' by discarding it & returning 'ACK'-.
  2195. * Respond to 'SYN+' by queuing it on new socket.
  2196. */
  2197. if (!msg_data_sz(msg)) {
  2198. struct msghdr m = {NULL,};
  2199. tsk_advance_rx_queue(sk);
  2200. __tipc_sendstream(new_sock, &m, 0);
  2201. } else {
  2202. __skb_dequeue(&sk->sk_receive_queue);
  2203. __skb_queue_head(&new_sk->sk_receive_queue, buf);
  2204. skb_set_owner_r(buf, new_sk);
  2205. }
  2206. release_sock(new_sk);
  2207. exit:
  2208. release_sock(sk);
  2209. return res;
  2210. }
  2211. /**
  2212. * tipc_shutdown - shutdown socket connection
  2213. * @sock: socket structure
  2214. * @how: direction to close (must be SHUT_RDWR)
  2215. *
  2216. * Terminates connection (if necessary), then purges socket's receive queue.
  2217. *
  2218. * Returns 0 on success, errno otherwise
  2219. */
  2220. static int tipc_shutdown(struct socket *sock, int how)
  2221. {
  2222. struct sock *sk = sock->sk;
  2223. int res;
  2224. if (how != SHUT_RDWR)
  2225. return -EINVAL;
  2226. lock_sock(sk);
  2227. __tipc_shutdown(sock, TIPC_CONN_SHUTDOWN);
  2228. sk->sk_shutdown = SEND_SHUTDOWN;
  2229. if (sk->sk_state == TIPC_DISCONNECTING) {
  2230. /* Discard any unreceived messages */
  2231. __skb_queue_purge(&sk->sk_receive_queue);
  2232. /* Wake up anyone sleeping in poll */
  2233. sk->sk_state_change(sk);
  2234. res = 0;
  2235. } else {
  2236. res = -ENOTCONN;
  2237. }
  2238. release_sock(sk);
  2239. return res;
  2240. }
  2241. static void tipc_sk_timeout(struct timer_list *t)
  2242. {
  2243. struct sock *sk = from_timer(sk, t, sk_timer);
  2244. struct tipc_sock *tsk = tipc_sk(sk);
  2245. u32 peer_port = tsk_peer_port(tsk);
  2246. u32 peer_node = tsk_peer_node(tsk);
  2247. u32 own_node = tsk_own_node(tsk);
  2248. u32 own_port = tsk->portid;
  2249. struct net *net = sock_net(sk);
  2250. struct sk_buff *skb = NULL;
  2251. bh_lock_sock(sk);
  2252. if (!tipc_sk_connected(sk))
  2253. goto exit;
  2254. /* Try again later if socket is busy */
  2255. if (sock_owned_by_user(sk)) {
  2256. sk_reset_timer(sk, &sk->sk_timer, jiffies + HZ / 20);
  2257. goto exit;
  2258. }
  2259. if (tsk->probe_unacked) {
  2260. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  2261. tipc_node_remove_conn(net, peer_node, peer_port);
  2262. sk->sk_state_change(sk);
  2263. goto exit;
  2264. }
  2265. /* Send new probe */
  2266. skb = tipc_msg_create(CONN_MANAGER, CONN_PROBE, INT_H_SIZE, 0,
  2267. peer_node, own_node, peer_port, own_port,
  2268. TIPC_OK);
  2269. tsk->probe_unacked = true;
  2270. sk_reset_timer(sk, &sk->sk_timer, jiffies + CONN_PROBING_INTV);
  2271. exit:
  2272. bh_unlock_sock(sk);
  2273. if (skb)
  2274. tipc_node_xmit_skb(net, skb, peer_node, own_port);
  2275. sock_put(sk);
  2276. }
  2277. static int tipc_sk_publish(struct tipc_sock *tsk, uint scope,
  2278. struct tipc_name_seq const *seq)
  2279. {
  2280. struct sock *sk = &tsk->sk;
  2281. struct net *net = sock_net(sk);
  2282. struct publication *publ;
  2283. u32 key;
  2284. if (tipc_sk_connected(sk))
  2285. return -EINVAL;
  2286. key = tsk->portid + tsk->pub_count + 1;
  2287. if (key == tsk->portid)
  2288. return -EADDRINUSE;
  2289. publ = tipc_nametbl_publish(net, seq->type, seq->lower, seq->upper,
  2290. scope, tsk->portid, key);
  2291. if (unlikely(!publ))
  2292. return -EINVAL;
  2293. list_add(&publ->pport_list, &tsk->publications);
  2294. tsk->pub_count++;
  2295. tsk->published = 1;
  2296. return 0;
  2297. }
  2298. static int tipc_sk_withdraw(struct tipc_sock *tsk, uint scope,
  2299. struct tipc_name_seq const *seq)
  2300. {
  2301. struct net *net = sock_net(&tsk->sk);
  2302. struct publication *publ;
  2303. struct publication *safe;
  2304. int rc = -EINVAL;
  2305. list_for_each_entry_safe(publ, safe, &tsk->publications, pport_list) {
  2306. if (seq) {
  2307. if (publ->scope != scope)
  2308. continue;
  2309. if (publ->type != seq->type)
  2310. continue;
  2311. if (publ->lower != seq->lower)
  2312. continue;
  2313. if (publ->upper != seq->upper)
  2314. break;
  2315. tipc_nametbl_withdraw(net, publ->type, publ->lower,
  2316. publ->ref, publ->key);
  2317. rc = 0;
  2318. break;
  2319. }
  2320. tipc_nametbl_withdraw(net, publ->type, publ->lower,
  2321. publ->ref, publ->key);
  2322. rc = 0;
  2323. }
  2324. if (list_empty(&tsk->publications))
  2325. tsk->published = 0;
  2326. return rc;
  2327. }
  2328. /* tipc_sk_reinit: set non-zero address in all existing sockets
  2329. * when we go from standalone to network mode.
  2330. */
  2331. void tipc_sk_reinit(struct net *net)
  2332. {
  2333. struct tipc_net *tn = net_generic(net, tipc_net_id);
  2334. struct rhashtable_iter iter;
  2335. struct tipc_sock *tsk;
  2336. struct tipc_msg *msg;
  2337. rhashtable_walk_enter(&tn->sk_rht, &iter);
  2338. do {
  2339. rhashtable_walk_start(&iter);
  2340. while ((tsk = rhashtable_walk_next(&iter)) && !IS_ERR(tsk)) {
  2341. spin_lock_bh(&tsk->sk.sk_lock.slock);
  2342. msg = &tsk->phdr;
  2343. msg_set_prevnode(msg, tn->own_addr);
  2344. msg_set_orignode(msg, tn->own_addr);
  2345. spin_unlock_bh(&tsk->sk.sk_lock.slock);
  2346. }
  2347. rhashtable_walk_stop(&iter);
  2348. } while (tsk == ERR_PTR(-EAGAIN));
  2349. }
  2350. static struct tipc_sock *tipc_sk_lookup(struct net *net, u32 portid)
  2351. {
  2352. struct tipc_net *tn = net_generic(net, tipc_net_id);
  2353. struct tipc_sock *tsk;
  2354. rcu_read_lock();
  2355. tsk = rhashtable_lookup_fast(&tn->sk_rht, &portid, tsk_rht_params);
  2356. if (tsk)
  2357. sock_hold(&tsk->sk);
  2358. rcu_read_unlock();
  2359. return tsk;
  2360. }
  2361. static int tipc_sk_insert(struct tipc_sock *tsk)
  2362. {
  2363. struct sock *sk = &tsk->sk;
  2364. struct net *net = sock_net(sk);
  2365. struct tipc_net *tn = net_generic(net, tipc_net_id);
  2366. u32 remaining = (TIPC_MAX_PORT - TIPC_MIN_PORT) + 1;
  2367. u32 portid = prandom_u32() % remaining + TIPC_MIN_PORT;
  2368. while (remaining--) {
  2369. portid++;
  2370. if ((portid < TIPC_MIN_PORT) || (portid > TIPC_MAX_PORT))
  2371. portid = TIPC_MIN_PORT;
  2372. tsk->portid = portid;
  2373. sock_hold(&tsk->sk);
  2374. if (!rhashtable_lookup_insert_fast(&tn->sk_rht, &tsk->node,
  2375. tsk_rht_params))
  2376. return 0;
  2377. sock_put(&tsk->sk);
  2378. }
  2379. return -1;
  2380. }
  2381. static void tipc_sk_remove(struct tipc_sock *tsk)
  2382. {
  2383. struct sock *sk = &tsk->sk;
  2384. struct tipc_net *tn = net_generic(sock_net(sk), tipc_net_id);
  2385. if (!rhashtable_remove_fast(&tn->sk_rht, &tsk->node, tsk_rht_params)) {
  2386. WARN_ON(refcount_read(&sk->sk_refcnt) == 1);
  2387. __sock_put(sk);
  2388. }
  2389. }
  2390. static const struct rhashtable_params tsk_rht_params = {
  2391. .nelem_hint = 192,
  2392. .head_offset = offsetof(struct tipc_sock, node),
  2393. .key_offset = offsetof(struct tipc_sock, portid),
  2394. .key_len = sizeof(u32), /* portid */
  2395. .max_size = 1048576,
  2396. .min_size = 256,
  2397. .automatic_shrinking = true,
  2398. };
  2399. int tipc_sk_rht_init(struct net *net)
  2400. {
  2401. struct tipc_net *tn = net_generic(net, tipc_net_id);
  2402. return rhashtable_init(&tn->sk_rht, &tsk_rht_params);
  2403. }
  2404. void tipc_sk_rht_destroy(struct net *net)
  2405. {
  2406. struct tipc_net *tn = net_generic(net, tipc_net_id);
  2407. /* Wait for socket readers to complete */
  2408. synchronize_net();
  2409. rhashtable_destroy(&tn->sk_rht);
  2410. }
  2411. static int tipc_sk_join(struct tipc_sock *tsk, struct tipc_group_req *mreq)
  2412. {
  2413. struct net *net = sock_net(&tsk->sk);
  2414. struct tipc_group *grp = tsk->group;
  2415. struct tipc_msg *hdr = &tsk->phdr;
  2416. struct tipc_name_seq seq;
  2417. int rc;
  2418. if (mreq->type < TIPC_RESERVED_TYPES)
  2419. return -EACCES;
  2420. if (mreq->scope > TIPC_NODE_SCOPE)
  2421. return -EINVAL;
  2422. if (grp)
  2423. return -EACCES;
  2424. grp = tipc_group_create(net, tsk->portid, mreq);
  2425. if (!grp)
  2426. return -ENOMEM;
  2427. tsk->group = grp;
  2428. msg_set_lookup_scope(hdr, mreq->scope);
  2429. msg_set_nametype(hdr, mreq->type);
  2430. msg_set_dest_droppable(hdr, true);
  2431. seq.type = mreq->type;
  2432. seq.lower = mreq->instance;
  2433. seq.upper = seq.lower;
  2434. tipc_nametbl_build_group(net, grp, mreq->type, mreq->scope);
  2435. rc = tipc_sk_publish(tsk, mreq->scope, &seq);
  2436. if (rc) {
  2437. tipc_group_delete(net, grp);
  2438. tsk->group = NULL;
  2439. }
  2440. /* Eliminate any risk that a broadcast overtakes sent JOINs */
  2441. tsk->mc_method.rcast = true;
  2442. tsk->mc_method.mandatory = true;
  2443. tipc_group_join(net, grp, &tsk->sk.sk_rcvbuf);
  2444. return rc;
  2445. }
  2446. static int tipc_sk_leave(struct tipc_sock *tsk)
  2447. {
  2448. struct net *net = sock_net(&tsk->sk);
  2449. struct tipc_group *grp = tsk->group;
  2450. struct tipc_name_seq seq;
  2451. int scope;
  2452. if (!grp)
  2453. return -EINVAL;
  2454. tipc_group_self(grp, &seq, &scope);
  2455. tipc_group_delete(net, grp);
  2456. tsk->group = NULL;
  2457. tipc_sk_withdraw(tsk, scope, &seq);
  2458. return 0;
  2459. }
  2460. /**
  2461. * tipc_setsockopt - set socket option
  2462. * @sock: socket structure
  2463. * @lvl: option level
  2464. * @opt: option identifier
  2465. * @ov: pointer to new option value
  2466. * @ol: length of option value
  2467. *
  2468. * For stream sockets only, accepts and ignores all IPPROTO_TCP options
  2469. * (to ease compatibility).
  2470. *
  2471. * Returns 0 on success, errno otherwise
  2472. */
  2473. static int tipc_setsockopt(struct socket *sock, int lvl, int opt,
  2474. char __user *ov, unsigned int ol)
  2475. {
  2476. struct sock *sk = sock->sk;
  2477. struct tipc_sock *tsk = tipc_sk(sk);
  2478. struct tipc_group_req mreq;
  2479. u32 value = 0;
  2480. int res = 0;
  2481. if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
  2482. return 0;
  2483. if (lvl != SOL_TIPC)
  2484. return -ENOPROTOOPT;
  2485. switch (opt) {
  2486. case TIPC_IMPORTANCE:
  2487. case TIPC_SRC_DROPPABLE:
  2488. case TIPC_DEST_DROPPABLE:
  2489. case TIPC_CONN_TIMEOUT:
  2490. if (ol < sizeof(value))
  2491. return -EINVAL;
  2492. if (get_user(value, (u32 __user *)ov))
  2493. return -EFAULT;
  2494. break;
  2495. case TIPC_GROUP_JOIN:
  2496. if (ol < sizeof(mreq))
  2497. return -EINVAL;
  2498. if (copy_from_user(&mreq, ov, sizeof(mreq)))
  2499. return -EFAULT;
  2500. break;
  2501. default:
  2502. if (ov || ol)
  2503. return -EINVAL;
  2504. }
  2505. lock_sock(sk);
  2506. switch (opt) {
  2507. case TIPC_IMPORTANCE:
  2508. res = tsk_set_importance(tsk, value);
  2509. break;
  2510. case TIPC_SRC_DROPPABLE:
  2511. if (sock->type != SOCK_STREAM)
  2512. tsk_set_unreliable(tsk, value);
  2513. else
  2514. res = -ENOPROTOOPT;
  2515. break;
  2516. case TIPC_DEST_DROPPABLE:
  2517. tsk_set_unreturnable(tsk, value);
  2518. break;
  2519. case TIPC_CONN_TIMEOUT:
  2520. tipc_sk(sk)->conn_timeout = value;
  2521. break;
  2522. case TIPC_MCAST_BROADCAST:
  2523. tsk->mc_method.rcast = false;
  2524. tsk->mc_method.mandatory = true;
  2525. break;
  2526. case TIPC_MCAST_REPLICAST:
  2527. tsk->mc_method.rcast = true;
  2528. tsk->mc_method.mandatory = true;
  2529. break;
  2530. case TIPC_GROUP_JOIN:
  2531. res = tipc_sk_join(tsk, &mreq);
  2532. break;
  2533. case TIPC_GROUP_LEAVE:
  2534. res = tipc_sk_leave(tsk);
  2535. break;
  2536. default:
  2537. res = -EINVAL;
  2538. }
  2539. release_sock(sk);
  2540. return res;
  2541. }
  2542. /**
  2543. * tipc_getsockopt - get socket option
  2544. * @sock: socket structure
  2545. * @lvl: option level
  2546. * @opt: option identifier
  2547. * @ov: receptacle for option value
  2548. * @ol: receptacle for length of option value
  2549. *
  2550. * For stream sockets only, returns 0 length result for all IPPROTO_TCP options
  2551. * (to ease compatibility).
  2552. *
  2553. * Returns 0 on success, errno otherwise
  2554. */
  2555. static int tipc_getsockopt(struct socket *sock, int lvl, int opt,
  2556. char __user *ov, int __user *ol)
  2557. {
  2558. struct sock *sk = sock->sk;
  2559. struct tipc_sock *tsk = tipc_sk(sk);
  2560. struct tipc_name_seq seq;
  2561. int len, scope;
  2562. u32 value;
  2563. int res;
  2564. if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
  2565. return put_user(0, ol);
  2566. if (lvl != SOL_TIPC)
  2567. return -ENOPROTOOPT;
  2568. res = get_user(len, ol);
  2569. if (res)
  2570. return res;
  2571. lock_sock(sk);
  2572. switch (opt) {
  2573. case TIPC_IMPORTANCE:
  2574. value = tsk_importance(tsk);
  2575. break;
  2576. case TIPC_SRC_DROPPABLE:
  2577. value = tsk_unreliable(tsk);
  2578. break;
  2579. case TIPC_DEST_DROPPABLE:
  2580. value = tsk_unreturnable(tsk);
  2581. break;
  2582. case TIPC_CONN_TIMEOUT:
  2583. value = tsk->conn_timeout;
  2584. /* no need to set "res", since already 0 at this point */
  2585. break;
  2586. case TIPC_NODE_RECVQ_DEPTH:
  2587. value = 0; /* was tipc_queue_size, now obsolete */
  2588. break;
  2589. case TIPC_SOCK_RECVQ_DEPTH:
  2590. value = skb_queue_len(&sk->sk_receive_queue);
  2591. break;
  2592. case TIPC_GROUP_JOIN:
  2593. seq.type = 0;
  2594. if (tsk->group)
  2595. tipc_group_self(tsk->group, &seq, &scope);
  2596. value = seq.type;
  2597. break;
  2598. default:
  2599. res = -EINVAL;
  2600. }
  2601. release_sock(sk);
  2602. if (res)
  2603. return res; /* "get" failed */
  2604. if (len < sizeof(value))
  2605. return -EINVAL;
  2606. if (copy_to_user(ov, &value, sizeof(value)))
  2607. return -EFAULT;
  2608. return put_user(sizeof(value), ol);
  2609. }
  2610. static int tipc_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
  2611. {
  2612. struct sock *sk = sock->sk;
  2613. struct tipc_sioc_ln_req lnr;
  2614. void __user *argp = (void __user *)arg;
  2615. switch (cmd) {
  2616. case SIOCGETLINKNAME:
  2617. if (copy_from_user(&lnr, argp, sizeof(lnr)))
  2618. return -EFAULT;
  2619. if (!tipc_node_get_linkname(sock_net(sk),
  2620. lnr.bearer_id & 0xffff, lnr.peer,
  2621. lnr.linkname, TIPC_MAX_LINK_NAME)) {
  2622. if (copy_to_user(argp, &lnr, sizeof(lnr)))
  2623. return -EFAULT;
  2624. return 0;
  2625. }
  2626. return -EADDRNOTAVAIL;
  2627. default:
  2628. return -ENOIOCTLCMD;
  2629. }
  2630. }
  2631. static int tipc_socketpair(struct socket *sock1, struct socket *sock2)
  2632. {
  2633. struct tipc_sock *tsk2 = tipc_sk(sock2->sk);
  2634. struct tipc_sock *tsk1 = tipc_sk(sock1->sk);
  2635. u32 onode = tipc_own_addr(sock_net(sock1->sk));
  2636. tsk1->peer.family = AF_TIPC;
  2637. tsk1->peer.addrtype = TIPC_ADDR_ID;
  2638. tsk1->peer.scope = TIPC_NODE_SCOPE;
  2639. tsk1->peer.addr.id.ref = tsk2->portid;
  2640. tsk1->peer.addr.id.node = onode;
  2641. tsk2->peer.family = AF_TIPC;
  2642. tsk2->peer.addrtype = TIPC_ADDR_ID;
  2643. tsk2->peer.scope = TIPC_NODE_SCOPE;
  2644. tsk2->peer.addr.id.ref = tsk1->portid;
  2645. tsk2->peer.addr.id.node = onode;
  2646. tipc_sk_finish_conn(tsk1, tsk2->portid, onode);
  2647. tipc_sk_finish_conn(tsk2, tsk1->portid, onode);
  2648. return 0;
  2649. }
  2650. /* Protocol switches for the various types of TIPC sockets */
  2651. static const struct proto_ops msg_ops = {
  2652. .owner = THIS_MODULE,
  2653. .family = AF_TIPC,
  2654. .release = tipc_release,
  2655. .bind = tipc_bind,
  2656. .connect = tipc_connect,
  2657. .socketpair = tipc_socketpair,
  2658. .accept = sock_no_accept,
  2659. .getname = tipc_getname,
  2660. .poll = tipc_poll,
  2661. .ioctl = tipc_ioctl,
  2662. .listen = sock_no_listen,
  2663. .shutdown = tipc_shutdown,
  2664. .setsockopt = tipc_setsockopt,
  2665. .getsockopt = tipc_getsockopt,
  2666. .sendmsg = tipc_sendmsg,
  2667. .recvmsg = tipc_recvmsg,
  2668. .mmap = sock_no_mmap,
  2669. .sendpage = sock_no_sendpage
  2670. };
  2671. static const struct proto_ops packet_ops = {
  2672. .owner = THIS_MODULE,
  2673. .family = AF_TIPC,
  2674. .release = tipc_release,
  2675. .bind = tipc_bind,
  2676. .connect = tipc_connect,
  2677. .socketpair = tipc_socketpair,
  2678. .accept = tipc_accept,
  2679. .getname = tipc_getname,
  2680. .poll = tipc_poll,
  2681. .ioctl = tipc_ioctl,
  2682. .listen = tipc_listen,
  2683. .shutdown = tipc_shutdown,
  2684. .setsockopt = tipc_setsockopt,
  2685. .getsockopt = tipc_getsockopt,
  2686. .sendmsg = tipc_send_packet,
  2687. .recvmsg = tipc_recvmsg,
  2688. .mmap = sock_no_mmap,
  2689. .sendpage = sock_no_sendpage
  2690. };
  2691. static const struct proto_ops stream_ops = {
  2692. .owner = THIS_MODULE,
  2693. .family = AF_TIPC,
  2694. .release = tipc_release,
  2695. .bind = tipc_bind,
  2696. .connect = tipc_connect,
  2697. .socketpair = tipc_socketpair,
  2698. .accept = tipc_accept,
  2699. .getname = tipc_getname,
  2700. .poll = tipc_poll,
  2701. .ioctl = tipc_ioctl,
  2702. .listen = tipc_listen,
  2703. .shutdown = tipc_shutdown,
  2704. .setsockopt = tipc_setsockopt,
  2705. .getsockopt = tipc_getsockopt,
  2706. .sendmsg = tipc_sendstream,
  2707. .recvmsg = tipc_recvstream,
  2708. .mmap = sock_no_mmap,
  2709. .sendpage = sock_no_sendpage
  2710. };
  2711. static const struct net_proto_family tipc_family_ops = {
  2712. .owner = THIS_MODULE,
  2713. .family = AF_TIPC,
  2714. .create = tipc_sk_create
  2715. };
  2716. static struct proto tipc_proto = {
  2717. .name = "TIPC",
  2718. .owner = THIS_MODULE,
  2719. .obj_size = sizeof(struct tipc_sock),
  2720. .sysctl_rmem = sysctl_tipc_rmem
  2721. };
  2722. /**
  2723. * tipc_socket_init - initialize TIPC socket interface
  2724. *
  2725. * Returns 0 on success, errno otherwise
  2726. */
  2727. int tipc_socket_init(void)
  2728. {
  2729. int res;
  2730. res = proto_register(&tipc_proto, 1);
  2731. if (res) {
  2732. pr_err("Failed to register TIPC protocol type\n");
  2733. goto out;
  2734. }
  2735. res = sock_register(&tipc_family_ops);
  2736. if (res) {
  2737. pr_err("Failed to register TIPC socket type\n");
  2738. proto_unregister(&tipc_proto);
  2739. goto out;
  2740. }
  2741. out:
  2742. return res;
  2743. }
  2744. /**
  2745. * tipc_socket_stop - stop TIPC socket interface
  2746. */
  2747. void tipc_socket_stop(void)
  2748. {
  2749. sock_unregister(tipc_family_ops.family);
  2750. proto_unregister(&tipc_proto);
  2751. }
  2752. /* Caller should hold socket lock for the passed tipc socket. */
  2753. static int __tipc_nl_add_sk_con(struct sk_buff *skb, struct tipc_sock *tsk)
  2754. {
  2755. u32 peer_node;
  2756. u32 peer_port;
  2757. struct nlattr *nest;
  2758. peer_node = tsk_peer_node(tsk);
  2759. peer_port = tsk_peer_port(tsk);
  2760. nest = nla_nest_start(skb, TIPC_NLA_SOCK_CON);
  2761. if (nla_put_u32(skb, TIPC_NLA_CON_NODE, peer_node))
  2762. goto msg_full;
  2763. if (nla_put_u32(skb, TIPC_NLA_CON_SOCK, peer_port))
  2764. goto msg_full;
  2765. if (tsk->conn_type != 0) {
  2766. if (nla_put_flag(skb, TIPC_NLA_CON_FLAG))
  2767. goto msg_full;
  2768. if (nla_put_u32(skb, TIPC_NLA_CON_TYPE, tsk->conn_type))
  2769. goto msg_full;
  2770. if (nla_put_u32(skb, TIPC_NLA_CON_INST, tsk->conn_instance))
  2771. goto msg_full;
  2772. }
  2773. nla_nest_end(skb, nest);
  2774. return 0;
  2775. msg_full:
  2776. nla_nest_cancel(skb, nest);
  2777. return -EMSGSIZE;
  2778. }
  2779. /* Caller should hold socket lock for the passed tipc socket. */
  2780. static int __tipc_nl_add_sk(struct sk_buff *skb, struct netlink_callback *cb,
  2781. struct tipc_sock *tsk)
  2782. {
  2783. int err;
  2784. void *hdr;
  2785. struct nlattr *attrs;
  2786. struct net *net = sock_net(skb->sk);
  2787. struct tipc_net *tn = net_generic(net, tipc_net_id);
  2788. struct sock *sk = &tsk->sk;
  2789. hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
  2790. &tipc_genl_family, NLM_F_MULTI, TIPC_NL_SOCK_GET);
  2791. if (!hdr)
  2792. goto msg_cancel;
  2793. attrs = nla_nest_start(skb, TIPC_NLA_SOCK);
  2794. if (!attrs)
  2795. goto genlmsg_cancel;
  2796. if (nla_put_u32(skb, TIPC_NLA_SOCK_REF, tsk->portid))
  2797. goto attr_msg_cancel;
  2798. if (nla_put_u32(skb, TIPC_NLA_SOCK_ADDR, tn->own_addr))
  2799. goto attr_msg_cancel;
  2800. if (tipc_sk_connected(sk)) {
  2801. err = __tipc_nl_add_sk_con(skb, tsk);
  2802. if (err)
  2803. goto attr_msg_cancel;
  2804. } else if (!list_empty(&tsk->publications)) {
  2805. if (nla_put_flag(skb, TIPC_NLA_SOCK_HAS_PUBL))
  2806. goto attr_msg_cancel;
  2807. }
  2808. nla_nest_end(skb, attrs);
  2809. genlmsg_end(skb, hdr);
  2810. return 0;
  2811. attr_msg_cancel:
  2812. nla_nest_cancel(skb, attrs);
  2813. genlmsg_cancel:
  2814. genlmsg_cancel(skb, hdr);
  2815. msg_cancel:
  2816. return -EMSGSIZE;
  2817. }
  2818. int tipc_nl_sk_dump(struct sk_buff *skb, struct netlink_callback *cb)
  2819. {
  2820. int err;
  2821. struct tipc_sock *tsk;
  2822. const struct bucket_table *tbl;
  2823. struct rhash_head *pos;
  2824. struct net *net = sock_net(skb->sk);
  2825. struct tipc_net *tn = net_generic(net, tipc_net_id);
  2826. u32 tbl_id = cb->args[0];
  2827. u32 prev_portid = cb->args[1];
  2828. rcu_read_lock();
  2829. tbl = rht_dereference_rcu((&tn->sk_rht)->tbl, &tn->sk_rht);
  2830. for (; tbl_id < tbl->size; tbl_id++) {
  2831. rht_for_each_entry_rcu(tsk, pos, tbl, tbl_id, node) {
  2832. spin_lock_bh(&tsk->sk.sk_lock.slock);
  2833. if (prev_portid && prev_portid != tsk->portid) {
  2834. spin_unlock_bh(&tsk->sk.sk_lock.slock);
  2835. continue;
  2836. }
  2837. err = __tipc_nl_add_sk(skb, cb, tsk);
  2838. if (err) {
  2839. prev_portid = tsk->portid;
  2840. spin_unlock_bh(&tsk->sk.sk_lock.slock);
  2841. goto out;
  2842. }
  2843. prev_portid = 0;
  2844. spin_unlock_bh(&tsk->sk.sk_lock.slock);
  2845. }
  2846. }
  2847. out:
  2848. rcu_read_unlock();
  2849. cb->args[0] = tbl_id;
  2850. cb->args[1] = prev_portid;
  2851. return skb->len;
  2852. }
  2853. /* Caller should hold socket lock for the passed tipc socket. */
  2854. static int __tipc_nl_add_sk_publ(struct sk_buff *skb,
  2855. struct netlink_callback *cb,
  2856. struct publication *publ)
  2857. {
  2858. void *hdr;
  2859. struct nlattr *attrs;
  2860. hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
  2861. &tipc_genl_family, NLM_F_MULTI, TIPC_NL_PUBL_GET);
  2862. if (!hdr)
  2863. goto msg_cancel;
  2864. attrs = nla_nest_start(skb, TIPC_NLA_PUBL);
  2865. if (!attrs)
  2866. goto genlmsg_cancel;
  2867. if (nla_put_u32(skb, TIPC_NLA_PUBL_KEY, publ->key))
  2868. goto attr_msg_cancel;
  2869. if (nla_put_u32(skb, TIPC_NLA_PUBL_TYPE, publ->type))
  2870. goto attr_msg_cancel;
  2871. if (nla_put_u32(skb, TIPC_NLA_PUBL_LOWER, publ->lower))
  2872. goto attr_msg_cancel;
  2873. if (nla_put_u32(skb, TIPC_NLA_PUBL_UPPER, publ->upper))
  2874. goto attr_msg_cancel;
  2875. nla_nest_end(skb, attrs);
  2876. genlmsg_end(skb, hdr);
  2877. return 0;
  2878. attr_msg_cancel:
  2879. nla_nest_cancel(skb, attrs);
  2880. genlmsg_cancel:
  2881. genlmsg_cancel(skb, hdr);
  2882. msg_cancel:
  2883. return -EMSGSIZE;
  2884. }
  2885. /* Caller should hold socket lock for the passed tipc socket. */
  2886. static int __tipc_nl_list_sk_publ(struct sk_buff *skb,
  2887. struct netlink_callback *cb,
  2888. struct tipc_sock *tsk, u32 *last_publ)
  2889. {
  2890. int err;
  2891. struct publication *p;
  2892. if (*last_publ) {
  2893. list_for_each_entry(p, &tsk->publications, pport_list) {
  2894. if (p->key == *last_publ)
  2895. break;
  2896. }
  2897. if (p->key != *last_publ) {
  2898. /* We never set seq or call nl_dump_check_consistent()
  2899. * this means that setting prev_seq here will cause the
  2900. * consistence check to fail in the netlink callback
  2901. * handler. Resulting in the last NLMSG_DONE message
  2902. * having the NLM_F_DUMP_INTR flag set.
  2903. */
  2904. cb->prev_seq = 1;
  2905. *last_publ = 0;
  2906. return -EPIPE;
  2907. }
  2908. } else {
  2909. p = list_first_entry(&tsk->publications, struct publication,
  2910. pport_list);
  2911. }
  2912. list_for_each_entry_from(p, &tsk->publications, pport_list) {
  2913. err = __tipc_nl_add_sk_publ(skb, cb, p);
  2914. if (err) {
  2915. *last_publ = p->key;
  2916. return err;
  2917. }
  2918. }
  2919. *last_publ = 0;
  2920. return 0;
  2921. }
  2922. int tipc_nl_publ_dump(struct sk_buff *skb, struct netlink_callback *cb)
  2923. {
  2924. int err;
  2925. u32 tsk_portid = cb->args[0];
  2926. u32 last_publ = cb->args[1];
  2927. u32 done = cb->args[2];
  2928. struct net *net = sock_net(skb->sk);
  2929. struct tipc_sock *tsk;
  2930. if (!tsk_portid) {
  2931. struct nlattr **attrs;
  2932. struct nlattr *sock[TIPC_NLA_SOCK_MAX + 1];
  2933. err = tipc_nlmsg_parse(cb->nlh, &attrs);
  2934. if (err)
  2935. return err;
  2936. if (!attrs[TIPC_NLA_SOCK])
  2937. return -EINVAL;
  2938. err = nla_parse_nested(sock, TIPC_NLA_SOCK_MAX,
  2939. attrs[TIPC_NLA_SOCK],
  2940. tipc_nl_sock_policy, NULL);
  2941. if (err)
  2942. return err;
  2943. if (!sock[TIPC_NLA_SOCK_REF])
  2944. return -EINVAL;
  2945. tsk_portid = nla_get_u32(sock[TIPC_NLA_SOCK_REF]);
  2946. }
  2947. if (done)
  2948. return 0;
  2949. tsk = tipc_sk_lookup(net, tsk_portid);
  2950. if (!tsk)
  2951. return -EINVAL;
  2952. lock_sock(&tsk->sk);
  2953. err = __tipc_nl_list_sk_publ(skb, cb, tsk, &last_publ);
  2954. if (!err)
  2955. done = 1;
  2956. release_sock(&tsk->sk);
  2957. sock_put(&tsk->sk);
  2958. cb->args[0] = tsk_portid;
  2959. cb->args[1] = last_publ;
  2960. cb->args[2] = done;
  2961. return skb->len;
  2962. }