nodeinfo.c 30.3 KB
Newer Older
1 2 3
/*
 * nodeinfo.c: Helper routines for OS specific node information
 *
4
 * Copyright (C) 2006-2008, 2010-2012 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 * Copyright (C) 2006 Daniel P. Berrange
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

24
#include <config.h>
J
Jim Meyering 已提交
25

26 27 28
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
29
#include <stdint.h>
30
#include <errno.h>
31
#include <dirent.h>
E
Eric Blake 已提交
32
#include <sys/utsname.h>
33
#include <sched.h>
34
#include "conf/domain_conf.h"
35 36 37 38 39

#if HAVE_NUMACTL
# define NUMA_VERSION1_COMPATIBILITY 1
# include <numa.h>
#endif
40

41 42
#include "c-ctype.h"
#include "memory.h"
43
#include "nodeinfo.h"
44
#include "physmem.h"
45
#include "util.h"
46
#include "logging.h"
47
#include "virterror_internal.h"
48
#include "count-one-bits.h"
E
Eric Blake 已提交
49
#include "intprops.h"
E
Eric Blake 已提交
50
#include "virfile.h"
51

52 53 54

#define VIR_FROM_THIS VIR_FROM_NONE

55
#define nodeReportError(code, ...)                                      \
56
    virReportErrorHelper(VIR_FROM_NONE, code, __FILE__,                 \
57
                         __FUNCTION__, __LINE__, __VA_ARGS__)
58

59
#ifdef __linux__
60
# define CPUINFO_PATH "/proc/cpuinfo"
61
# define SYSFS_SYSTEM_PATH "/sys/devices/system"
62
# define PROCSTAT_PATH "/proc/stat"
63
# define MEMINFO_PATH "/proc/meminfo"
64 65

# define LINUX_NB_CPU_STATS 4
66 67
# define LINUX_NB_MEMORY_STATS_ALL 4
# define LINUX_NB_MEMORY_STATS_CELL 2
68

69
/* NB, this is not static as we need to call it from the testsuite */
70
int linuxNodeInfoCPUPopulate(FILE *cpuinfo,
71
                             const char *sysfs_dir,
72
                             virNodeInfoPtr nodeinfo);
73

74 75
static int linuxNodeGetCPUStats(FILE *procstat,
                                int cpuNum,
76
                                virNodeCPUStatsPtr params,
77
                                int *nparams);
78 79
static int linuxNodeGetMemoryStats(FILE *meminfo,
                                   int cellNum,
80
                                   virNodeMemoryStatsPtr params,
81
                                   int *nparams);
82

E
Eric Blake 已提交
83
/* Return the positive decimal contents of the given
E
Eric Blake 已提交
84
 * DIR/cpu%u/FILE, or -1 on error.  If MISSING_OK and the
E
Eric Blake 已提交
85
 * file could not be found, return 1 instead of an error; this is
86 87
 * because some machines cannot hot-unplug cpu0, or because
 * hot-unplugging is disabled.  */
E
Eric Blake 已提交
88
static int
E
Eric Blake 已提交
89 90
virNodeGetCpuValue(const char *dir, unsigned int cpu, const char *file,
                   bool missing_ok)
E
Eric Blake 已提交
91 92 93 94 95 96 97
{
    char *path;
    FILE *pathfp;
    int value = -1;
    char value_str[INT_BUFSIZE_BOUND(value)];
    char *tmp;

E
Eric Blake 已提交
98
    if (virAsprintf(&path, "%s/cpu%u/%s", dir, cpu, file) < 0) {
E
Eric Blake 已提交
99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
        virReportOOMError();
        return -1;
    }

    pathfp = fopen(path, "r");
    if (pathfp == NULL) {
        if (missing_ok && errno == ENOENT)
            value = 1;
        else
            virReportSystemError(errno, _("cannot open %s"), path);
        goto cleanup;
    }

    if (fgets(value_str, sizeof(value_str), pathfp) == NULL) {
        virReportSystemError(errno, _("cannot read from %s"), path);
        goto cleanup;
    }
    if (virStrToLong_i(value_str, &tmp, 10, &value) < 0) {
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                        _("could not convert '%s' to an integer"),
                        value_str);
        goto cleanup;
    }

cleanup:
124
    VIR_FORCE_FCLOSE(pathfp);
E
Eric Blake 已提交
125 126 127 128 129
    VIR_FREE(path);

    return value;
}

