nodeinfo.c 29.7 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 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225
static int
virNodeParseNode(const char *sysfs_dir)
{
    char *file = NULL;
    char *possible = NULL;
    char *tmp;
    int ret = -1;

    if (virAsprintf(&file, "%s/node/possible", sysfs_dir) < 0) {
        virReportOOMError();
        goto cleanup;
    }
    /* Assume that a missing node/possible file implies no NUMA
     * support, and hence all cpus belong to the same node.  */
    if (!virFileExists(file)) {
        ret = 1;
        goto cleanup;
    }
    if (virFileReadAll(file, 1024, &possible) < 0)
        goto cleanup;
    if (virStrToLong_i(possible, &tmp, 10, &ret) < 0 ||
        (*tmp == '-' && virStrToLong_i(tmp+1, &tmp, 10, &ret) < 0) ||
        *tmp != '\n') {
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                        _("failed to parse possible nodes '%s'"), possible);
        goto cleanup;
    }
    ret++;

cleanup:
    VIR_FREE(file);
    VIR_FREE(possible);
    return ret;
}

226
int linuxNodeInfoCPUPopulate(FILE *cpuinfo,
227
                             const char *sysfs_dir,
228
                             virNodeInfoPtr nodeinfo)
229
{
230
    char line[1024];
231 232
    DIR *cpudir = NULL;
    struct dirent *cpudirent = NULL;
C
Chris Lalancette 已提交
233
    unsigned int cpu;
H
Hu Tao 已提交
234
    unsigned long core, sock, cur_threads;
235 236
    cpu_set_t core_mask;
    cpu_set_t socket_mask;
E
Eric Blake 已提交
237
    int online;
238 239
    int ret = -1;
    char *sysfs_cpudir = NULL;
240
    unsigned int cpu_cores = 0;
241 242 243

    nodeinfo->cpus = 0;
    nodeinfo->mhz = 0;
244
    nodeinfo->cores = 0;
245

246 247
    /* Start with parsing /proc/cpuinfo; although it tends to have
     * fewer details.  Hyperthreads are ignored at this stage.  */
248
    while (fgets(line, sizeof(line), cpuinfo) != NULL) {
E
Eric Blake 已提交
249
# if defined(__x86_64__) || \
250 251
    defined(__amd64__)  || \
    defined(__i386__)
E
Eric Blake 已提交
252
        char *buf = line;
253
        if (STRPREFIX(buf, "cpu MHz")) {
254 255
            char *p;
            unsigned int ui;
256

257
            buf += 9;
258
            while (*buf && c_isspace(*buf))
259
                buf++;
260

261
            if (*buf != ':' || !buf[1]) {
262
                nodeReportError(VIR_ERR_INTERNAL_ERROR,
263
                                "%s", _("parsing cpu MHz from cpuinfo"));
264
                goto cleanup;
265
            }
266

267
            if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0
268
                /* Accept trailing fractional part.  */
269
                && (*p == '\0' || *p == '.' || c_isspace(*p)))
270
                nodeinfo->mhz = ui;
271
        }
272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290

        if (STRPREFIX(buf, "cpu cores")) {
            char *p;
            unsigned int ui;

            buf += 9;
            while (*buf && c_isspace(*buf))
                buf++;

            if (*buf != ':' || !buf[1]) {
                nodeReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("parsing cpu cores from cpuinfo"));
                return -1;
            }

            if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0
                && (*p == '\0' || c_isspace(*p)))
                cpu_cores = ui;
         }
E
Eric Blake 已提交
291
# elif defined(__powerpc__) || \
292
      defined(__powerpc64__)
E
Eric Blake 已提交
293
        char *buf = line;
294
        if (STRPREFIX(buf, "clock")) {
295 296
            char *p;
            unsigned int ui;
297

298 299 300
            buf += 5;
            while (*buf && c_isspace(*buf))
                buf++;
301

302 303
            if (*buf != ':' || !buf[1]) {
                nodeReportError(VIR_ERR_INTERNAL_ERROR,
304
                                "%s", _("parsing cpu MHz from cpuinfo"));
305
                goto cleanup;
306
            }
307

308 309 310 311
            if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0
                /* Accept trailing fractional part.  */
                && (*p == '\0' || *p == '.' || c_isspace(*p)))
                nodeinfo->mhz = ui;
