src/common.c

   1 #include "common.h"
   2
   3 char *global_host_prefix = "";
   4 int enable_ksm = 1;
   5
   6 volatile sig_atomic_t netdata_exit = 0;
   7
   8 // time(NULL) in milliseconds
   9 unsigned long long timems(void) {
  10         struct timeval now;
  11         gettimeofday(&now, NULL);
  12         return now.tv_sec * 1000000ULL + now.tv_usec;
  13 }
  14
  15 int usecsleep(unsigned long long usec) {
  16
  17 #ifndef NETDATA_WITH_USLEEP
  18         // we expect microseconds (1.000.000 per second)
  19         // but timespec is nanoseconds (1.000.000.000 per second)
  20         struct timespec req = { .tv_sec = usec / 1000000, .tv_nsec = (usec % 1000000) * 1000 }, rem;
  21
  22         while(nanosleep(&req, &rem) == -1) {
  23                 if(likely(errno == EINTR)) {
  24                         info("nanosleep() interrupted (while sleeping for %llu microseconds).", usec);
  25                         req.tv_sec = rem.tv_sec;
  26                         req.tv_nsec = rem.tv_nsec;
  27                 }
  28                 else {
  29                         error("Cannot nanosleep() for %llu microseconds.", usec);
  30                         break;
  31                 }
  32         }
  33
  34         return 0;
  35 #else
  36         int ret = usleep(usec);
  37         if(unlikely(ret == -1 && errno == EINVAL)) {
  38                 // on certain systems, usec has to be up to 999999
  39                 if(usec > 999999) {
  40                         int counter = usec / 999999;
  41                         while(counter--)
  42                                 usleep(999999);
  43
  44                         usleep(usec % 999999);
  45                 }
  46                 else {
  47                         error("Cannot usleep() for %llu microseconds.", usec);
  48                         return ret;
  49                 }
  50         }
  51
  52         if(ret != 0)
  53                 error("usleep() failed for %llu microseconds.", usec);
  54
  55         return ret;
  56 #endif
  57 }
  58
  59 unsigned char netdata_map_chart_names[256] = {
  60                 [0] = '\0', //
  61                 [1] = '_', //
  62                 [2] = '_', //
  63                 [3] = '_', //
  64                 [4] = '_', //
  65                 [5] = '_', //
  66                 [6] = '_', //
  67                 [7] = '_', //
  68                 [8] = '_', //
  69                 [9] = '_', //
  70                 [10] = '_', //
  71                 [11] = '_', //
  72                 [12] = '_', //
  73                 [13] = '_', //
  74                 [14] = '_', //
  75                 [15] = '_', //
  76                 [16] = '_', //
  77                 [17] = '_', //
  78                 [18] = '_', //
  79                 [19] = '_', //
  80                 [20] = '_', //
  81                 [21] = '_', //
  82                 [22] = '_', //
  83                 [23] = '_', //
  84                 [24] = '_', //
  85                 [25] = '_', //
  86                 [26] = '_', //
  87                 [27] = '_', //
  88                 [28] = '_', //
  89                 [29] = '_', //
  90                 [30] = '_', //
  91                 [31] = '_', //
  92                 [32] = '_', //
  93                 [33] = '_', // !
  94                 [34] = '_', // "
  95                 [35] = '_', // #
  96                 [36] = '_', // $
  97                 [37] = '_', // %
  98                 [38] = '_', // &
  99                 [39] = '_', // '
 100                 [40] = '_', // (
 101                 [41] = '_', // )
 102                 [42] = '_', // *
 103                 [43] = '_', // +
 104                 [44] = '.', // ,
 105                 [45] = '-', // -
 106                 [46] = '.', // .
 107                 [47] = '/', // /
 108                 [48] = '0', // 0
 109                 [49] = '1', // 1
 110                 [50] = '2', // 2
 111                 [51] = '3', // 3
 112                 [52] = '4', // 4
 113                 [53] = '5', // 5
 114                 [54] = '6', // 6
 115                 [55] = '7', // 7
 116                 [56] = '8', // 8
 117                 [57] = '9', // 9
 118                 [58] = '_', // :
 119                 [59] = '_', // ;
 120                 [60] = '_', // <
 121                 [61] = '_', // =
 122                 [62] = '_', // >
 123                 [63] = '_', // ?
 124                 [64] = '_', // @
 125                 [65] = 'a', // A
 126                 [66] = 'b', // B
 127                 [67] = 'c', // C
 128                 [68] = 'd', // D
 129                 [69] = 'e', // E
 130                 [70] = 'f', // F
 131                 [71] = 'g', // G
 132                 [72] = 'h', // H
 133                 [73] = 'i', // I
 134                 [74] = 'j', // J
 135                 [75] = 'k', // K
 136                 [76] = 'l', // L
 137                 [77] = 'm', // M
 138                 [78] = 'n', // N
 139                 [79] = 'o', // O
 140                 [80] = 'p', // P
 141                 [81] = 'q', // Q
 142                 [82] = 'r', // R
 143                 [83] = 's', // S
 144                 [84] = 't', // T
 145                 [85] = 'u', // U
 146                 [86] = 'v', // V
 147                 [87] = 'w', // W
 148                 [88] = 'x', // X
 149                 [89] = 'y', // Y
 150                 [90] = 'z', // Z
 151                 [91] = '_', // [
 152                 [92] = '/', // backslash
 153                 [93] = '_', // ]
 154                 [94] = '_', // ^
 155                 [95] = '_', // _
 156                 [96] = '_', // `
 157                 [97] = 'a', // a
 158                 [98] = 'b', // b
 159                 [99] = 'c', // c
 160                 [100] = 'd', // d
 161                 [101] = 'e', // e
 162                 [102] = 'f', // f
 163                 [103] = 'g', // g
 164                 [104] = 'h', // h
 165                 [105] = 'i', // i
 166                 [106] = 'j', // j
 167                 [107] = 'k', // k
 168                 [108] = 'l', // l
 169                 [109] = 'm', // m
 170                 [110] = 'n', // n
 171                 [111] = 'o', // o
 172                 [112] = 'p', // p
 173                 [113] = 'q', // q
 174                 [114] = 'r', // r
 175                 [115] = 's', // s
 176                 [116] = 't', // t
 177                 [117] = 'u', // u
 178                 [118] = 'v', // v
 179                 [119] = 'w', // w
 180                 [120] = 'x', // x
 181                 [121] = 'y', // y
 182                 [122] = 'z', // z
 183                 [123] = '_', // {
 184                 [124] = '_', // |
 185                 [125] = '_', // }
 186                 [126] = '_', // ~
 187                 [127] = '_', //
 188                 [128] = '_', //
 189                 [129] = '_', //
 190                 [130] = '_', //
 191                 [131] = '_', //
 192                 [132] = '_', //
 193                 [133] = '_', //
 194                 [134] = '_', //
 195                 [135] = '_', //
 196                 [136] = '_', //
 197                 [137] = '_', //
 198                 [138] = '_', //
 199                 [139] = '_', //
 200                 [140] = '_', //
 201                 [141] = '_', //
 202                 [142] = '_', //
 203                 [143] = '_', //
 204                 [144] = '_', //
 205                 [145] = '_', //
 206                 [146] = '_', //
 207                 [147] = '_', //
 208                 [148] = '_', //
 209                 [149] = '_', //
 210                 [150] = '_', //
 211                 [151] = '_', //
 212                 [152] = '_', //
 213                 [153] = '_', //
 214                 [154] = '_', //
 215                 [155] = '_', //
 216                 [156] = '_', //
 217                 [157] = '_', //
 218                 [158] = '_', //
 219                 [159] = '_', //
 220                 [160] = '_', //
 221                 [161] = '_', //
 222                 [162] = '_', //
 223                 [163] = '_', //
 224                 [164] = '_', //
 225                 [165] = '_', //
 226                 [166] = '_', //
 227                 [167] = '_', //
 228                 [168] = '_', //
 229                 [169] = '_', //
 230                 [170] = '_', //
 231                 [171] = '_', //
 232                 [172] = '_', //
 233                 [173] = '_', //
 234                 [174] = '_', //
 235                 [175] = '_', //
 236                 [176] = '_', //
 237                 [177] = '_', //
 238                 [178] = '_', //
 239                 [179] = '_', //
 240                 [180] = '_', //
 241                 [181] = '_', //
 242                 [182] = '_', //
 243                 [183] = '_', //
 244                 [184] = '_', //
 245                 [185] = '_', //
 246                 [186] = '_', //
 247                 [187] = '_', //
 248                 [188] = '_', //
 249                 [189] = '_', //
 250                 [190] = '_', //
 251                 [191] = '_', //
 252                 [192] = '_', //
 253                 [193] = '_', //
 254                 [194] = '_', //
 255                 [195] = '_', //
 256                 [196] = '_', //
 257                 [197] = '_', //
 258                 [198] = '_', //
 259                 [199] = '_', //
 260                 [200] = '_', //
 261                 [201] = '_', //
 262                 [202] = '_', //
 263                 [203] = '_', //
 264                 [204] = '_', //
 265                 [205] = '_', //
 266                 [206] = '_', //
 267                 [207] = '_', //
 268                 [208] = '_', //
 269                 [209] = '_', //
 270                 [210] = '_', //
 271                 [211] = '_', //
 272                 [212] = '_', //
 273                 [213] = '_', //
 274                 [214] = '_', //
 275                 [215] = '_', //
 276                 [216] = '_', //
 277                 [217] = '_', //
 278                 [218] = '_', //
 279                 [219] = '_', //
 280                 [220] = '_', //
 281                 [221] = '_', //
 282                 [222] = '_', //
 283                 [223] = '_', //
 284                 [224] = '_', //
 285                 [225] = '_', //
 286                 [226] = '_', //
 287                 [227] = '_', //
 288                 [228] = '_', //
 289                 [229] = '_', //
 290                 [230] = '_', //
 291                 [231] = '_', //
 292                 [232] = '_', //
 293                 [233] = '_', //
 294                 [234] = '_', //
 295                 [235] = '_', //
 296                 [236] = '_', //
 297                 [237] = '_', //
 298                 [238] = '_', //
 299                 [239] = '_', //
 300                 [240] = '_', //
 301                 [241] = '_', //
 302                 [242] = '_', //
 303                 [243] = '_', //
 304                 [244] = '_', //
 305                 [245] = '_', //
 306                 [246] = '_', //
 307                 [247] = '_', //
 308                 [248] = '_', //
 309                 [249] = '_', //
 310                 [250] = '_', //
 311                 [251] = '_', //
 312                 [252] = '_', //
 313                 [253] = '_', //
 314                 [254] = '_', //
 315                 [255] = '_'  //
 316 };
 317
 318 // make sure the supplied string
 319 // is good for a netdata chart/dimension ID/NAME
 320 void netdata_fix_chart_name(char *s) {
 321         while((*s = netdata_map_chart_names[(unsigned char)*s])) s++;
 322 }
 323
 324 unsigned char netdata_map_chart_ids[256] = {
 325                 [0] = '\0', //
 326                 [1] = '_', //
 327                 [2] = '_', //
 328                 [3] = '_', //
 329                 [4] = '_', //
 330                 [5] = '_', //
 331                 [6] = '_', //
 332                 [7] = '_', //
 333                 [8] = '_', //
 334                 [9] = '_', //
 335                 [10] = '_', //
 336                 [11] = '_', //
 337                 [12] = '_', //
 338                 [13] = '_', //
 339                 [14] = '_', //
 340                 [15] = '_', //
 341                 [16] = '_', //
 342                 [17] = '_', //
 343                 [18] = '_', //
 344                 [19] = '_', //
 345                 [20] = '_', //
 346                 [21] = '_', //
 347                 [22] = '_', //
 348                 [23] = '_', //
 349                 [24] = '_', //
 350                 [25] = '_', //
 351                 [26] = '_', //
 352                 [27] = '_', //
 353                 [28] = '_', //
 354                 [29] = '_', //
 355                 [30] = '_', //
 356                 [31] = '_', //
 357                 [32] = '_', //
 358                 [33] = '_', // !
 359                 [34] = '_', // "
 360                 [35] = '_', // #
 361                 [36] = '_', // $
 362                 [37] = '_', // %
 363                 [38] = '_', // &
 364                 [39] = '_', // '
 365                 [40] = '_', // (
 366                 [41] = '_', // )
 367                 [42] = '_', // *
 368                 [43] = '_', // +
 369                 [44] = '.', // ,
 370                 [45] = '-', // -
 371                 [46] = '.', // .
 372                 [47] = '_', // /
 373                 [48] = '0', // 0
 374                 [49] = '1', // 1
 375                 [50] = '2', // 2
 376                 [51] = '3', // 3
 377                 [52] = '4', // 4
 378                 [53] = '5', // 5
 379                 [54] = '6', // 6
 380                 [55] = '7', // 7
 381                 [56] = '8', // 8
 382                 [57] = '9', // 9
 383                 [58] = '_', // :
 384                 [59] = '_', // ;
 385                 [60] = '_', // <
 386                 [61] = '_', // =
 387                 [62] = '_', // >
 388                 [63] = '_', // ?
 389                 [64] = '_', // @
 390                 [65] = 'a', // A
 391                 [66] = 'b', // B
 392                 [67] = 'c', // C
 393                 [68] = 'd', // D
 394                 [69] = 'e', // E
 395                 [70] = 'f', // F
 396                 [71] = 'g', // G
 397                 [72] = 'h', // H
 398                 [73] = 'i', // I
 399                 [74] = 'j', // J
 400                 [75] = 'k', // K
 401                 [76] = 'l', // L
 402                 [77] = 'm', // M
 403                 [78] = 'n', // N
 404                 [79] = 'o', // O
 405                 [80] = 'p', // P
 406                 [81] = 'q', // Q
 407                 [82] = 'r', // R
 408                 [83] = 's', // S
 409                 [84] = 't', // T
 410                 [85] = 'u', // U
 411                 [86] = 'v', // V
 412                 [87] = 'w', // W
 413                 [88] = 'x', // X
 414                 [89] = 'y', // Y
 415                 [90] = 'z', // Z
 416                 [91] = '_', // [
 417                 [92] = '/', // backslash
 418                 [93] = '_', // ]
 419                 [94] = '_', // ^
 420                 [95] = '_', // _
 421                 [96] = '_', // `
 422                 [97] = 'a', // a
 423                 [98] = 'b', // b
 424                 [99] = 'c', // c
 425                 [100] = 'd', // d
 426                 [101] = 'e', // e
 427                 [102] = 'f', // f
 428                 [103] = 'g', // g
 429                 [104] = 'h', // h
 430                 [105] = 'i', // i
 431                 [106] = 'j', // j
 432                 [107] = 'k', // k
 433                 [108] = 'l', // l
 434                 [109] = 'm', // m
 435                 [110] = 'n', // n
 436                 [111] = 'o', // o
 437                 [112] = 'p', // p
 438                 [113] = 'q', // q
 439                 [114] = 'r', // r
 440                 [115] = 's', // s
 441                 [116] = 't', // t
 442                 [117] = 'u', // u
 443                 [118] = 'v', // v
 444                 [119] = 'w', // w
 445                 [120] = 'x', // x
 446                 [121] = 'y', // y
 447                 [122] = 'z', // z
 448                 [123] = '_', // {
 449                 [124] = '_', // |
 450                 [125] = '_', // }
 451                 [126] = '_', // ~
 452                 [127] = '_', //
 453                 [128] = '_', //
 454                 [129] = '_', //
 455                 [130] = '_', //
 456                 [131] = '_', //
 457                 [132] = '_', //
 458                 [133] = '_', //
 459                 [134] = '_', //
 460                 [135] = '_', //
 461                 [136] = '_', //
 462                 [137] = '_', //
 463                 [138] = '_', //
 464                 [139] = '_', //
 465                 [140] = '_', //
 466                 [141] = '_', //
 467                 [142] = '_', //
 468                 [143] = '_', //
 469                 [144] = '_', //
 470                 [145] = '_', //
 471                 [146] = '_', //
 472                 [147] = '_', //
 473                 [148] = '_', //
 474                 [149] = '_', //
 475                 [150] = '_', //
 476                 [151] = '_', //
 477                 [152] = '_', //
 478                 [153] = '_', //
 479                 [154] = '_', //
 480                 [155] = '_', //
 481                 [156] = '_', //
 482                 [157] = '_', //
 483                 [158] = '_', //
 484                 [159] = '_', //
 485                 [160] = '_', //
 486                 [161] = '_', //
 487                 [162] = '_', //
 488                 [163] = '_', //
 489                 [164] = '_', //
 490                 [165] = '_', //
 491                 [166] = '_', //
 492                 [167] = '_', //
 493                 [168] = '_', //
 494                 [169] = '_', //
 495                 [170] = '_', //
 496                 [171] = '_', //
 497                 [172] = '_', //
 498                 [173] = '_', //
 499                 [174] = '_', //
 500                 [175] = '_', //
 501                 [176] = '_', //
 502                 [177] = '_', //
 503                 [178] = '_', //
 504                 [179] = '_', //
 505                 [180] = '_', //
 506                 [181] = '_', //
 507                 [182] = '_', //
 508                 [183] = '_', //
 509                 [184] = '_', //
 510                 [185] = '_', //
 511                 [186] = '_', //
 512                 [187] = '_', //
 513                 [188] = '_', //
 514                 [189] = '_', //
 515                 [190] = '_', //
 516                 [191] = '_', //
 517                 [192] = '_', //
 518                 [193] = '_', //
 519                 [194] = '_', //
 520                 [195] = '_', //
 521                 [196] = '_', //
 522                 [197] = '_', //
 523                 [198] = '_', //
 524                 [199] = '_', //
 525                 [200] = '_', //
 526                 [201] = '_', //
 527                 [202] = '_', //
 528                 [203] = '_', //
 529                 [204] = '_', //
 530                 [205] = '_', //
 531                 [206] = '_', //
 532                 [207] = '_', //
 533                 [208] = '_', //
 534                 [209] = '_', //
 535                 [210] = '_', //
 536                 [211] = '_', //
 537                 [212] = '_', //
 538                 [213] = '_', //
 539                 [214] = '_', //
 540                 [215] = '_', //
 541                 [216] = '_', //
 542                 [217] = '_', //
 543                 [218] = '_', //
 544                 [219] = '_', //
 545                 [220] = '_', //
 546                 [221] = '_', //
 547                 [222] = '_', //
 548                 [223] = '_', //
 549                 [224] = '_', //
 550                 [225] = '_', //
 551                 [226] = '_', //
 552                 [227] = '_', //
 553                 [228] = '_', //
 554                 [229] = '_', //
 555                 [230] = '_', //
 556                 [231] = '_', //
 557                 [232] = '_', //
 558                 [233] = '_', //
 559                 [234] = '_', //
 560                 [235] = '_', //
 561                 [236] = '_', //
 562                 [237] = '_', //
 563                 [238] = '_', //
 564                 [239] = '_', //
 565                 [240] = '_', //
 566                 [241] = '_', //
 567                 [242] = '_', //
 568                 [243] = '_', //
 569                 [244] = '_', //
 570                 [245] = '_', //
 571                 [246] = '_', //
 572                 [247] = '_', //
 573                 [248] = '_', //
 574                 [249] = '_', //
 575                 [250] = '_', //
 576                 [251] = '_', //
 577                 [252] = '_', //
 578                 [253] = '_', //
 579                 [254] = '_', //
 580                 [255] = '_'  //
 581 };
 582
 583 // make sure the supplied string
 584 // is good for a netdata chart/dimension ID/NAME
 585 void netdata_fix_chart_id(char *s) {
 586         while((*s = netdata_map_chart_ids[(unsigned char)*s])) s++;
 587 }
 588
 589 /*
 590 // http://stackoverflow.com/questions/7666509/hash-function-for-string
 591 uint32_t simple_hash(const char *name)
 592 {
 593         const char *s = name;
 594         uint32_t hash = 5381;
 595         int i;
 596
 597         while((i = *s++)) hash = ((hash << 5) + hash) + i;
 598
 599         // fprintf(stderr, "HASH: %lu %s\n", hash, name);
 600
 601         return hash;
 602 }
 603 */
 604
 605
 606 // http://isthe.com/chongo/tech/comp/fnv/#FNV-1a
 607 uint32_t simple_hash(const char *name) {
 608         unsigned char *s = (unsigned char *)name;
 609         uint32_t hval = 0x811c9dc5;
 610
 611         // FNV-1a algorithm
 612         while (*s) {
 613                 // multiply by the 32 bit FNV magic prime mod 2^32
 614                 // NOTE: No need to optimize with left shifts.
 615                 //       GCC will use imul instruction anyway.
 616                 //       Tested with 'gcc -O3 -S'
 617                 //hval += (hval<<1) + (hval<<4) + (hval<<7) + (hval<<8) + (hval<<24);
 618                 hval *= 16777619;
 619
 620                 // xor the bottom with the current octet
 621                 hval ^= (uint32_t)*s++;
 622         }
 623
 624         // fprintf(stderr, "HASH: %u = %s\n", hval, name);
 625         return hval;
 626 }
 627
 628 uint32_t simple_uhash(const char *name) {
 629         unsigned char *s = (unsigned char *)name;
 630         uint32_t hval = 0x811c9dc5, c;
 631
 632         // FNV-1a algorithm
 633         while((c = *s++)) {
 634                 if(unlikely(c >= 'A' && c <= 'Z')) c += 'a' - 'A';
 635                 hval *= 16777619;
 636                 hval ^= c;
 637         }
 638         return hval;
 639 }
 640
 641 /*
 642 // http://eternallyconfuzzled.com/tuts/algorithms/jsw_tut_hashing.aspx
 643 // one at a time hash
 644 uint32_t simple_hash(const char *name) {
 645         unsigned char *s = (unsigned char *)name;
 646         uint32_t h = 0;
 647
 648         while(*s) {
 649                 h += *s++;
 650                 h += (h << 10);
 651                 h ^= (h >> 6);
 652         }
 653
 654         h += (h << 3);
 655         h ^= (h >> 11);
 656         h += (h << 15);
 657
 658         // fprintf(stderr, "HASH: %u = %s\n", h, name);
 659
 660         return h;
 661 }
 662 */
 663
 664 void strreverse(char* begin, char* end)
 665 {
 666         char aux;
 667
 668         while (end > begin)
 669         {
 670                 // clearer code.
 671                 aux = *end;
 672                 *end-- = *begin;
 673                 *begin++ = aux;
 674         }
 675 }
 676
 677 char *mystrsep(char **ptr, char *s)
 678 {
 679         char *p = "";
 680         while ( p && !p[0] && *ptr ) p = strsep(ptr, s);
 681         return(p);
 682 }
 683
 684 char *trim(char *s)
 685 {
 686         // skip leading spaces
 687         // and 'comments' as well!?
 688         while(*s && isspace(*s)) s++;
 689         if(!*s || *s == '#') return NULL;
 690
 691         // skip tailing spaces
 692         // this way is way faster. Writes only one NUL char.
 693         ssize_t l = strlen(s);
 694   if (--l >= 0)
 695         {
 696                 char *p = s + l;
 697                 while (p > s && isspace(*p)) p--;
 698                 *++p = '\0';
 699         }
 700
 701         if(!*s) return NULL;
 702
 703         return s;
 704 }
 705
 706 void *mymmap(const char *filename, size_t size, int flags, int ksm) {
 707         static int log_madvise_1 = 1;
 708 #ifdef MADV_MERGEABLE
 709         static int log_madvise_2 = 1, log_madvise_3 = 1;
 710 #endif
 711         int fd;
 712         void *mem = NULL;
 713
 714         errno = 0;
 715         fd = open(filename, O_RDWR|O_CREAT|O_NOATIME, 0664);
 716         if(fd != -1) {
 717                 if(lseek(fd, size, SEEK_SET) == (off_t)size) {
 718                         if(write(fd, "", 1) == 1) {
 719                                 if(ftruncate(fd, size))
 720                                         error("Cannot truncate file '%s' to size %zu. Will use the larger file.", filename, size);
 721
 722 #ifdef MADV_MERGEABLE
 723                                 if(flags & MAP_SHARED || !enable_ksm || !ksm) {
 724 #endif
 725                                         mem = mmap(NULL, size, PROT_READ|PROT_WRITE, flags, fd, 0);
 726                                         if(mem != MAP_FAILED) {
 727                                                 int advise = MADV_SEQUENTIAL|MADV_DONTFORK;
 728                                                 if(flags & MAP_SHARED) advise |= MADV_WILLNEED;
 729
 730                                                 if(madvise(mem, size, advise) != 0 && log_madvise_1) {
 731                                                         error("Cannot advise the kernel about the memory usage of file '%s'.", filename);
 732                                                         log_madvise_1--;
 733                                                 }
 734                                         }
 735 #ifdef MADV_MERGEABLE
 736                                 }
 737                                 else {
 738 /*
 739                                         // test - load the file into memory
 740                                         mem = calloc(1, size);
 741                                         if(mem) {
 742                                                 if(lseek(fd, 0, SEEK_SET) == 0) {
 743                                                         if(read(fd, mem, size) != (ssize_t)size)
 744                                                                 error("Cannot read from file '%s'", filename);
 745                                                 }
 746                                                 else
 747                                                         error("Cannot seek to beginning of file '%s'.", filename);
 748                                         }
 749 */
 750                                         mem = mmap(NULL, size, PROT_READ|PROT_WRITE, flags|MAP_ANONYMOUS, -1, 0);
 751                                         if(mem != MAP_FAILED) {
 752                                                 if(lseek(fd, 0, SEEK_SET) == 0) {
 753                                                         if(read(fd, mem, size) != (ssize_t)size)
 754                                                                 error("Cannot read from file '%s'", filename);
 755                                                 }
 756                                                 else
 757                                                         error("Cannot seek to beginning of file '%s'.", filename);
 758
 759                                                 // don't use MADV_SEQUENTIAL|MADV_DONTFORK, they disable MADV_MERGEABLE
 760                                                 if(madvise(mem, size, MADV_SEQUENTIAL|MADV_DONTFORK) != 0 && log_madvise_2) {
 761                                                         error("Cannot advise the kernel about the memory usage (MADV_SEQUENTIAL|MADV_DONTFORK) of file '%s'.", filename);
 762                                                         log_madvise_2--;
 763                                                 }
 764
 765                                                 if(madvise(mem, size, MADV_MERGEABLE) != 0 && log_madvise_3) {
 766                                                         error("Cannot advise the kernel about the memory usage (MADV_MERGEABLE) of file '%s'.", filename);
 767                                                         log_madvise_3--;
 768                                                 }
 769                                         }
 770                                         else
 771                                                 error("Cannot allocate PRIVATE ANONYMOUS memory for KSM for file '%s'.", filename);
 772                                 }
 773 #endif
 774                         }
 775                         else error("Cannot write to file '%s' at position %zu.", filename, size);
 776                 }
 777                 else error("Cannot seek file '%s' to size %zu.", filename, size);
 778
 779                 close(fd);
 780         }
 781         else error("Cannot create/open file '%s'.", filename);
 782
 783         return mem;
 784 }
 785
 786 int savememory(const char *filename, void *mem, size_t size)
 787 {
 788         char tmpfilename[FILENAME_MAX + 1];
 789
 790         snprintfz(tmpfilename, FILENAME_MAX, "%s.%ld.tmp", filename, (long)getpid());
 791
 792         int fd = open(tmpfilename, O_RDWR|O_CREAT|O_NOATIME, 0664);
 793         if(fd < 0) {
 794                 error("Cannot create/open file '%s'.", filename);
 795                 return -1;
 796         }
 797
 798         if(write(fd, mem, size) != (ssize_t)size) {
 799                 error("Cannot write to file '%s' %ld bytes.", filename, (long)size);
 800                 close(fd);
 801                 return -1;
 802         }
 803
 804         close(fd);
 805
 806         if(rename(tmpfilename, filename)) {
 807                 error("Cannot rename '%s' to '%s'", tmpfilename, filename);
 808                 return -1;
 809         }
 810
 811         return 0;
 812 }
 813
 814 int fd_is_valid(int fd) {
 815     return fcntl(fd, F_GETFD) != -1 || errno != EBADF;
 816 }
 817
 818 /*
 819  ***************************************************************************
 820  * Get number of clock ticks per second.
 821  ***************************************************************************
 822  */
 823 unsigned int hz;
 824
 825 void get_HZ(void)
 826 {
 827         long ticks;
 828
 829         if ((ticks = sysconf(_SC_CLK_TCK)) == -1) {
 830                 perror("sysconf");
 831         }
 832
 833         hz = (unsigned int) ticks;
 834 }
 835
 836 pid_t gettid(void)
 837 {
 838         return syscall(SYS_gettid);
 839 }
 840
 841 char *fgets_trim_len(char *buf, size_t buf_size, FILE *fp, size_t *len) {
 842         char *s = fgets(buf, buf_size, fp);
 843         if(!s) return NULL;
 844
 845         char *t = s;
 846         if(*t != '\0') {
 847                 // find the string end
 848                 while (*++t != '\0');
 849
 850                 // trim trailing spaces/newlines/tabs
 851                 while (--t > s && *t == '\n')
 852                         *t = '\0';
 853         }
 854
 855         if(len)
 856                 *len = t - s + 1;
 857
 858         return s;
 859 }
 860
 861 char *strncpyz(char *dst, const char *src, size_t n) {
 862         char *p = dst;
 863
 864         while(*src && n--)
 865                 *dst++ = *src++;
 866
 867         *dst = '\0';
 868
 869         return p;
 870 }
 871
 872 int vsnprintfz(char *dst, size_t n, const char *fmt, va_list args) {
 873         int size;
 874
 875         size = vsnprintf(dst, n, fmt, args);
 876
 877         if(unlikely((size_t)size > n)) {
 878                 // there is bug in vsnprintf() and it returns
 879                 // a number higher to len, but it does not
 880                 // overflow the buffer.
 881                 size = n;
 882         }
 883
 884         dst[size] = '\0';
 885         return size;
 886 }
 887
 888 int snprintfz(char *dst, size_t n, const char *fmt, ...) {
 889         va_list args;
 890
 891         va_start(args, fmt);
 892         int ret = vsnprintfz(dst, n, fmt, args);
 893         va_end(args);
 894
 895         return ret;
 896 }