E
Eric Blake 已提交
130 131
static unsigned long
virNodeCountThreadSiblings(const char *dir, unsigned int cpu)
132 133
{
    unsigned long ret = 0;
C
Chris Lalancette 已提交
134 135
    char *path;
    FILE *pathfp;
136 137 138
    char str[1024];
    int i;

139
    if (virAsprintf(&path, "%s/cpu%u/topology/thread_siblings",
E
Eric Blake 已提交
140
                    dir, cpu) < 0) {
141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
        virReportOOMError();
        return 0;
    }

    pathfp = fopen(path, "r");
    if (pathfp == NULL) {
        virReportSystemError(errno, _("cannot open %s"), path);
        VIR_FREE(path);
        return 0;
    }

    if (fgets(str, sizeof(str), pathfp) == NULL) {
        virReportSystemError(errno, _("cannot read from %s"), path);
        goto cleanup;
    }

    i = 0;
    while (str[i] != '\0') {
C
Chris Lalancette 已提交
159
        if (c_isdigit(str[i]))
160
            ret += count_one_bits(str[i] - '0');
C
Chris Lalancette 已提交
161 162 163 164
        else if (str[i] >= 'A' && str[i] <= 'F')
            ret += count_one_bits(str[i] - 'A' + 10);
        else if (str[i] >= 'a' && str[i] <= 'f')
            ret += count_one_bits(str[i] - 'a' + 10);
165 166 167 168
        i++;
    }

cleanup:
169
    VIR_FORCE_FCLOSE(pathfp);
170 171 172 173 174
    VIR_FREE(path);

    return ret;
}

E
Eric Blake 已提交
175 176
static int
virNodeParseSocket(const char *dir, unsigned int cpu)
177
{
E
Eric Blake 已提交
178 179
    int ret = virNodeGetCpuValue(dir, cpu, "topology/physical_package_id",
                                 false);
E
Eric Blake 已提交
180
# if defined(__powerpc__) || \
181 182 183 184
    defined(__powerpc64__) || \
    defined(__s390__) || \
    defined(__s390x__)
    /* ppc and s390(x) has -1 */
185 186
    if (ret < 0)
        ret = 0;
E
Eric Blake 已提交
187
# endif
188
    return ret;
189 190
}

191 192
/* parses a node entry, returning number of processors in the node and
 * filling arguments */
193
static int
194 195 196
virNodeParseNode(const char *node, int *sockets, int *cores, int *threads)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_NONNULL(4)
197 198
{
    int ret = -1;
199 200 201 202 203 204 205 206 207 208 209 210
    int processors = 0;
    DIR *cpudir = NULL;
    struct dirent *cpudirent = NULL;
    int sock_max = 0;
    cpu_set_t sock_map;
    int sock;
    cpu_set_t *core_maps = NULL;
    int core;
    int i;
    int siblings;
    unsigned int cpu;
    int online;
211

212 213 214 215 216 217
    *threads = 0;
    *cores = 0;
    *sockets = 0;

    if (!(cpudir = opendir(node))) {
        virReportSystemError(errno, _("cannot opendir %s"), node);
218 219
        goto cleanup;
    }
220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239

    /* enumerate sockets in the node */
    CPU_ZERO(&sock_map);
    errno = 0;
    while ((cpudirent = readdir(cpudir))) {
        if (sscanf(cpudirent->d_name, "cpu%u", &cpu) != 1)
            continue;

        /* Parse socket */
        sock = virNodeParseSocket(node, cpu);
        CPU_SET(sock, &sock_map);

        if (sock > sock_max)
            sock_max = sock;

        errno = 0;
    }

    if (errno) {
        virReportSystemError(errno, _("problem reading %s"), node);
240 241
        goto cleanup;
    }
242 243 244 245 246 247

    sock_max++;

    /* allocate cpu maps for each socket */
    if (VIR_ALLOC_N(core_maps, sock_max) < 0) {
        virReportOOMError();
248
        goto cleanup;
249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298
    }

    for (i = 0; i < sock_max; i++)
        CPU_ZERO(&core_maps[i]);

    /* iterate over all CPU's in the node */
    rewinddir(cpudir);
    errno = 0;
    while ((cpudirent = readdir(cpudir))) {
        if (sscanf(cpudirent->d_name, "cpu%u", &cpu) != 1)
            continue;

        if ((online = virNodeGetCpuValue(node, cpu, "online", true)) < 0)
            goto cleanup;

        if (!online)
            continue;

        processors++;

        /* Parse socket */
        sock = virNodeParseSocket(node, cpu);
        if (!CPU_ISSET(sock, &sock_map)) {
            nodeReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("CPU socket topology has changed"));
            goto cleanup;
        }

        /* Parse core */
# if defined(__s390__) || \
    defined(__s390x__)
        /* logical cpu is equivalent to a core on s390 */
        core = cpu;
# else
        core = virNodeGetCpuValue(node, cpu, "topology/core_id", false);
# endif

        CPU_SET(core, &core_maps[sock]);

        if (!(siblings = virNodeCountThreadSiblings(node, cpu)))
            goto cleanup;

        if (siblings > *threads)
            *threads = siblings;

        errno = 0;
    }

    if (errno) {
        virReportSystemError(errno, _("problem reading %s"), node);
299 300
        goto cleanup;
    }