312 313 314 315 316
            /* 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. */
317
        }
318 319 320 321 322
# 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 已提交
323 324 325
# else
#  warning Parser for /proc/cpuinfo needs to be adapted for your architecture
# endif
326 327
    }

328 329
    /* OK, we've parsed clock speed out of /proc/cpuinfo. Get the
     * core, node, socket, thread and topology information from /sys
330
     */
331 332 333 334
    if (virAsprintf(&sysfs_cpudir, "%s/cpu", sysfs_dir) < 0) {
        virReportOOMError();
        goto cleanup;
    }
335
    cpudir = opendir(sysfs_cpudir);
336
    if (cpudir == NULL) {
337
        virReportSystemError(errno, _("cannot opendir %s"), sysfs_cpudir);
338
        goto cleanup;
339
    }
340 341 342 343 344

    CPU_ZERO(&core_mask);
    CPU_ZERO(&socket_mask);

    while ((cpudirent = readdir(cpudir))) {
C
Chris Lalancette 已提交
345
        if (sscanf(cpudirent->d_name, "cpu%u", &cpu) != 1)
346 347
            continue;

E
Eric Blake 已提交
348
        online = virNodeGetCpuValue(sysfs_cpudir, cpu, "online", true);
E
Eric Blake 已提交
349 350
        if (online < 0) {
            closedir(cpudir);
351
            goto cleanup;
E
Eric Blake 已提交
352 353 354
        }
        if (!online)
            continue;
355
        nodeinfo->cpus++;
E
Eric Blake 已提交
356

357
        /* Parse core */
358 359 360 361 362
# if defined(__s390__) || \
    defined(__s390x__)
    /* logical cpu is equivalent to a core on s390 */
        core = cpu;
# else
E
Eric Blake 已提交
363
        core = virNodeGetCpuValue(sysfs_cpudir, cpu, "topology/core_id", false);
364
# endif
365 366 367
        if (!CPU_ISSET(core, &core_mask)) {
            CPU_SET(core, &core_mask);
            nodeinfo->cores++;
368
        }
369 370

        /* Parse socket */
E
Eric Blake 已提交
371
        sock = virNodeParseSocket(sysfs_cpudir, cpu);
H
Hu Tao 已提交
372 373
        if (!CPU_ISSET(sock, &socket_mask)) {
            CPU_SET(sock, &socket_mask);
374 375 376
            nodeinfo->sockets++;
        }

E
Eric Blake 已提交
377
        cur_threads = virNodeCountThreadSiblings(sysfs_cpudir, cpu);
378 379
        if (cur_threads == 0) {
            closedir(cpudir);
380
            goto cleanup;
381 382 383 384
        }
        if (cur_threads > nodeinfo->threads)
            nodeinfo->threads = cur_threads;
    }
E
Eric Blake 已提交
385 386
    if (errno) {
        virReportSystemError(errno,
E
Eric Blake 已提交
387
                             _("problem reading %s"), sysfs_cpudir);
E
Eric Blake 已提交
388
        closedir(cpudir);
389 390 391 392 393 394
        goto cleanup;
    }
    if (closedir(cpudir) < 0) {
        virReportSystemError(errno,
                             _("problem closing %s"), sysfs_cpudir);
        goto cleanup;
E
Eric Blake 已提交
395
    }
396

397 398
    if ((nodeinfo->nodes = virNodeParseNode(sysfs_dir)) <= 0)
        goto cleanup;
399

400
    /* There should always be at least one cpu, socket, node, and thread. */
401 402 403
    if (nodeinfo->cpus == 0) {
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no CPUs found"));
404
        goto cleanup;
405
    }
C
Chris Lalancette 已提交
406
    if (nodeinfo->sockets == 0) {
407
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
408
                        "%s", _("no sockets found"));
409
        goto cleanup;
C
Chris Lalancette 已提交
410 411
    }
    if (nodeinfo->threads == 0) {
412
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
413
                        "%s", _("no threads found"));
