messenger.c 78 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197
  1. #include <linux/ceph/ceph_debug.h>
  2. #include <linux/crc32c.h>
  3. #include <linux/ctype.h>
  4. #include <linux/highmem.h>
  5. #include <linux/inet.h>
  6. #include <linux/kthread.h>
  7. #include <linux/net.h>
  8. #include <linux/slab.h>
  9. #include <linux/socket.h>
  10. #include <linux/string.h>
  11. #ifdef CONFIG_BLOCK
  12. #include <linux/bio.h>
  13. #endif /* CONFIG_BLOCK */
  14. #include <linux/dns_resolver.h>
  15. #include <net/tcp.h>
  16. #include <linux/ceph/libceph.h>
  17. #include <linux/ceph/messenger.h>
  18. #include <linux/ceph/decode.h>
  19. #include <linux/ceph/pagelist.h>
  20. #include <linux/export.h>
  21. #define list_entry_next(pos, member) \
  22. list_entry(pos->member.next, typeof(*pos), member)
  23. /*
  24. * Ceph uses the messenger to exchange ceph_msg messages with other
  25. * hosts in the system. The messenger provides ordered and reliable
  26. * delivery. We tolerate TCP disconnects by reconnecting (with
  27. * exponential backoff) in the case of a fault (disconnection, bad
  28. * crc, protocol error). Acks allow sent messages to be discarded by
  29. * the sender.
  30. */
  31. /*
  32. * We track the state of the socket on a given connection using
  33. * values defined below. The transition to a new socket state is
  34. * handled by a function which verifies we aren't coming from an
  35. * unexpected state.
  36. *
  37. * --------
  38. * | NEW* | transient initial state
  39. * --------
  40. * | con_sock_state_init()
  41. * v
  42. * ----------
  43. * | CLOSED | initialized, but no socket (and no
  44. * ---------- TCP connection)
  45. * ^ \
  46. * | \ con_sock_state_connecting()
  47. * | ----------------------
  48. * | \
  49. * + con_sock_state_closed() \
  50. * |+--------------------------- \
  51. * | \ \ \
  52. * | ----------- \ \
  53. * | | CLOSING | socket event; \ \
  54. * | ----------- await close \ \
  55. * | ^ \ |
  56. * | | \ |
  57. * | + con_sock_state_closing() \ |
  58. * | / \ | |
  59. * | / --------------- | |
  60. * | / \ v v
  61. * | / --------------
  62. * | / -----------------| CONNECTING | socket created, TCP
  63. * | | / -------------- connect initiated
  64. * | | | con_sock_state_connected()
  65. * | | v
  66. * -------------
  67. * | CONNECTED | TCP connection established
  68. * -------------
  69. *
  70. * State values for ceph_connection->sock_state; NEW is assumed to be 0.
  71. */
  72. #define CON_SOCK_STATE_NEW 0 /* -> CLOSED */
  73. #define CON_SOCK_STATE_CLOSED 1 /* -> CONNECTING */
  74. #define CON_SOCK_STATE_CONNECTING 2 /* -> CONNECTED or -> CLOSING */
  75. #define CON_SOCK_STATE_CONNECTED 3 /* -> CLOSING or -> CLOSED */
  76. #define CON_SOCK_STATE_CLOSING 4 /* -> CLOSED */
  77. /*
  78. * connection states
  79. */
  80. #define CON_STATE_CLOSED 1 /* -> PREOPEN */
  81. #define CON_STATE_PREOPEN 2 /* -> CONNECTING, CLOSED */
  82. #define CON_STATE_CONNECTING 3 /* -> NEGOTIATING, CLOSED */
  83. #define CON_STATE_NEGOTIATING 4 /* -> OPEN, CLOSED */
  84. #define CON_STATE_OPEN 5 /* -> STANDBY, CLOSED */
  85. #define CON_STATE_STANDBY 6 /* -> PREOPEN, CLOSED */
  86. /*
  87. * ceph_connection flag bits
  88. */
  89. #define CON_FLAG_LOSSYTX 0 /* we can close channel or drop
  90. * messages on errors */
  91. #define CON_FLAG_KEEPALIVE_PENDING 1 /* we need to send a keepalive */
  92. #define CON_FLAG_WRITE_PENDING 2 /* we have data ready to send */
  93. #define CON_FLAG_SOCK_CLOSED 3 /* socket state changed to closed */
  94. #define CON_FLAG_BACKOFF 4 /* need to retry queuing delayed work */
  95. static bool con_flag_valid(unsigned long con_flag)
  96. {
  97. switch (con_flag) {
  98. case CON_FLAG_LOSSYTX:
  99. case CON_FLAG_KEEPALIVE_PENDING:
  100. case CON_FLAG_WRITE_PENDING:
  101. case CON_FLAG_SOCK_CLOSED:
  102. case CON_FLAG_BACKOFF:
  103. return true;
  104. default:
  105. return false;
  106. }
  107. }
  108. static void con_flag_clear(struct ceph_connection *con, unsigned long con_flag)
  109. {
  110. BUG_ON(!con_flag_valid(con_flag));
  111. clear_bit(con_flag, &con->flags);
  112. }
  113. static void con_flag_set(struct ceph_connection *con, unsigned long con_flag)
  114. {
  115. BUG_ON(!con_flag_valid(con_flag));
  116. set_bit(con_flag, &con->flags);
  117. }
  118. static bool con_flag_test(struct ceph_connection *con, unsigned long con_flag)
  119. {
  120. BUG_ON(!con_flag_valid(con_flag));
  121. return test_bit(con_flag, &con->flags);
  122. }
  123. static bool con_flag_test_and_clear(struct ceph_connection *con,
  124. unsigned long con_flag)
  125. {
  126. BUG_ON(!con_flag_valid(con_flag));
  127. return test_and_clear_bit(con_flag, &con->flags);
  128. }
  129. static bool con_flag_test_and_set(struct ceph_connection *con,
  130. unsigned long con_flag)
  131. {
  132. BUG_ON(!con_flag_valid(con_flag));
  133. return test_and_set_bit(con_flag, &con->flags);
  134. }
  135. /* static tag bytes (protocol control messages) */
  136. static char tag_msg = CEPH_MSGR_TAG_MSG;
  137. static char tag_ack = CEPH_MSGR_TAG_ACK;
  138. static char tag_keepalive = CEPH_MSGR_TAG_KEEPALIVE;
  139. #ifdef CONFIG_LOCKDEP
  140. static struct lock_class_key socket_class;
  141. #endif
  142. /*
  143. * When skipping (ignoring) a block of input we read it into a "skip
  144. * buffer," which is this many bytes in size.
  145. */
  146. #define SKIP_BUF_SIZE 1024
  147. static void queue_con(struct ceph_connection *con);
  148. static void con_work(struct work_struct *);
  149. static void con_fault(struct ceph_connection *con);
  150. /*
  151. * Nicely render a sockaddr as a string. An array of formatted
  152. * strings is used, to approximate reentrancy.
  153. */
  154. #define ADDR_STR_COUNT_LOG 5 /* log2(# address strings in array) */
  155. #define ADDR_STR_COUNT (1 << ADDR_STR_COUNT_LOG)
  156. #define ADDR_STR_COUNT_MASK (ADDR_STR_COUNT - 1)
  157. #define MAX_ADDR_STR_LEN 64 /* 54 is enough */
  158. static char addr_str[ADDR_STR_COUNT][MAX_ADDR_STR_LEN];
  159. static atomic_t addr_str_seq = ATOMIC_INIT(0);
  160. static struct page *zero_page; /* used in certain error cases */
  161. const char *ceph_pr_addr(const struct sockaddr_storage *ss)
  162. {
  163. int i;
  164. char *s;
  165. struct sockaddr_in *in4 = (struct sockaddr_in *) ss;
  166. struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) ss;
  167. i = atomic_inc_return(&addr_str_seq) & ADDR_STR_COUNT_MASK;
  168. s = addr_str[i];
  169. switch (ss->ss_family) {
  170. case AF_INET:
  171. snprintf(s, MAX_ADDR_STR_LEN, "%pI4:%hu", &in4->sin_addr,
  172. ntohs(in4->sin_port));
  173. break;
  174. case AF_INET6:
  175. snprintf(s, MAX_ADDR_STR_LEN, "[%pI6c]:%hu", &in6->sin6_addr,
  176. ntohs(in6->sin6_port));
  177. break;
  178. default:
  179. snprintf(s, MAX_ADDR_STR_LEN, "(unknown sockaddr family %hu)",
  180. ss->ss_family);
  181. }
  182. return s;
  183. }
  184. EXPORT_SYMBOL(ceph_pr_addr);
  185. static void encode_my_addr(struct ceph_messenger *msgr)
  186. {
  187. memcpy(&msgr->my_enc_addr, &msgr->inst.addr, sizeof(msgr->my_enc_addr));
  188. ceph_encode_addr(&msgr->my_enc_addr);
  189. }
  190. /*
  191. * work queue for all reading and writing to/from the socket.
  192. */
  193. static struct workqueue_struct *ceph_msgr_wq;
  194. static void _ceph_msgr_exit(void)
  195. {
  196. if (ceph_msgr_wq) {
  197. destroy_workqueue(ceph_msgr_wq);
  198. ceph_msgr_wq = NULL;
  199. }
  200. BUG_ON(zero_page == NULL);
  201. kunmap(zero_page);
  202. page_cache_release(zero_page);
  203. zero_page = NULL;
  204. }
  205. int ceph_msgr_init(void)
  206. {
  207. BUG_ON(zero_page != NULL);
  208. zero_page = ZERO_PAGE(0);
  209. page_cache_get(zero_page);
  210. ceph_msgr_wq = alloc_workqueue("ceph-msgr", WQ_NON_REENTRANT, 0);
  211. if (ceph_msgr_wq)
  212. return 0;
  213. pr_err("msgr_init failed to create workqueue\n");
  214. _ceph_msgr_exit();
  215. return -ENOMEM;
  216. }
  217. EXPORT_SYMBOL(ceph_msgr_init);
  218. void ceph_msgr_exit(void)
  219. {
  220. BUG_ON(ceph_msgr_wq == NULL);
  221. _ceph_msgr_exit();
  222. }
  223. EXPORT_SYMBOL(ceph_msgr_exit);
  224. void ceph_msgr_flush(void)
  225. {
  226. flush_workqueue(ceph_msgr_wq);
  227. }
  228. EXPORT_SYMBOL(ceph_msgr_flush);
  229. /* Connection socket state transition functions */
  230. static void con_sock_state_init(struct ceph_connection *con)
  231. {
  232. int old_state;
  233. old_state = atomic_xchg(&con->sock_state, CON_SOCK_STATE_CLOSED);
  234. if (WARN_ON(old_state != CON_SOCK_STATE_NEW))
  235. printk("%s: unexpected old state %d\n", __func__, old_state);
  236. dout("%s con %p sock %d -> %d\n", __func__, con, old_state,
  237. CON_SOCK_STATE_CLOSED);
  238. }
  239. static void con_sock_state_connecting(struct ceph_connection *con)
  240. {
  241. int old_state;
  242. old_state = atomic_xchg(&con->sock_state, CON_SOCK_STATE_CONNECTING);
  243. if (WARN_ON(old_state != CON_SOCK_STATE_CLOSED))
  244. printk("%s: unexpected old state %d\n", __func__, old_state);
  245. dout("%s con %p sock %d -> %d\n", __func__, con, old_state,
  246. CON_SOCK_STATE_CONNECTING);
  247. }
  248. static void con_sock_state_connected(struct ceph_connection *con)
  249. {
  250. int old_state;
  251. old_state = atomic_xchg(&con->sock_state, CON_SOCK_STATE_CONNECTED);
  252. if (WARN_ON(old_state != CON_SOCK_STATE_CONNECTING))
  253. printk("%s: unexpected old state %d\n", __func__, old_state);
  254. dout("%s con %p sock %d -> %d\n", __func__, con, old_state,
  255. CON_SOCK_STATE_CONNECTED);
  256. }
  257. static void con_sock_state_closing(struct ceph_connection *con)
  258. {
  259. int old_state;
  260. old_state = atomic_xchg(&con->sock_state, CON_SOCK_STATE_CLOSING);
  261. if (WARN_ON(old_state != CON_SOCK_STATE_CONNECTING &&
  262. old_state != CON_SOCK_STATE_CONNECTED &&
  263. old_state != CON_SOCK_STATE_CLOSING))
  264. printk("%s: unexpected old state %d\n", __func__, old_state);
  265. dout("%s con %p sock %d -> %d\n", __func__, con, old_state,
  266. CON_SOCK_STATE_CLOSING);
  267. }
  268. static void con_sock_state_closed(struct ceph_connection *con)
  269. {
  270. int old_state;
  271. old_state = atomic_xchg(&con->sock_state, CON_SOCK_STATE_CLOSED);
  272. if (WARN_ON(old_state != CON_SOCK_STATE_CONNECTED &&
  273. old_state != CON_SOCK_STATE_CLOSING &&
  274. old_state != CON_SOCK_STATE_CONNECTING &&
  275. old_state != CON_SOCK_STATE_CLOSED))
  276. printk("%s: unexpected old state %d\n", __func__, old_state);
  277. dout("%s con %p sock %d -> %d\n", __func__, con, old_state,
  278. CON_SOCK_STATE_CLOSED);
  279. }
  280. /*
  281. * socket callback functions
  282. */
  283. /* data available on socket, or listen socket received a connect */
  284. static void ceph_sock_data_ready(struct sock *sk, int count_unused)
  285. {
  286. struct ceph_connection *con = sk->sk_user_data;
  287. if (atomic_read(&con->msgr->stopping)) {
  288. return;
  289. }
  290. if (sk->sk_state != TCP_CLOSE_WAIT) {
  291. dout("%s on %p state = %lu, queueing work\n", __func__,
  292. con, con->state);
  293. queue_con(con);
  294. }
  295. }
  296. /* socket has buffer space for writing */
  297. static void ceph_sock_write_space(struct sock *sk)
  298. {
  299. struct ceph_connection *con = sk->sk_user_data;
  300. /* only queue to workqueue if there is data we want to write,
  301. * and there is sufficient space in the socket buffer to accept
  302. * more data. clear SOCK_NOSPACE so that ceph_sock_write_space()
  303. * doesn't get called again until try_write() fills the socket
  304. * buffer. See net/ipv4/tcp_input.c:tcp_check_space()
  305. * and net/core/stream.c:sk_stream_write_space().
  306. */
  307. if (con_flag_test(con, CON_FLAG_WRITE_PENDING)) {
  308. if (sk_stream_wspace(sk) >= sk_stream_min_wspace(sk)) {
  309. dout("%s %p queueing write work\n", __func__, con);
  310. clear_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
  311. queue_con(con);
  312. }
  313. } else {
  314. dout("%s %p nothing to write\n", __func__, con);
  315. }
  316. }
  317. /* socket's state has changed */
  318. static void ceph_sock_state_change(struct sock *sk)
  319. {
  320. struct ceph_connection *con = sk->sk_user_data;
  321. dout("%s %p state = %lu sk_state = %u\n", __func__,
  322. con, con->state, sk->sk_state);
  323. switch (sk->sk_state) {
  324. case TCP_CLOSE:
  325. dout("%s TCP_CLOSE\n", __func__);
  326. case TCP_CLOSE_WAIT:
  327. dout("%s TCP_CLOSE_WAIT\n", __func__);
  328. con_sock_state_closing(con);
  329. con_flag_set(con, CON_FLAG_SOCK_CLOSED);
  330. queue_con(con);
  331. break;
  332. case TCP_ESTABLISHED:
  333. dout("%s TCP_ESTABLISHED\n", __func__);
  334. con_sock_state_connected(con);
  335. queue_con(con);
  336. break;
  337. default: /* Everything else is uninteresting */
  338. break;
  339. }
  340. }
  341. /*
  342. * set up socket callbacks
  343. */
  344. static void set_sock_callbacks(struct socket *sock,
  345. struct ceph_connection *con)
  346. {
  347. struct sock *sk = sock->sk;
  348. sk->sk_user_data = con;
  349. sk->sk_data_ready = ceph_sock_data_ready;
  350. sk->sk_write_space = ceph_sock_write_space;
  351. sk->sk_state_change = ceph_sock_state_change;
  352. }
  353. /*
  354. * socket helpers
  355. */
  356. /*
  357. * initiate connection to a remote socket.
  358. */
  359. static int ceph_tcp_connect(struct ceph_connection *con)
  360. {
  361. struct sockaddr_storage *paddr = &con->peer_addr.in_addr;
  362. struct socket *sock;
  363. int ret;
  364. BUG_ON(con->sock);
  365. ret = sock_create_kern(con->peer_addr.in_addr.ss_family, SOCK_STREAM,
  366. IPPROTO_TCP, &sock);
  367. if (ret)
  368. return ret;
  369. sock->sk->sk_allocation = GFP_NOFS;
  370. #ifdef CONFIG_LOCKDEP
  371. lockdep_set_class(&sock->sk->sk_lock, &socket_class);
  372. #endif
  373. set_sock_callbacks(sock, con);
  374. dout("connect %s\n", ceph_pr_addr(&con->peer_addr.in_addr));
  375. con_sock_state_connecting(con);
  376. ret = sock->ops->connect(sock, (struct sockaddr *)paddr, sizeof(*paddr),
  377. O_NONBLOCK);
  378. if (ret == -EINPROGRESS) {
  379. dout("connect %s EINPROGRESS sk_state = %u\n",
  380. ceph_pr_addr(&con->peer_addr.in_addr),
  381. sock->sk->sk_state);
  382. } else if (ret < 0) {
  383. pr_err("connect %s error %d\n",
  384. ceph_pr_addr(&con->peer_addr.in_addr), ret);
  385. sock_release(sock);
  386. con->error_msg = "connect error";
  387. return ret;
  388. }
  389. con->sock = sock;
  390. return 0;
  391. }
  392. static int ceph_tcp_recvmsg(struct socket *sock, void *buf, size_t len)
  393. {
  394. struct kvec iov = {buf, len};
  395. struct msghdr msg = { .msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL };
  396. int r;
  397. r = kernel_recvmsg(sock, &msg, &iov, 1, len, msg.msg_flags);
  398. if (r == -EAGAIN)
  399. r = 0;
  400. return r;
  401. }
  402. static int ceph_tcp_recvpage(struct socket *sock, struct page *page,
  403. int page_offset, size_t length)
  404. {
  405. void *kaddr;
  406. int ret;
  407. BUG_ON(page_offset + length > PAGE_SIZE);
  408. kaddr = kmap(page);
  409. BUG_ON(!kaddr);
  410. ret = ceph_tcp_recvmsg(sock, kaddr + page_offset, length);
  411. kunmap(page);
  412. return ret;
  413. }
  414. /*
  415. * write something. @more is true if caller will be sending more data
  416. * shortly.
  417. */
  418. static int ceph_tcp_sendmsg(struct socket *sock, struct kvec *iov,
  419. size_t kvlen, size_t len, int more)
  420. {
  421. struct msghdr msg = { .msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL };
  422. int r;
  423. if (more)
  424. msg.msg_flags |= MSG_MORE;
  425. else
  426. msg.msg_flags |= MSG_EOR; /* superfluous, but what the hell */
  427. r = kernel_sendmsg(sock, &msg, iov, kvlen, len);
  428. if (r == -EAGAIN)
  429. r = 0;
  430. return r;
  431. }
  432. static int ceph_tcp_sendpage(struct socket *sock, struct page *page,
  433. int offset, size_t size, bool more)
  434. {
  435. int flags = MSG_DONTWAIT | MSG_NOSIGNAL | (more ? MSG_MORE : MSG_EOR);
  436. int ret;
  437. ret = kernel_sendpage(sock, page, offset, size, flags);
  438. if (ret == -EAGAIN)
  439. ret = 0;
  440. return ret;
  441. }
  442. /*
  443. * Shutdown/close the socket for the given connection.
  444. */
  445. static int con_close_socket(struct ceph_connection *con)
  446. {
  447. int rc = 0;
  448. dout("con_close_socket on %p sock %p\n", con, con->sock);
  449. if (con->sock) {
  450. rc = con->sock->ops->shutdown(con->sock, SHUT_RDWR);
  451. sock_release(con->sock);
  452. con->sock = NULL;
  453. }
  454. /*
  455. * Forcibly clear the SOCK_CLOSED flag. It gets set
  456. * independent of the connection mutex, and we could have
  457. * received a socket close event before we had the chance to
  458. * shut the socket down.
  459. */
  460. con_flag_clear(con, CON_FLAG_SOCK_CLOSED);
  461. con_sock_state_closed(con);
  462. return rc;
  463. }
  464. /*
  465. * Reset a connection. Discard all incoming and outgoing messages
  466. * and clear *_seq state.
  467. */
  468. static void ceph_msg_remove(struct ceph_msg *msg)
  469. {
  470. list_del_init(&msg->list_head);
  471. BUG_ON(msg->con == NULL);
  472. msg->con->ops->put(msg->con);
  473. msg->con = NULL;
  474. ceph_msg_put(msg);
  475. }
  476. static void ceph_msg_remove_list(struct list_head *head)
  477. {
  478. while (!list_empty(head)) {
  479. struct ceph_msg *msg = list_first_entry(head, struct ceph_msg,
  480. list_head);
  481. ceph_msg_remove(msg);
  482. }
  483. }
  484. static void reset_connection(struct ceph_connection *con)
  485. {
  486. /* reset connection, out_queue, msg_ and connect_seq */
  487. /* discard existing out_queue and msg_seq */
  488. dout("reset_connection %p\n", con);
  489. ceph_msg_remove_list(&con->out_queue);
  490. ceph_msg_remove_list(&con->out_sent);
  491. if (con->in_msg) {
  492. BUG_ON(con->in_msg->con != con);
  493. con->in_msg->con = NULL;
  494. ceph_msg_put(con->in_msg);
  495. con->in_msg = NULL;
  496. con->ops->put(con);
  497. }
  498. con->connect_seq = 0;
  499. con->out_seq = 0;
  500. if (con->out_msg) {
  501. ceph_msg_put(con->out_msg);
  502. con->out_msg = NULL;
  503. }
  504. con->in_seq = 0;
  505. con->in_seq_acked = 0;
  506. }
  507. /*
  508. * mark a peer down. drop any open connections.
  509. */
  510. void ceph_con_close(struct ceph_connection *con)
  511. {
  512. mutex_lock(&con->mutex);
  513. dout("con_close %p peer %s\n", con,
  514. ceph_pr_addr(&con->peer_addr.in_addr));
  515. con->state = CON_STATE_CLOSED;
  516. con_flag_clear(con, CON_FLAG_LOSSYTX); /* so we retry next connect */
  517. con_flag_clear(con, CON_FLAG_KEEPALIVE_PENDING);
  518. con_flag_clear(con, CON_FLAG_WRITE_PENDING);
  519. con_flag_clear(con, CON_FLAG_BACKOFF);
  520. reset_connection(con);
  521. con->peer_global_seq = 0;
  522. cancel_delayed_work(&con->work);
  523. con_close_socket(con);
  524. mutex_unlock(&con->mutex);
  525. }
  526. EXPORT_SYMBOL(ceph_con_close);
  527. /*
  528. * Reopen a closed connection, with a new peer address.
  529. */
  530. void ceph_con_open(struct ceph_connection *con,
  531. __u8 entity_type, __u64 entity_num,
  532. struct ceph_entity_addr *addr)
  533. {
  534. mutex_lock(&con->mutex);
  535. dout("con_open %p %s\n", con, ceph_pr_addr(&addr->in_addr));
  536. WARN_ON(con->state != CON_STATE_CLOSED);
  537. con->state = CON_STATE_PREOPEN;
  538. con->peer_name.type = (__u8) entity_type;
  539. con->peer_name.num = cpu_to_le64(entity_num);
  540. memcpy(&con->peer_addr, addr, sizeof(*addr));
  541. con->delay = 0; /* reset backoff memory */
  542. mutex_unlock(&con->mutex);
  543. queue_con(con);
  544. }
  545. EXPORT_SYMBOL(ceph_con_open);
  546. /*
  547. * return true if this connection ever successfully opened
  548. */
  549. bool ceph_con_opened(struct ceph_connection *con)
  550. {
  551. return con->connect_seq > 0;
  552. }
  553. /*
  554. * initialize a new connection.
  555. */
  556. void ceph_con_init(struct ceph_connection *con, void *private,
  557. const struct ceph_connection_operations *ops,
  558. struct ceph_messenger *msgr)
  559. {
  560. dout("con_init %p\n", con);
  561. memset(con, 0, sizeof(*con));
  562. con->private = private;
  563. con->ops = ops;
  564. con->msgr = msgr;
  565. con_sock_state_init(con);
  566. mutex_init(&con->mutex);
  567. INIT_LIST_HEAD(&con->out_queue);
  568. INIT_LIST_HEAD(&con->out_sent);
  569. INIT_DELAYED_WORK(&con->work, con_work);
  570. con->state = CON_STATE_CLOSED;
  571. }
  572. EXPORT_SYMBOL(ceph_con_init);
  573. /*
  574. * We maintain a global counter to order connection attempts. Get
  575. * a unique seq greater than @gt.
  576. */
  577. static u32 get_global_seq(struct ceph_messenger *msgr, u32 gt)
  578. {
  579. u32 ret;
  580. spin_lock(&msgr->global_seq_lock);
  581. if (msgr->global_seq < gt)
  582. msgr->global_seq = gt;
  583. ret = ++msgr->global_seq;
  584. spin_unlock(&msgr->global_seq_lock);
  585. return ret;
  586. }
  587. static void con_out_kvec_reset(struct ceph_connection *con)
  588. {
  589. con->out_kvec_left = 0;
  590. con->out_kvec_bytes = 0;
  591. con->out_kvec_cur = &con->out_kvec[0];
  592. }
  593. static void con_out_kvec_add(struct ceph_connection *con,
  594. size_t size, void *data)
  595. {
  596. int index;
  597. index = con->out_kvec_left;
  598. BUG_ON(index >= ARRAY_SIZE(con->out_kvec));
  599. con->out_kvec[index].iov_len = size;
  600. con->out_kvec[index].iov_base = data;
  601. con->out_kvec_left++;
  602. con->out_kvec_bytes += size;
  603. }
  604. #ifdef CONFIG_BLOCK
  605. static void init_bio_iter(struct bio *bio, struct bio **bio_iter,
  606. unsigned int *bio_seg)
  607. {
  608. if (!bio) {
  609. *bio_iter = NULL;
  610. *bio_seg = 0;
  611. return;
  612. }
  613. *bio_iter = bio;
  614. *bio_seg = (unsigned int) bio->bi_idx;
  615. }
  616. static void iter_bio_next(struct bio **bio_iter, unsigned int *seg)
  617. {
  618. if (*bio_iter == NULL)
  619. return;
  620. BUG_ON(*seg >= (*bio_iter)->bi_vcnt);
  621. (*seg)++;
  622. if (*seg == (*bio_iter)->bi_vcnt)
  623. init_bio_iter((*bio_iter)->bi_next, bio_iter, seg);
  624. }
  625. #endif
  626. /*
  627. * For a pagelist, a piece is whatever remains to be consumed in the
  628. * first page in the list, or the front of the next page.
  629. */
  630. static void ceph_msg_data_pagelist_cursor_init(struct ceph_msg_data *data)
  631. {
  632. struct ceph_msg_data_cursor *cursor = &data->cursor;
  633. struct ceph_pagelist *pagelist;
  634. struct page *page;
  635. BUG_ON(data->type != CEPH_MSG_DATA_PAGELIST);
  636. pagelist = data->pagelist;
  637. BUG_ON(!pagelist);
  638. if (!pagelist->length)
  639. return; /* pagelist can be assigned but empty */
  640. BUG_ON(list_empty(&pagelist->head));
  641. page = list_first_entry(&pagelist->head, struct page, lru);
  642. cursor->page = page;
  643. cursor->offset = 0;
  644. cursor->last_piece = pagelist->length <= PAGE_SIZE;
  645. }
  646. static struct page *ceph_msg_data_pagelist_next(struct ceph_msg_data *data,
  647. size_t *page_offset,
  648. size_t *length)
  649. {
  650. struct ceph_msg_data_cursor *cursor = &data->cursor;
  651. struct ceph_pagelist *pagelist;
  652. size_t piece_end;
  653. BUG_ON(data->type != CEPH_MSG_DATA_PAGELIST);
  654. pagelist = data->pagelist;
  655. BUG_ON(!pagelist);
  656. BUG_ON(!cursor->page);
  657. BUG_ON(cursor->offset >= pagelist->length);
  658. if (cursor->last_piece) {
  659. /* pagelist offset is always 0 */
  660. piece_end = pagelist->length & ~PAGE_MASK;
  661. if (!piece_end)
  662. piece_end = PAGE_SIZE;
  663. } else {
  664. piece_end = PAGE_SIZE;
  665. }
  666. *page_offset = cursor->offset & ~PAGE_MASK;
  667. *length = piece_end - *page_offset;
  668. return data->cursor.page;
  669. }
  670. static bool ceph_msg_data_pagelist_advance(struct ceph_msg_data *data,
  671. size_t bytes)
  672. {
  673. struct ceph_msg_data_cursor *cursor = &data->cursor;
  674. struct ceph_pagelist *pagelist;
  675. BUG_ON(data->type != CEPH_MSG_DATA_PAGELIST);
  676. pagelist = data->pagelist;
  677. BUG_ON(!pagelist);
  678. BUG_ON(!cursor->page);
  679. BUG_ON(cursor->offset + bytes > pagelist->length);
  680. BUG_ON((cursor->offset & ~PAGE_MASK) + bytes > PAGE_SIZE);
  681. /* Advance the cursor offset */
  682. cursor->offset += bytes;
  683. /* pagelist offset is always 0 */
  684. if (!bytes || cursor->offset & ~PAGE_MASK)
  685. return false; /* more bytes to process in the current page */
  686. /* Move on to the next page */
  687. BUG_ON(list_is_last(&cursor->page->lru, &pagelist->head));
  688. cursor->page = list_entry_next(cursor->page, lru);
  689. /* cursor offset is at page boundary; pagelist offset is always 0 */
  690. if (pagelist->length - cursor->offset <= PAGE_SIZE)
  691. cursor->last_piece = true;
  692. return true;
  693. }
  694. /*
  695. * Message data is handled (sent or received) in pieces, where each
  696. * piece resides on a single page. The network layer might not
  697. * consume an entire piece at once. A data item's cursor keeps
  698. * track of which piece is next to process and how much remains to
  699. * be processed in that piece. It also tracks whether the current
  700. * piece is the last one in the data item.
  701. */
  702. static void ceph_msg_data_cursor_init(struct ceph_msg_data *data)
  703. {
  704. switch (data->type) {
  705. case CEPH_MSG_DATA_PAGELIST:
  706. ceph_msg_data_pagelist_cursor_init(data);
  707. break;
  708. case CEPH_MSG_DATA_NONE:
  709. case CEPH_MSG_DATA_PAGES:
  710. #ifdef CONFIG_BLOCK
  711. case CEPH_MSG_DATA_BIO:
  712. #endif /* CONFIG_BLOCK */
  713. default:
  714. /* BUG(); */
  715. break;
  716. }
  717. }
  718. /*
  719. * Return the page containing the next piece to process for a given
  720. * data item, and supply the page offset and length of that piece.
  721. * Indicate whether this is the last piece in this data item.
  722. */
  723. static struct page *ceph_msg_data_next(struct ceph_msg_data *data,
  724. size_t *page_offset,
  725. size_t *length,
  726. bool *last_piece)
  727. {
  728. struct page *page;
  729. switch (data->type) {
  730. case CEPH_MSG_DATA_PAGELIST:
  731. page = ceph_msg_data_pagelist_next(data, page_offset, length);
  732. break;
  733. case CEPH_MSG_DATA_NONE:
  734. case CEPH_MSG_DATA_PAGES:
  735. #ifdef CONFIG_BLOCK
  736. case CEPH_MSG_DATA_BIO:
  737. #endif /* CONFIG_BLOCK */
  738. default:
  739. page = NULL;
  740. break;
  741. }
  742. BUG_ON(!page);
  743. BUG_ON(*page_offset + *length > PAGE_SIZE);
  744. BUG_ON(!*length);
  745. if (last_piece)
  746. *last_piece = data->cursor.last_piece;
  747. return page;
  748. }
  749. /*
  750. * Returns true if the result moves the cursor on to the next piece
  751. * of the data item.
  752. */
  753. static bool ceph_msg_data_advance(struct ceph_msg_data *data, size_t bytes)
  754. {
  755. bool new_piece;
  756. switch (data->type) {
  757. case CEPH_MSG_DATA_PAGELIST:
  758. new_piece = ceph_msg_data_pagelist_advance(data, bytes);
  759. break;
  760. case CEPH_MSG_DATA_NONE:
  761. case CEPH_MSG_DATA_PAGES:
  762. #ifdef CONFIG_BLOCK
  763. case CEPH_MSG_DATA_BIO:
  764. #endif /* CONFIG_BLOCK */
  765. default:
  766. BUG();
  767. break;
  768. }
  769. return new_piece;
  770. }
  771. static void prepare_message_data(struct ceph_msg *msg,
  772. struct ceph_msg_pos *msg_pos)
  773. {
  774. BUG_ON(!msg);
  775. BUG_ON(!msg->hdr.data_len);
  776. /* initialize page iterator */
  777. msg_pos->page = 0;
  778. if (ceph_msg_has_pages(msg))
  779. msg_pos->page_pos = msg->p.alignment;
  780. else
  781. msg_pos->page_pos = 0;
  782. #ifdef CONFIG_BLOCK
  783. if (ceph_msg_has_bio(msg))
  784. init_bio_iter(msg->b.bio, &msg->b.bio_iter, &msg->b.bio_seg);
  785. #endif
  786. msg_pos->data_pos = 0;
  787. /* Initialize data cursors */
  788. if (ceph_msg_has_pagelist(msg))
  789. ceph_msg_data_cursor_init(&msg->l);
  790. if (ceph_msg_has_trail(msg))
  791. ceph_msg_data_cursor_init(&msg->t);
  792. msg_pos->did_page_crc = false;
  793. }
  794. /*
  795. * Prepare footer for currently outgoing message, and finish things
  796. * off. Assumes out_kvec* are already valid.. we just add on to the end.
  797. */
  798. static void prepare_write_message_footer(struct ceph_connection *con)
  799. {
  800. struct ceph_msg *m = con->out_msg;
  801. int v = con->out_kvec_left;
  802. m->footer.flags |= CEPH_MSG_FOOTER_COMPLETE;
  803. dout("prepare_write_message_footer %p\n", con);
  804. con->out_kvec_is_msg = true;
  805. con->out_kvec[v].iov_base = &m->footer;
  806. con->out_kvec[v].iov_len = sizeof(m->footer);
  807. con->out_kvec_bytes += sizeof(m->footer);
  808. con->out_kvec_left++;
  809. con->out_more = m->more_to_follow;
  810. con->out_msg_done = true;
  811. }
  812. /*
  813. * Prepare headers for the next outgoing message.
  814. */
  815. static void prepare_write_message(struct ceph_connection *con)
  816. {
  817. struct ceph_msg *m;
  818. u32 crc;
  819. con_out_kvec_reset(con);
  820. con->out_kvec_is_msg = true;
  821. con->out_msg_done = false;
  822. /* Sneak an ack in there first? If we can get it into the same
  823. * TCP packet that's a good thing. */
  824. if (con->in_seq > con->in_seq_acked) {
  825. con->in_seq_acked = con->in_seq;
  826. con_out_kvec_add(con, sizeof (tag_ack), &tag_ack);
  827. con->out_temp_ack = cpu_to_le64(con->in_seq_acked);
  828. con_out_kvec_add(con, sizeof (con->out_temp_ack),
  829. &con->out_temp_ack);
  830. }
  831. BUG_ON(list_empty(&con->out_queue));
  832. m = list_first_entry(&con->out_queue, struct ceph_msg, list_head);
  833. con->out_msg = m;
  834. BUG_ON(m->con != con);
  835. /* put message on sent list */
  836. ceph_msg_get(m);
  837. list_move_tail(&m->list_head, &con->out_sent);
  838. /*
  839. * only assign outgoing seq # if we haven't sent this message
  840. * yet. if it is requeued, resend with it's original seq.
  841. */
  842. if (m->needs_out_seq) {
  843. m->hdr.seq = cpu_to_le64(++con->out_seq);
  844. m->needs_out_seq = false;
  845. }
  846. dout("prepare_write_message %p seq %lld type %d len %d+%d+%d (%zd)\n",
  847. m, con->out_seq, le16_to_cpu(m->hdr.type),
  848. le32_to_cpu(m->hdr.front_len), le32_to_cpu(m->hdr.middle_len),
  849. le32_to_cpu(m->hdr.data_len), m->p.length);
  850. BUG_ON(le32_to_cpu(m->hdr.front_len) != m->front.iov_len);
  851. /* tag + hdr + front + middle */
  852. con_out_kvec_add(con, sizeof (tag_msg), &tag_msg);
  853. con_out_kvec_add(con, sizeof (m->hdr), &m->hdr);
  854. con_out_kvec_add(con, m->front.iov_len, m->front.iov_base);
  855. if (m->middle)
  856. con_out_kvec_add(con, m->middle->vec.iov_len,
  857. m->middle->vec.iov_base);
  858. /* fill in crc (except data pages), footer */
  859. crc = crc32c(0, &m->hdr, offsetof(struct ceph_msg_header, crc));
  860. con->out_msg->hdr.crc = cpu_to_le32(crc);
  861. con->out_msg->footer.flags = 0;
  862. crc = crc32c(0, m->front.iov_base, m->front.iov_len);
  863. con->out_msg->footer.front_crc = cpu_to_le32(crc);
  864. if (m->middle) {
  865. crc = crc32c(0, m->middle->vec.iov_base,
  866. m->middle->vec.iov_len);
  867. con->out_msg->footer.middle_crc = cpu_to_le32(crc);
  868. } else
  869. con->out_msg->footer.middle_crc = 0;
  870. dout("%s front_crc %u middle_crc %u\n", __func__,
  871. le32_to_cpu(con->out_msg->footer.front_crc),
  872. le32_to_cpu(con->out_msg->footer.middle_crc));
  873. /* is there a data payload? */
  874. con->out_msg->footer.data_crc = 0;
  875. if (m->hdr.data_len) {
  876. prepare_message_data(con->out_msg, &con->out_msg_pos);
  877. con->out_more = 1; /* data + footer will follow */
  878. } else {
  879. /* no, queue up footer too and be done */
  880. prepare_write_message_footer(con);
  881. }
  882. con_flag_set(con, CON_FLAG_WRITE_PENDING);
  883. }
  884. /*
  885. * Prepare an ack.
  886. */
  887. static void prepare_write_ack(struct ceph_connection *con)
  888. {
  889. dout("prepare_write_ack %p %llu -> %llu\n", con,
  890. con->in_seq_acked, con->in_seq);
  891. con->in_seq_acked = con->in_seq;
  892. con_out_kvec_reset(con);
  893. con_out_kvec_add(con, sizeof (tag_ack), &tag_ack);
  894. con->out_temp_ack = cpu_to_le64(con->in_seq_acked);
  895. con_out_kvec_add(con, sizeof (con->out_temp_ack),
  896. &con->out_temp_ack);
  897. con->out_more = 1; /* more will follow.. eventually.. */
  898. con_flag_set(con, CON_FLAG_WRITE_PENDING);
  899. }
  900. /*
  901. * Prepare to write keepalive byte.
  902. */
  903. static void prepare_write_keepalive(struct ceph_connection *con)
  904. {
  905. dout("prepare_write_keepalive %p\n", con);
  906. con_out_kvec_reset(con);
  907. con_out_kvec_add(con, sizeof (tag_keepalive), &tag_keepalive);
  908. con_flag_set(con, CON_FLAG_WRITE_PENDING);
  909. }
  910. /*
  911. * Connection negotiation.
  912. */
  913. static struct ceph_auth_handshake *get_connect_authorizer(struct ceph_connection *con,
  914. int *auth_proto)
  915. {
  916. struct ceph_auth_handshake *auth;
  917. if (!con->ops->get_authorizer) {
  918. con->out_connect.authorizer_protocol = CEPH_AUTH_UNKNOWN;
  919. con->out_connect.authorizer_len = 0;
  920. return NULL;
  921. }
  922. /* Can't hold the mutex while getting authorizer */
  923. mutex_unlock(&con->mutex);
  924. auth = con->ops->get_authorizer(con, auth_proto, con->auth_retry);
  925. mutex_lock(&con->mutex);
  926. if (IS_ERR(auth))
  927. return auth;
  928. if (con->state != CON_STATE_NEGOTIATING)
  929. return ERR_PTR(-EAGAIN);
  930. con->auth_reply_buf = auth->authorizer_reply_buf;
  931. con->auth_reply_buf_len = auth->authorizer_reply_buf_len;
  932. return auth;
  933. }
  934. /*
  935. * We connected to a peer and are saying hello.
  936. */
  937. static void prepare_write_banner(struct ceph_connection *con)
  938. {
  939. con_out_kvec_add(con, strlen(CEPH_BANNER), CEPH_BANNER);
  940. con_out_kvec_add(con, sizeof (con->msgr->my_enc_addr),
  941. &con->msgr->my_enc_addr);
  942. con->out_more = 0;
  943. con_flag_set(con, CON_FLAG_WRITE_PENDING);
  944. }
  945. static int prepare_write_connect(struct ceph_connection *con)
  946. {
  947. unsigned int global_seq = get_global_seq(con->msgr, 0);
  948. int proto;
  949. int auth_proto;
  950. struct ceph_auth_handshake *auth;
  951. switch (con->peer_name.type) {
  952. case CEPH_ENTITY_TYPE_MON:
  953. proto = CEPH_MONC_PROTOCOL;
  954. break;
  955. case CEPH_ENTITY_TYPE_OSD:
  956. proto = CEPH_OSDC_PROTOCOL;
  957. break;
  958. case CEPH_ENTITY_TYPE_MDS:
  959. proto = CEPH_MDSC_PROTOCOL;
  960. break;
  961. default:
  962. BUG();
  963. }
  964. dout("prepare_write_connect %p cseq=%d gseq=%d proto=%d\n", con,
  965. con->connect_seq, global_seq, proto);
  966. con->out_connect.features = cpu_to_le64(con->msgr->supported_features);
  967. con->out_connect.host_type = cpu_to_le32(CEPH_ENTITY_TYPE_CLIENT);
  968. con->out_connect.connect_seq = cpu_to_le32(con->connect_seq);
  969. con->out_connect.global_seq = cpu_to_le32(global_seq);
  970. con->out_connect.protocol_version = cpu_to_le32(proto);
  971. con->out_connect.flags = 0;
  972. auth_proto = CEPH_AUTH_UNKNOWN;
  973. auth = get_connect_authorizer(con, &auth_proto);
  974. if (IS_ERR(auth))
  975. return PTR_ERR(auth);
  976. con->out_connect.authorizer_protocol = cpu_to_le32(auth_proto);
  977. con->out_connect.authorizer_len = auth ?
  978. cpu_to_le32(auth->authorizer_buf_len) : 0;
  979. con_out_kvec_add(con, sizeof (con->out_connect),
  980. &con->out_connect);
  981. if (auth && auth->authorizer_buf_len)
  982. con_out_kvec_add(con, auth->authorizer_buf_len,
  983. auth->authorizer_buf);
  984. con->out_more = 0;
  985. con_flag_set(con, CON_FLAG_WRITE_PENDING);
  986. return 0;
  987. }
  988. /*
  989. * write as much of pending kvecs to the socket as we can.
  990. * 1 -> done
  991. * 0 -> socket full, but more to do
  992. * <0 -> error
  993. */
  994. static int write_partial_kvec(struct ceph_connection *con)
  995. {
  996. int ret;
  997. dout("write_partial_kvec %p %d left\n", con, con->out_kvec_bytes);
  998. while (con->out_kvec_bytes > 0) {
  999. ret = ceph_tcp_sendmsg(con->sock, con->out_kvec_cur,
  1000. con->out_kvec_left, con->out_kvec_bytes,
  1001. con->out_more);
  1002. if (ret <= 0)
  1003. goto out;
  1004. con->out_kvec_bytes -= ret;
  1005. if (con->out_kvec_bytes == 0)
  1006. break; /* done */
  1007. /* account for full iov entries consumed */
  1008. while (ret >= con->out_kvec_cur->iov_len) {
  1009. BUG_ON(!con->out_kvec_left);
  1010. ret -= con->out_kvec_cur->iov_len;
  1011. con->out_kvec_cur++;
  1012. con->out_kvec_left--;
  1013. }
  1014. /* and for a partially-consumed entry */
  1015. if (ret) {
  1016. con->out_kvec_cur->iov_len -= ret;
  1017. con->out_kvec_cur->iov_base += ret;
  1018. }
  1019. }
  1020. con->out_kvec_left = 0;
  1021. con->out_kvec_is_msg = false;
  1022. ret = 1;
  1023. out:
  1024. dout("write_partial_kvec %p %d left in %d kvecs ret = %d\n", con,
  1025. con->out_kvec_bytes, con->out_kvec_left, ret);
  1026. return ret; /* done! */
  1027. }
  1028. static void out_msg_pos_next(struct ceph_connection *con, struct page *page,
  1029. size_t len, size_t sent, bool in_trail)
  1030. {
  1031. struct ceph_msg *msg = con->out_msg;
  1032. struct ceph_msg_pos *msg_pos = &con->out_msg_pos;
  1033. bool need_crc = false;
  1034. BUG_ON(!msg);
  1035. BUG_ON(!sent);
  1036. msg_pos->data_pos += sent;
  1037. msg_pos->page_pos += sent;
  1038. if (in_trail)
  1039. need_crc = ceph_msg_data_advance(&msg->t, sent);
  1040. else if (ceph_msg_has_pagelist(msg))
  1041. need_crc = ceph_msg_data_advance(&msg->l, sent);
  1042. BUG_ON(need_crc && sent != len);
  1043. if (sent < len)
  1044. return;
  1045. BUG_ON(sent != len);
  1046. msg_pos->page_pos = 0;
  1047. msg_pos->page++;
  1048. msg_pos->did_page_crc = false;
  1049. #ifdef CONFIG_BLOCK
  1050. if (ceph_msg_has_bio(msg))
  1051. iter_bio_next(&msg->b.bio_iter, &msg->b.bio_seg);
  1052. #endif
  1053. }
  1054. static void in_msg_pos_next(struct ceph_connection *con, size_t len,
  1055. size_t received)
  1056. {
  1057. struct ceph_msg *msg = con->in_msg;
  1058. struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
  1059. BUG_ON(!msg);
  1060. BUG_ON(!received);
  1061. msg_pos->data_pos += received;
  1062. msg_pos->page_pos += received;
  1063. if (received < len)
  1064. return;
  1065. BUG_ON(received != len);
  1066. msg_pos->page_pos = 0;
  1067. msg_pos->page++;
  1068. #ifdef CONFIG_BLOCK
  1069. if (msg->b.bio)
  1070. iter_bio_next(&msg->b.bio_iter, &msg->b.bio_seg);
  1071. #endif /* CONFIG_BLOCK */
  1072. }
  1073. static u32 ceph_crc32c_page(u32 crc, struct page *page,
  1074. unsigned int page_offset,
  1075. unsigned int length)
  1076. {
  1077. char *kaddr;
  1078. kaddr = kmap(page);
  1079. BUG_ON(kaddr == NULL);
  1080. crc = crc32c(crc, kaddr + page_offset, length);
  1081. kunmap(page);
  1082. return crc;
  1083. }
  1084. /*
  1085. * Write as much message data payload as we can. If we finish, queue
  1086. * up the footer.
  1087. * 1 -> done, footer is now queued in out_kvec[].
  1088. * 0 -> socket full, but more to do
  1089. * <0 -> error
  1090. */
  1091. static int write_partial_message_data(struct ceph_connection *con)
  1092. {
  1093. struct ceph_msg *msg = con->out_msg;
  1094. struct ceph_msg_pos *msg_pos = &con->out_msg_pos;
  1095. unsigned int data_len = le32_to_cpu(msg->hdr.data_len);
  1096. bool do_datacrc = !con->msgr->nocrc;
  1097. int ret;
  1098. int total_max_write;
  1099. bool in_trail = false;
  1100. size_t trail_len = 0;
  1101. size_t trail_off = data_len;
  1102. if (ceph_msg_has_trail(msg)) {
  1103. trail_len = msg->t.pagelist->length;
  1104. trail_off -= trail_len;
  1105. }
  1106. dout("%s %p msg %p page %d offset %d\n", __func__,
  1107. con, msg, msg_pos->page, msg_pos->page_pos);
  1108. /*
  1109. * Iterate through each page that contains data to be
  1110. * written, and send as much as possible for each.
  1111. *
  1112. * If we are calculating the data crc (the default), we will
  1113. * need to map the page. If we have no pages, they have
  1114. * been revoked, so use the zero page.
  1115. */
  1116. while (data_len > msg_pos->data_pos) {
  1117. struct page *page = NULL;
  1118. size_t page_offset;
  1119. size_t length;
  1120. int max_write = PAGE_SIZE;
  1121. int bio_offset = 0;
  1122. bool use_cursor = false;
  1123. bool last_piece = true; /* preserve existing behavior */
  1124. in_trail = in_trail || msg_pos->data_pos >= trail_off;
  1125. if (!in_trail)
  1126. total_max_write = trail_off - msg_pos->data_pos;
  1127. if (in_trail) {
  1128. BUG_ON(!ceph_msg_has_trail(msg));
  1129. use_cursor = true;
  1130. page = ceph_msg_data_next(&msg->t, &page_offset,
  1131. &length, &last_piece);
  1132. } else if (ceph_msg_has_pages(msg)) {
  1133. page = msg->p.pages[msg_pos->page];
  1134. } else if (ceph_msg_has_pagelist(msg)) {
  1135. use_cursor = true;
  1136. page = ceph_msg_data_next(&msg->l, &page_offset,
  1137. &length, &last_piece);
  1138. #ifdef CONFIG_BLOCK
  1139. } else if (ceph_msg_has_bio(msg)) {
  1140. struct bio_vec *bv;
  1141. bv = bio_iovec_idx(msg->b.bio_iter, msg->b.bio_seg);
  1142. page = bv->bv_page;
  1143. bio_offset = bv->bv_offset;
  1144. max_write = bv->bv_len;
  1145. #endif
  1146. } else {
  1147. page = zero_page;
  1148. }
  1149. if (!use_cursor)
  1150. length = min_t(int, max_write - msg_pos->page_pos,
  1151. total_max_write);
  1152. page_offset = msg_pos->page_pos + bio_offset;
  1153. if (do_datacrc && !msg_pos->did_page_crc) {
  1154. u32 crc = le32_to_cpu(msg->footer.data_crc);
  1155. crc = ceph_crc32c_page(crc, page, page_offset, length);
  1156. msg->footer.data_crc = cpu_to_le32(crc);
  1157. msg_pos->did_page_crc = true;
  1158. }
  1159. ret = ceph_tcp_sendpage(con->sock, page, page_offset,
  1160. length, last_piece);
  1161. if (ret <= 0)
  1162. goto out;
  1163. out_msg_pos_next(con, page, length, (size_t) ret, in_trail);
  1164. }
  1165. dout("%s %p msg %p done\n", __func__, con, msg);
  1166. /* prepare and queue up footer, too */
  1167. if (!do_datacrc)
  1168. msg->footer.flags |= CEPH_MSG_FOOTER_NOCRC;
  1169. con_out_kvec_reset(con);
  1170. prepare_write_message_footer(con);
  1171. ret = 1;
  1172. out:
  1173. return ret;
  1174. }
  1175. /*
  1176. * write some zeros
  1177. */
  1178. static int write_partial_skip(struct ceph_connection *con)
  1179. {
  1180. int ret;
  1181. while (con->out_skip > 0) {
  1182. size_t size = min(con->out_skip, (int) PAGE_CACHE_SIZE);
  1183. ret = ceph_tcp_sendpage(con->sock, zero_page, 0, size, true);
  1184. if (ret <= 0)
  1185. goto out;
  1186. con->out_skip -= ret;
  1187. }
  1188. ret = 1;
  1189. out:
  1190. return ret;
  1191. }
  1192. /*
  1193. * Prepare to read connection handshake, or an ack.
  1194. */
  1195. static void prepare_read_banner(struct ceph_connection *con)
  1196. {
  1197. dout("prepare_read_banner %p\n", con);
  1198. con->in_base_pos = 0;
  1199. }
  1200. static void prepare_read_connect(struct ceph_connection *con)
  1201. {
  1202. dout("prepare_read_connect %p\n", con);
  1203. con->in_base_pos = 0;
  1204. }
  1205. static void prepare_read_ack(struct ceph_connection *con)
  1206. {
  1207. dout("prepare_read_ack %p\n", con);
  1208. con->in_base_pos = 0;
  1209. }
  1210. static void prepare_read_tag(struct ceph_connection *con)
  1211. {
  1212. dout("prepare_read_tag %p\n", con);
  1213. con->in_base_pos = 0;
  1214. con->in_tag = CEPH_MSGR_TAG_READY;
  1215. }
  1216. /*
  1217. * Prepare to read a message.
  1218. */
  1219. static int prepare_read_message(struct ceph_connection *con)
  1220. {
  1221. dout("prepare_read_message %p\n", con);
  1222. BUG_ON(con->in_msg != NULL);
  1223. con->in_base_pos = 0;
  1224. con->in_front_crc = con->in_middle_crc = con->in_data_crc = 0;
  1225. return 0;
  1226. }
  1227. static int read_partial(struct ceph_connection *con,
  1228. int end, int size, void *object)
  1229. {
  1230. while (con->in_base_pos < end) {
  1231. int left = end - con->in_base_pos;
  1232. int have = size - left;
  1233. int ret = ceph_tcp_recvmsg(con->sock, object + have, left);
  1234. if (ret <= 0)
  1235. return ret;
  1236. con->in_base_pos += ret;
  1237. }
  1238. return 1;
  1239. }
  1240. /*
  1241. * Read all or part of the connect-side handshake on a new connection
  1242. */
  1243. static int read_partial_banner(struct ceph_connection *con)
  1244. {
  1245. int size;
  1246. int end;
  1247. int ret;
  1248. dout("read_partial_banner %p at %d\n", con, con->in_base_pos);
  1249. /* peer's banner */
  1250. size = strlen(CEPH_BANNER);
  1251. end = size;
  1252. ret = read_partial(con, end, size, con->in_banner);
  1253. if (ret <= 0)
  1254. goto out;
  1255. size = sizeof (con->actual_peer_addr);
  1256. end += size;
  1257. ret = read_partial(con, end, size, &con->actual_peer_addr);
  1258. if (ret <= 0)
  1259. goto out;
  1260. size = sizeof (con->peer_addr_for_me);
  1261. end += size;
  1262. ret = read_partial(con, end, size, &con->peer_addr_for_me);
  1263. if (ret <= 0)
  1264. goto out;
  1265. out:
  1266. return ret;
  1267. }
  1268. static int read_partial_connect(struct ceph_connection *con)
  1269. {
  1270. int size;
  1271. int end;
  1272. int ret;
  1273. dout("read_partial_connect %p at %d\n", con, con->in_base_pos);
  1274. size = sizeof (con->in_reply);
  1275. end = size;
  1276. ret = read_partial(con, end, size, &con->in_reply);
  1277. if (ret <= 0)
  1278. goto out;
  1279. size = le32_to_cpu(con->in_reply.authorizer_len);
  1280. end += size;
  1281. ret = read_partial(con, end, size, con->auth_reply_buf);
  1282. if (ret <= 0)
  1283. goto out;
  1284. dout("read_partial_connect %p tag %d, con_seq = %u, g_seq = %u\n",
  1285. con, (int)con->in_reply.tag,
  1286. le32_to_cpu(con->in_reply.connect_seq),
  1287. le32_to_cpu(con->in_reply.global_seq));
  1288. out:
  1289. return ret;
  1290. }
  1291. /*
  1292. * Verify the hello banner looks okay.
  1293. */
  1294. static int verify_hello(struct ceph_connection *con)
  1295. {
  1296. if (memcmp(con->in_banner, CEPH_BANNER, strlen(CEPH_BANNER))) {
  1297. pr_err("connect to %s got bad banner\n",
  1298. ceph_pr_addr(&con->peer_addr.in_addr));
  1299. con->error_msg = "protocol error, bad banner";
  1300. return -1;
  1301. }
  1302. return 0;
  1303. }
  1304. static bool addr_is_blank(struct sockaddr_storage *ss)
  1305. {
  1306. switch (ss->ss_family) {
  1307. case AF_INET:
  1308. return ((struct sockaddr_in *)ss)->sin_addr.s_addr == 0;
  1309. case AF_INET6:
  1310. return
  1311. ((struct sockaddr_in6 *)ss)->sin6_addr.s6_addr32[0] == 0 &&
  1312. ((struct sockaddr_in6 *)ss)->sin6_addr.s6_addr32[1] == 0 &&
  1313. ((struct sockaddr_in6 *)ss)->sin6_addr.s6_addr32[2] == 0 &&
  1314. ((struct sockaddr_in6 *)ss)->sin6_addr.s6_addr32[3] == 0;
  1315. }
  1316. return false;
  1317. }
  1318. static int addr_port(struct sockaddr_storage *ss)
  1319. {
  1320. switch (ss->ss_family) {
  1321. case AF_INET:
  1322. return ntohs(((struct sockaddr_in *)ss)->sin_port);
  1323. case AF_INET6:
  1324. return ntohs(((struct sockaddr_in6 *)ss)->sin6_port);
  1325. }
  1326. return 0;
  1327. }
  1328. static void addr_set_port(struct sockaddr_storage *ss, int p)
  1329. {
  1330. switch (ss->ss_family) {
  1331. case AF_INET:
  1332. ((struct sockaddr_in *)ss)->sin_port = htons(p);
  1333. break;
  1334. case AF_INET6:
  1335. ((struct sockaddr_in6 *)ss)->sin6_port = htons(p);
  1336. break;
  1337. }
  1338. }
  1339. /*
  1340. * Unlike other *_pton function semantics, zero indicates success.
  1341. */
  1342. static int ceph_pton(const char *str, size_t len, struct sockaddr_storage *ss,
  1343. char delim, const char **ipend)
  1344. {
  1345. struct sockaddr_in *in4 = (struct sockaddr_in *) ss;
  1346. struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) ss;
  1347. memset(ss, 0, sizeof(*ss));
  1348. if (in4_pton(str, len, (u8 *)&in4->sin_addr.s_addr, delim, ipend)) {
  1349. ss->ss_family = AF_INET;
  1350. return 0;
  1351. }
  1352. if (in6_pton(str, len, (u8 *)&in6->sin6_addr.s6_addr, delim, ipend)) {
  1353. ss->ss_family = AF_INET6;
  1354. return 0;
  1355. }
  1356. return -EINVAL;
  1357. }
  1358. /*
  1359. * Extract hostname string and resolve using kernel DNS facility.
  1360. */
  1361. #ifdef CONFIG_CEPH_LIB_USE_DNS_RESOLVER
  1362. static int ceph_dns_resolve_name(const char *name, size_t namelen,
  1363. struct sockaddr_storage *ss, char delim, const char **ipend)
  1364. {
  1365. const char *end, *delim_p;
  1366. char *colon_p, *ip_addr = NULL;
  1367. int ip_len, ret;
  1368. /*
  1369. * The end of the hostname occurs immediately preceding the delimiter or
  1370. * the port marker (':') where the delimiter takes precedence.
  1371. */
  1372. delim_p = memchr(name, delim, namelen);
  1373. colon_p = memchr(name, ':', namelen);
  1374. if (delim_p && colon_p)
  1375. end = delim_p < colon_p ? delim_p : colon_p;
  1376. else if (!delim_p && colon_p)
  1377. end = colon_p;
  1378. else {
  1379. end = delim_p;
  1380. if (!end) /* case: hostname:/ */
  1381. end = name + namelen;
  1382. }
  1383. if (end <= name)
  1384. return -EINVAL;
  1385. /* do dns_resolve upcall */
  1386. ip_len = dns_query(NULL, name, end - name, NULL, &ip_addr, NULL);
  1387. if (ip_len > 0)
  1388. ret = ceph_pton(ip_addr, ip_len, ss, -1, NULL);
  1389. else
  1390. ret = -ESRCH;
  1391. kfree(ip_addr);
  1392. *ipend = end;
  1393. pr_info("resolve '%.*s' (ret=%d): %s\n", (int)(end - name), name,
  1394. ret, ret ? "failed" : ceph_pr_addr(ss));
  1395. return ret;
  1396. }
  1397. #else
  1398. static inline int ceph_dns_resolve_name(const char *name, size_t namelen,
  1399. struct sockaddr_storage *ss, char delim, const char **ipend)
  1400. {
  1401. return -EINVAL;
  1402. }
  1403. #endif
  1404. /*
  1405. * Parse a server name (IP or hostname). If a valid IP address is not found
  1406. * then try to extract a hostname to resolve using userspace DNS upcall.
  1407. */
  1408. static int ceph_parse_server_name(const char *name, size_t namelen,
  1409. struct sockaddr_storage *ss, char delim, const char **ipend)
  1410. {
  1411. int ret;
  1412. ret = ceph_pton(name, namelen, ss, delim, ipend);
  1413. if (ret)
  1414. ret = ceph_dns_resolve_name(name, namelen, ss, delim, ipend);
  1415. return ret;
  1416. }
  1417. /*
  1418. * Parse an ip[:port] list into an addr array. Use the default
  1419. * monitor port if a port isn't specified.
  1420. */
  1421. int ceph_parse_ips(const char *c, const char *end,
  1422. struct ceph_entity_addr *addr,
  1423. int max_count, int *count)
  1424. {
  1425. int i, ret = -EINVAL;
  1426. const char *p = c;
  1427. dout("parse_ips on '%.*s'\n", (int)(end-c), c);
  1428. for (i = 0; i < max_count; i++) {
  1429. const char *ipend;
  1430. struct sockaddr_storage *ss = &addr[i].in_addr;
  1431. int port;
  1432. char delim = ',';
  1433. if (*p == '[') {
  1434. delim = ']';
  1435. p++;
  1436. }
  1437. ret = ceph_parse_server_name(p, end - p, ss, delim, &ipend);
  1438. if (ret)
  1439. goto bad;
  1440. ret = -EINVAL;
  1441. p = ipend;
  1442. if (delim == ']') {
  1443. if (*p != ']') {
  1444. dout("missing matching ']'\n");
  1445. goto bad;
  1446. }
  1447. p++;
  1448. }
  1449. /* port? */
  1450. if (p < end && *p == ':') {
  1451. port = 0;
  1452. p++;
  1453. while (p < end && *p >= '0' && *p <= '9') {
  1454. port = (port * 10) + (*p - '0');
  1455. p++;
  1456. }
  1457. if (port > 65535 || port == 0)
  1458. goto bad;
  1459. } else {
  1460. port = CEPH_MON_PORT;
  1461. }
  1462. addr_set_port(ss, port);
  1463. dout("parse_ips got %s\n", ceph_pr_addr(ss));
  1464. if (p == end)
  1465. break;
  1466. if (*p != ',')
  1467. goto bad;
  1468. p++;
  1469. }
  1470. if (p != end)
  1471. goto bad;
  1472. if (count)
  1473. *count = i + 1;
  1474. return 0;
  1475. bad:
  1476. pr_err("parse_ips bad ip '%.*s'\n", (int)(end - c), c);
  1477. return ret;
  1478. }
  1479. EXPORT_SYMBOL(ceph_parse_ips);
  1480. static int process_banner(struct ceph_connection *con)
  1481. {
  1482. dout("process_banner on %p\n", con);
  1483. if (verify_hello(con) < 0)
  1484. return -1;
  1485. ceph_decode_addr(&con->actual_peer_addr);
  1486. ceph_decode_addr(&con->peer_addr_for_me);
  1487. /*
  1488. * Make sure the other end is who we wanted. note that the other
  1489. * end may not yet know their ip address, so if it's 0.0.0.0, give
  1490. * them the benefit of the doubt.
  1491. */
  1492. if (memcmp(&con->peer_addr, &con->actual_peer_addr,
  1493. sizeof(con->peer_addr)) != 0 &&
  1494. !(addr_is_blank(&con->actual_peer_addr.in_addr) &&
  1495. con->actual_peer_addr.nonce == con->peer_addr.nonce)) {
  1496. pr_warning("wrong peer, want %s/%d, got %s/%d\n",
  1497. ceph_pr_addr(&con->peer_addr.in_addr),
  1498. (int)le32_to_cpu(con->peer_addr.nonce),
  1499. ceph_pr_addr(&con->actual_peer_addr.in_addr),
  1500. (int)le32_to_cpu(con->actual_peer_addr.nonce));
  1501. con->error_msg = "wrong peer at address";
  1502. return -1;
  1503. }
  1504. /*
  1505. * did we learn our address?
  1506. */
  1507. if (addr_is_blank(&con->msgr->inst.addr.in_addr)) {
  1508. int port = addr_port(&con->msgr->inst.addr.in_addr);
  1509. memcpy(&con->msgr->inst.addr.in_addr,
  1510. &con->peer_addr_for_me.in_addr,
  1511. sizeof(con->peer_addr_for_me.in_addr));
  1512. addr_set_port(&con->msgr->inst.addr.in_addr, port);
  1513. encode_my_addr(con->msgr);
  1514. dout("process_banner learned my addr is %s\n",
  1515. ceph_pr_addr(&con->msgr->inst.addr.in_addr));
  1516. }
  1517. return 0;
  1518. }
  1519. static int process_connect(struct ceph_connection *con)
  1520. {
  1521. u64 sup_feat = con->msgr->supported_features;
  1522. u64 req_feat = con->msgr->required_features;
  1523. u64 server_feat = le64_to_cpu(con->in_reply.features);
  1524. int ret;
  1525. dout("process_connect on %p tag %d\n", con, (int)con->in_tag);
  1526. switch (con->in_reply.tag) {
  1527. case CEPH_MSGR_TAG_FEATURES:
  1528. pr_err("%s%lld %s feature set mismatch,"
  1529. " my %llx < server's %llx, missing %llx\n",
  1530. ENTITY_NAME(con->peer_name),
  1531. ceph_pr_addr(&con->peer_addr.in_addr),
  1532. sup_feat, server_feat, server_feat & ~sup_feat);
  1533. con->error_msg = "missing required protocol features";
  1534. reset_connection(con);
  1535. return -1;
  1536. case CEPH_MSGR_TAG_BADPROTOVER:
  1537. pr_err("%s%lld %s protocol version mismatch,"
  1538. " my %d != server's %d\n",
  1539. ENTITY_NAME(con->peer_name),
  1540. ceph_pr_addr(&con->peer_addr.in_addr),
  1541. le32_to_cpu(con->out_connect.protocol_version),
  1542. le32_to_cpu(con->in_reply.protocol_version));
  1543. con->error_msg = "protocol version mismatch";
  1544. reset_connection(con);
  1545. return -1;
  1546. case CEPH_MSGR_TAG_BADAUTHORIZER:
  1547. con->auth_retry++;
  1548. dout("process_connect %p got BADAUTHORIZER attempt %d\n", con,
  1549. con->auth_retry);
  1550. if (con->auth_retry == 2) {
  1551. con->error_msg = "connect authorization failure";
  1552. return -1;
  1553. }
  1554. con->auth_retry = 1;
  1555. con_out_kvec_reset(con);
  1556. ret = prepare_write_connect(con);
  1557. if (ret < 0)
  1558. return ret;
  1559. prepare_read_connect(con);
  1560. break;
  1561. case CEPH_MSGR_TAG_RESETSESSION:
  1562. /*
  1563. * If we connected with a large connect_seq but the peer
  1564. * has no record of a session with us (no connection, or
  1565. * connect_seq == 0), they will send RESETSESION to indicate
  1566. * that they must have reset their session, and may have
  1567. * dropped messages.
  1568. */
  1569. dout("process_connect got RESET peer seq %u\n",
  1570. le32_to_cpu(con->in_reply.connect_seq));
  1571. pr_err("%s%lld %s connection reset\n",
  1572. ENTITY_NAME(con->peer_name),
  1573. ceph_pr_addr(&con->peer_addr.in_addr));
  1574. reset_connection(con);
  1575. con_out_kvec_reset(con);
  1576. ret = prepare_write_connect(con);
  1577. if (ret < 0)
  1578. return ret;
  1579. prepare_read_connect(con);
  1580. /* Tell ceph about it. */
  1581. mutex_unlock(&con->mutex);
  1582. pr_info("reset on %s%lld\n", ENTITY_NAME(con->peer_name));
  1583. if (con->ops->peer_reset)
  1584. con->ops->peer_reset(con);
  1585. mutex_lock(&con->mutex);
  1586. if (con->state != CON_STATE_NEGOTIATING)
  1587. return -EAGAIN;
  1588. break;
  1589. case CEPH_MSGR_TAG_RETRY_SESSION:
  1590. /*
  1591. * If we sent a smaller connect_seq than the peer has, try
  1592. * again with a larger value.
  1593. */
  1594. dout("process_connect got RETRY_SESSION my seq %u, peer %u\n",
  1595. le32_to_cpu(con->out_connect.connect_seq),
  1596. le32_to_cpu(con->in_reply.connect_seq));
  1597. con->connect_seq = le32_to_cpu(con->in_reply.connect_seq);
  1598. con_out_kvec_reset(con);
  1599. ret = prepare_write_connect(con);
  1600. if (ret < 0)
  1601. return ret;
  1602. prepare_read_connect(con);
  1603. break;
  1604. case CEPH_MSGR_TAG_RETRY_GLOBAL:
  1605. /*
  1606. * If we sent a smaller global_seq than the peer has, try
  1607. * again with a larger value.
  1608. */
  1609. dout("process_connect got RETRY_GLOBAL my %u peer_gseq %u\n",
  1610. con->peer_global_seq,
  1611. le32_to_cpu(con->in_reply.global_seq));
  1612. get_global_seq(con->msgr,
  1613. le32_to_cpu(con->in_reply.global_seq));
  1614. con_out_kvec_reset(con);
  1615. ret = prepare_write_connect(con);
  1616. if (ret < 0)
  1617. return ret;
  1618. prepare_read_connect(con);
  1619. break;
  1620. case CEPH_MSGR_TAG_READY:
  1621. if (req_feat & ~server_feat) {
  1622. pr_err("%s%lld %s protocol feature mismatch,"
  1623. " my required %llx > server's %llx, need %llx\n",
  1624. ENTITY_NAME(con->peer_name),
  1625. ceph_pr_addr(&con->peer_addr.in_addr),
  1626. req_feat, server_feat, req_feat & ~server_feat);
  1627. con->error_msg = "missing required protocol features";
  1628. reset_connection(con);
  1629. return -1;
  1630. }
  1631. WARN_ON(con->state != CON_STATE_NEGOTIATING);
  1632. con->state = CON_STATE_OPEN;
  1633. con->peer_global_seq = le32_to_cpu(con->in_reply.global_seq);
  1634. con->connect_seq++;
  1635. con->peer_features = server_feat;
  1636. dout("process_connect got READY gseq %d cseq %d (%d)\n",
  1637. con->peer_global_seq,
  1638. le32_to_cpu(con->in_reply.connect_seq),
  1639. con->connect_seq);
  1640. WARN_ON(con->connect_seq !=
  1641. le32_to_cpu(con->in_reply.connect_seq));
  1642. if (con->in_reply.flags & CEPH_MSG_CONNECT_LOSSY)
  1643. con_flag_set(con, CON_FLAG_LOSSYTX);
  1644. con->delay = 0; /* reset backoff memory */
  1645. prepare_read_tag(con);
  1646. break;
  1647. case CEPH_MSGR_TAG_WAIT:
  1648. /*
  1649. * If there is a connection race (we are opening
  1650. * connections to each other), one of us may just have
  1651. * to WAIT. This shouldn't happen if we are the
  1652. * client.
  1653. */
  1654. pr_err("process_connect got WAIT as client\n");
  1655. con->error_msg = "protocol error, got WAIT as client";
  1656. return -1;
  1657. default:
  1658. pr_err("connect protocol error, will retry\n");
  1659. con->error_msg = "protocol error, garbage tag during connect";
  1660. return -1;
  1661. }
  1662. return 0;
  1663. }
  1664. /*
  1665. * read (part of) an ack
  1666. */
  1667. static int read_partial_ack(struct ceph_connection *con)
  1668. {
  1669. int size = sizeof (con->in_temp_ack);
  1670. int end = size;
  1671. return read_partial(con, end, size, &con->in_temp_ack);
  1672. }
  1673. /*
  1674. * We can finally discard anything that's been acked.
  1675. */
  1676. static void process_ack(struct ceph_connection *con)
  1677. {
  1678. struct ceph_msg *m;
  1679. u64 ack = le64_to_cpu(con->in_temp_ack);
  1680. u64 seq;
  1681. while (!list_empty(&con->out_sent)) {
  1682. m = list_first_entry(&con->out_sent, struct ceph_msg,
  1683. list_head);
  1684. seq = le64_to_cpu(m->hdr.seq);
  1685. if (seq > ack)
  1686. break;
  1687. dout("got ack for seq %llu type %d at %p\n", seq,
  1688. le16_to_cpu(m->hdr.type), m);
  1689. m->ack_stamp = jiffies;
  1690. ceph_msg_remove(m);
  1691. }
  1692. prepare_read_tag(con);
  1693. }
  1694. static int read_partial_message_section(struct ceph_connection *con,
  1695. struct kvec *section,
  1696. unsigned int sec_len, u32 *crc)
  1697. {
  1698. int ret, left;
  1699. BUG_ON(!section);
  1700. while (section->iov_len < sec_len) {
  1701. BUG_ON(section->iov_base == NULL);
  1702. left = sec_len - section->iov_len;
  1703. ret = ceph_tcp_recvmsg(con->sock, (char *)section->iov_base +
  1704. section->iov_len, left);
  1705. if (ret <= 0)
  1706. return ret;
  1707. section->iov_len += ret;
  1708. }
  1709. if (section->iov_len == sec_len)
  1710. *crc = crc32c(0, section->iov_base, section->iov_len);
  1711. return 1;
  1712. }
  1713. static int ceph_con_in_msg_alloc(struct ceph_connection *con, int *skip);
  1714. static int read_partial_message_pages(struct ceph_connection *con,
  1715. struct page **pages,
  1716. unsigned int data_len, bool do_datacrc)
  1717. {
  1718. struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
  1719. struct page *page;
  1720. size_t page_offset;
  1721. size_t length;
  1722. unsigned int left;
  1723. int ret;
  1724. /* (page) data */
  1725. BUG_ON(pages == NULL);
  1726. page = pages[msg_pos->page];
  1727. page_offset = msg_pos->page_pos;
  1728. BUG_ON(msg_pos->data_pos >= data_len);
  1729. left = data_len - msg_pos->data_pos;
  1730. BUG_ON(page_offset >= PAGE_SIZE);
  1731. length = min_t(unsigned int, PAGE_SIZE - page_offset, left);
  1732. ret = ceph_tcp_recvpage(con->sock, page, page_offset, length);
  1733. if (ret <= 0)
  1734. return ret;
  1735. if (do_datacrc)
  1736. con->in_data_crc = ceph_crc32c_page(con->in_data_crc, page,
  1737. page_offset, ret);
  1738. in_msg_pos_next(con, length, ret);
  1739. return ret;
  1740. }
  1741. #ifdef CONFIG_BLOCK
  1742. static int read_partial_message_bio(struct ceph_connection *con,
  1743. unsigned int data_len, bool do_datacrc)
  1744. {
  1745. struct ceph_msg *msg = con->in_msg;
  1746. struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
  1747. struct bio_vec *bv;
  1748. struct page *page;
  1749. size_t page_offset;
  1750. size_t length;
  1751. unsigned int left;
  1752. int ret;
  1753. BUG_ON(!msg);
  1754. BUG_ON(!msg->b.bio_iter);
  1755. bv = bio_iovec_idx(msg->b.bio_iter, msg->b.bio_seg);
  1756. page = bv->bv_page;
  1757. page_offset = bv->bv_offset + msg_pos->page_pos;
  1758. BUG_ON(msg_pos->data_pos >= data_len);
  1759. left = data_len - msg_pos->data_pos;
  1760. BUG_ON(msg_pos->page_pos >= bv->bv_len);
  1761. length = min_t(unsigned int, bv->bv_len - msg_pos->page_pos, left);
  1762. ret = ceph_tcp_recvpage(con->sock, page, page_offset, length);
  1763. if (ret <= 0)
  1764. return ret;
  1765. if (do_datacrc)
  1766. con->in_data_crc = ceph_crc32c_page(con->in_data_crc, page,
  1767. page_offset, ret);
  1768. in_msg_pos_next(con, length, ret);
  1769. return ret;
  1770. }
  1771. #endif
  1772. static int read_partial_msg_data(struct ceph_connection *con)
  1773. {
  1774. struct ceph_msg *msg = con->in_msg;
  1775. struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
  1776. const bool do_datacrc = !con->msgr->nocrc;
  1777. unsigned int data_len;
  1778. int ret;
  1779. BUG_ON(!msg);
  1780. data_len = le32_to_cpu(con->in_hdr.data_len);
  1781. while (msg_pos->data_pos < data_len) {
  1782. if (ceph_msg_has_pages(msg)) {
  1783. ret = read_partial_message_pages(con, msg->p.pages,
  1784. data_len, do_datacrc);
  1785. if (ret <= 0)
  1786. return ret;
  1787. #ifdef CONFIG_BLOCK
  1788. } else if (ceph_msg_has_bio(msg)) {
  1789. ret = read_partial_message_bio(con,
  1790. data_len, do_datacrc);
  1791. if (ret <= 0)
  1792. return ret;
  1793. #endif
  1794. } else {
  1795. BUG_ON(1);
  1796. }
  1797. }
  1798. return 1; /* must return > 0 to indicate success */
  1799. }
  1800. /*
  1801. * read (part of) a message.
  1802. */
  1803. static int read_partial_message(struct ceph_connection *con)
  1804. {
  1805. struct ceph_msg *m = con->in_msg;
  1806. int size;
  1807. int end;
  1808. int ret;
  1809. unsigned int front_len, middle_len, data_len;
  1810. bool do_datacrc = !con->msgr->nocrc;
  1811. u64 seq;
  1812. u32 crc;
  1813. dout("read_partial_message con %p msg %p\n", con, m);
  1814. /* header */
  1815. size = sizeof (con->in_hdr);
  1816. end = size;
  1817. ret = read_partial(con, end, size, &con->in_hdr);
  1818. if (ret <= 0)
  1819. return ret;
  1820. crc = crc32c(0, &con->in_hdr, offsetof(struct ceph_msg_header, crc));
  1821. if (cpu_to_le32(crc) != con->in_hdr.crc) {
  1822. pr_err("read_partial_message bad hdr "
  1823. " crc %u != expected %u\n",
  1824. crc, con->in_hdr.crc);
  1825. return -EBADMSG;
  1826. }
  1827. front_len = le32_to_cpu(con->in_hdr.front_len);
  1828. if (front_len > CEPH_MSG_MAX_FRONT_LEN)
  1829. return -EIO;
  1830. middle_len = le32_to_cpu(con->in_hdr.middle_len);
  1831. if (middle_len > CEPH_MSG_MAX_MIDDLE_LEN)
  1832. return -EIO;
  1833. data_len = le32_to_cpu(con->in_hdr.data_len);
  1834. if (data_len > CEPH_MSG_MAX_DATA_LEN)
  1835. return -EIO;
  1836. /* verify seq# */
  1837. seq = le64_to_cpu(con->in_hdr.seq);
  1838. if ((s64)seq - (s64)con->in_seq < 1) {
  1839. pr_info("skipping %s%lld %s seq %lld expected %lld\n",
  1840. ENTITY_NAME(con->peer_name),
  1841. ceph_pr_addr(&con->peer_addr.in_addr),
  1842. seq, con->in_seq + 1);
  1843. con->in_base_pos = -front_len - middle_len - data_len -
  1844. sizeof(m->footer);
  1845. con->in_tag = CEPH_MSGR_TAG_READY;
  1846. return 0;
  1847. } else if ((s64)seq - (s64)con->in_seq > 1) {
  1848. pr_err("read_partial_message bad seq %lld expected %lld\n",
  1849. seq, con->in_seq + 1);
  1850. con->error_msg = "bad message sequence # for incoming message";
  1851. return -EBADMSG;
  1852. }
  1853. /* allocate message? */
  1854. if (!con->in_msg) {
  1855. int skip = 0;
  1856. dout("got hdr type %d front %d data %d\n", con->in_hdr.type,
  1857. front_len, data_len);
  1858. ret = ceph_con_in_msg_alloc(con, &skip);
  1859. if (ret < 0)
  1860. return ret;
  1861. if (skip) {
  1862. /* skip this message */
  1863. dout("alloc_msg said skip message\n");
  1864. BUG_ON(con->in_msg);
  1865. con->in_base_pos = -front_len - middle_len - data_len -
  1866. sizeof(m->footer);
  1867. con->in_tag = CEPH_MSGR_TAG_READY;
  1868. con->in_seq++;
  1869. return 0;
  1870. }
  1871. BUG_ON(!con->in_msg);
  1872. BUG_ON(con->in_msg->con != con);
  1873. m = con->in_msg;
  1874. m->front.iov_len = 0; /* haven't read it yet */
  1875. if (m->middle)
  1876. m->middle->vec.iov_len = 0;
  1877. /* prepare for data payload, if any */
  1878. if (data_len)
  1879. prepare_message_data(con->in_msg, &con->in_msg_pos);
  1880. }
  1881. /* front */
  1882. ret = read_partial_message_section(con, &m->front, front_len,
  1883. &con->in_front_crc);
  1884. if (ret <= 0)
  1885. return ret;
  1886. /* middle */
  1887. if (m->middle) {
  1888. ret = read_partial_message_section(con, &m->middle->vec,
  1889. middle_len,
  1890. &con->in_middle_crc);
  1891. if (ret <= 0)
  1892. return ret;
  1893. }
  1894. /* (page) data */
  1895. if (data_len) {
  1896. ret = read_partial_msg_data(con);
  1897. if (ret <= 0)
  1898. return ret;
  1899. }
  1900. /* footer */
  1901. size = sizeof (m->footer);
  1902. end += size;
  1903. ret = read_partial(con, end, size, &m->footer);
  1904. if (ret <= 0)
  1905. return ret;
  1906. dout("read_partial_message got msg %p %d (%u) + %d (%u) + %d (%u)\n",
  1907. m, front_len, m->footer.front_crc, middle_len,
  1908. m->footer.middle_crc, data_len, m->footer.data_crc);
  1909. /* crc ok? */
  1910. if (con->in_front_crc != le32_to_cpu(m->footer.front_crc)) {
  1911. pr_err("read_partial_message %p front crc %u != exp. %u\n",
  1912. m, con->in_front_crc, m->footer.front_crc);
  1913. return -EBADMSG;
  1914. }
  1915. if (con->in_middle_crc != le32_to_cpu(m->footer.middle_crc)) {
  1916. pr_err("read_partial_message %p middle crc %u != exp %u\n",
  1917. m, con->in_middle_crc, m->footer.middle_crc);
  1918. return -EBADMSG;
  1919. }
  1920. if (do_datacrc &&
  1921. (m->footer.flags & CEPH_MSG_FOOTER_NOCRC) == 0 &&
  1922. con->in_data_crc != le32_to_cpu(m->footer.data_crc)) {
  1923. pr_err("read_partial_message %p data crc %u != exp. %u\n", m,
  1924. con->in_data_crc, le32_to_cpu(m->footer.data_crc));
  1925. return -EBADMSG;
  1926. }
  1927. return 1; /* done! */
  1928. }
  1929. /*
  1930. * Process message. This happens in the worker thread. The callback should
  1931. * be careful not to do anything that waits on other incoming messages or it
  1932. * may deadlock.
  1933. */
  1934. static void process_message(struct ceph_connection *con)
  1935. {
  1936. struct ceph_msg *msg;
  1937. BUG_ON(con->in_msg->con != con);
  1938. con->in_msg->con = NULL;
  1939. msg = con->in_msg;
  1940. con->in_msg = NULL;
  1941. con->ops->put(con);
  1942. /* if first message, set peer_name */
  1943. if (con->peer_name.type == 0)
  1944. con->peer_name = msg->hdr.src;
  1945. con->in_seq++;
  1946. mutex_unlock(&con->mutex);
  1947. dout("===== %p %llu from %s%lld %d=%s len %d+%d (%u %u %u) =====\n",
  1948. msg, le64_to_cpu(msg->hdr.seq),
  1949. ENTITY_NAME(msg->hdr.src),
  1950. le16_to_cpu(msg->hdr.type),
  1951. ceph_msg_type_name(le16_to_cpu(msg->hdr.type)),
  1952. le32_to_cpu(msg->hdr.front_len),
  1953. le32_to_cpu(msg->hdr.data_len),
  1954. con->in_front_crc, con->in_middle_crc, con->in_data_crc);
  1955. con->ops->dispatch(con, msg);
  1956. mutex_lock(&con->mutex);
  1957. }
  1958. /*
  1959. * Write something to the socket. Called in a worker thread when the
  1960. * socket appears to be writeable and we have something ready to send.
  1961. */
  1962. static int try_write(struct ceph_connection *con)
  1963. {
  1964. int ret = 1;
  1965. dout("try_write start %p state %lu\n", con, con->state);
  1966. more:
  1967. dout("try_write out_kvec_bytes %d\n", con->out_kvec_bytes);
  1968. /* open the socket first? */
  1969. if (con->state == CON_STATE_PREOPEN) {
  1970. BUG_ON(con->sock);
  1971. con->state = CON_STATE_CONNECTING;
  1972. con_out_kvec_reset(con);
  1973. prepare_write_banner(con);
  1974. prepare_read_banner(con);
  1975. BUG_ON(con->in_msg);
  1976. con->in_tag = CEPH_MSGR_TAG_READY;
  1977. dout("try_write initiating connect on %p new state %lu\n",
  1978. con, con->state);
  1979. ret = ceph_tcp_connect(con);
  1980. if (ret < 0) {
  1981. con->error_msg = "connect error";
  1982. goto out;
  1983. }
  1984. }
  1985. more_kvec:
  1986. /* kvec data queued? */
  1987. if (con->out_skip) {
  1988. ret = write_partial_skip(con);
  1989. if (ret <= 0)
  1990. goto out;
  1991. }
  1992. if (con->out_kvec_left) {
  1993. ret = write_partial_kvec(con);
  1994. if (ret <= 0)
  1995. goto out;
  1996. }
  1997. /* msg pages? */
  1998. if (con->out_msg) {
  1999. if (con->out_msg_done) {
  2000. ceph_msg_put(con->out_msg);
  2001. con->out_msg = NULL; /* we're done with this one */
  2002. goto do_next;
  2003. }
  2004. ret = write_partial_message_data(con);
  2005. if (ret == 1)
  2006. goto more_kvec; /* we need to send the footer, too! */
  2007. if (ret == 0)
  2008. goto out;
  2009. if (ret < 0) {
  2010. dout("try_write write_partial_message_data err %d\n",
  2011. ret);
  2012. goto out;
  2013. }
  2014. }
  2015. do_next:
  2016. if (con->state == CON_STATE_OPEN) {
  2017. /* is anything else pending? */
  2018. if (!list_empty(&con->out_queue)) {
  2019. prepare_write_message(con);
  2020. goto more;
  2021. }
  2022. if (con->in_seq > con->in_seq_acked) {
  2023. prepare_write_ack(con);
  2024. goto more;
  2025. }
  2026. if (con_flag_test_and_clear(con, CON_FLAG_KEEPALIVE_PENDING)) {
  2027. prepare_write_keepalive(con);
  2028. goto more;
  2029. }
  2030. }
  2031. /* Nothing to do! */
  2032. con_flag_clear(con, CON_FLAG_WRITE_PENDING);
  2033. dout("try_write nothing else to write.\n");
  2034. ret = 0;
  2035. out:
  2036. dout("try_write done on %p ret %d\n", con, ret);
  2037. return ret;
  2038. }
  2039. /*
  2040. * Read what we can from the socket.
  2041. */
  2042. static int try_read(struct ceph_connection *con)
  2043. {
  2044. int ret = -1;
  2045. more:
  2046. dout("try_read start on %p state %lu\n", con, con->state);
  2047. if (con->state != CON_STATE_CONNECTING &&
  2048. con->state != CON_STATE_NEGOTIATING &&
  2049. con->state != CON_STATE_OPEN)
  2050. return 0;
  2051. BUG_ON(!con->sock);
  2052. dout("try_read tag %d in_base_pos %d\n", (int)con->in_tag,
  2053. con->in_base_pos);
  2054. if (con->state == CON_STATE_CONNECTING) {
  2055. dout("try_read connecting\n");
  2056. ret = read_partial_banner(con);
  2057. if (ret <= 0)
  2058. goto out;
  2059. ret = process_banner(con);
  2060. if (ret < 0)
  2061. goto out;
  2062. con->state = CON_STATE_NEGOTIATING;
  2063. /*
  2064. * Received banner is good, exchange connection info.
  2065. * Do not reset out_kvec, as sending our banner raced
  2066. * with receiving peer banner after connect completed.
  2067. */
  2068. ret = prepare_write_connect(con);
  2069. if (ret < 0)
  2070. goto out;
  2071. prepare_read_connect(con);
  2072. /* Send connection info before awaiting response */
  2073. goto out;
  2074. }
  2075. if (con->state == CON_STATE_NEGOTIATING) {
  2076. dout("try_read negotiating\n");
  2077. ret = read_partial_connect(con);
  2078. if (ret <= 0)
  2079. goto out;
  2080. ret = process_connect(con);
  2081. if (ret < 0)
  2082. goto out;
  2083. goto more;
  2084. }
  2085. WARN_ON(con->state != CON_STATE_OPEN);
  2086. if (con->in_base_pos < 0) {
  2087. /*
  2088. * skipping + discarding content.
  2089. *
  2090. * FIXME: there must be a better way to do this!
  2091. */
  2092. static char buf[SKIP_BUF_SIZE];
  2093. int skip = min((int) sizeof (buf), -con->in_base_pos);
  2094. dout("skipping %d / %d bytes\n", skip, -con->in_base_pos);
  2095. ret = ceph_tcp_recvmsg(con->sock, buf, skip);
  2096. if (ret <= 0)
  2097. goto out;
  2098. con->in_base_pos += ret;
  2099. if (con->in_base_pos)
  2100. goto more;
  2101. }
  2102. if (con->in_tag == CEPH_MSGR_TAG_READY) {
  2103. /*
  2104. * what's next?
  2105. */
  2106. ret = ceph_tcp_recvmsg(con->sock, &con->in_tag, 1);
  2107. if (ret <= 0)
  2108. goto out;
  2109. dout("try_read got tag %d\n", (int)con->in_tag);
  2110. switch (con->in_tag) {
  2111. case CEPH_MSGR_TAG_MSG:
  2112. prepare_read_message(con);
  2113. break;
  2114. case CEPH_MSGR_TAG_ACK:
  2115. prepare_read_ack(con);
  2116. break;
  2117. case CEPH_MSGR_TAG_CLOSE:
  2118. con_close_socket(con);
  2119. con->state = CON_STATE_CLOSED;
  2120. goto out;
  2121. default:
  2122. goto bad_tag;
  2123. }
  2124. }
  2125. if (con->in_tag == CEPH_MSGR_TAG_MSG) {
  2126. ret = read_partial_message(con);
  2127. if (ret <= 0) {
  2128. switch (ret) {
  2129. case -EBADMSG:
  2130. con->error_msg = "bad crc";
  2131. ret = -EIO;
  2132. break;
  2133. case -EIO:
  2134. con->error_msg = "io error";
  2135. break;
  2136. }
  2137. goto out;
  2138. }
  2139. if (con->in_tag == CEPH_MSGR_TAG_READY)
  2140. goto more;
  2141. process_message(con);
  2142. if (con->state == CON_STATE_OPEN)
  2143. prepare_read_tag(con);
  2144. goto more;
  2145. }
  2146. if (con->in_tag == CEPH_MSGR_TAG_ACK) {
  2147. ret = read_partial_ack(con);
  2148. if (ret <= 0)
  2149. goto out;
  2150. process_ack(con);
  2151. goto more;
  2152. }
  2153. out:
  2154. dout("try_read done on %p ret %d\n", con, ret);
  2155. return ret;
  2156. bad_tag:
  2157. pr_err("try_read bad con->in_tag = %d\n", (int)con->in_tag);
  2158. con->error_msg = "protocol error, garbage tag";
  2159. ret = -1;
  2160. goto out;
  2161. }
  2162. /*
  2163. * Atomically queue work on a connection after the specified delay.
  2164. * Bump @con reference to avoid races with connection teardown.
  2165. * Returns 0 if work was queued, or an error code otherwise.
  2166. */
  2167. static int queue_con_delay(struct ceph_connection *con, unsigned long delay)
  2168. {
  2169. if (!con->ops->get(con)) {
  2170. dout("%s %p ref count 0\n", __func__, con);
  2171. return -ENOENT;
  2172. }
  2173. if (!queue_delayed_work(ceph_msgr_wq, &con->work, delay)) {
  2174. dout("%s %p - already queued\n", __func__, con);
  2175. con->ops->put(con);
  2176. return -EBUSY;
  2177. }
  2178. dout("%s %p %lu\n", __func__, con, delay);
  2179. return 0;
  2180. }
  2181. static void queue_con(struct ceph_connection *con)
  2182. {
  2183. (void) queue_con_delay(con, 0);
  2184. }
  2185. static bool con_sock_closed(struct ceph_connection *con)
  2186. {
  2187. if (!con_flag_test_and_clear(con, CON_FLAG_SOCK_CLOSED))
  2188. return false;
  2189. #define CASE(x) \
  2190. case CON_STATE_ ## x: \
  2191. con->error_msg = "socket closed (con state " #x ")"; \
  2192. break;
  2193. switch (con->state) {
  2194. CASE(CLOSED);
  2195. CASE(PREOPEN);
  2196. CASE(CONNECTING);
  2197. CASE(NEGOTIATING);
  2198. CASE(OPEN);
  2199. CASE(STANDBY);
  2200. default:
  2201. pr_warning("%s con %p unrecognized state %lu\n",
  2202. __func__, con, con->state);
  2203. con->error_msg = "unrecognized con state";
  2204. BUG();
  2205. break;
  2206. }
  2207. #undef CASE
  2208. return true;
  2209. }
  2210. static bool con_backoff(struct ceph_connection *con)
  2211. {
  2212. int ret;
  2213. if (!con_flag_test_and_clear(con, CON_FLAG_BACKOFF))
  2214. return false;
  2215. ret = queue_con_delay(con, round_jiffies_relative(con->delay));
  2216. if (ret) {
  2217. dout("%s: con %p FAILED to back off %lu\n", __func__,
  2218. con, con->delay);
  2219. BUG_ON(ret == -ENOENT);
  2220. con_flag_set(con, CON_FLAG_BACKOFF);
  2221. }
  2222. return true;
  2223. }
  2224. /* Finish fault handling; con->mutex must *not* be held here */
  2225. static void con_fault_finish(struct ceph_connection *con)
  2226. {
  2227. /*
  2228. * in case we faulted due to authentication, invalidate our
  2229. * current tickets so that we can get new ones.
  2230. */
  2231. if (con->auth_retry && con->ops->invalidate_authorizer) {
  2232. dout("calling invalidate_authorizer()\n");
  2233. con->ops->invalidate_authorizer(con);
  2234. }
  2235. if (con->ops->fault)
  2236. con->ops->fault(con);
  2237. }
  2238. /*
  2239. * Do some work on a connection. Drop a connection ref when we're done.
  2240. */
  2241. static void con_work(struct work_struct *work)
  2242. {
  2243. struct ceph_connection *con = container_of(work, struct ceph_connection,
  2244. work.work);
  2245. bool fault;
  2246. mutex_lock(&con->mutex);
  2247. while (true) {
  2248. int ret;
  2249. if ((fault = con_sock_closed(con))) {
  2250. dout("%s: con %p SOCK_CLOSED\n", __func__, con);
  2251. break;
  2252. }
  2253. if (con_backoff(con)) {
  2254. dout("%s: con %p BACKOFF\n", __func__, con);
  2255. break;
  2256. }
  2257. if (con->state == CON_STATE_STANDBY) {
  2258. dout("%s: con %p STANDBY\n", __func__, con);
  2259. break;
  2260. }
  2261. if (con->state == CON_STATE_CLOSED) {
  2262. dout("%s: con %p CLOSED\n", __func__, con);
  2263. BUG_ON(con->sock);
  2264. break;
  2265. }
  2266. if (con->state == CON_STATE_PREOPEN) {
  2267. dout("%s: con %p PREOPEN\n", __func__, con);
  2268. BUG_ON(con->sock);
  2269. }
  2270. ret = try_read(con);
  2271. if (ret < 0) {
  2272. if (ret == -EAGAIN)
  2273. continue;
  2274. con->error_msg = "socket error on read";
  2275. fault = true;
  2276. break;
  2277. }
  2278. ret = try_write(con);
  2279. if (ret < 0) {
  2280. if (ret == -EAGAIN)
  2281. continue;
  2282. con->error_msg = "socket error on write";
  2283. fault = true;
  2284. }
  2285. break; /* If we make it to here, we're done */
  2286. }
  2287. if (fault)
  2288. con_fault(con);
  2289. mutex_unlock(&con->mutex);
  2290. if (fault)
  2291. con_fault_finish(con);
  2292. con->ops->put(con);
  2293. }
  2294. /*
  2295. * Generic error/fault handler. A retry mechanism is used with
  2296. * exponential backoff
  2297. */
  2298. static void con_fault(struct ceph_connection *con)
  2299. {
  2300. pr_warning("%s%lld %s %s\n", ENTITY_NAME(con->peer_name),
  2301. ceph_pr_addr(&con->peer_addr.in_addr), con->error_msg);
  2302. dout("fault %p state %lu to peer %s\n",
  2303. con, con->state, ceph_pr_addr(&con->peer_addr.in_addr));
  2304. WARN_ON(con->state != CON_STATE_CONNECTING &&
  2305. con->state != CON_STATE_NEGOTIATING &&
  2306. con->state != CON_STATE_OPEN);
  2307. con_close_socket(con);
  2308. if (con_flag_test(con, CON_FLAG_LOSSYTX)) {
  2309. dout("fault on LOSSYTX channel, marking CLOSED\n");
  2310. con->state = CON_STATE_CLOSED;
  2311. return;
  2312. }
  2313. if (con->in_msg) {
  2314. BUG_ON(con->in_msg->con != con);
  2315. con->in_msg->con = NULL;
  2316. ceph_msg_put(con->in_msg);
  2317. con->in_msg = NULL;
  2318. con->ops->put(con);
  2319. }
  2320. /* Requeue anything that hasn't been acked */
  2321. list_splice_init(&con->out_sent, &con->out_queue);
  2322. /* If there are no messages queued or keepalive pending, place
  2323. * the connection in a STANDBY state */
  2324. if (list_empty(&con->out_queue) &&
  2325. !con_flag_test(con, CON_FLAG_KEEPALIVE_PENDING)) {
  2326. dout("fault %p setting STANDBY clearing WRITE_PENDING\n", con);
  2327. con_flag_clear(con, CON_FLAG_WRITE_PENDING);
  2328. con->state = CON_STATE_STANDBY;
  2329. } else {
  2330. /* retry after a delay. */
  2331. con->state = CON_STATE_PREOPEN;
  2332. if (con->delay == 0)
  2333. con->delay = BASE_DELAY_INTERVAL;
  2334. else if (con->delay < MAX_DELAY_INTERVAL)
  2335. con->delay *= 2;
  2336. con_flag_set(con, CON_FLAG_BACKOFF);
  2337. queue_con(con);
  2338. }
  2339. }
  2340. /*
  2341. * initialize a new messenger instance
  2342. */
  2343. void ceph_messenger_init(struct ceph_messenger *msgr,
  2344. struct ceph_entity_addr *myaddr,
  2345. u32 supported_features,
  2346. u32 required_features,
  2347. bool nocrc)
  2348. {
  2349. msgr->supported_features = supported_features;
  2350. msgr->required_features = required_features;
  2351. spin_lock_init(&msgr->global_seq_lock);
  2352. if (myaddr)
  2353. msgr->inst.addr = *myaddr;
  2354. /* select a random nonce */
  2355. msgr->inst.addr.type = 0;
  2356. get_random_bytes(&msgr->inst.addr.nonce, sizeof(msgr->inst.addr.nonce));
  2357. encode_my_addr(msgr);
  2358. msgr->nocrc = nocrc;
  2359. atomic_set(&msgr->stopping, 0);
  2360. dout("%s %p\n", __func__, msgr);
  2361. }
  2362. EXPORT_SYMBOL(ceph_messenger_init);
  2363. static void clear_standby(struct ceph_connection *con)
  2364. {
  2365. /* come back from STANDBY? */
  2366. if (con->state == CON_STATE_STANDBY) {
  2367. dout("clear_standby %p and ++connect_seq\n", con);
  2368. con->state = CON_STATE_PREOPEN;
  2369. con->connect_seq++;
  2370. WARN_ON(con_flag_test(con, CON_FLAG_WRITE_PENDING));
  2371. WARN_ON(con_flag_test(con, CON_FLAG_KEEPALIVE_PENDING));
  2372. }
  2373. }
  2374. /*
  2375. * Queue up an outgoing message on the given connection.
  2376. */
  2377. void ceph_con_send(struct ceph_connection *con, struct ceph_msg *msg)
  2378. {
  2379. /* set src+dst */
  2380. msg->hdr.src = con->msgr->inst.name;
  2381. BUG_ON(msg->front.iov_len != le32_to_cpu(msg->hdr.front_len));
  2382. msg->needs_out_seq = true;
  2383. mutex_lock(&con->mutex);
  2384. if (con->state == CON_STATE_CLOSED) {
  2385. dout("con_send %p closed, dropping %p\n", con, msg);
  2386. ceph_msg_put(msg);
  2387. mutex_unlock(&con->mutex);
  2388. return;
  2389. }
  2390. BUG_ON(msg->con != NULL);
  2391. msg->con = con->ops->get(con);
  2392. BUG_ON(msg->con == NULL);
  2393. BUG_ON(!list_empty(&msg->list_head));
  2394. list_add_tail(&msg->list_head, &con->out_queue);
  2395. dout("----- %p to %s%lld %d=%s len %d+%d+%d -----\n", msg,
  2396. ENTITY_NAME(con->peer_name), le16_to_cpu(msg->hdr.type),
  2397. ceph_msg_type_name(le16_to_cpu(msg->hdr.type)),
  2398. le32_to_cpu(msg->hdr.front_len),
  2399. le32_to_cpu(msg->hdr.middle_len),
  2400. le32_to_cpu(msg->hdr.data_len));
  2401. clear_standby(con);
  2402. mutex_unlock(&con->mutex);
  2403. /* if there wasn't anything waiting to send before, queue
  2404. * new work */
  2405. if (con_flag_test_and_set(con, CON_FLAG_WRITE_PENDING) == 0)
  2406. queue_con(con);
  2407. }
  2408. EXPORT_SYMBOL(ceph_con_send);
  2409. /*
  2410. * Revoke a message that was previously queued for send
  2411. */
  2412. void ceph_msg_revoke(struct ceph_msg *msg)
  2413. {
  2414. struct ceph_connection *con = msg->con;
  2415. if (!con)
  2416. return; /* Message not in our possession */
  2417. mutex_lock(&con->mutex);
  2418. if (!list_empty(&msg->list_head)) {
  2419. dout("%s %p msg %p - was on queue\n", __func__, con, msg);
  2420. list_del_init(&msg->list_head);
  2421. BUG_ON(msg->con == NULL);
  2422. msg->con->ops->put(msg->con);
  2423. msg->con = NULL;
  2424. msg->hdr.seq = 0;
  2425. ceph_msg_put(msg);
  2426. }
  2427. if (con->out_msg == msg) {
  2428. dout("%s %p msg %p - was sending\n", __func__, con, msg);
  2429. con->out_msg = NULL;
  2430. if (con->out_kvec_is_msg) {
  2431. con->out_skip = con->out_kvec_bytes;
  2432. con->out_kvec_is_msg = false;
  2433. }
  2434. msg->hdr.seq = 0;
  2435. ceph_msg_put(msg);
  2436. }
  2437. mutex_unlock(&con->mutex);
  2438. }
  2439. /*
  2440. * Revoke a message that we may be reading data into
  2441. */
  2442. void ceph_msg_revoke_incoming(struct ceph_msg *msg)
  2443. {
  2444. struct ceph_connection *con;
  2445. BUG_ON(msg == NULL);
  2446. if (!msg->con) {
  2447. dout("%s msg %p null con\n", __func__, msg);
  2448. return; /* Message not in our possession */
  2449. }
  2450. con = msg->con;
  2451. mutex_lock(&con->mutex);
  2452. if (con->in_msg == msg) {
  2453. unsigned int front_len = le32_to_cpu(con->in_hdr.front_len);
  2454. unsigned int middle_len = le32_to_cpu(con->in_hdr.middle_len);
  2455. unsigned int data_len = le32_to_cpu(con->in_hdr.data_len);
  2456. /* skip rest of message */
  2457. dout("%s %p msg %p revoked\n", __func__, con, msg);
  2458. con->in_base_pos = con->in_base_pos -
  2459. sizeof(struct ceph_msg_header) -
  2460. front_len -
  2461. middle_len -
  2462. data_len -
  2463. sizeof(struct ceph_msg_footer);
  2464. ceph_msg_put(con->in_msg);
  2465. con->in_msg = NULL;
  2466. con->in_tag = CEPH_MSGR_TAG_READY;
  2467. con->in_seq++;
  2468. } else {
  2469. dout("%s %p in_msg %p msg %p no-op\n",
  2470. __func__, con, con->in_msg, msg);
  2471. }
  2472. mutex_unlock(&con->mutex);
  2473. }
  2474. /*
  2475. * Queue a keepalive byte to ensure the tcp connection is alive.
  2476. */
  2477. void ceph_con_keepalive(struct ceph_connection *con)
  2478. {
  2479. dout("con_keepalive %p\n", con);
  2480. mutex_lock(&con->mutex);
  2481. clear_standby(con);
  2482. mutex_unlock(&con->mutex);
  2483. if (con_flag_test_and_set(con, CON_FLAG_KEEPALIVE_PENDING) == 0 &&
  2484. con_flag_test_and_set(con, CON_FLAG_WRITE_PENDING) == 0)
  2485. queue_con(con);
  2486. }
  2487. EXPORT_SYMBOL(ceph_con_keepalive);
  2488. static void ceph_msg_data_init(struct ceph_msg_data *data)
  2489. {
  2490. data->type = CEPH_MSG_DATA_NONE;
  2491. }
  2492. void ceph_msg_data_set_pages(struct ceph_msg *msg, struct page **pages,
  2493. size_t length, size_t alignment)
  2494. {
  2495. BUG_ON(!pages);
  2496. BUG_ON(!length);
  2497. BUG_ON(msg->p.type != CEPH_MSG_DATA_NONE);
  2498. msg->p.type = CEPH_MSG_DATA_PAGES;
  2499. msg->p.pages = pages;
  2500. msg->p.length = length;
  2501. msg->p.alignment = alignment & ~PAGE_MASK;
  2502. }
  2503. EXPORT_SYMBOL(ceph_msg_data_set_pages);
  2504. void ceph_msg_data_set_pagelist(struct ceph_msg *msg,
  2505. struct ceph_pagelist *pagelist)
  2506. {
  2507. BUG_ON(!pagelist);
  2508. BUG_ON(!pagelist->length);
  2509. BUG_ON(msg->l.type != CEPH_MSG_DATA_NONE);
  2510. msg->l.type = CEPH_MSG_DATA_PAGELIST;
  2511. msg->l.pagelist = pagelist;
  2512. }
  2513. EXPORT_SYMBOL(ceph_msg_data_set_pagelist);
  2514. void ceph_msg_data_set_bio(struct ceph_msg *msg, struct bio *bio)
  2515. {
  2516. BUG_ON(!bio);
  2517. BUG_ON(msg->b.type != CEPH_MSG_DATA_NONE);
  2518. msg->b.type = CEPH_MSG_DATA_BIO;
  2519. msg->b.bio = bio;
  2520. }
  2521. EXPORT_SYMBOL(ceph_msg_data_set_bio);
  2522. void ceph_msg_data_set_trail(struct ceph_msg *msg, struct ceph_pagelist *trail)
  2523. {
  2524. BUG_ON(!trail);
  2525. BUG_ON(!trail->length);
  2526. BUG_ON(msg->b.type != CEPH_MSG_DATA_NONE);
  2527. msg->t.type = CEPH_MSG_DATA_PAGELIST;
  2528. msg->t.pagelist = trail;
  2529. }
  2530. EXPORT_SYMBOL(ceph_msg_data_set_trail);
  2531. /*
  2532. * construct a new message with given type, size
  2533. * the new msg has a ref count of 1.
  2534. */
  2535. struct ceph_msg *ceph_msg_new(int type, int front_len, gfp_t flags,
  2536. bool can_fail)
  2537. {
  2538. struct ceph_msg *m;
  2539. m = kzalloc(sizeof(*m), flags);
  2540. if (m == NULL)
  2541. goto out;
  2542. m->hdr.type = cpu_to_le16(type);
  2543. m->hdr.priority = cpu_to_le16(CEPH_MSG_PRIO_DEFAULT);
  2544. m->hdr.front_len = cpu_to_le32(front_len);
  2545. INIT_LIST_HEAD(&m->list_head);
  2546. kref_init(&m->kref);
  2547. ceph_msg_data_init(&m->p);
  2548. ceph_msg_data_init(&m->l);
  2549. ceph_msg_data_init(&m->b);
  2550. ceph_msg_data_init(&m->t);
  2551. /* front */
  2552. m->front_max = front_len;
  2553. if (front_len) {
  2554. if (front_len > PAGE_CACHE_SIZE) {
  2555. m->front.iov_base = __vmalloc(front_len, flags,
  2556. PAGE_KERNEL);
  2557. m->front_is_vmalloc = true;
  2558. } else {
  2559. m->front.iov_base = kmalloc(front_len, flags);
  2560. }
  2561. if (m->front.iov_base == NULL) {
  2562. dout("ceph_msg_new can't allocate %d bytes\n",
  2563. front_len);
  2564. goto out2;
  2565. }
  2566. } else {
  2567. m->front.iov_base = NULL;
  2568. }
  2569. m->front.iov_len = front_len;
  2570. dout("ceph_msg_new %p front %d\n", m, front_len);
  2571. return m;
  2572. out2:
  2573. ceph_msg_put(m);
  2574. out:
  2575. if (!can_fail) {
  2576. pr_err("msg_new can't create type %d front %d\n", type,
  2577. front_len);
  2578. WARN_ON(1);
  2579. } else {
  2580. dout("msg_new can't create type %d front %d\n", type,
  2581. front_len);
  2582. }
  2583. return NULL;
  2584. }
  2585. EXPORT_SYMBOL(ceph_msg_new);
  2586. /*
  2587. * Allocate "middle" portion of a message, if it is needed and wasn't
  2588. * allocated by alloc_msg. This allows us to read a small fixed-size
  2589. * per-type header in the front and then gracefully fail (i.e.,
  2590. * propagate the error to the caller based on info in the front) when
  2591. * the middle is too large.
  2592. */
  2593. static int ceph_alloc_middle(struct ceph_connection *con, struct ceph_msg *msg)
  2594. {
  2595. int type = le16_to_cpu(msg->hdr.type);
  2596. int middle_len = le32_to_cpu(msg->hdr.middle_len);
  2597. dout("alloc_middle %p type %d %s middle_len %d\n", msg, type,
  2598. ceph_msg_type_name(type), middle_len);
  2599. BUG_ON(!middle_len);
  2600. BUG_ON(msg->middle);
  2601. msg->middle = ceph_buffer_new(middle_len, GFP_NOFS);
  2602. if (!msg->middle)
  2603. return -ENOMEM;
  2604. return 0;
  2605. }
  2606. /*
  2607. * Allocate a message for receiving an incoming message on a
  2608. * connection, and save the result in con->in_msg. Uses the
  2609. * connection's private alloc_msg op if available.
  2610. *
  2611. * Returns 0 on success, or a negative error code.
  2612. *
  2613. * On success, if we set *skip = 1:
  2614. * - the next message should be skipped and ignored.
  2615. * - con->in_msg == NULL
  2616. * or if we set *skip = 0:
  2617. * - con->in_msg is non-null.
  2618. * On error (ENOMEM, EAGAIN, ...),
  2619. * - con->in_msg == NULL
  2620. */
  2621. static int ceph_con_in_msg_alloc(struct ceph_connection *con, int *skip)
  2622. {
  2623. struct ceph_msg_header *hdr = &con->in_hdr;
  2624. int middle_len = le32_to_cpu(hdr->middle_len);
  2625. struct ceph_msg *msg;
  2626. int ret = 0;
  2627. BUG_ON(con->in_msg != NULL);
  2628. BUG_ON(!con->ops->alloc_msg);
  2629. mutex_unlock(&con->mutex);
  2630. msg = con->ops->alloc_msg(con, hdr, skip);
  2631. mutex_lock(&con->mutex);
  2632. if (con->state != CON_STATE_OPEN) {
  2633. if (msg)
  2634. ceph_msg_put(msg);
  2635. return -EAGAIN;
  2636. }
  2637. if (msg) {
  2638. BUG_ON(*skip);
  2639. con->in_msg = msg;
  2640. con->in_msg->con = con->ops->get(con);
  2641. BUG_ON(con->in_msg->con == NULL);
  2642. } else {
  2643. /*
  2644. * Null message pointer means either we should skip
  2645. * this message or we couldn't allocate memory. The
  2646. * former is not an error.
  2647. */
  2648. if (*skip)
  2649. return 0;
  2650. con->error_msg = "error allocating memory for incoming message";
  2651. return -ENOMEM;
  2652. }
  2653. memcpy(&con->in_msg->hdr, &con->in_hdr, sizeof(con->in_hdr));
  2654. if (middle_len && !con->in_msg->middle) {
  2655. ret = ceph_alloc_middle(con, con->in_msg);
  2656. if (ret < 0) {
  2657. ceph_msg_put(con->in_msg);
  2658. con->in_msg = NULL;
  2659. }
  2660. }
  2661. return ret;
  2662. }
  2663. /*
  2664. * Free a generically kmalloc'd message.
  2665. */
  2666. void ceph_msg_kfree(struct ceph_msg *m)
  2667. {
  2668. dout("msg_kfree %p\n", m);
  2669. if (m->front_is_vmalloc)
  2670. vfree(m->front.iov_base);
  2671. else
  2672. kfree(m->front.iov_base);
  2673. kfree(m);
  2674. }
  2675. /*
  2676. * Drop a msg ref. Destroy as needed.
  2677. */
  2678. void ceph_msg_last_put(struct kref *kref)
  2679. {
  2680. struct ceph_msg *m = container_of(kref, struct ceph_msg, kref);
  2681. dout("ceph_msg_put last one on %p\n", m);
  2682. WARN_ON(!list_empty(&m->list_head));
  2683. /* drop middle, data, if any */
  2684. if (m->middle) {
  2685. ceph_buffer_put(m->middle);
  2686. m->middle = NULL;
  2687. }
  2688. if (ceph_msg_has_pages(m)) {
  2689. m->p.length = 0;
  2690. m->p.pages = NULL;
  2691. }
  2692. if (ceph_msg_has_pagelist(m)) {
  2693. ceph_pagelist_release(m->l.pagelist);
  2694. kfree(m->l.pagelist);
  2695. m->l.pagelist = NULL;
  2696. }
  2697. if (ceph_msg_has_trail(m))
  2698. m->t.pagelist = NULL;
  2699. if (m->pool)
  2700. ceph_msgpool_put(m->pool, m);
  2701. else
  2702. ceph_msg_kfree(m);
  2703. }
  2704. EXPORT_SYMBOL(ceph_msg_last_put);
  2705. void ceph_msg_dump(struct ceph_msg *msg)
  2706. {
  2707. pr_debug("msg_dump %p (front_max %d length %zd)\n", msg,
  2708. msg->front_max, msg->p.length);
  2709. print_hex_dump(KERN_DEBUG, "header: ",
  2710. DUMP_PREFIX_OFFSET, 16, 1,
  2711. &msg->hdr, sizeof(msg->hdr), true);
  2712. print_hex_dump(KERN_DEBUG, " front: ",
  2713. DUMP_PREFIX_OFFSET, 16, 1,
  2714. msg->front.iov_base, msg->front.iov_len, true);
  2715. if (msg->middle)
  2716. print_hex_dump(KERN_DEBUG, "middle: ",
  2717. DUMP_PREFIX_OFFSET, 16, 1,
  2718. msg->middle->vec.iov_base,
  2719. msg->middle->vec.iov_len, true);
  2720. print_hex_dump(KERN_DEBUG, "footer: ",
  2721. DUMP_PREFIX_OFFSET, 16, 1,
  2722. &msg->footer, sizeof(msg->footer), true);
  2723. }
  2724. EXPORT_SYMBOL(ceph_msg_dump);