301 302 303 304 305 306 307 308 309 310 311 312 313 314

    /* finalize the returned data */
    *sockets = CPU_COUNT(&sock_map);

    for (i = 0; i < sock_max; i++) {
        if (!CPU_ISSET(i, &sock_map))
            continue;

        core = CPU_COUNT(&core_maps[i]);
        if (core > *cores)
            *cores = core;
    }

    ret = processors;
315 316

cleanup:
317 318 319 320 321 322 323
    /* don't shadow a more serious error */
    if (cpudir && closedir(cpudir) < 0 && ret >= 0) {
        virReportSystemError(errno, _("problem closing %s"), node);
        ret = -1;
    }
    VIR_FREE(core_maps);

324 325 326
    return ret;
}

327
int linuxNodeInfoCPUPopulate(FILE *cpuinfo,
328
                             const char *sysfs_dir,
329
                             virNodeInfoPtr nodeinfo)
330
{
331
    char line[1024];
332 333 334 335
    DIR *nodedir = NULL;
    struct dirent *nodedirent = NULL;
    int cpus, cores, socks, threads;
    unsigned int node;
336
    int ret = -1;
337
    char *sysfs_nodedir = NULL;
338
    char *sysfs_cpudir = NULL;
339 340 341

    nodeinfo->cpus = 0;
    nodeinfo->mhz = 0;
342
    nodeinfo->cores = 0;
343
    nodeinfo->nodes = 0;
344

345
    /* Start with parsing CPU clock speed from /proc/cpuinfo */
346
    while (fgets(line, sizeof(line), cpuinfo) != NULL) {
E
Eric Blake 已提交
347
# if defined(__x86_64__) || \
348 349
    defined(__amd64__)  || \
    defined(__i386__)
E
Eric Blake 已提交
350
        char *buf = line;
351
        if (STRPREFIX(buf, "cpu MHz")) {
352 353
            char *p;
            unsigned int ui;
354

355
            buf += 7;
356
            while (*buf && c_isspace(*buf))
357
                buf++;
358

359
            if (*buf != ':' || !buf[1]) {
360 361
                nodeReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                _("parsing cpu MHz from cpuinfo"));
362
                goto cleanup;
363
            }
364

365
            if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0 &&
366
                /* Accept trailing fractional part.  */
367
                (*p == '\0' || *p == '.' || c_isspace(*p)))
368
                nodeinfo->mhz = ui;
369
        }
370

E
Eric Blake 已提交
371
# elif defined(__powerpc__) || \
372
      defined(__powerpc64__)
E
Eric Blake 已提交
373
        char *buf = line;
374
        if (STRPREFIX(buf, "clock")) {
375 376
            char *p;
            unsigned int ui;
377

378 379 380
            buf += 5;
            while (*buf && c_isspace(*buf))
                buf++;
381

382
            if (*buf != ':' || !buf[1]) {
383 384
                nodeReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                _("parsing cpu MHz from cpuinfo"));
385
                goto cleanup;
386
            }
387

388
            if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0 &&
389
                /* Accept trailing fractional part.  */
390
                (*p == '\0' || *p == '.' || c_isspace(*p)))
391
                nodeinfo->mhz = ui;
392 393 394 395 396
            /* No other interesting infos are available in /proc/cpuinfo.
             * However, there is a line identifying processor's version,
             * identification and machine, but we don't want it to be caught
             * and parsed in next iteration, because it is not in expected
             * format and thus lead to error. */
397
        }
398 399 400 401 402
# elif defined(__s390__) || \
      defined(__s390x__)
        /* s390x has no realistic value for CPU speed,
         * assign a value of zero to signify this */
        nodeinfo->mhz = 0;
E
Eric Blake 已提交
403 404 405
# else
#  warning Parser for /proc/cpuinfo needs to be adapted for your architecture
# endif
406 407
    }

408 409
    /* OK, we've parsed clock speed out of /proc/cpuinfo. Get the
     * core, node, socket, thread and topology information from /sys
410
     */
411
    if (virAsprintf(&sysfs_nodedir, "%s/node", sysfs_dir) < 0) {
412 413 414
        virReportOOMError();
        goto cleanup;
    }