414
        goto cleanup;
C
Chris Lalancette 已提交
415 416
    }

417 418 419 420 421 422 423 424 425 426
    /* Platform like AMD Magny Cours has two NUMA nodes each package, and
     * the two nodes share the same core ID set, it results in the cores
     * number calculated from sysfs is not the actual cores number. Use
     * "cpu cores" in /proc/cpuinfo as the cores number instead in this case.
     * More details about the problem:
     * https://www.redhat.com/archives/libvir-list/2012-May/msg00607.html
     */
    if (cpu_cores && (cpu_cores > nodeinfo->cores))
        nodeinfo->cores = cpu_cores;

427 428 429 430 431 432 433 434 435 436
    /* nodeinfo->sockets is supposed to be a number of sockets per NUMA node,
     * however if NUMA nodes are not composed of whole sockets, we just lie
     * about the number of NUMA nodes and force apps to check capabilities XML
     * for the actual NUMA topology.
     */
    if (nodeinfo->sockets % nodeinfo->nodes == 0)
        nodeinfo->sockets /= nodeinfo->nodes;
    else
        nodeinfo->nodes = 1;

437 438 439 440 441
    ret = 0;

cleanup:
    VIR_FREE(sysfs_cpudir);
    return ret;
442 443
}

444 445 446 447
# define TICK_TO_NSEC (1000ull * 1000ull * 1000ull / sysconf(_SC_CLK_TCK))

int linuxNodeGetCPUStats(FILE *procstat,
                         int cpuNum,
448
                         virNodeCPUStatsPtr params,
449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464
                         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) {
465 466 467
        virReportInvalidArg(*nparams,
                            _("nparams in %s must be equal to %d"),
                            __FUNCTION__, LINUX_NB_CPU_STATS);
468 469 470
        goto cleanup;
    }

471
    if (cpuNum == VIR_NODE_CPU_STATS_ALL_CPUS) {
472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491
        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++) {
492
                virNodeCPUStatsPtr param = &params[i];
493 494 495

                switch (i) {
                case 0: /* fill kernel cpu time here */
496
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_KERNEL) == NULL) {
497 498 499 500 501 502 503 504
                        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 */
505
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_USER) == NULL) {
506 507 508 509 510 511 512 513
                        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 */
514
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_IDLE) == NULL) {
515 516 517 518 519 520 521 522
                        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 */
523
                    if (virStrcpyStatic(param->field, VIR_NODE_CPU_STATS_IOWAIT) == NULL) {
524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540
                        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;
        }
    }

541 542 543
    virReportInvalidArg(cpuNum,
                        _("Invalid cpuNum in %s"),
                        __FUNCTION__);
544

545 546 547 548 549 550
cleanup:
    return ret;
}

int linuxNodeGetMemoryStats(FILE *meminfo,
                            int cellNum,
551
                            virNodeMemoryStatsPtr params,
552 553 554 555 556 557 558
                            int *nparams)
{
    int ret = -1;
    int i = 0, j = 0, k = 0;
    int found = 0;
    int nr_param;
    char line[1024];
559
    char meminfo_hdr[VIR_NODE_MEMORY_STATS_FIELD_LENGTH];
560 561 562 563 564
    unsigned long val;
    struct field_conv {
        const char *meminfo_hdr;  // meminfo header
        const char *field;        // MemoryStats field name
    } field_conv[] = {
565 566 567 568
        {"MemTotal:", VIR_NODE_MEMORY_STATS_TOTAL},
        {"MemFree:",  VIR_NODE_MEMORY_STATS_FREE},
        {"Buffers:",  VIR_NODE_MEMORY_STATS_BUFFERS},
        {"Cached:",   VIR_NODE_MEMORY_STATS_CACHED},
569 570 571
        {NULL,        NULL}
    };

572
    if (cellNum == VIR_NODE_MEMORY_STATS_ALL_CELLS) {
573 574 575 576 577 578 579 580 581 582 583 584 585
        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) {
586 587 588
        virReportInvalidArg(nparams,
                            _("nparams in %s must be %d"),
                            __FUNCTION__, nr_param);
589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625
        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)) {
626
                virNodeMemoryStatsPtr param = &params[k++];
627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649

                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;

650 651 652
cleanup:
    return ret;
}
653 654 655 656 657 658 659 660 661 662 663

