jbuff.cpp 36 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570
  1. /*##############################################################################
  2. HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems®.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. ############################################################################## */
  13. #include "platform.h"
  14. #include <stdio.h>
  15. #include <stdlib.h>
  16. #include <stdarg.h>
  17. #include <assert.h>
  18. #include <string.h>
  19. #include <ctype.h>
  20. #include <time.h>
  21. #include <math.h>
  22. #ifndef WIN32
  23. #include <sys/mman.h>
  24. #define LARGEMEM_USE_MMAP_SIZE 0x10000 // in largemem use mmap for chunks bigger than 64K
  25. #endif
  26. #include "jbuff.hpp"
  27. #include "jexcept.hpp"
  28. #include "jmisc.hpp"
  29. #include "jutil.hpp"
  30. #include "jvmem.hpp"
  31. #ifdef _DEBUG
  32. #define KILL_CLEARS_MEMORY
  33. //#define TRACE_LARGEMEM
  34. //#define TRACE_LARGEMEM_ALLOC
  35. #define TRACE_LARGEMEM_OOM
  36. #endif
  37. #if 1
  38. #define ChunkSize 0x10000
  39. #define DOUBLE_LIMIT 0x7fffffff // avoid doubling hitting 0 and infinite loop
  40. #else
  41. #define ChunkSize 2048
  42. #define DOUBLE_LIMIT 4096
  43. #endif
  44. #define FIRST_CHUNK_SIZE 8
  45. #define DETACH_GRANULARITY 16
  46. #ifdef _DEBUG
  47. #define CHECKREADPOS(len) assertex(readPos+(len)<=length())
  48. #else
  49. #define CHECKREADPOS(len)
  50. #endif
  51. //-----------------------------------------------------------------------
  52. static inline size32_t checkMemoryBufferOverflow(size32_t curLen, size32_t inc)
  53. {
  54. size_t newLen = (size_t)curLen + (size_t)inc;
  55. if (newLen > MEMBUFFER_MAXLEN)
  56. RaiseOutOfMemException(-10, curLen, inc, false, "Exceeded maximum size");
  57. return (size32_t)newLen;
  58. }
  59. jlib_decl void *checked_realloc(void *orig, size_t newlen, size_t origlen,int errcode)
  60. {
  61. if (newlen==0) {
  62. free(orig);
  63. return NULL;
  64. }
  65. if (orig==NULL)
  66. return checked_malloc(newlen,errcode);
  67. void *ret = realloc(orig, newlen);
  68. if (!ret)
  69. RaiseOutOfMemException(errcode, newlen, origlen);
  70. return ret;
  71. }
  72. class jlib_thrown_decl COutOfMemException: public CInterface, implements IOutOfMemException
  73. {
  74. int errcode;
  75. size_t wanted;
  76. size_t got;
  77. static int recursion;
  78. bool expected;
  79. StringBuffer errorMsg;
  80. public:
  81. IMPLEMENT_IINTERFACE;
  82. COutOfMemException(int _errcode, size_t _wanted, size_t _got, bool _expected, const char *errMsg)
  83. {
  84. errcode = _errcode;
  85. wanted = _wanted;
  86. expected = _expected;
  87. got = _got;
  88. if (nullptr == errMsg)
  89. errorMsg.append("Out of Memory");
  90. else
  91. errorMsg.append(errMsg);
  92. // DebugBreak();
  93. if ((recursion++==0)&&!expected) {
  94. // Bit risky if *very* out of memory so protect against recursion and catch exceptions
  95. try {
  96. // try to log
  97. PROGLOG("Jbuff: %s (%d,%" I64F "d,%" I64F "dk)",errorMsg.str(),_errcode,(unsigned __int64)wanted,(unsigned __int64) (got/1024));
  98. PrintStackReport();
  99. PrintMemoryReport();
  100. }
  101. catch (...) {
  102. }
  103. }
  104. recursion--;
  105. };
  106. int errorCode() const { return errcode; }
  107. StringBuffer & errorMessage(StringBuffer &str) const
  108. {
  109. str.append("Jbuff: ").append(errorMsg.str()).append(" (").append((unsigned __int64)wanted);
  110. if (got)
  111. str.append(',').append((unsigned __int64)got);
  112. return str.append(")");
  113. }
  114. MessageAudience errorAudience() const { return MSGAUD_user; }
  115. };
  116. int COutOfMemException::recursion=0;
  117. IOutOfMemException *createOutOfMemException(int errcode, size_t wanted, size_t got, bool expected, const char *errMsg)
  118. {
  119. return new COutOfMemException(errcode, wanted, got, expected, errMsg);
  120. }
  121. void RaiseOutOfMemException(int errcode, size_t wanted, size_t got, bool expected, const char *errMsg)
  122. {
  123. throw createOutOfMemException(errcode, wanted, got, expected, errMsg);
  124. }
  125. MemoryAttr::MemoryAttr(size_t _len)
  126. {
  127. ptr = checked_malloc(_len,-1);
  128. len = _len;
  129. }
  130. MemoryAttr::MemoryAttr(size_t _len, const void * _ptr)
  131. {
  132. len = 0;
  133. ptr = NULL;
  134. set(_len, _ptr);
  135. }
  136. MemoryAttr::MemoryAttr(const MemoryAttr & src)
  137. {
  138. len = 0;
  139. ptr = NULL;
  140. set(src.length(), src.get());
  141. }
  142. void MemoryAttr::set(size_t _len, const void * _ptr)
  143. {
  144. memcpy(allocate(_len), _ptr, _len);
  145. }
  146. void MemoryAttr::setOwn(size_t _len, void * _ptr)
  147. {
  148. free(ptr);
  149. len = _len;
  150. ptr = _ptr;
  151. }
  152. void MemoryAttr::clear()
  153. {
  154. free(ptr);
  155. ptr = NULL;
  156. len = 0;
  157. }
  158. void * MemoryAttr::detach()
  159. {
  160. void * ret=ptr;
  161. ptr = NULL;
  162. len = 0;
  163. return ret;
  164. }
  165. int MemoryAttr::compare(const MemoryAttr & m1, const MemoryAttr & m2)
  166. {
  167. size_t len1 = m1.length();
  168. size_t len2 = m2.length();
  169. size_t len = len1;
  170. if (len1 > len2)
  171. len = len2;
  172. int compare = memcmp(m1.get(), m2.get(), len);
  173. if (compare == 0)
  174. compare = (len1 > len2) ? +1 : (len1 < len2) ? -1 : 0;
  175. return compare;
  176. }
  177. void * MemoryAttr::allocate(size_t _len)
  178. {
  179. if (_len==len)
  180. return ptr;
  181. clear();
  182. ptr = checked_malloc(_len,-2);
  183. len = _len;
  184. return ptr;
  185. }
  186. void * MemoryAttr::ensure(size_t _len)
  187. {
  188. if (_len <=len)
  189. return ptr;
  190. return reallocate(_len);
  191. }
  192. void * MemoryAttr::reallocate(size_t _len)
  193. {
  194. if (_len==len)
  195. return ptr;
  196. ptr = checked_realloc(ptr, _len, len, -9);
  197. len = _len;
  198. return ptr;
  199. }
  200. //===========================================================================
  201. void MemoryBuffer::_realloc(size32_t newLen)
  202. {
  203. if (newLen > maxLen)
  204. {
  205. assertex(ownBuffer);
  206. size32_t newMax = maxLen;
  207. //double up to a certain size, otherwise go up in chunks.
  208. if (newLen < DOUBLE_LIMIT)
  209. {
  210. if (newMax == 0)
  211. newMax = FIRST_CHUNK_SIZE;
  212. while (newLen > newMax)
  213. {
  214. size32_t newMaxTmp = checkMemoryBufferOverflow(newMax, newMax);
  215. newMax = newMaxTmp;
  216. }
  217. }
  218. else
  219. {
  220. size32_t newMaxTmp = checkMemoryBufferOverflow((newLen & ~(ChunkSize-1)), ChunkSize);
  221. /*** ((Size + 1) + (ChunkSize - 1)) & ~(ChunkSize-1) ***/
  222. newMax = newMaxTmp;
  223. }
  224. buffer =(char *)checked_realloc(buffer, newMax, maxLen, -7);
  225. maxLen = newMax;
  226. }
  227. }
  228. void MemoryBuffer::_reallocExact(size32_t newLen)
  229. {
  230. if (newLen > maxLen)
  231. {
  232. assertex(ownBuffer);
  233. buffer =(char *)checked_realloc(buffer, newLen, maxLen, -8);
  234. maxLen = newLen;
  235. }
  236. }
  237. void MemoryBuffer::init()
  238. {
  239. buffer = NULL;
  240. curLen = 0;
  241. maxLen = 0;
  242. ownBuffer = true;
  243. readPos = 0;
  244. swapEndian = false;
  245. }
  246. void *MemoryBuffer::insertDirect(unsigned offset, size32_t insertLen)
  247. {
  248. assertex(offset<=curLen);
  249. unsigned newLen = checkMemoryBufferOverflow(curLen, insertLen);
  250. _realloc(newLen);
  251. memmove(buffer + offset + insertLen, buffer + offset, curLen - offset);
  252. curLen += insertLen;
  253. return buffer+offset;
  254. }
  255. void * MemoryBuffer::ensureCapacity(unsigned max)
  256. {
  257. if (maxLen - curLen < max)
  258. {
  259. unsigned newLen = checkMemoryBufferOverflow(curLen, max);
  260. _realloc(newLen);
  261. }
  262. return buffer + curLen;
  263. }
  264. void MemoryBuffer::kill()
  265. {
  266. if (ownBuffer)
  267. free(buffer);
  268. }
  269. MemoryBuffer & MemoryBuffer::_remove(unsigned start, unsigned len)
  270. {
  271. if (start > curLen) start = curLen;
  272. if (start + len > curLen) len = curLen - start;
  273. unsigned start2 = start + len;
  274. memmove(buffer + start, buffer + start2, curLen - start2);
  275. setLength(curLen - len);
  276. return *this;
  277. }
  278. void * MemoryBuffer::reserve(unsigned size)
  279. {
  280. unsigned newLen = checkMemoryBufferOverflow(curLen, size);
  281. _realloc(newLen);
  282. void * ret = buffer + curLen;
  283. curLen += size;
  284. return ret;
  285. }
  286. void * MemoryBuffer::reserveTruncate(unsigned size)
  287. {
  288. unsigned newLen = checkMemoryBufferOverflow(curLen, size);
  289. curLen += size;
  290. _reallocExact(newLen);
  291. truncate();
  292. return buffer + curLen - size;
  293. }
  294. void MemoryBuffer::truncate()
  295. {
  296. if (maxLen>curLen) {
  297. if (curLen==0) {
  298. free(buffer);
  299. buffer = NULL;
  300. }
  301. else
  302. buffer = (char *)realloc(buffer, curLen);
  303. maxLen = curLen;
  304. }
  305. }
  306. void MemoryBuffer::resetBuffer()
  307. {
  308. kill();
  309. init();
  310. }
  311. MemoryBuffer & MemoryBuffer::_reverse()
  312. {
  313. unsigned max = curLen/2;
  314. char * end = buffer + curLen;
  315. unsigned idx;
  316. for (idx = 0; idx < max; idx++)
  317. {
  318. char temp = buffer[idx];
  319. end--;
  320. buffer[idx] = *end;
  321. *end = temp;
  322. }
  323. return *this;
  324. }
  325. void MemoryBuffer::setBuffer(size_t len, void * _buffer, bool takeOwnership)
  326. {
  327. assertex((size32_t)len == len);
  328. kill();
  329. buffer = (char *) _buffer;
  330. if (len) assertex(buffer);
  331. curLen = maxLen = (size32_t)len;
  332. ownBuffer = takeOwnership;
  333. readPos = 0;
  334. }
  335. void *MemoryBuffer::detach()
  336. {
  337. void *ret;
  338. if (ownBuffer) {
  339. if (maxLen>curLen+DETACH_GRANULARITY)
  340. buffer = (char *)realloc(buffer,curLen);
  341. ret = buffer;
  342. }
  343. else {
  344. ret = memcpy(checked_malloc(curLen,-3), buffer, curLen);
  345. }
  346. init();
  347. return ret;
  348. }
  349. void *MemoryBuffer::detachOwn()
  350. {
  351. assertex(ownBuffer);
  352. void *ret = buffer;
  353. init();
  354. return ret;
  355. }
  356. void MemoryBuffer::setLength(unsigned len)
  357. {
  358. if (len > curLen)
  359. {
  360. _realloc(len);
  361. memset(buffer + curLen, 0, len-curLen);
  362. }
  363. else
  364. {
  365. #ifdef KILL_CLEARS_MEMORY
  366. if (curLen)
  367. memset(buffer + len, 'x', curLen-len);
  368. #endif
  369. }
  370. curLen = len;
  371. }
  372. void MemoryBuffer::setWritePos(unsigned len)
  373. {
  374. if (len > curLen)
  375. _realloc(len);
  376. curLen = len;
  377. }
  378. #define SWAP(x, y, t) { t t_##x = x; x = y; y = t_##x; }
  379. void MemoryBuffer::swapWith(MemoryBuffer & other)
  380. {
  381. //swap two string buffers. Used for efficiently moving a string on in a pipeline etc.
  382. SWAP(buffer, other.buffer, char *);
  383. SWAP(curLen, other.curLen, size32_t);
  384. SWAP(maxLen, other.maxLen, size32_t);
  385. SWAP(readPos, other.readPos, size32_t);
  386. SWAP(swapEndian, other.swapEndian, bool);
  387. }
  388. //-----------------------------------------------------------------------
  389. MemoryBuffer::MemoryBuffer(size_t initial)
  390. {
  391. assertex((size32_t)initial == initial);
  392. init();
  393. _realloc((size32_t)initial);
  394. }
  395. MemoryBuffer::MemoryBuffer(MemoryBuffer & value __attribute__((unused)))
  396. {
  397. assertex(!"This should never be used");
  398. }
  399. MemoryBuffer::MemoryBuffer(size_t len, const void * newBuffer)
  400. {
  401. init();
  402. assertex((size32_t)len == len);
  403. append((size32_t)len, newBuffer);
  404. }
  405. MemoryBuffer & MemoryBuffer::append(char value)
  406. {
  407. unsigned newLen = checkMemoryBufferOverflow(curLen, 1);
  408. _realloc(newLen);
  409. buffer[curLen] = value;
  410. ++curLen;
  411. return *this;
  412. }
  413. MemoryBuffer & MemoryBuffer::append(unsigned char value)
  414. {
  415. unsigned newLen = checkMemoryBufferOverflow(curLen, 1);
  416. _realloc(newLen);
  417. buffer[curLen] = value;
  418. ++curLen;
  419. return *this;
  420. }
  421. MemoryBuffer & MemoryBuffer::append(bool value)
  422. {
  423. unsigned newLen = checkMemoryBufferOverflow(curLen, 1);
  424. _realloc(newLen);
  425. buffer[curLen] = (value==0)?0:1;
  426. ++curLen;
  427. return *this;
  428. }
  429. MemoryBuffer & MemoryBuffer::append(const char * value)
  430. {
  431. if (value)
  432. return append((size32_t)strlen(value)+1,value);
  433. else
  434. return append((char)0);
  435. }
  436. MemoryBuffer & MemoryBuffer::append(const unsigned char * value)
  437. {
  438. return append((const char *) value);
  439. }
  440. MemoryBuffer & MemoryBuffer::append(unsigned len, const void * value)
  441. {
  442. unsigned newLen = checkMemoryBufferOverflow(curLen, len);
  443. _realloc(newLen);
  444. memcpy(buffer + curLen, value, len);
  445. curLen += len;
  446. return *this;
  447. }
  448. MemoryBuffer & MemoryBuffer::append(double value)
  449. {
  450. return appendEndian(sizeof(value), &value);
  451. }
  452. MemoryBuffer & MemoryBuffer::append(float value)
  453. {
  454. return appendEndian(sizeof(value), &value);
  455. }
  456. MemoryBuffer & MemoryBuffer::append(short value)
  457. {
  458. return appendEndian(sizeof(value), &value);
  459. }
  460. MemoryBuffer & MemoryBuffer::append(unsigned short value)
  461. {
  462. return appendEndian(sizeof(value), &value);
  463. }
  464. MemoryBuffer & MemoryBuffer::append(int value)
  465. {
  466. return appendEndian(sizeof(value), &value);
  467. }
  468. MemoryBuffer & MemoryBuffer::append(unsigned value)
  469. {
  470. return appendEndian(sizeof(value), &value);
  471. }
  472. #if 0
  473. MemoryBuffer & MemoryBuffer::append(long value)
  474. {
  475. return appendEndian(sizeof(value), &value);
  476. }
  477. MemoryBuffer & MemoryBuffer::append(unsigned long value)
  478. {
  479. return appendEndian(sizeof(value), &value);
  480. }
  481. #endif
  482. MemoryBuffer & MemoryBuffer::append(__int64 value)
  483. {
  484. return appendEndian(sizeof(value), &value);
  485. }
  486. MemoryBuffer & MemoryBuffer::append(unsigned __int64 value)
  487. {
  488. return appendEndian(sizeof(value), &value);
  489. }
  490. MemoryBuffer & MemoryBuffer::appendPacked(unsigned __int64 value)
  491. {
  492. //Append bytes with the top bit set until the value is less than 0x80
  493. while (value >= 0x80)
  494. {
  495. byte next = ((byte)value) | 0x80;
  496. append(next);
  497. value >>= 7;
  498. }
  499. return append((byte)value);
  500. }
  501. MemoryBuffer & MemoryBuffer::append(const MemoryBuffer & value)
  502. {
  503. size32_t SourceLen = value.length();
  504. unsigned newLen = checkMemoryBufferOverflow(curLen, SourceLen);
  505. _realloc(newLen);
  506. memcpy(buffer + curLen, value.toByteArray(), SourceLen);
  507. curLen += SourceLen;
  508. return *this;
  509. }
  510. MemoryBuffer & MemoryBuffer::appendBytes(unsigned char value, unsigned count)
  511. {
  512. unsigned newLen = checkMemoryBufferOverflow(curLen, count);
  513. _realloc(newLen);
  514. memset(buffer+curLen, value, count);
  515. curLen+=count;
  516. return *this;
  517. }
  518. MemoryBuffer & MemoryBuffer::appendEndian(size32_t len, const void * value)
  519. {
  520. unsigned newLen = checkMemoryBufferOverflow(curLen, len);
  521. _realloc(newLen);
  522. if (swapEndian)
  523. _cpyrevn(buffer + curLen, value, len);
  524. else
  525. memcpy(buffer + curLen, value, len);
  526. curLen += len;
  527. return *this;
  528. }
  529. MemoryBuffer & MemoryBuffer::appendSwap(size32_t len, const void * value)
  530. {
  531. unsigned newLen = checkMemoryBufferOverflow(curLen, len);
  532. _realloc(newLen);
  533. _cpyrevn(buffer + curLen, value, len);
  534. curLen += len;
  535. return *this;
  536. }
  537. MemoryBuffer &MemoryBuffer::appendFile(const char *fileName)
  538. {
  539. char buf[1024];
  540. int h = _open(fileName, _O_BINARY | _O_RDONLY | _O_SEQUENTIAL);
  541. if (h == HFILE_ERROR)
  542. throw MakeStringException(0, "MemoryBuffer: Error reading file : %s", fileName);
  543. append(fileName);
  544. unsigned fileSize = _lseek(h, 0, FILE_END);
  545. _lseek(h, 0, FILE_BEGIN);
  546. append(fileSize);
  547. int r;
  548. while ((r = _read(h, buf, 1024)) != 0)
  549. {
  550. if (-1==r) throw makeErrnoException("MemoryBuffer::appendFile");
  551. append(r, buf);
  552. }
  553. _close(h);
  554. return *this;
  555. }
  556. MemoryBuffer & MemoryBuffer::read(char & value)
  557. {
  558. CHECKREADPOS(sizeof(value));
  559. value = buffer[readPos++];
  560. return *this;
  561. }
  562. MemoryBuffer & MemoryBuffer::read(unsigned char & value)
  563. {
  564. CHECKREADPOS(sizeof(value));
  565. value = buffer[readPos++];
  566. return *this;
  567. }
  568. MemoryBuffer & MemoryBuffer::read(bool & value)
  569. {
  570. CHECKREADPOS(sizeof(value));
  571. char _value = buffer[readPos++];
  572. value = (_value==0 ? false : true);
  573. return *this;
  574. }
  575. MemoryBuffer & MemoryBuffer::read(StringAttr & value)
  576. {
  577. char * src = buffer + readPos;
  578. size32_t len = (size32_t)strlen(src);
  579. CHECKREADPOS(len+1);
  580. value.set(src, len);
  581. readPos += (len+1);
  582. return *this;
  583. }
  584. MemoryBuffer & MemoryBuffer::read(StringBuffer & value)
  585. {
  586. char * src = buffer + readPos;
  587. size32_t len = (size32_t)strlen(src);
  588. CHECKREADPOS(len+1);
  589. value.append(len, src);
  590. readPos += (len+1);
  591. return *this;
  592. }
  593. MemoryBuffer & MemoryBuffer::read(const char * &value)
  594. {
  595. value = buffer+readPos;
  596. size32_t len = (size32_t)strlen(value);
  597. CHECKREADPOS(len+1);
  598. readPos += (len+1);
  599. return *this;
  600. }
  601. MemoryBuffer & MemoryBuffer::read(size32_t len, void * value)
  602. {
  603. CHECKREADPOS(len);
  604. memcpy(value, buffer + readPos, len);
  605. readPos += len;
  606. return *this;
  607. }
  608. MemoryBuffer & MemoryBuffer::read(double & value)
  609. {
  610. return readEndian(sizeof(value), &value);
  611. }
  612. MemoryBuffer & MemoryBuffer::read(float & value)
  613. {
  614. return readEndian(sizeof(value), &value);
  615. }
  616. MemoryBuffer & MemoryBuffer::read(short & value)
  617. {
  618. return readEndian(sizeof(value), &value);
  619. }
  620. MemoryBuffer & MemoryBuffer::read(unsigned short & value)
  621. {
  622. return readEndian(sizeof(value), &value);
  623. }
  624. MemoryBuffer & MemoryBuffer::read(int & value)
  625. {
  626. return readEndian(sizeof(value), &value);
  627. }
  628. MemoryBuffer & MemoryBuffer::read(unsigned & value)
  629. {
  630. return readEndian(sizeof(value), &value);
  631. }
  632. #if 0
  633. MemoryBuffer & MemoryBuffer::read(unsigned long & value)
  634. {
  635. return readEndian(sizeof(value), &value);
  636. }
  637. MemoryBuffer & MemoryBuffer::read(long & value)
  638. {
  639. return readEndian(sizeof(value), &value);
  640. }
  641. #endif
  642. MemoryBuffer & MemoryBuffer::read(unsigned __int64 & value)
  643. {
  644. return readEndian(sizeof(value), &value);
  645. }
  646. MemoryBuffer & MemoryBuffer::read(__int64 & value)
  647. {
  648. return readEndian(sizeof(value), &value);
  649. }
  650. const byte * MemoryBuffer::readDirect(size32_t len)
  651. {
  652. CHECKREADPOS(len);
  653. const byte * ret = (const byte *)buffer + readPos;
  654. readPos += len;
  655. return ret;
  656. }
  657. unsigned __int64 MemoryBuffer::readPacked()
  658. {
  659. unsigned __int64 value = 0;
  660. unsigned shift = 0;
  661. loop
  662. {
  663. byte next;
  664. read(next);
  665. value = value | (((unsigned __int64)(next & 0x7f)) << shift);
  666. if (!(next & 0x80))
  667. break;
  668. shift += 7;
  669. }
  670. return value;
  671. }
  672. MemoryBuffer & MemoryBuffer::readPacked(unsigned & value)
  673. {
  674. unsigned __int64 serializedValue = readPacked();
  675. dbgassertex((unsigned)serializedValue == serializedValue);
  676. value = (unsigned)serializedValue;
  677. return *this;
  678. }
  679. MemoryBuffer & MemoryBuffer::readPacked(unsigned __int64 & value)
  680. {
  681. value = readPacked();
  682. return *this;
  683. }
  684. MemoryBuffer & MemoryBuffer::skip(unsigned len)
  685. {
  686. CHECKREADPOS(len);
  687. readPos += len;
  688. return *this;
  689. }
  690. void MemoryBuffer::writeDirect(size32_t pos,size32_t len,const void *buf)
  691. {
  692. assertex(pos+len<=curLen); // does not extend
  693. memcpy(buffer+pos,buf,len);
  694. }
  695. void MemoryBuffer::writeEndianDirect(size32_t pos,size32_t len,const void *buf)
  696. {
  697. assertex(pos+len<=curLen); // does not extend
  698. if (swapEndian)
  699. _cpyrevn(buffer+pos,buf,len);
  700. else
  701. memcpy(buffer+pos,buf,len);
  702. }
  703. MemoryBuffer & MemoryBuffer::readEndian(size32_t len, void * value)
  704. {
  705. CHECKREADPOS(len);
  706. if (swapEndian)
  707. _cpyrevn(value, buffer + readPos, len);
  708. else
  709. memcpy(value, buffer + readPos, len);
  710. readPos += len;
  711. return *this;
  712. }
  713. MemoryBuffer & MemoryBuffer::readSwap(size32_t len, void * value)
  714. {
  715. CHECKREADPOS(len);
  716. _cpyrevn(value, buffer + readPos, len);
  717. readPos += len;
  718. return *this;
  719. }
  720. MemoryBuffer &MemoryBuffer::readFile(StringAttr &fileName)
  721. {
  722. read(fileName);
  723. unsigned fileSize;
  724. read(fileSize);
  725. int h = _open(fileName.get(), _O_WRONLY|_O_CREAT|_O_TRUNC|_O_BINARY|_O_SEQUENTIAL, _S_IREAD | _S_IWRITE);
  726. if (h == HFILE_ERROR)
  727. throw MakeStringException(0, "MemoryBuffer: Unable to create file : %s, error=%d", fileName.get(), GetLastError());
  728. CHECKREADPOS(fileSize);
  729. int w;
  730. while (fileSize) {
  731. w = _write(h, buffer+readPos, fileSize);
  732. if (w == 0) {
  733. _close(h);
  734. throw MakeStringException(0, "MemoryBuffer: Disk full writing %d to file : %s", fileSize, fileName.get());
  735. }
  736. if (w == -1)
  737. {
  738. _close(h);
  739. throw MakeStringException(0, "MemoryBuffer: Error writing to file : %s, error=%d", fileName.get(), GetLastError());
  740. }
  741. readPos += (size32_t)w;
  742. fileSize -= (size32_t)w;
  743. }
  744. _close(h);
  745. return *this;
  746. }
  747. MemoryBuffer & MemoryBuffer::rewrite(size32_t pos)
  748. {
  749. assertex(pos<=maxLen);
  750. curLen = pos;
  751. if (readPos>pos)
  752. readPos = pos;
  753. return *this;
  754. }
  755. MemoryBuffer & MemoryBuffer::reset(size32_t pos)
  756. {
  757. CHECKREADPOS(pos-readPos);
  758. readPos = pos;
  759. return *this;
  760. }
  761. #if 0
  762. void MemoryBuffer::getBytes(int srcBegin, int srcEnd, char * target)
  763. {
  764. memcpy(target, buffer + srcBegin, srcEnd - srcBegin);
  765. }
  766. MemoryBuffer & MemoryBuffer::remove(unsigned start, unsigned len)
  767. {
  768. return (MemoryBuffer &)_remove(start, len);
  769. }
  770. #endif
  771. int MemoryBuffer::setEndian(int endian)
  772. {
  773. assertex((endian == __LITTLE_ENDIAN) || (endian == __BIG_ENDIAN));
  774. bool wasSwapped = setSwapEndian(endian != __BYTE_ORDER);
  775. return wasSwapped ? (__BYTE_ORDER ^ __LITTLE_ENDIAN ^ __BIG_ENDIAN) : __BYTE_ORDER;
  776. }
  777. bool MemoryBuffer::setSwapEndian(bool swap)
  778. {
  779. bool saved = swapEndian;
  780. swapEndian = swap;
  781. return saved;
  782. }
  783. MemoryBuffer & serialize(MemoryBuffer & buffer, const MemoryAttr & value)
  784. {
  785. size32_t length = (size32_t)value.length();
  786. buffer.append(length).append(length, value.get());
  787. return buffer;
  788. }
  789. MemoryBuffer & deserialize(MemoryBuffer & buffer, MemoryAttr & value)
  790. {
  791. unsigned length;
  792. buffer.read(length);
  793. void * target = value.allocate(length);
  794. buffer.read(length, target);
  795. return buffer;
  796. }
  797. MemoryBuffer & serialize(MemoryBuffer & buffer, const char * value)
  798. {
  799. if (value)
  800. {
  801. unsigned length = (size32_t)strlen(value);
  802. buffer.append(length).append(length, value);
  803. }
  804. else
  805. buffer.append((unsigned)-1);
  806. return buffer;
  807. }
  808. MemoryBuffer & deserialize(MemoryBuffer & buffer, StringAttr & value)
  809. {
  810. unsigned length;
  811. buffer.read(length);
  812. if (length == (unsigned)-1)
  813. value.clear();
  814. else
  815. {
  816. char * target = (char *)checked_malloc(length+1,-4);
  817. buffer.read(length, target);
  818. target[length] = 0;
  819. value.setown(target);
  820. }
  821. return buffer;
  822. }
  823. // =====================================================================================================
  824. const char * MemoryAttr2IStringVal::str() const
  825. {
  826. UNIMPLEMENTED;
  827. }
  828. // =====================================================================================================
  829. static memsize_t LMsemlimit=0;
  830. static memsize_t LMtotal=0;
  831. static CriticalSection LMsemsect;
  832. static Owned<ILargeMemLimitNotify> LMnotify;
  833. static bool LMlocked = false;
  834. void setLargeMemLimitNotify(memsize_t size,ILargeMemLimitNotify *notify)
  835. {
  836. CriticalBlock block(LMsemsect);
  837. LMsemlimit = size;
  838. LMnotify.set(notify);
  839. if (LMlocked&&(LMtotal<LMsemlimit))
  840. LMlocked = false;
  841. }
  842. inline void incLargeMemTotal(memsize_t sz)
  843. {
  844. if (sz) {
  845. CriticalBlock block(LMsemsect);
  846. LMtotal += sz;
  847. #ifdef TRACE_LARGEMEM
  848. if ((LMtotal/0x100000)!=((LMtotal-sz)/0x100000))
  849. PROGLOG("LARGEMEM(+): %" I64F "d",(offset_t)LMtotal);
  850. #endif
  851. if (!LMlocked&&LMnotify.get()&&(LMtotal>=LMsemlimit)) {
  852. LMlocked = true;
  853. DBGLOG("LargeMemTotal limit exceeded: %" I64F "d",(offset_t)LMtotal);
  854. if (!LMnotify->take(LMtotal)) {
  855. LMtotal -= sz;
  856. LMlocked = false;
  857. throw createOutOfMemException(-9,sz, LMtotal);
  858. }
  859. DBGLOG("LargeMem taken");
  860. }
  861. }
  862. }
  863. inline void decLargeMemTotal(memsize_t sz)
  864. {
  865. if (sz) {
  866. CriticalBlock block(LMsemsect);
  867. LMtotal -= sz;
  868. #ifdef TRACE_LARGEMEM
  869. if ((LMtotal/0x100000)!=((LMtotal+sz)/0x100000))
  870. PROGLOG("LARGEMEM(-): %" I64F "d",(offset_t)LMtotal);
  871. #endif
  872. if (LMlocked) {
  873. if (LMtotal<LMsemlimit) {
  874. DBGLOG("LargeMemTotal limit reduced to %" I64F "d",(offset_t)LMtotal);
  875. LMlocked = false;
  876. if (LMnotify.get())
  877. LMnotify->give(LMtotal);
  878. }
  879. }
  880. }
  881. }
  882. void CLargeMemoryAllocator::allocchunkmem()
  883. {
  884. #ifdef LARGEMEM_USE_MMAP_SIZE
  885. size32_t masize = VMPAGEROUND(chunk.max);
  886. if (masize>=LARGEMEM_USE_MMAP_SIZE) { // use mmap
  887. chunk.base = (byte *) mmap(NULL,masize,PROT_READ|PROT_WRITE,MAP_PRIVATE|MAP_NORESERVE|MAP_ANONYMOUS,-1,0);
  888. if (chunk.base == (byte *)MAP_FAILED)
  889. chunk.base = NULL;
  890. #ifdef TRACE_LARGEMEM_ALLOC
  891. PROGLOG("CLargeMemoryAllocator::allocchunkmem mmaped %d at %p",masize,chunk.base);
  892. #endif
  893. return;
  894. }
  895. #endif
  896. chunk.base = (byte *)malloc(chunk.max);
  897. #ifdef TRACE_LARGEMEM_ALLOC
  898. PROGLOG("CLargeMemoryAllocator::allocchunkmem malloced %d at %p",chunk.max,chunk.base);
  899. #endif
  900. }
  901. void CLargeMemoryAllocator::disposechunkmem()
  902. {
  903. #ifdef LARGEMEM_USE_MMAP_SIZE
  904. size32_t masize = VMPAGEROUND(chunk.max);
  905. if (masize>=LARGEMEM_USE_MMAP_SIZE) { // use mmap
  906. munmap(chunk.base,masize);
  907. return;
  908. }
  909. #endif
  910. free(chunk.base);
  911. }
  912. bool CLargeMemoryAllocator::newchunk(size32_t sz,size32_t extra,bool exceptionwanted)
  913. {
  914. size32_t newchunksz = (sz>chunkmin)?sz:chunkmin;
  915. if (maxallocated()+newchunksz+extra>totalmax) {
  916. #ifdef TRACE_LARGEMEM_OOM
  917. PrintStackReport();
  918. PROGLOG("OOM.1 wanted sz=%d, extra = %d, maxallocated=%" I64F "d, newchunksz=%u, totalmax=%" I64F "d",sz,extra,(offset_t)maxallocated(),newchunksz,(offset_t)totalmax);
  919. #endif
  920. if (exceptionwanted) {
  921. throw createOutOfMemException(-5,sz, maxallocated(),true);
  922. }
  923. return false;
  924. }
  925. if (chunk.size) {
  926. Chunk *p = new Chunk;
  927. *p = chunk;
  928. chunk.prev = p;
  929. atot += chunk.size;
  930. }
  931. else if (chunk.max) {
  932. decLargeMemTotal(chunk.max);
  933. amax -= chunk.max;
  934. disposechunkmem();
  935. }
  936. chunk.max = newchunksz;
  937. allocchunkmem();
  938. chunk.size = 0;
  939. if (!chunk.base) {
  940. // restore prev
  941. if (chunk.prev) {
  942. Chunk *p = chunk.prev;
  943. chunk = *p;
  944. atot -= chunk.size;
  945. delete p;
  946. }
  947. else
  948. chunk.max = 0;
  949. #ifdef TRACE_LARGEMEM_OOM
  950. PrintStackReport();
  951. PROGLOG("OOM.2 wanted sz=%d, extra = %d, maxallocated=%" I64F "d, newchunksz=%u, totalmax=%" I64F "d",sz,extra,(offset_t)maxallocated(),newchunksz,(offset_t)totalmax);
  952. #endif
  953. if (throwexception) {
  954. throw createOutOfMemException(-6,sz, maxallocated(),true);
  955. }
  956. return false;
  957. }
  958. amax += chunk.max;
  959. incLargeMemTotal(newchunksz);
  960. return true;
  961. }
  962. void CLargeMemoryAllocator::reset()
  963. {
  964. decLargeMemTotal(maxallocated());
  965. disposechunkmem();
  966. while (chunk.prev) {
  967. Chunk *p = chunk.prev;
  968. chunk = *chunk.prev;
  969. delete p;
  970. disposechunkmem();
  971. }
  972. chunk.max = 0;
  973. chunk.base = NULL;
  974. chunk.size = 0;
  975. atot = 0;
  976. amax = 0;
  977. }
  978. void CLargeMemoryAllocator::reduceSize(memsize_t amount)
  979. {
  980. if (amount<=chunk.size) {
  981. chunk.size-=amount;
  982. return;
  983. }
  984. memsize_t reduced = 0;
  985. do {
  986. amount -= chunk.size;
  987. reduced += chunk.max;
  988. disposechunkmem();
  989. amax -= chunk.max;
  990. Chunk *p = chunk.prev;
  991. chunk = *p;
  992. atot -= chunk.size;
  993. delete p;
  994. } while (amount>chunk.size);
  995. chunk.size-=amount;
  996. decLargeMemTotal(reduced);
  997. }
  998. void CLargeMemoryAllocator::setSize(memsize_t pos)
  999. {
  1000. memsize_t sz = allocated();
  1001. assertex(sz>=pos);
  1002. reduceSize(sz-pos);
  1003. }
  1004. byte *CLargeMemoryAllocator::next(memsize_t pos,size32_t &size) // this should not be used for small jumps as it is slow
  1005. {
  1006. memsize_t sz = allocated();
  1007. if (sz<=pos) {
  1008. size = 0;
  1009. return NULL;
  1010. }
  1011. memsize_t dif = sz-pos; // how much to go back
  1012. Chunk *p = &chunk;
  1013. while (dif>p->size) {
  1014. dif -= p->size;
  1015. p = p->prev;
  1016. }
  1017. size = (size32_t)dif; // must be smaller than chunk
  1018. return p->base+p->size-dif;
  1019. }
  1020. CLargeMemoryAllocator::CLargeMemoryAllocator()
  1021. {
  1022. // values overwritten by init
  1023. throwexception = true;
  1024. totalmax = 0;
  1025. chunkmin = 0x1000;
  1026. chunk.prev = NULL;
  1027. chunk.max = 0;
  1028. chunk.base = NULL;
  1029. chunk.size = 0;
  1030. atot = 0;
  1031. amax = 0;
  1032. }
  1033. void CLargeMemoryAllocator::init(memsize_t _totalmax,size32_t _chunkmin,bool _throwexception)
  1034. {
  1035. throwexception = _throwexception;
  1036. totalmax = _totalmax;
  1037. chunkmin = _chunkmin;
  1038. chunk.prev = NULL;
  1039. chunk.max = 0;
  1040. chunk.base = NULL;
  1041. chunk.size = 0;
  1042. atot = 0;
  1043. amax = 0;
  1044. }
  1045. MemoryBuffer &CLargeMemoryAllocator::serialize(MemoryBuffer &mb)
  1046. {
  1047. memsize_t al = allocated();
  1048. size32_t sz = (size32_t)al;
  1049. if (sz!=al)
  1050. throw MakeStringException(-1,"CLargeMemoryAllocator::serialize overflow");
  1051. byte *d = (byte *)mb.reserveTruncate(sz)+sz;
  1052. Chunk *p = &chunk;
  1053. while (sz&&p) {
  1054. size32_t s = p->size;
  1055. d -= s;
  1056. memcpy(d,p->base,s);
  1057. p = p->prev;
  1058. sz -= s;
  1059. }
  1060. return mb;
  1061. }
  1062. MemoryBuffer &CLargeMemoryAllocator::deserialize(MemoryBuffer &mb,size32_t sz, size32_t extra)
  1063. {
  1064. mb.read(sz,alloc(sz,extra));
  1065. return mb;
  1066. }
  1067. void *CLargeMemoryAllocator::nextBuffer(void *prev,size32_t &sz)
  1068. { // not fast
  1069. Chunk *p = NULL;
  1070. Chunk *n = &chunk;
  1071. while (n&&(n->base!=prev)) {
  1072. p = n;
  1073. n = n->prev;
  1074. }
  1075. if (!p) {
  1076. sz = 0;
  1077. return NULL;
  1078. }
  1079. sz = p->size;
  1080. return p->base;
  1081. }
  1082. void CJMallocLargeMemoryAllocator::allocchunkmem()
  1083. {
  1084. chunk.base = (byte *)allocator->allocMem(chunk.max);
  1085. #ifdef TRACE_LARGEMEM_ALLOC
  1086. PROGLOG("CJMallocLargeMemoryAllocator::allocchunkmem malloced %d at %p",chunk.max,chunk.base);
  1087. #endif
  1088. }
  1089. void CJMallocLargeMemoryAllocator::disposechunkmem()
  1090. {
  1091. allocator->freeMem(chunk.base);
  1092. }
  1093. CFixedSizeAllocator::CFixedSizeAllocator()
  1094. {
  1095. chunklist = NULL;
  1096. }
  1097. CFixedSizeAllocator::CFixedSizeAllocator(size32_t _allocsize,size32_t _chunksize)
  1098. {
  1099. chunklist = NULL;
  1100. init(_allocsize,_chunksize);
  1101. }
  1102. void CFixedSizeAllocator::init(size32_t _allocsize,size32_t _chunksize)
  1103. {
  1104. kill();
  1105. allocsize = _allocsize;
  1106. assertex(allocsize);
  1107. if (allocsize<sizeof(void *))
  1108. allocsize = sizeof(void *);
  1109. chunksize = _chunksize;
  1110. if (chunksize<allocsize*16)
  1111. chunksize = allocsize+sizeof(void *); // give up on sublety
  1112. }
  1113. void CFixedSizeAllocator::kill()
  1114. {
  1115. while (chunklist) {
  1116. void *p = chunklist;
  1117. chunklist = *(void **)p;
  1118. freeChunk(p);
  1119. }
  1120. freelist = NULL;
  1121. numalloc = 0;
  1122. numfree = 0;
  1123. chunklist = NULL;
  1124. }
  1125. CFixedSizeAllocator::~CFixedSizeAllocator()
  1126. {
  1127. kill();
  1128. }
  1129. void *CFixedSizeAllocator::allocChunk()
  1130. {
  1131. return checked_malloc(chunksize,-5); // don't try to be clever and allocate less (fragmentation)
  1132. }
  1133. void CFixedSizeAllocator::freeChunk(void *p)
  1134. {
  1135. free(p);
  1136. }
  1137. void *CFixedSizeAllocator::alloc()
  1138. {
  1139. NonReentrantSpinBlock block(lock);
  1140. void *ret;
  1141. if (numfree) {
  1142. numfree--;
  1143. ret = freelist;
  1144. freelist = *(void **)freelist;
  1145. }
  1146. else {
  1147. void **newchunk = (void **)allocChunk();
  1148. unsigned num = (chunksize-sizeof(void *))/allocsize;
  1149. assertex(num);
  1150. *newchunk = chunklist;
  1151. chunklist = (void *)newchunk;
  1152. newchunk++;
  1153. ret = (void *)newchunk;
  1154. numfree+=num-1;
  1155. while (--num) { // we could do this on the fly but I think this marginally better
  1156. newchunk = (void **)(((byte *)newchunk)+allocsize);
  1157. *newchunk = freelist;
  1158. freelist = (void *)newchunk;
  1159. }
  1160. }
  1161. numalloc++;
  1162. return ret;
  1163. }
  1164. void CFixedSizeAllocator::dealloc(void *blk)
  1165. {
  1166. if (blk) {
  1167. NonReentrantSpinBlock block(lock);
  1168. *(void **)blk = freelist;
  1169. freelist = blk;
  1170. numfree++;
  1171. numalloc--;
  1172. }
  1173. }
  1174. void CFixedSizeAllocator::stats(size32_t &sizealloc, size32_t &sizeunused)
  1175. {
  1176. NonReentrantSpinBlock block(lock);
  1177. sizealloc = numalloc*allocsize;
  1178. sizeunused = numfree*allocsize;
  1179. }
  1180. //============================================================
  1181. #define LARGEST_CONTIGUOUS_BLOCK (0xffff0000)
  1182. void CContiguousLargeMemoryAllocator::init(size32_t _totalmax,size32_t _chunkmin,bool _throwexception)
  1183. {
  1184. throwexception = _throwexception;
  1185. totalmax = (_totalmax<LARGEST_CONTIGUOUS_BLOCK)?VMPAGEROUND(_totalmax):LARGEST_CONTIGUOUS_BLOCK;
  1186. chunkmin = _chunkmin;
  1187. ofs = 0;
  1188. mapped = 0;
  1189. base = NULL;
  1190. #ifdef WIN32
  1191. LARGE_INTEGER li;
  1192. li.QuadPart = totalmax;
  1193. hmap = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE|SEC_RESERVE, li.HighPart, li.LowPart, NULL);
  1194. #endif
  1195. }
  1196. CContiguousLargeMemoryAllocator::CContiguousLargeMemoryAllocator()
  1197. {
  1198. // values overwritten by init
  1199. throwexception = true;
  1200. totalmax = 0;
  1201. chunkmin = 0x1000;
  1202. ofs = 0;
  1203. mapped = 0;
  1204. base = NULL;
  1205. }
  1206. CContiguousLargeMemoryAllocator::~CContiguousLargeMemoryAllocator()
  1207. {
  1208. reset();
  1209. #ifdef WIN32
  1210. if (hmap) {
  1211. if (base)
  1212. UnmapViewOfFile(base);
  1213. CloseHandle(hmap);
  1214. }
  1215. #endif
  1216. }
  1217. void *CContiguousLargeMemoryAllocator::getBase()
  1218. {
  1219. if (!base) {
  1220. #ifdef WIN32
  1221. if (hmap)
  1222. base = (byte *) MapViewOfFile(hmap, FILE_MAP_READ|FILE_MAP_WRITE, 0, 0, totalmax);
  1223. else
  1224. base = NULL;
  1225. #else
  1226. base = (byte *) mmap(NULL,totalmax,PROT_NONE,MAP_PRIVATE|MAP_NORESERVE|MAP_ANONYMOUS,-1,0);
  1227. // create initially with no access
  1228. if (base == (byte *)MAP_FAILED)
  1229. base = NULL;
  1230. #endif
  1231. }
  1232. return base;
  1233. }
  1234. bool CContiguousLargeMemoryAllocator::map(size32_t tot,size32_t sz)
  1235. {
  1236. getBase();
  1237. if (!base||(tot>totalmax)) {
  1238. outOfMem(sz);
  1239. return false;
  1240. }
  1241. if (tot>mapped) {
  1242. void * a = base+mapped;
  1243. size32_t tomap = VMPAGEROUND(chunkmin);
  1244. #ifdef WIN32
  1245. if (VirtualAlloc(a,tomap,MEM_COMMIT,PAGE_READWRITE)!=a) {
  1246. outOfMem(sz);
  1247. return false;
  1248. }
  1249. #else
  1250. if (mprotect(a,tomap,PROT_READ|PROT_WRITE)<0) {
  1251. int err = errno;
  1252. if ((err==ENOMEM)||(err==EFAULT)) {
  1253. outOfMem(sz);
  1254. return false;
  1255. }
  1256. WARNLOG("CContiguousLargeMemoryAllocator:map madvise err=%d",err);
  1257. }
  1258. #endif
  1259. mapped = mapped+tomap;
  1260. }
  1261. return true;
  1262. }
  1263. void CContiguousLargeMemoryAllocator::unmap()
  1264. {
  1265. // ensures above ofs is unmapped
  1266. size32_t ch = VMPAGEROUND(chunkmin);
  1267. size32_t newmapped = ((ofs+ch-1)/ch)*ch;
  1268. if (newmapped<mapped) {
  1269. void * a = base+newmapped;
  1270. #ifdef WIN32
  1271. if (newmapped==0) { // free completely
  1272. if (base) {
  1273. UnmapViewOfFile(base);
  1274. base = NULL;
  1275. }
  1276. }
  1277. else {
  1278. VirtualFree(a,mapped-newmapped,MEM_DECOMMIT); // can't fail
  1279. }
  1280. #else
  1281. if (newmapped==0) { // free completely
  1282. if (base) {
  1283. munmap(base,totalmax);
  1284. base = NULL;
  1285. }
  1286. }
  1287. else {
  1288. if (mprotect(a,mapped-newmapped,PROT_NONE)<0)
  1289. WARNLOG("CContiguousLargeMemoryAllocator:unmap mprotect err=%d",errno);
  1290. // if (madvise(a,mapped-newmapped,MADV_DONTNEED)<0) // not sure if this does anything but tell it anyway
  1291. // WARNLOG("CContiguousLargeMemoryAllocator:unmap madvise err=%d",errno);
  1292. }
  1293. #endif
  1294. mapped = newmapped;
  1295. }
  1296. }
  1297. void CContiguousLargeMemoryAllocator::reset()
  1298. {
  1299. reduceSize(ofs);
  1300. }
  1301. void CContiguousLargeMemoryAllocator::setSize(size32_t pos)
  1302. {
  1303. assertex(ofs>=pos);
  1304. reduceSize(ofs-pos);
  1305. }
  1306. void CContiguousLargeMemoryAllocator::reduceSize(size32_t amount)
  1307. {
  1308. assertex(ofs>=amount);
  1309. ofs-=amount;
  1310. unmap();
  1311. }
  1312. void *CContiguousLargeMemoryAllocator::nextBuffer(void *prev,size32_t &sz)
  1313. {
  1314. // have to be careful as approaches 4GB
  1315. byte *p = prev?((byte *)prev):base;
  1316. size32_t o = p-base;
  1317. size32_t r = (o<ofs)?ofs-o:0;
  1318. sz = (r<=chunkmin)?0:(r-chunkmin);
  1319. if (sz==0)
  1320. return NULL;
  1321. if (sz>chunkmin)
  1322. sz = chunkmin;
  1323. return p+chunkmin;
  1324. }
  1325. byte *CContiguousLargeMemoryAllocator::next(size32_t pos,size32_t &size)
  1326. {
  1327. if (ofs<=pos) {
  1328. size = 0;
  1329. return NULL;
  1330. }
  1331. size = ofs-pos;
  1332. return base+pos;
  1333. }
  1334. MemoryBuffer &CContiguousLargeMemoryAllocator::serialize(MemoryBuffer &mb)
  1335. {
  1336. memcpy(mb.reserveTruncate(ofs),base,ofs);
  1337. return mb;
  1338. }
  1339. MemoryBuffer &CContiguousLargeMemoryAllocator::deserialize(MemoryBuffer &mb,size32_t sz, size32_t extra)
  1340. {
  1341. mb.read(sz,alloc(sz,extra));
  1342. return mb;
  1343. }
  1344. void CContiguousLargeMemoryAllocator::outOfMem(size32_t sz)
  1345. {
  1346. if (throwexception) {
  1347. throw createOutOfMemException(-6,sz, ofs,true);
  1348. }
  1349. }