415

416 417 418 419
    if (!(nodedir = opendir(sysfs_nodedir))) {
        /* the host isn't probably running a NUMA architecture */
        goto fallback;
    }
420

421 422 423
    errno = 0;
    while ((nodedirent = readdir(nodedir))) {
        if (sscanf(nodedirent->d_name, "node%u", &node) != 1)
424 425
            continue;

426 427 428 429 430
        nodeinfo->nodes++;

        if (virAsprintf(&sysfs_cpudir, "%s/node/%s",
                        sysfs_dir, nodedirent->d_name) < 0) {
            virReportOOMError();
431
            goto cleanup;
E
Eric Blake 已提交
432 433
        }

434 435 436
        if ((cpus = virNodeParseNode(sysfs_cpudir, &socks,
                                     &cores, &threads)) < 0)
            goto cleanup;
437

438
        VIR_FREE(sysfs_cpudir);
439

440 441 442 443 444 445 446 447 448 449 450 451
        nodeinfo->cpus += cpus;

        if (socks > nodeinfo->sockets)
            nodeinfo->sockets = socks;

        if (cores > nodeinfo->cores)
            nodeinfo->cores = cores;

        if (threads > nodeinfo->threads)
            nodeinfo->threads = threads;

        errno = 0;
452
    }
453

E
Eric Blake 已提交
454
    if (errno) {
455
        virReportSystemError(errno, _("problem reading %s"), sysfs_nodedir);
456 457
        goto cleanup;
    }
458 459 460 461 462 463 464 465 466

    if (nodeinfo->cpus && nodeinfo->nodes)
        goto done;

fallback:
    VIR_FREE(sysfs_cpudir);

    if (virAsprintf(&sysfs_cpudir, "%s/cpu", sysfs_dir) < 0) {
        virReportOOMError();
467
        goto cleanup;
E
Eric Blake 已提交
468
    }
469

470
    if ((cpus = virNodeParseNode(sysfs_cpudir, &socks, &cores, &threads)) < 0)
471
        goto cleanup;
472

473 474 475 476 477 478 479
    nodeinfo->nodes = 1;
    nodeinfo->cpus = cpus;
    nodeinfo->sockets = socks;
    nodeinfo->cores = cores;
    nodeinfo->threads = threads;

done:
480
    /* There should always be at least one cpu, socket, node, and thread. */
481
    if (nodeinfo->cpus == 0) {
482
        nodeReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("no CPUs found"));
483
        goto cleanup;
484
    }
485

C
Chris Lalancette 已提交
486
    if (nodeinfo->sockets == 0) {
487
        nodeReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("no sockets found"));
488
        goto cleanup;
C
Chris Lalancette 已提交
489
    }
490

C
Chris Lalancette 已提交
491
    if (nodeinfo->threads == 0) {
492
        nodeReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("no threads found"));
493
        goto cleanup;
C
Chris Lalancette 已提交
494 495
    }

496 497 498
    ret = 0;

cleanup:
499 500 501 502 503 504 505
    /* don't shadow a more serious error */
    if (nodedir && closedir(nodedir) < 0 && ret >= 0) {
        virReportSystemError(errno, _("problem closing %s"), sysfs_nodedir);
        ret = -1;
    }

    VIR_FREE(sysfs_nodedir);
506 507
    VIR_FREE(sysfs_cpudir);
    return ret;
508 509
}

510 511 512 513
# define TICK_TO_NSEC (1000ull * 1000ull * 1000ull / sysconf(_SC_CLK_TCK))