/*
 * 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;
664
    int max_id = 0, i;
665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686

    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 已提交
687
    VIR_FREE(str);
688 689 690 691 692 693 694
    return map;

error:
    VIR_FREE(str);
    VIR_FREE(map);
    return NULL;
}
695 696
#endif

697
int nodeGetInfo(virConnectPtr conn ATTRIBUTE_UNUSED, virNodeInfoPtr nodeinfo) {
698 699
    struct utsname info;

E
Eric Blake 已提交
700
    memset(nodeinfo, 0, sizeof(*nodeinfo));
701 702
    uname(&info);

C
Chris Lalancette 已提交
703 704
    if (virStrcpyStatic(nodeinfo->model, info.machine) == NULL)
        return -1;
705

706
#ifdef __linux__
707
    {
708
    int ret = -1;
709
    FILE *cpuinfo = fopen(CPUINFO_PATH, "r");
710
    if (!cpuinfo) {
711
        virReportSystemError(errno,
712
                             _("cannot open %s"), CPUINFO_PATH);
713 714
        return -1;
    }
715

716
    ret = linuxNodeInfoCPUPopulate(cpuinfo, SYSFS_SYSTEM_PATH, nodeinfo);
717 718
    if (ret < 0)
        goto cleanup;
719

720 721
    /* Convert to KB. */
    nodeinfo->memory = physmem_total () / 1024;
722

723 724
cleanup:
    VIR_FORCE_FCLOSE(cpuinfo);
725
    return ret;
726
    }
727 728
#else
    /* XXX Solaris will need an impl later if they port QEMU driver */
O
Osier Yang 已提交
729
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
730
                    _("node info not implemented on this platform"));
731 732 733
    return -1;
#endif
}
734

735
int nodeGetCPUStats(virConnectPtr conn ATTRIBUTE_UNUSED,
736
                    int cpuNum ATTRIBUTE_UNUSED,
737
                    virNodeCPUStatsPtr params ATTRIBUTE_UNUSED,
738
                    int *nparams ATTRIBUTE_UNUSED,
739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757
                    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 已提交
758
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
759 760 761 762 763
                    _("node CPU stats not implemented on this platform"));
    return -1;
#endif
}

