nodeinfo.c 29.5 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 */
E
Eric Blake 已提交
358
        core = virNodeGetCpuValue(sysfs_cpudir, cpu, "topology/core_id", false);
359 360 361
        if (!CPU_ISSET(core, &core_mask)) {
            CPU_SET(core, &core_mask);
            nodeinfo->cores++;
362
        }
363 364

        /* Parse socket */
E
Eric Blake 已提交
365
        sock = virNodeParseSocket(sysfs_cpudir, cpu);
H
Hu Tao 已提交
366 367
        if (!CPU_ISSET(sock, &socket_mask)) {
            CPU_SET(sock, &socket_mask);
368 369 370
            nodeinfo->sockets++;
        }

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

391 392
    if ((nodeinfo->nodes = virNodeParseNode(sysfs_dir)) <= 0)
        goto cleanup;
393

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

411 412 413 414 415 416 417 418 419 420
    /* 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;

421 422 423 424 425 426 427 428 429 430
    /* 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;

431 432 433 434 435
    ret = 0;

cleanup:
    VIR_FREE(sysfs_cpudir);
    return ret;
436 437
}

438 439 440 441
# define TICK_TO_NSEC (1000ull * 1000ull * 1000ull / sysconf(_SC_CLK_TCK))

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

465
    if (cpuNum == VIR_NODE_CPU_STATS_ALL_CPUS) {
466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485
        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++) {
486
                virNodeCPUStatsPtr param = &params[i];
487 488 489

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

535 536 537
    virReportInvalidArg(cpuNum,
                        _("Invalid cpuNum in %s"),
                        __FUNCTION__);
538

539 540 541 542 543 544
cleanup:
    return ret;
}

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

566
    if (cellNum == VIR_NODE_MEMORY_STATS_ALL_CELLS) {
567 568 569 570 571 572 573 574 575 576 577 578 579
        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) {
580 581 582
        virReportInvalidArg(nparams,
                            _("nparams in %s must be %d"),
                            __FUNCTION__, nr_param);
583 584 585 586 587 588 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
        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)) {
620
                virNodeMemoryStatsPtr param = &params[k++];
621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643

                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;

644 645 646
cleanup:
    return ret;
}
647 648 649 650 651 652 653 654 655 656 657

/*
 * 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;
658
    int max_id = 0, i;
659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680

    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 已提交
681
    VIR_FREE(str);
682 683 684 685 686 687 688
    return map;

error:
    VIR_FREE(str);
    VIR_FREE(map);
    return NULL;
}
689 690
#endif

691
int nodeGetInfo(virConnectPtr conn ATTRIBUTE_UNUSED, virNodeInfoPtr nodeinfo) {
692 693
    struct utsname info;

E
Eric Blake 已提交
694
    memset(nodeinfo, 0, sizeof(*nodeinfo));
695 696
    uname(&info);

C
Chris Lalancette 已提交
697 698
    if (virStrcpyStatic(nodeinfo->model, info.machine) == NULL)
        return -1;
699

700
#ifdef __linux__
701
    {
702
    int ret = -1;
703
    FILE *cpuinfo = fopen(CPUINFO_PATH, "r");
704
    if (!cpuinfo) {
705
        virReportSystemError(errno,
706
                             _("cannot open %s"), CPUINFO_PATH);
707 708
        return -1;
    }
709

710
    ret = linuxNodeInfoCPUPopulate(cpuinfo, SYSFS_SYSTEM_PATH, nodeinfo);
711 712
    if (ret < 0)
        goto cleanup;
713

714 715
    /* Convert to KB. */
    nodeinfo->memory = physmem_total () / 1024;
716

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

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

758
int nodeGetMemoryStats(virConnectPtr conn ATTRIBUTE_UNUSED,
759
                       int cellNum ATTRIBUTE_UNUSED,
760
                       virNodeMemoryStatsPtr params ATTRIBUTE_UNUSED,
761
                       int *nparams ATTRIBUTE_UNUSED,
762 763 764 765 766 767 768 769 770 771
                       unsigned int flags)
{
    virCheckFlags(0, -1);

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

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

789
# if HAVE_NUMACTL
790
            if (cellNum > numa_max_node()) {
791 792 793
                virReportInvalidArg(cellNum,
                                    _("cellNum in %s must be less than or equal to %d"),
                                    __FUNCTION__, numa_max_node());
794 795
                return -1;
            }
796
# endif
797

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

825 826 827 828 829 830 831 832 833
char *
nodeGetCPUmap(virConnectPtr conn ATTRIBUTE_UNUSED,
              int *max_id ATTRIBUTE_UNUSED,
              const char *mapname ATTRIBUTE_UNUSED)
{
#ifdef __linux__
    char *path;
    char *cpumap;

834
    if (virAsprintf(&path, SYSFS_SYSTEM_PATH "/cpu/%s", mapname) < 0) {
835 836 837 838 839 840 841 842 843 844 845 846 847 848
        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
}

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

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

        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);
921
    VIR_FREE(allonesmask);
922 923
    return ret;
}
924 925 926


int
927
nodeGetCellsFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED,
928 929 930 931 932 933 934 935 936
                       unsigned long long *freeMems,
                       int startCell,
                       int maxCells)
{
    int n, lastCell, numCells;
    int ret = -1;
    int maxCell;

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

cleanup:
    return ret;
}

unsigned long long
969
nodeGetFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED)
970 971 972 973 974
{
    unsigned long long freeMem = 0;
    int n;

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

cleanup:
    return freeMem;
}

994
#else
995 996 997 998
int nodeCapsInitNUMA(virCapsPtr caps ATTRIBUTE_UNUSED) {
    return 0;
}

999
int nodeGetCellsFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED,
1000 1001 1002 1003
                              unsigned long long *freeMems ATTRIBUTE_UNUSED,
                              int startCell ATTRIBUTE_UNUSED,
                              int maxCells ATTRIBUTE_UNUSED)
{
O
Osier Yang 已提交
1004
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
1005 1006 1007 1008
                    _("NUMA memory information not available on this platform"));
    return -1;
}

1009
unsigned long long nodeGetFreeMemory(virConnectPtr conn ATTRIBUTE_UNUSED)
1010
{
O
Osier Yang 已提交
1011
    nodeReportError(VIR_ERR_NO_SUPPORT, "%s",
1012 1013 1014
                    _("NUMA memory information not available on this platform"));
    return 0;
}
1015
#endif