int linuxNodeGetCPUStats(FILE *procstat,
                         int cpuNum,
514
                         virNodeCPUStatsPtr params,
515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530
                         int *nparams)
{
    int ret = -1;
    char line[1024];
    unsigned long long usr, ni, sys, idle, iowait;
    unsigned long long irq, softirq, steal, guest, guest_nice;
    char cpu_header[3 + INT_BUFSIZE_BOUND(cpuNum)];

    if ((*nparams) == 0) {
        /* Current number of cpu stats supported by linux */
        *nparams = LINUX_NB_CPU_STATS;
        ret = 0;
        goto cleanup;
    }

    if ((*nparams) != LINUX_NB_CPU_STATS) {
531 532 533
        virReportInvalidArg(*nparams,
                            _("nparams in %s must be equal to %d"),
                            __FUNCTION__, LINUX_NB_CPU_STATS);
534 535 536
        goto cleanup;
    }

537
    if (cpuNum == VIR_NODE_CPU_STATS_ALL_CPUS) {
538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
        strcpy(cpu_header, "cpu");
    } else {
        snprintf(cpu_header, sizeof(cpu_header), "cpu%d", cpuNum);
    }

    while (fgets(line, sizeof(line), procstat) != NULL) {
        char *buf = line;

        if (STRPREFIX(buf, cpu_header)) { /* aka logical CPU time */
            int i;

            if (sscanf(buf,
                       "%*s %llu %llu %llu %llu %llu" // user ~ iowait
                       "%llu %llu %llu %llu %llu",    // irq  ~ guest_nice
                       &usr, &ni, &sys, &idle, &iowait,
                       &irq, &softirq, &steal, &guest, &guest_nice) < 4) {
                continue;
            }

            for (i = 0; i < *nparams; i++) {
558
                virNodeCPUStatsPtr param = &params[i];
559 560 561

                switch (i) {
                case 0: /* fill kernel cpu time here */
562
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_KERNEL) == NULL) {
563 564 565 566 567 568 569 570
                        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                                        "%s", _("Field kernel cpu time too long for destination"));
                        goto cleanup;
                    }
                    param->value = (sys + irq + softirq) * TICK_TO_NSEC;
                    break;

                case 1: /* fill user cpu time here */
571
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_USER) == NULL) {
572 573 574 575 576 577 578 579
                        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                                        "%s", _("Field kernel cpu time too long for destination"));
                        goto cleanup;
                    }
                    param->value = (usr + ni) * TICK_TO_NSEC;
                    break;

                case 2: /* fill idle cpu time here */
580
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_IDLE) == NULL) {
581 582 583 584 585 586 587 588
                        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                                        "%s", _("Field kernel cpu time too long for destination"));
                        goto cleanup;
                    }
                    param->value = idle * TICK_TO_NSEC;
                    break;

                case 3: /* fill iowait cpu time here */
589
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_IOWAIT) == NULL) {
590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606
                        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                                        "%s", _("Field kernel cpu time too long for destination"));
                        goto cleanup;
                    }
                    param->value = iowait * TICK_TO_NSEC;
                    break;

                default:
                    break;
                    /* should not hit here */
                }
            }
            ret = 0;
            goto cleanup;
        }
    }

607 608 609
    virReportInvalidArg(cpuNum,
                        _("Invalid cpuNum in %s"),
                        __FUNCTION__);
610

611 612 613 614 615 616
cleanup:
    return ret;
}

int linuxNodeGetMemoryStats(FILE *meminfo,
                            int cellNum,
617
                            virNodeMemoryStatsPtr params,
618 619 620 621 622 623 624
                            int *nparams)
{
    int ret = -1;
    int i = 0, j = 0, k = 0;
    int found = 0;
    int nr_param;
    char line[1024];
625
    char meminfo_hdr[VIR_NODE_MEMORY_STATS_FIELD_LENGTH];
626 627 628 629 630
    unsigned long val;
    struct field_conv {
        const char *meminfo_hdr;  // meminfo header
        const char *field;        // MemoryStats field name
    } field_conv[] = {
631 632 633 634
        {"MemTotal:", VIR_NODE_MEMORY_STATS_TOTAL},
        {"MemFree:",  VIR_NODE_MEMORY_STATS_FREE},
        {"Buffers:",  VIR_NODE_MEMORY_STATS_BUFFERS},
        {"Cached:",   VIR_NODE_MEMORY_STATS_CACHED},
635 636 637
        {NULL,        NULL}
    };

638
    if (cellNum == VIR_NODE_MEMORY_STATS_ALL_CELLS) {
639 640 641 642 643 644 645 646 647 648 649 650 651
        nr_param = LINUX_NB_MEMORY_STATS_ALL;
    } else {
        nr_param = LINUX_NB_MEMORY_STATS_CELL;
    }

    if ((*nparams) == 0) {
        /* Current number of memory stats supported by linux */
        *nparams = nr_param;
        ret = 0;
        goto cleanup;
    }

    if ((*nparams) != nr_param) {
652 653 654
        virReportInvalidArg(nparams,
                            _("nparams in %s must be %d"),
                            __FUNCTION__, nr_param);
655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691
        goto cleanup;
    }

    while (fgets(line, sizeof(line), meminfo) != NULL) {
        char *buf = line;

        if (STRPREFIX(buf, "Node ")) {
            /*
             * /sys/devices/system/node/nodeX/meminfo format is below.
             * So, skip prefix "Node XX ".
             *
             * Node 0 MemTotal:        8386980 kB
             * Node 0 MemFree:         5300920 kB
             *         :
             */
            char *p;

            p = buf;
            for (i = 0; i < 2; i++) {
                p = strchr(p, ' ');
                if (p == NULL) {
                    nodeReportError(VIR_ERR_INTERNAL_ERROR,
                                    "%s", _("no prefix found"));
                    goto cleanup;
                }
                p++;
            }
            buf = p;
        }

        if (sscanf(buf, "%s %lu kB", meminfo_hdr, &val) < 2)
            continue;

        for (j = 0; field_conv[j].meminfo_hdr != NULL; j++) {
            struct field_conv *convp = &field_conv[j];

            if (STREQ(meminfo_hdr, convp->meminfo_hdr)) {
692
                virNodeMemoryStatsPtr param = &params[k++];
693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715

                if (virStrcpyStatic(param->field, convp->field) == NULL) {
                    nodeReportError(VIR_ERR_INTERNAL_ERROR,
                                    "%s", _("Field kernel memory too long for destination"));
                    goto cleanup;
                }
                param->value = val;
                found++;
                break;
            }
        }
        if (found >= nr_param)
            break;
    }

    if (found == 0) {
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no available memory line found"));
        goto cleanup;
    }

    ret = 0;

716 717 718
cleanup:
    return ret;
}
719 720 721 722 723 724 725 726 727 728 729