764
int nodeGetMemoryStats(virConnectPtr conn ATTRIBUTE_UNUSED,
765
                       int cellNum ATTRIBUTE_UNUSED,
766
                       virNodeMemoryStatsPtr params ATTRIBUTE_UNUSED,
767
                       int *nparams ATTRIBUTE_UNUSED,
768 769 770 771 772 773 774 775 776 777
                       unsigned int flags)
{
    virCheckFlags(0, -1);

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

778
        if (cellNum == VIR_NODE_MEMORY_STATS_ALL_CELLS) {
779 780 781 782 783 784
            meminfo_path = strdup(MEMINFO_PATH);
            if (!meminfo_path) {
                virReportOOMError();
                return -1;
            }
        } else {
785
# if HAVE_NUMACTL
786
            if (numa_available() < 0) {
787
# endif
O
Osier Yang 已提交
788
                nodeReportError(VIR_ERR_INTERNAL_ERROR,
789 790
                                "%s", _("NUMA not supported on this host"));
                return -1;
791
# if HAVE_NUMACTL
792
            }
793
# endif
794

795
# if HAVE_NUMACTL
796
            if (cellNum > numa_max_node()) {
797 798 799
                virReportInvalidArg(cellNum,
                                    _("cellNum in %s must be less than or equal to %d"),
                                    __FUNCTION__, numa_max_node());
800 801
                return -1;
            }
802
# endif
803

804 805
            if (virAsprintf(&meminfo_path, "%s/node/node%d/meminfo",
                            SYSFS_SYSTEM_PATH, cellNum) < 0) {
806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824
                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 已提交
825
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
826 827 828 829 830
                    _("node memory stats not implemented on this platform"));
    return -1;
#endif
}

831 832 833 834 835 836 837 838 839
char *
nodeGetCPUmap(virConnectPtr conn ATTRIBUTE_UNUSED,
              int *max_id ATTRIBUTE_UNUSED,
              const char *mapname ATTRIBUTE_UNUSED)
{
#ifdef __linux__
    char *path;
    char *cpumap;

840
    if (virAsprintf(&path, SYSFS_SYSTEM_PATH "/cpu/%s", mapname) < 0) {
841 842 843 844 845 846 847 848 849 850 851 852 853 854
        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
}

855 856 857 858 859 860 861 862 863 864 865 866
#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
867
nodeCapsInitNUMA(virCapsPtr caps)
868 869
{
    int n;
870
    unsigned long *mask = NULL;
871
    unsigned long *allonesmask = NULL;
872 873 874 875 876 877 878 879
    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;
880
    if (VIR_ALLOC_N(mask, mask_n_bytes / sizeof(*mask)) < 0)
881
        goto cleanup;
882
    if (VIR_ALLOC_N(allonesmask, mask_n_bytes / sizeof(*mask)) < 0)
883 884
        goto cleanup;
    memset(allonesmask, 0xff, mask_n_bytes);
885 886 887 888

    for (n = 0 ; n <= numa_max_node() ; n++) {
        int i;
        int ncpus;
889
        /* The first time this returns -1, ENOENT if node doesn't exist... */
890 891
        if (numa_node_to_cpus(n, mask, mask_n_bytes) < 0) {
            VIR_WARN("NUMA topology for cell %d of %d not available, ignoring",
892 893 894 895 896 897 898
                     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);
899 900
            continue;
        }
901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926

        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);
927
    VIR_FREE(allonesmask);
928 929
    return ret;
}
930 931 932


int
933
nodeGetCellsFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED,
934 935 936 937 938 939 940 941 942
                       unsigned long long *freeMems,
                       int startCell,
                       int maxCells)
{
    int n, lastCell, numCells;
    int ret = -1;
    int maxCell;

    if (numa_available() < 0) {
O
Osier Yang 已提交
943
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
944 945 946 947 948
                        "%s", _("NUMA not supported on this host"));
        goto cleanup;
    }
    maxCell = numa_max_node();
    if (startCell > maxCell) {
949
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
950 951 952 953 954 955 956 957 958 959 960
                        _("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) {
961
            nodeReportError(VIR_ERR_INTERNAL_ERROR,
962 963
                           _("Failed to query NUMA free memory for node: %d"),
                           n);
964 965 966 967 968 969 970 971 972 973 974
            goto cleanup;
        }
        freeMems[numCells++] = mem;
    }
    ret = numCells;

cleanup:
    return ret;
}

unsigned long long
975
nodeGetFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED)
976 977 978 979 980
{
    unsigned long long freeMem = 0;
    int n;

    if (numa_available() < 0) {
O
Osier Yang 已提交
981
        nodeReportError(VIR_ERR_INTERNAL_ERROR,
982 983 984 985 986 987 988
                        "%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) {
989
            nodeReportError(VIR_ERR_INTERNAL_ERROR,
990 991 992 993 994 995 996 997 998 999
                            "%s", _("Failed to query NUMA free memory"));
            goto cleanup;
        }
        freeMem += mem;
    }

cleanup:
    return freeMem;
}

1000
#else
1001 1002 1003 1004
int nodeCapsInitNUMA(virCapsPtr caps ATTRIBUTE_UNUSED) {
    return 0;
}

1005
int nodeGetCellsFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED,
1006 1007 1008 1009
                              unsigned long long *freeMems ATTRIBUTE_UNUSED,
                              int startCell ATTRIBUTE_UNUSED,
                              int maxCells ATTRIBUTE_UNUSED)
{
O
Osier Yang 已提交
1010
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
1011 1012 1013 1014
                    _("NUMA memory information not available on this platform"));
    return -1;
}

1015
unsigned long long nodeGetFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED)
1016
{
O
Osier Yang 已提交
1017
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
1018 1019 1020
                    _("NUMA memory information not available on this platform"));
    return 0;
}
1021
#endif