/*
 * Linux maintains cpu bit map. For example, if cpuid=5's flag is not set
 * and max cpu is 7. The map file shows 0-4,6-7. This function parses
 * it and returns cpumap.
 */
static char *
linuxParseCPUmap(int *max_cpuid, const char *path)
{
    char *map = NULL;
    char *str = NULL;
730
    int max_id = 0, i;
731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752

    if (virFileReadAll(path, 5 * VIR_DOMAIN_CPUMASK_LEN, &str) < 0) {
        virReportOOMError();
        goto error;
    }

    if (VIR_ALLOC_N(map, VIR_DOMAIN_CPUMASK_LEN) < 0) {
        virReportOOMError();
        goto error;
    }
    if (virDomainCpuSetParse(str, 0, map,
                             VIR_DOMAIN_CPUMASK_LEN) < 0) {
        goto error;
    }

    for (i = 0; i < VIR_DOMAIN_CPUMASK_LEN; i++) {
        if (map[i]) {
            max_id = i;
        }
    }
    *max_cpuid = max_id;

H
Hu Tao 已提交
753
    VIR_FREE(str);
754 755 756 757 758 759 760
    return map;

error:
    VIR_FREE(str);
    VIR_FREE(map);
    return NULL;
}
761 762
#endif

763
int nodeGetInfo(virConnectPtr conn ATTRIBUTE_UNUSED, virNodeInfoPtr nodeinfo) {
764 765
    struct utsname info;

E
Eric Blake 已提交
766
    memset(nodeinfo, 0, sizeof(*nodeinfo));
767 768
    uname(&info);

C
Chris Lalancette 已提交
769 770
    if (virStrcpyStatic(nodeinfo->model, info.machine) == NULL)
        return -1;
771

772
#ifdef __linux__
773
    {
774
    int ret = -1;
775
    FILE *cpuinfo = fopen(CPUINFO_PATH, "r");
776
    if (!cpuinfo) {
777
        virReportSystemError(errno,
778
                             _("cannot open %s"), CPUINFO_PATH);
779 780
        return -1;
    }
781

782
    ret = linuxNodeInfoCPUPopulate(cpuinfo, SYSFS_SYSTEM_PATH, nodeinfo);
783 784
    if (ret < 0)
        goto cleanup;
785

786 787
    /* Convert to KB. */
    nodeinfo->memory = physmem_total () / 1024;
788

789 790
cleanup:
    VIR_FORCE_FCLOSE(cpuinfo);
791
    return ret;
792
    }
793 794
#else
    /* XXX Solaris will need an impl later if they port QEMU driver */
O
Osier Yang 已提交
795
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
796
                    _("node info not implemented on this platform"));
797 798 799
    return -1;
#endif
}
800

801
int nodeGetCPUStats(virConnectPtr conn ATTRIBUTE_UNUSED,
802
                    int cpuNum ATTRIBUTE_UNUSED,
803
                    virNodeCPUStatsPtr params ATTRIBUTE_UNUSED,
804
                    int *nparams ATTRIBUTE_UNUSED,
805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823
                    unsigned int flags)
{
    virCheckFlags(0, -1);

#ifdef __linux__
    {
        int ret;
        FILE *procstat = fopen(PROCSTAT_PATH, "r");
        if (!procstat) {
            virReportSystemError(errno,
                                 _("cannot open %s"), PROCSTAT_PATH);
            return -1;
        }
        ret = linuxNodeGetCPUStats(procstat, cpuNum, params, nparams);
        VIR_FORCE_FCLOSE(procstat);

        return ret;
    }
#else
O
Osier Yang 已提交
824
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
825 826 827 828 829
                    _("node CPU stats not implemented on this platform"));
    return -1;
#endif
}

830
int nodeGetMemoryStats(virConnectPtr conn ATTRIBUTE_UNUSED,
831
                       int cellNum ATTRIBUTE_UNUSED,
832
                       virNodeMemoryStatsPtr params ATTRIBUTE_UNUSED,
833
                       int *nparams ATTRIBUTE_UNUSED,
834 835 836 837 838 839 840 841 842 843
                       unsigned int flags)
{
    virCheckFlags(0, -1);

#ifdef __linux__
    {
        int ret;
        char *meminfo_path = NULL;
        FILE *meminfo;

844
        if (cellNum == VIR_NODE_MEMORY_STATS_ALL_CELLS) {
845 846 847 848 849 850
            meminfo_path = strdup(MEMINFO_PATH);
            if (!meminfo_path) {
                virReportOOMError();
                return -1;
            }
        } else {
851
# if HAVE_NUMACTL
852
            if (numa_available() < 0) {
853
# endif
O
Osier Yang 已提交
854
                nodeReportError(VIR_ERR_INTERNAL_ERROR,
855 856
                                "%s", _("NUMA not supported on this host"));
                return -1;
857
# if HAVE_NUMACTL
858
            }
859
# endif
860

861
# if HAVE_NUMACTL
862
            if (cellNum > numa_max_node()) {
863 864 865
                virReportInvalidArg(cellNum,
                                    _("cellNum in %s must be less than or equal to %d"),
                                    __FUNCTION__, numa_max_node());
866 867
                return -1;
            }
868
# endif
869

870 871
            if (virAsprintf(&meminfo_path, "%s/node/node%d/meminfo",
                            SYSFS_SYSTEM_PATH, cellNum) < 0) {
872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890
                virReportOOMError();
                return -1;
            }
        }
        meminfo = fopen(meminfo_path, "r");

        if (!meminfo) {
            virReportSystemError(errno,
                                 _("cannot open %s"), meminfo_path);
            VIR_FREE(meminfo_path);
            return -1;
        }
        ret = linuxNodeGetMemoryStats(meminfo, cellNum, params, nparams);
        VIR_FORCE_FCLOSE(meminfo);
        VIR_FREE(meminfo_path);

        return ret;
    }
#else
O
Osier Yang 已提交
891
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
892 893 894 895 896
                    _("node memory stats not implemented on this platform"));
    return -1;
#endif
}

897 898 899 900 901 902 903 904 905
char *
nodeGetCPUmap(virConnectPtr conn ATTRIBUTE_UNUSED,
              int *max_id ATTRIBUTE_UNUSED,
              const char *mapname ATTRIBUTE_UNUSED)
{
#ifdef __linux__
    char *path;
    char *cpumap;

906
    if (virAsprintf(&path, SYSFS_SYSTEM_PATH "/cpu/%s", mapname) < 0) {
907 908 909 910 911 912 913 914 915 916 917 918 919 920
        virReportOOMError();
        return NULL;
    }

    cpumap = linuxParseCPUmap(max_id, path);
    VIR_FREE(path);
    return cpumap;
#else
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
                    _("node cpumap not implemented on this platform"));
    return NULL;
#endif
}

921 922 923 924 925 926 927 928 929 930 931 932
#if HAVE_NUMACTL
# if LIBNUMA_API_VERSION <= 1
#  define NUMA_MAX_N_CPUS 4096
# else
#  define NUMA_MAX_N_CPUS (numa_all_cpus_ptr->size)
# endif

# define n_bits(var) (8 * sizeof(var))
# define MASK_CPU_ISSET(mask, cpu) \
  (((mask)[((cpu) / n_bits(*(mask)))] >> ((cpu) % n_bits(*(mask)))) & 1)

int
933
nodeCapsInitNUMA(virCapsPtr caps)
934 935
{
    int n;
936
    unsigned long *mask = NULL;
937
    unsigned long *allonesmask = NULL;
938 939 940 941 942 943 944 945
    int *cpus = NULL;
    int ret = -1;
    int max_n_cpus = NUMA_MAX_N_CPUS;

    if (numa_available() < 0)
        return 0;

    int mask_n_bytes = max_n_cpus / 8;
946
    if (VIR_ALLOC_N(mask, mask_n_bytes / sizeof(*mask)) < 0)
947
        goto cleanup;
948
    if (VIR_ALLOC_N(allonesmask, mask_n_bytes / sizeof(*mask)) < 0)
949 950
        goto cleanup;
    memset(allonesmask, 0xff, mask_n_bytes);
951 952 953 954

    for (n = 0 ; n <= numa_max_node() ; n++) {
        int i;
        int ncpus;
955
        /* The first time this returns -1, ENOENT if node doesn't exist... */
956 957
        if (numa_node_to_cpus(n, mask, mask_n_bytes) < 0) {
            VIR_WARN("NUMA topology for cell %d of %d not available, ignoring",
958 959 960 961 962 963 964
                     n, numa_max_node()+1);
            continue;
        }
        /* second, third... times it returns an all-1's mask */
        if (memcmp(mask, allonesmask, mask_n_bytes) == 0) {
            VIR_DEBUG("NUMA topology for cell %d of %d is all ones, ignoring",
                      n, numa_max_node()+1);
965 966
            continue;
        }
967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992

        for (ncpus = 0, i = 0 ; i < max_n_cpus ; i++)
            if (MASK_CPU_ISSET(mask, i))
                ncpus++;

        if (VIR_ALLOC_N(cpus, ncpus) < 0)
            goto cleanup;

        for (ncpus = 0, i = 0 ; i < max_n_cpus ; i++)
            if (MASK_CPU_ISSET(mask, i))
                cpus[ncpus++] = i;

        if (virCapabilitiesAddHostNUMACell(caps,
                                           n,
                                           ncpus,
                                           cpus) < 0)
            goto cleanup;

        VIR_FREE(cpus);
    }

    ret = 0;

cleanup:
    VIR_FREE(cpus);
    VIR_FREE(mask);
993
    VIR_FREE(allonesmask);
994 995
    return ret;
}
996 997 998


int
999
nodeGetCellsFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED,
1000 1001 1002 1003 1004 1005 1006 1007 1008
                       unsigned long long *freeMems,
                       int startCell,
                       int maxCells)
{
    int n, lastCell, numCells;
    int ret = -1;
    int maxCell;

    if (numa_available() < 0) {
O
Osier Yang 已提交
1009
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
1010 1011 1012 1013 1014
                        "%s", _("NUMA not supported on this host"));
        goto cleanup;
    }
    maxCell = numa_max_node();
    if (startCell > maxCell) {
1015
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026
                        _("start cell %d out of range (0-%d)"),
                        startCell, maxCell);
        goto cleanup;
    }
    lastCell = startCell + maxCells - 1;
    if (lastCell > maxCell)
        lastCell = maxCell;

    for (numCells = 0, n = startCell ; n <= lastCell ; n++) {
        long long mem;
        if (numa_node_size64(n, &mem) < 0) {
1027
            nodeReportError(VIR_ERR_INTERNAL_ERROR,
1028 1029
                           _("Failed to query NUMA free memory for node: %d"),
                           n);
1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040
            goto cleanup;
        }
        freeMems[numCells++] = mem;
    }
    ret = numCells;

cleanup:
    return ret;
}

unsigned long long
1041
nodeGetFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED)
1042 1043 1044 1045 1046
{
    unsigned long long freeMem = 0;
    int n;

    if (numa_available() < 0) {
O
Osier Yang 已提交
1047
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
1048 1049 1050 1051 1052 1053 1054
                        "%s", _("NUMA not supported on this host"));
        goto cleanup;
    }

    for (n = 0 ; n <= numa_max_node() ; n++) {
        long long mem;
        if (numa_node_size64(n, &mem) < 0) {
1055
            nodeReportError(VIR_ERR_INTERNAL_ERROR,
1056 1057 1058 1059 1060 1061 1062 1063 1064 1065
                            "%s", _("Failed to query NUMA free memory"));
            goto cleanup;
        }
        freeMem += mem;
    }

cleanup:
    return freeMem;
}

1066
#else
1067 1068 1069 1070
int nodeCapsInitNUMA(virCapsPtr caps ATTRIBUTE_UNUSED) {
    return 0;
}

1071
int nodeGetCellsFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED,
1072 1073 1074 1075
                              unsigned long long *freeMems ATTRIBUTE_UNUSED,
                              int startCell ATTRIBUTE_UNUSED,
                              int maxCells ATTRIBUTE_UNUSED)
{
O
Osier Yang 已提交
1076
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
1077 1078 1079 1080
                    _("NUMA memory information not available on this platform"));
    return -1;
}

1081
unsigned long long nodeGetFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED)
1082
{
O
Osier Yang 已提交
1083
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
1084 1085 1086
                    _("NUMA memory information not available on this platform"));
    return 0;
}
1087
#endif