nodeinfo.c 65.6 KB
Newer Older
1 2 3
/*
 * nodeinfo.c: Helper routines for OS specific node information
 *
4
 * Copyright (C) 2006-2008, 2010-2015 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16 17
 * 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
18
 * License along with this library.  If not, see
O
Osier Yang 已提交
19
 * <http://www.gnu.org/licenses/>.
20 21 22 23
 *
 * 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 "conf/domain_conf.h"
34 35
#include <fcntl.h>
#include <sys/ioctl.h>
M
Michal Privoznik 已提交
36
#include <unistd.h>
37 38 39 40

#if HAVE_LINUX_KVM_H
# include <linux/kvm.h>
#endif
41

42
#if defined(__FreeBSD__) || defined(__APPLE__)
43
# include <sys/time.h>
R
Roman Bogorodskiy 已提交
44 45
# include <sys/types.h>
# include <sys/sysctl.h>
46
# include <sys/resource.h>
R
Roman Bogorodskiy 已提交
47 48
#endif

49
#include "c-ctype.h"
50
#include "viralloc.h"
51
#include "nodeinfopriv.h"
52
#include "nodeinfo.h"
53
#include "physmem.h"
54
#include "virerror.h"
55
#include "count-one-bits.h"
E
Eric Blake 已提交
56
#include "intprops.h"
57
#include "virarch.h"
E
Eric Blake 已提交
58
#include "virfile.h"
59
#include "virtypedparam.h"
60
#include "virstring.h"
61
#include "virnuma.h"
62
#include "virlog.h"
63 64 65

#define VIR_FROM_THIS VIR_FROM_NONE

66 67
VIR_LOG_INIT("nodeinfo");

68

69
#if defined(__FreeBSD__) || defined(__APPLE__)
R
Roman Bogorodskiy 已提交
70
static int
71
appleFreebsdNodeGetCPUCount(void)
R
Roman Bogorodskiy 已提交
72 73 74 75 76 77 78 79 80 81 82 83
{
    int ncpu_mib[2] = { CTL_HW, HW_NCPU };
    unsigned long ncpu;
    size_t ncpu_len = sizeof(ncpu);

    if (sysctl(ncpu_mib, 2, &ncpu, &ncpu_len, NULL, 0) == -1) {
        virReportSystemError(errno, "%s", _("Cannot obtain CPU count"));
        return -1;
    }

    return ncpu;
}
84 85 86
#endif /* defined(__FreeBSD__) || defined(__APPLE__) */

#ifdef __FreeBSD__
87
# define BSD_CPU_STATS_ALL 4
88 89
# define BSD_MEMORY_STATS_ALL 4

90 91 92 93 94 95 96 97 98 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 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182
# define TICK_TO_NSEC (1000ull * 1000ull * 1000ull / (stathz ? stathz : hz))

static int
freebsdNodeGetCPUStats(int cpuNum,
                       virNodeCPUStatsPtr params,
                       int *nparams)
{
    const char *sysctl_name;
    long *cpu_times;
    struct clockinfo clkinfo;
    size_t i, j, cpu_times_size, clkinfo_size;
    int cpu_times_num, offset, hz, stathz, ret = -1;
    struct field_cpu_map {
        const char *field;
        int idx[CPUSTATES];
    } cpu_map[] = {
        {VIR_NODE_CPU_STATS_KERNEL, {CP_SYS}},
        {VIR_NODE_CPU_STATS_USER, {CP_USER, CP_NICE}},
        {VIR_NODE_CPU_STATS_IDLE, {CP_IDLE}},
        {VIR_NODE_CPU_STATS_INTR, {CP_INTR}},
        {NULL, {0}}
    };

    if ((*nparams) == 0) {
        *nparams = BSD_CPU_STATS_ALL;
        return 0;
    }

    if ((*nparams) != BSD_CPU_STATS_ALL) {
        virReportInvalidArg(*nparams,
                            _("nparams in %s must be equal to %d"),
                            __FUNCTION__, BSD_CPU_STATS_ALL);
        return -1;
    }

    clkinfo_size = sizeof(clkinfo);
    if (sysctlbyname("kern.clockrate", &clkinfo, &clkinfo_size, NULL, 0) < 0) {
        virReportSystemError(errno,
                             _("sysctl failed for '%s'"),
                             "kern.clockrate");
        return -1;
    }

    stathz = clkinfo.stathz;
    hz = clkinfo.hz;

    if (cpuNum == VIR_NODE_CPU_STATS_ALL_CPUS) {
        sysctl_name = "kern.cp_time";
        cpu_times_num = 1;
        offset = 0;
    } else {
        sysctl_name = "kern.cp_times";
        cpu_times_num = appleFreebsdNodeGetCPUCount();

        if (cpuNum >= cpu_times_num) {
            virReportInvalidArg(cpuNum,
                                _("Invalid cpuNum in %s"),
                                __FUNCTION__);
            return -1;
        }

        offset = cpu_times_num * CPUSTATES;
    }

    cpu_times_size = sizeof(long) * cpu_times_num * CPUSTATES;

    if (VIR_ALLOC_N(cpu_times, cpu_times_num * CPUSTATES) < 0)
        goto cleanup;

    if (sysctlbyname(sysctl_name, cpu_times, &cpu_times_size, NULL, 0) < 0) {
        virReportSystemError(errno,
                             _("sysctl failed for '%s'"),
                             sysctl_name);
        goto cleanup;
    }

    for (i = 0; cpu_map[i].field != NULL; i++) {
        virNodeCPUStatsPtr param = &params[i];

        if (virStrcpyStatic(param->field, cpu_map[i].field) == NULL) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Field '%s' too long for destination"),
                           cpu_map[i].field);
            goto cleanup;
        }

        param->value = 0;
        for (j = 0; j < ARRAY_CARDINALITY(cpu_map[i].idx); j++)
            param->value += cpu_times[offset + cpu_map[i].idx[j]] * TICK_TO_NSEC;
    }

    ret = 0;

183
 cleanup:
184 185 186 187 188
    VIR_FREE(cpu_times);

    return ret;
}

189 190
static int
freebsdNodeGetMemoryStats(virNodeMemoryStatsPtr params,
A
Andrea Bolognani 已提交
191
                          int *nparams)
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 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262
{
    size_t i, j = 0;
    unsigned long pagesize = getpagesize() >> 10;
    long bufpages;
    size_t bufpages_size = sizeof(bufpages);
    struct field_sysctl_map {
        const char *field;
        const char *sysctl_name;
    } sysctl_map[] = {
        {VIR_NODE_MEMORY_STATS_TOTAL, "vm.stats.vm.v_page_count"},
        {VIR_NODE_MEMORY_STATS_FREE, "vm.stats.vm.v_free_count"},
        {VIR_NODE_MEMORY_STATS_CACHED, "vm.stats.vm.v_cache_count"},
        {NULL, NULL}
    };

    if ((*nparams) == 0) {
        *nparams = BSD_MEMORY_STATS_ALL;
        return 0;
    }

    if ((*nparams) != BSD_MEMORY_STATS_ALL) {
        virReportInvalidArg(nparams,
                            _("nparams in %s must be %d"),
                            __FUNCTION__, BSD_MEMORY_STATS_ALL);
        return -1;
    }

    for (i = 0; sysctl_map[i].field != NULL; i++) {
        u_int value;
        size_t value_size = sizeof(value);
        virNodeMemoryStatsPtr param;

        if (sysctlbyname(sysctl_map[i].sysctl_name, &value,
                         &value_size, NULL, 0) < 0) {
            virReportSystemError(errno,
                                 _("sysctl failed for '%s'"),
                                 sysctl_map[i].sysctl_name);
            return -1;
        }

        param = &params[j++];
        if (virStrcpyStatic(param->field, sysctl_map[i].field) == NULL) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Field '%s' too long for destination"),
                           sysctl_map[i].field);
            return -1;
        }
        param->value = (unsigned long long)value * pagesize;
    }

    {
        virNodeMemoryStatsPtr param = &params[j++];

        if (sysctlbyname("vfs.bufspace", &bufpages, &bufpages_size, NULL, 0) < 0) {
            virReportSystemError(errno,
                                 _("sysctl failed for '%s'"),
                                 "vfs.bufspace");
            return -1;
        }
        if (virStrcpyStatic(param->field, VIR_NODE_MEMORY_STATS_BUFFERS) == NULL) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Field '%s' too long for destination"),
                           VIR_NODE_MEMORY_STATS_BUFFERS);
            return -1;
        }
        param->value = (unsigned long long)bufpages >> 10;
    }

    return 0;
}
#endif /* __FreeBSD__ */
R
Roman Bogorodskiy 已提交
263

264
#ifdef __linux__
265
# define SYSFS_SYSTEM_PATH "/sys/devices/system"
266
# define CPUINFO_PATH "/proc/cpuinfo"
267
# define PROCSTAT_PATH "/proc/stat"
268
# define MEMINFO_PATH "/proc/meminfo"
269
# define SYSFS_MEMORY_SHARED_PATH "/sys/kernel/mm/ksm"
270
# define SYSFS_THREAD_SIBLINGS_LIST_LENGTH_MAX 8192
271 272

# define LINUX_NB_CPU_STATS 4
273 274
# define LINUX_NB_MEMORY_STATS_ALL 4
# define LINUX_NB_MEMORY_STATS_CELL 2
275

276 277 278 279 280 281 282 283 284 285
static const char *sysfs_system_path = SYSFS_SYSTEM_PATH;

void linuxNodeInfoSetSysFSSystemPath(const char *path)
{
    if (path)
        sysfs_system_path = path;
    else
        sysfs_system_path = SYSFS_SYSTEM_PATH;
}

E
Eric Blake 已提交
286
/* Return the positive decimal contents of the given
287 288 289 290 291
 * DIR/cpu%u/FILE, or -1 on error.  If DEFAULT_VALUE is non-negative
 * and the file could not be found, return that instead of an error;
 * this is useful for machines that cannot hot-unplug cpu0, or where
 * hot-unplugging is disabled, or where the kernel is too old
 * to support NUMA cells, etc.  */
E
Eric Blake 已提交
292
static int
E
Eric Blake 已提交
293
virNodeGetCpuValue(const char *dir, unsigned int cpu, const char *file,
294
                   int default_value)
E
Eric Blake 已提交
295 296 297 298 299 300 301
{
    char *path;
    FILE *pathfp;
    int value = -1;
    char value_str[INT_BUFSIZE_BOUND(value)];
    char *tmp;

302
    if (virAsprintf(&path, "%s/cpu%u/%s", dir, cpu, file) < 0)
E
Eric Blake 已提交
303 304 305 306
        return -1;

    pathfp = fopen(path, "r");
    if (pathfp == NULL) {
307 308
        if (default_value >= 0 && errno == ENOENT)
            value = default_value;
E
Eric Blake 已提交
309 310 311 312 313 314 315 316 317 318
        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) {
319 320 321
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("could not convert '%s' to an integer"),
                       value_str);
E
Eric Blake 已提交
322 323 324
        goto cleanup;
    }

325
 cleanup:
326
    VIR_FORCE_FCLOSE(pathfp);
E
Eric Blake 已提交
327 328 329 330 331
    VIR_FREE(path);

    return value;
}

E
Eric Blake 已提交
332 333
static unsigned long
virNodeCountThreadSiblings(const char *dir, unsigned int cpu)
334 335
{
    unsigned long ret = 0;
C
Chris Lalancette 已提交
336
    char *path;
337
    char *str = NULL;
338
    size_t i;
339

340
    if (virAsprintf(&path, "%s/cpu%u/topology/thread_siblings",
341
                    dir, cpu) < 0)
342 343
        return 0;

344
    if (!virFileExists(path)) {
345 346
        /* If file doesn't exist, then pretend our only
         * sibling is ourself */
347
        ret = 1;
348
        goto cleanup;
349
    }
350

351
    if (virFileReadAll(path, SYSFS_THREAD_SIBLINGS_LIST_LENGTH_MAX, &str) < 0)
352 353
        goto cleanup;

J
Ján Tomko 已提交
354 355 356
    for (i = 0; str[i] != '\0'; i++) {
        if (c_isxdigit(str[i]))
            ret += count_one_bits(virHexToBin(str[i]));
357 358
    }

359
 cleanup:
360
    VIR_FREE(str);
361 362 363 364
    VIR_FREE(path);
    return ret;
}

E
Eric Blake 已提交
365
static int
366 367 368
virNodeParseSocket(const char *dir,
                   virArch arch,
                   unsigned int cpu)
369
{
370 371
    int ret = virNodeGetCpuValue(dir, cpu, "topology/physical_package_id", 0);

372 373
    if (ARCH_IS_ARM(arch) || ARCH_IS_PPC(arch) || ARCH_IS_S390(arch)) {
        /* arm, ppc and s390(x) has -1 */
374 375 376 377
        if (ret < 0)
            ret = 0;
    }

378
    return ret;
379 380
}

381 382
/* parses a node entry, returning number of processors in the node and
 * filling arguments */
383
static int
384
ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(3)
385 386 387
ATTRIBUTE_NONNULL(4) ATTRIBUTE_NONNULL(6)
ATTRIBUTE_NONNULL(7) ATTRIBUTE_NONNULL(8)
ATTRIBUTE_NONNULL(9)
388
virNodeParseNode(const char *node,
389
                 virArch arch,
390 391
                 virBitmapPtr present_cpus_map,
                 virBitmapPtr online_cpus_map,
392
                 int threads_per_subcore,
393 394 395 396
                 int *sockets,
                 int *cores,
                 int *threads,
                 int *offline)
397
{
398 399 400
    /* Biggest value we can expect to be used as either socket id
     * or core id. Bitmaps will need to be sized accordingly */
    const int ID_MAX = 4095;
401
    int ret = -1;
402 403 404
    int processors = 0;
    DIR *cpudir = NULL;
    struct dirent *cpudirent = NULL;
405
    virBitmapPtr node_cpus_map = NULL;
406 407
    virBitmapPtr sockets_map = NULL;
    virBitmapPtr *cores_maps = NULL;
408
    int npresent_cpus = virBitmapSize(present_cpus_map);
409 410 411
    int sock_max = 0;
    int sock;
    int core;
412
    size_t i;
413 414
    int siblings;
    unsigned int cpu;
N
Natanael Copa 已提交
415
    int direrr;
416

417 418 419 420 421 422
    *threads = 0;
    *cores = 0;
    *sockets = 0;

    if (!(cpudir = opendir(node))) {
        virReportSystemError(errno, _("cannot opendir %s"), node);
423 424
        goto cleanup;
    }
425

426 427 428 429
    /* Keep track of the CPUs that belong to the current node */
    if (!(node_cpus_map = virBitmapNew(npresent_cpus)))
        goto cleanup;

430
    /* enumerate sockets in the node */
431 432 433
    if (!(sockets_map = virBitmapNew(ID_MAX + 1)))
        goto cleanup;

N
Natanael Copa 已提交
434
    while ((direrr = virDirRead(cpudir, &cpudirent, node)) > 0) {
435 436 437
        if (sscanf(cpudirent->d_name, "cpu%u", &cpu) != 1)
            continue;

438
        if (!virBitmapIsBitSet(present_cpus_map, cpu))
439 440
            continue;

441 442 443 444
        /* Mark this CPU as part of the current node */
        if (virBitmapSetBit(node_cpus_map, cpu) < 0)
            goto cleanup;

445
        if (!virBitmapIsBitSet(online_cpus_map, cpu))
446 447
            continue;

448
        /* Parse socket */
449
        if ((sock = virNodeParseSocket(node, arch, cpu)) < 0)
450
            goto cleanup;
451 452 453 454 455 456 457 458 459
        if (sock > ID_MAX) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Socket %d can't be handled (max socket is %d)"),
                           sock, ID_MAX);
            goto cleanup;
        }

        if (virBitmapSetBit(sockets_map, sock) < 0)
            goto cleanup;
460 461 462 463 464

        if (sock > sock_max)
            sock_max = sock;
    }

N
Natanael Copa 已提交
465
    if (direrr < 0)
466
        goto cleanup;
467 468 469

    sock_max++;

470 471
    /* allocate cores maps for each socket */
    if (VIR_ALLOC_N(cores_maps, sock_max) < 0)
472
        goto cleanup;
473 474

    for (i = 0; i < sock_max; i++)
475 476
        if (!(cores_maps[i] = virBitmapNew(ID_MAX + 1)))
            goto cleanup;
477

478 479
    /* Iterate over all CPUs in the node, in ascending order */
    for (cpu = 0; cpu < npresent_cpus; cpu++) {
480

481 482
        /* Skip CPUs that are not part of the current node */
        if (!virBitmapIsBitSet(node_cpus_map, cpu))
483 484
            continue;

485
        if (!virBitmapIsBitSet(online_cpus_map, cpu)) {
486 487 488 489 490 491 492 493 494 495 496 497
            if (threads_per_subcore > 0 &&
                cpu % threads_per_subcore != 0 &&
                virBitmapIsBitSet(online_cpus_map,
                                  cpu - (cpu % threads_per_subcore))) {
                /* Secondary offline threads are counted as online when
                 * subcores are in use and the corresponding primary
                 * thread is online */
                processors++;
            } else {
                /* But they are counted as offline otherwise */
                (*offline)++;
            }
498
            continue;
499
        }
500 501 502 503

        processors++;

        /* Parse socket */
504
        if ((sock = virNodeParseSocket(node, arch, cpu)) < 0)
505
            goto cleanup;
506
        if (!virBitmapIsBitSet(sockets_map, sock)) {
507 508
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("CPU socket topology has changed"));
509 510 511 512
            goto cleanup;
        }

        /* Parse core */
513 514 515 516
        if (ARCH_IS_S390(arch)) {
            /* logical cpu is equivalent to a core on s390 */
            core = cpu;
        } else {
517 518 519
            if ((core = virNodeGetCpuValue(node, cpu,
                                           "topology/core_id", 0)) < 0)
                goto cleanup;
520
        }
521 522 523 524 525 526
        if (core > ID_MAX) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Core %d can't be handled (max core is %d)"),
                           core, ID_MAX);
            goto cleanup;
        }
527

528 529
        if (virBitmapSetBit(cores_maps[sock], core) < 0)
            goto cleanup;
530 531 532 533 534 535 536 537 538

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

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

    /* finalize the returned data */
539
    *sockets = virBitmapCountBits(sockets_map);
540 541

    for (i = 0; i < sock_max; i++) {
542
        if (!virBitmapIsBitSet(sockets_map, i))
543 544
            continue;

545
        core = virBitmapCountBits(cores_maps[i]);
546 547 548 549
        if (core > *cores)
            *cores = core;
    }

550 551 552 553 554 555
    if (threads_per_subcore > 0) {
        /* The thread count ignores offline threads, which means that only
         * only primary threads have been considered so far. If subcores
         * are in use, we need to also account for secondary threads */
        *threads *= threads_per_subcore;
    }
556
    ret = processors;
557

558
 cleanup:
559 560 561 562 563
    /* don't shadow a more serious error */
    if (cpudir && closedir(cpudir) < 0 && ret >= 0) {
        virReportSystemError(errno, _("problem closing %s"), node);
        ret = -1;
    }
564 565 566 567 568
    if (cores_maps)
        for (i = 0; i < sock_max; i++)
            virBitmapFree(cores_maps[i]);
    VIR_FREE(cores_maps);
    virBitmapFree(sockets_map);
569
    virBitmapFree(node_cpus_map);
570

571 572 573
    return ret;
}

574 575 576 577 578
/* Check whether the host subcore configuration is valid.
 *
 * A valid configuration is one where no secondary thread is online;
 * the primary thread in a subcore is always the first one */
static bool
579
nodeHasValidSubcoreConfiguration(int threads_per_subcore)
580 581 582 583 584 585 586 587 588
{
    virBitmapPtr online_cpus = NULL;
    int cpu = -1;
    bool ret = false;

    /* No point in checking if subcores are not in use */
    if (threads_per_subcore <= 0)
        goto cleanup;

589
    if (!(online_cpus = nodeGetOnlineCPUBitmap()))
590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607
        goto cleanup;

    while ((cpu = virBitmapNextSetBit(online_cpus, cpu)) >= 0) {

        /* A single online secondary thread is enough to
         * make the configuration invalid */
        if (cpu % threads_per_subcore != 0)
            goto cleanup;
    }

    ret = true;

 cleanup:
    virBitmapFree(online_cpus);

    return ret;
}

A
Andrea Bolognani 已提交
608
int
609
linuxNodeInfoCPUPopulate(FILE *cpuinfo,
A
Andrea Bolognani 已提交
610 611
                         virArch arch,
                         virNodeInfoPtr nodeinfo)
612
{
613 614
    virBitmapPtr present_cpus_map = NULL;
    virBitmapPtr online_cpus_map = NULL;
615
    char line[1024];
616 617
    DIR *nodedir = NULL;
    struct dirent *nodedirent = NULL;
618
    int cpus, cores, socks, threads, offline = 0;
619
    int threads_per_subcore = 0;
620
    unsigned int node;
621
    int ret = -1;
622
    char *sysfs_nodedir = NULL;
623
    char *sysfs_cpudir = NULL;
N
Natanael Copa 已提交
624
    int direrr;
625

626
    /* Start with parsing CPU clock speed from /proc/cpuinfo */
627
    while (fgets(line, sizeof(line), cpuinfo) != NULL) {
628 629 630 631 632 633 634 635 636 637 638 639 640 641 642
        if (ARCH_IS_X86(arch)) {
            char *buf = line;
            if (STRPREFIX(buf, "cpu MHz")) {
                char *p;
                unsigned int ui;

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

                if (*buf != ':' || !buf[1]) {
                    virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                   _("parsing cpu MHz from cpuinfo"));
                    goto cleanup;
                }
643

644 645 646 647
                if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0 &&
                    /* Accept trailing fractional part.  */
                    (*p == '\0' || *p == '.' || c_isspace(*p)))
                    nodeinfo->mhz = ui;
648
            }
649

650 651 652 653 654
        } else if (ARCH_IS_PPC(arch)) {
            char *buf = line;
            if (STRPREFIX(buf, "clock")) {
                char *p;
                unsigned int ui;
655

656 657 658
                buf += 5;
                while (*buf && c_isspace(*buf))
                    buf++;
659

660 661 662 663 664
                if (*buf != ':' || !buf[1]) {
                    virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                   _("parsing cpu MHz from cpuinfo"));
                    goto cleanup;
                }
665

666 667 668 669 670 671 672 673 674
                if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0 &&
                    /* Accept trailing fractional part.  */
                    (*p == '\0' || *p == '.' || c_isspace(*p)))
                    nodeinfo->mhz = ui;
                /* 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. */
675
            }
676 677 678 679 680
        } else if (ARCH_IS_ARM(arch)) {
            char *buf = line;
            if (STRPREFIX(buf, "BogoMIPS")) {
                char *p;
                unsigned int ui;
681

682 683 684
                buf += 8;
                while (*buf && c_isspace(*buf))
                    buf++;
685

686 687 688 689 690
                if (*buf != ':' || !buf[1]) {
                    virReportError(VIR_ERR_INTERNAL_ERROR,
                                   "%s", _("parsing cpu MHz from cpuinfo"));
                    goto cleanup;
                }
691

692 693 694 695
                if (virStrToLong_ui(buf+1, &p, 10, &ui) == 0
                    /* Accept trailing fractional part.  */
                    && (*p == '\0' || *p == '.' || c_isspace(*p)))
                    nodeinfo->mhz = ui;
696
            }
697 698 699 700 701 702 703
        } else if (ARCH_IS_S390(arch)) {
            /* s390x has no realistic value for CPU speed,
             * assign a value of zero to signify this */
            nodeinfo->mhz = 0;
        } else {
            VIR_WARN("Parser for /proc/cpuinfo needs to be adapted for your architecture");
            break;
704
        }
705 706
    }

707 708
    /* Get information about what CPUs are present in the host and what
     * CPUs are online, so that we don't have to so for each node */
709
    present_cpus_map = nodeGetPresentCPUBitmap();
710 711
    if (!present_cpus_map)
        goto cleanup;
712
    online_cpus_map = nodeGetOnlineCPUBitmap();
713 714 715
    if (!online_cpus_map)
        goto cleanup;

716 717
    /* OK, we've parsed clock speed out of /proc/cpuinfo. Get the
     * core, node, socket, thread and topology information from /sys
718
     */
719
    if (virAsprintf(&sysfs_nodedir, "%s/node", sysfs_system_path) < 0)
720
        goto cleanup;
721

722 723 724 725
    if (!(nodedir = opendir(sysfs_nodedir))) {
        /* the host isn't probably running a NUMA architecture */
        goto fallback;
    }
726

727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753
    /* PPC-KVM needs the secondary threads of a core to be offline on the
     * host. The kvm scheduler brings the secondary threads online in the
     * guest context. Moreover, P8 processor has split-core capability
     * where, there can be 1,2 or 4 subcores per core. The primaries of the
     * subcores alone will be online on the host for a subcore in the
     * host. Even though the actual threads per core for P8 processor is 8,
     * depending on the subcores_per_core = 1, 2 or 4, the threads per
     * subcore will vary accordingly to 8, 4 and 2 repectively.
     * So, On host threads_per_core what is arrived at from sysfs in the
     * current logic is actually the subcores_per_core. Threads per subcore
     * can only be obtained from the kvm device. For example, on P8 wih 1
     * core having 8 threads, sub_cores_percore=4, the threads 0,2,4 & 6
     * will be online. The sysfs reflects this and in the current logic
     * variable 'threads' will be 4 which is nothing but subcores_per_core.
     * If the user tampers the cpu online/offline states using chcpu or other
     * means, then it is an unsupported configuration for kvm.
     * The code below tries to keep in mind
     *  - when the libvirtd is run inside a KVM guest or Phyp based guest.
     *  - Or on the kvm host where user manually tampers the cpu states to
     *    offline/online randomly.
     * On hosts other than POWER this will be 0, in which case a simpler
     * thread-counting logic will be used  */
    if ((threads_per_subcore = nodeGetThreadsPerSubcore(arch)) < 0)
        goto cleanup;

    /* If the subcore configuration is not valid, just pretend subcores
     * are not in use and count threads one by one */
754
    if (!nodeHasValidSubcoreConfiguration(threads_per_subcore))
755 756
        threads_per_subcore = 0;

N
Natanael Copa 已提交
757
    while ((direrr = virDirRead(nodedir, &nodedirent, sysfs_nodedir)) > 0) {
758
        if (sscanf(nodedirent->d_name, "node%u", &node) != 1)
759 760
            continue;

761 762 763
        nodeinfo->nodes++;

        if (virAsprintf(&sysfs_cpudir, "%s/node/%s",
764
                        sysfs_system_path, nodedirent->d_name) < 0)
765
            goto cleanup;
E
Eric Blake 已提交
766

767 768 769
        if ((cpus = virNodeParseNode(sysfs_cpudir, arch,
                                     present_cpus_map,
                                     online_cpus_map,
770
                                     threads_per_subcore,
771
                                     &socks, &cores,
772
                                     &threads, &offline)) < 0)
773
            goto cleanup;
774

775
        VIR_FREE(sysfs_cpudir);
776

777 778 779 780 781 782 783 784 785 786
        nodeinfo->cpus += cpus;

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

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

        if (threads > nodeinfo->threads)
            nodeinfo->threads = threads;
787
    }
788

N
Natanael Copa 已提交
789
    if (direrr < 0)
790
        goto cleanup;
791 792 793 794

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

795
 fallback:
796 797
    VIR_FREE(sysfs_cpudir);

798
    if (virAsprintf(&sysfs_cpudir, "%s/cpu", sysfs_system_path) < 0)
799
        goto cleanup;
800

801 802 803
    if ((cpus = virNodeParseNode(sysfs_cpudir, arch,
                                 present_cpus_map,
                                 online_cpus_map,
804
                                 threads_per_subcore,
805
                                 &socks, &cores,
806
                                 &threads, &offline)) < 0)
807
        goto cleanup;
808

809 810 811 812 813 814
    nodeinfo->nodes = 1;
    nodeinfo->cpus = cpus;
    nodeinfo->sockets = socks;
    nodeinfo->cores = cores;
    nodeinfo->threads = threads;

815
 done:
816
    /* There should always be at least one cpu, socket, node, and thread. */
817
    if (nodeinfo->cpus == 0) {
818
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("no CPUs found"));
819
        goto cleanup;
820
    }
821

C
Chris Lalancette 已提交
822
    if (nodeinfo->sockets == 0) {
823
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("no sockets found"));
824
        goto cleanup;
C
Chris Lalancette 已提交
825
    }
826

C
Chris Lalancette 已提交
827
    if (nodeinfo->threads == 0) {
828
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("no threads found"));
829
        goto cleanup;
C
Chris Lalancette 已提交
830 831
    }

832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848
    /* Now check if the topology makes sense. There are machines that don't
     * expose their real number of nodes or for example the AMD Bulldozer
     * architecture that exposes their Clustered integer core modules as both
     * threads and cores. This approach throws off our detection. Unfortunately
     * the nodeinfo structure isn't designed to carry the full topology so
     * we're going to lie about the detected topology to notify the user
     * to check the host capabilities for the actual topology. */
    if ((nodeinfo->nodes *
         nodeinfo->sockets *
         nodeinfo->cores *
         nodeinfo->threads) != (nodeinfo->cpus + offline)) {
        nodeinfo->nodes = 1;
        nodeinfo->sockets = 1;
        nodeinfo->cores = nodeinfo->cpus + offline;
        nodeinfo->threads = 1;
    }

849 850
    ret = 0;

851
 cleanup:
852 853 854 855 856 857
    /* don't shadow a more serious error */
    if (nodedir && closedir(nodedir) < 0 && ret >= 0) {
        virReportSystemError(errno, _("problem closing %s"), sysfs_nodedir);
        ret = -1;
    }

858 859
    virBitmapFree(present_cpus_map);
    virBitmapFree(online_cpus_map);
860
    VIR_FREE(sysfs_nodedir);
861 862
    VIR_FREE(sysfs_cpudir);
    return ret;
863 864
}

J
Ján Tomko 已提交
865 866 867 868 869 870 871
static int
virNodeCPUStatsAssign(virNodeCPUStatsPtr param,
                      const char *name,
                      unsigned long long value)
{
    if (virStrcpyStatic(param->field, name) == NULL) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
872 873
                       "%s", _("kernel cpu time field is too long"
                               " for the destination"));
J
Ján Tomko 已提交
874 875 876 877 878 879
        return -1;
    }
    param->value = value;
    return 0;
}

880 881
# define TICK_TO_NSEC (1000ull * 1000ull * 1000ull / sysconf(_SC_CLK_TCK))

J
Ján Tomko 已提交
882
int
883 884 885 886
linuxNodeGetCPUStats(FILE *procstat,
                     int cpuNum,
                     virNodeCPUStatsPtr params,
                     int *nparams)
887 888 889 890 891
{
    int ret = -1;
    char line[1024];
    unsigned long long usr, ni, sys, idle, iowait;
    unsigned long long irq, softirq, steal, guest, guest_nice;
892
    char cpu_header[4 + INT_BUFSIZE_BOUND(cpuNum)];
893 894 895 896 897 898 899 900 901

    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) {
902 903 904
        virReportInvalidArg(*nparams,
                            _("nparams in %s must be equal to %d"),
                            __FUNCTION__, LINUX_NB_CPU_STATS);
905 906 907
        goto cleanup;
    }

908
    if (cpuNum == VIR_NODE_CPU_STATS_ALL_CPUS) {
909
        strcpy(cpu_header, "cpu ");
910
    } else {
911
        snprintf(cpu_header, sizeof(cpu_header), "cpu%d ", cpuNum);
912 913 914 915 916 917 918 919 920 921 922 923 924 925
    }

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

        if (STRPREFIX(buf, cpu_header)) { /* aka logical CPU time */
            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;
            }

J
Ján Tomko 已提交
926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941
            if (virNodeCPUStatsAssign(&params[0], VIR_NODE_CPU_STATS_KERNEL,
                                      (sys + irq + softirq) * TICK_TO_NSEC) < 0)
                goto cleanup;

            if (virNodeCPUStatsAssign(&params[1], VIR_NODE_CPU_STATS_USER,
                                      (usr + ni) * TICK_TO_NSEC) < 0)
                goto cleanup;

            if (virNodeCPUStatsAssign(&params[2], VIR_NODE_CPU_STATS_IDLE,
                                      idle * TICK_TO_NSEC) < 0)
                goto cleanup;

            if (virNodeCPUStatsAssign(&params[3], VIR_NODE_CPU_STATS_IOWAIT,
                                      iowait * TICK_TO_NSEC) < 0)
                goto cleanup;

942 943 944 945 946
            ret = 0;
            goto cleanup;
        }
    }

947 948 949
    virReportInvalidArg(cpuNum,
                        _("Invalid cpuNum in %s"),
                        __FUNCTION__);
950

951
 cleanup:
952 953 954
    return ret;
}

955 956 957 958 959
static int
linuxNodeGetMemoryStats(FILE *meminfo,
                        int cellNum,
                        virNodeMemoryStatsPtr params,
                        int *nparams)
960 961
{
    int ret = -1;
962
    size_t i = 0, j = 0, k = 0;
963 964 965
    int found = 0;
    int nr_param;
    char line[1024];
966
    char meminfo_hdr[VIR_NODE_MEMORY_STATS_FIELD_LENGTH];
967 968 969 970 971
    unsigned long val;
    struct field_conv {
        const char *meminfo_hdr;  // meminfo header
        const char *field;        // MemoryStats field name
    } field_conv[] = {
972 973 974 975
        {"MemTotal:", VIR_NODE_MEMORY_STATS_TOTAL},
        {"MemFree:",  VIR_NODE_MEMORY_STATS_FREE},
        {"Buffers:",  VIR_NODE_MEMORY_STATS_BUFFERS},
        {"Cached:",   VIR_NODE_MEMORY_STATS_CACHED},
976 977 978
        {NULL,        NULL}
    };

979
    if (cellNum == VIR_NODE_MEMORY_STATS_ALL_CELLS) {
980 981 982 983 984 985 986 987 988 989 990 991 992
        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) {
993 994 995
        virReportInvalidArg(nparams,
                            _("nparams in %s must be %d"),
                            __FUNCTION__, nr_param);
996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016
        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) {
1017 1018
                    virReportError(VIR_ERR_INTERNAL_ERROR,
                                   "%s", _("no prefix found"));
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032
                    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)) {
1033
                virNodeMemoryStatsPtr param = &params[k++];
1034 1035

                if (virStrcpyStatic(param->field, convp->field) == NULL) {
1036 1037
                    virReportError(VIR_ERR_INTERNAL_ERROR,
                                   "%s", _("Field kernel memory too long for destination"));
1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049
                    goto cleanup;
                }
                param->value = val;
                found++;
                break;
            }
        }
        if (found >= nr_param)
            break;
    }

    if (found == 0) {
1050 1051
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("no available memory line found"));
1052 1053 1054 1055 1056
        goto cleanup;
    }

    ret = 0;

1057
 cleanup:
1058 1059
    return ret;
}
1060

1061
static char *
1062
linuxGetCPUGlobalPath(const char *file)
1063 1064 1065
{
    char *path = NULL;

1066
    if (virAsprintf(&path, "%s/cpu/%s", sysfs_system_path, file) < 0)
1067
        return NULL;
1068

1069 1070
    return path;
}
1071

1072
static char *
1073
linuxGetCPUPresentPath(void)
1074
{
1075
    return linuxGetCPUGlobalPath("present");
1076 1077
}

1078
static char *
1079
linuxGetCPUOnlinePath(void)
1080
{
1081
    return linuxGetCPUGlobalPath("online");
1082 1083
}

1084 1085
/* Determine the number of CPUs (maximum CPU id + 1) from a file containing
 * a list of CPU ids, like the Linux sysfs cpu/present file */
1086
static int
1087
linuxParseCPUCount(const char *path)
1088 1089 1090 1091 1092
{
    char *str = NULL;
    char *tmp;
    int ret = -1;

E
Eric Blake 已提交
1093
    if (virFileReadAll(path, 5 * VIR_DOMAIN_CPUMASK_LEN, &str) < 0)
1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107
        goto cleanup;

    tmp = str;
    do {
        if (virStrToLong_i(tmp, &tmp, 10, &ret) < 0 ||
            !strchr(",-\n", *tmp)) {
            virReportError(VIR_ERR_NO_SUPPORT,
                           _("failed to parse %s"), path);
            ret = -1;
            goto cleanup;
        }
    } while (*tmp++ != '\n');
    ret++;

1108
 cleanup:
1109 1110 1111 1112
    VIR_FREE(str);
    return ret;
}

1113
/*
1114 1115 1116
 * Linux maintains cpu bit map under cpu/online. 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.
1117
 */
H
Hu Tao 已提交
1118
static virBitmapPtr
1119
linuxParseCPUmap(int max_cpuid, const char *path)
1120
{
H
Hu Tao 已提交
1121
    virBitmapPtr map = NULL;
1122 1123
    char *str = NULL;

1124
    if (virFileReadAll(path, 5 * VIR_DOMAIN_CPUMASK_LEN, &str) < 0)
1125 1126
        goto error;

1127
    if (virBitmapParse(str, 0, &map, max_cpuid) < 0)
1128 1129
        goto error;

H
Hu Tao 已提交
1130
    VIR_FREE(str);
1131 1132
    return map;

1133
 error:
1134
    VIR_FREE(str);
H
Hu Tao 已提交
1135
    virBitmapFree(map);
1136 1137
    return NULL;
}
E
Eric Blake 已提交
1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156


static virBitmapPtr
virNodeGetSiblingsList(const char *dir, int cpu_id)
{
    char *path = NULL;
    char *buf = NULL;
    virBitmapPtr ret = NULL;

    if (virAsprintf(&path, "%s/cpu%u/topology/thread_siblings_list",
                    dir, cpu_id) < 0)
        goto cleanup;

    if (virFileReadAll(path, SYSFS_THREAD_SIBLINGS_LIST_LENGTH_MAX, &buf) < 0)
        goto cleanup;

    if (virBitmapParse(buf, 0, &ret, virNumaGetMaxCPUs()) < 0)
        goto cleanup;

1157
 cleanup:
E
Eric Blake 已提交
1158 1159 1160 1161
    VIR_FREE(buf);
    VIR_FREE(path);
    return ret;
}
1162 1163
#endif

A
Andrea Bolognani 已提交
1164
int
1165
nodeGetInfo(virNodeInfoPtr nodeinfo)
1166 1167
{
    virArch hostarch = virArchFromHost();
1168
    unsigned long long memorybytes;
1169

1170 1171
    memset(nodeinfo, 0, sizeof(*nodeinfo));

1172
    if (virStrcpyStatic(nodeinfo->model, virArchToString(hostarch)) == NULL)
C
Chris Lalancette 已提交
1173
        return -1;
1174

1175 1176 1177 1178
    if (nodeGetMemory(&memorybytes, NULL) < 0)
        return -1;
    nodeinfo->memory = memorybytes / 1024;

1179
#ifdef __linux__
1180
    {
1181
    int ret = -1;
1182
    FILE *cpuinfo = fopen(CPUINFO_PATH, "r");
1183

1184
    if (!cpuinfo) {
1185
        virReportSystemError(errno,
1186
                             _("cannot open %s"), CPUINFO_PATH);
1187 1188
        return -1;
    }
1189

1190
    ret = linuxNodeInfoCPUPopulate(cpuinfo, hostarch, nodeinfo);
1191 1192
    if (ret < 0)
        goto cleanup;
1193

1194
 cleanup:
1195
    VIR_FORCE_FCLOSE(cpuinfo);
1196
    return ret;
1197
    }
1198
#elif defined(__FreeBSD__) || defined(__APPLE__)
R
Roman Bogorodskiy 已提交
1199 1200 1201 1202 1203
    {
    nodeinfo->nodes = 1;
    nodeinfo->sockets = 1;
    nodeinfo->threads = 1;

1204
    nodeinfo->cpus = appleFreebsdNodeGetCPUCount();
R
Roman Bogorodskiy 已提交
1205 1206 1207 1208 1209 1210 1211 1212
    if (nodeinfo->cpus == -1)
        return -1;

    nodeinfo->cores = nodeinfo->cpus;

    unsigned long cpu_freq;
    size_t cpu_freq_len = sizeof(cpu_freq);

1213
# ifdef __FreeBSD__
R
Roman Bogorodskiy 已提交
1214 1215 1216 1217 1218 1219
    if (sysctlbyname("dev.cpu.0.freq", &cpu_freq, &cpu_freq_len, NULL, 0) < 0) {
        virReportSystemError(errno, "%s", _("cannot obtain CPU freq"));
        return -1;
    }

    nodeinfo->mhz = cpu_freq;
1220 1221 1222 1223 1224 1225 1226 1227
# else
    if (sysctlbyname("hw.cpufrequency", &cpu_freq, &cpu_freq_len, NULL, 0) < 0) {
        virReportSystemError(errno, "%s", _("cannot obtain CPU freq"));
        return -1;
    }

    nodeinfo->mhz = cpu_freq / 1000000;
# endif
R
Roman Bogorodskiy 已提交
1228 1229 1230

    return 0;
    }
1231 1232
#else
    /* XXX Solaris will need an impl later if they port QEMU driver */
1233 1234
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
                   _("node info not implemented on this platform"));
1235 1236 1237
    return -1;
#endif
}
1238

A
Andrea Bolognani 已提交
1239 1240 1241 1242 1243
int
nodeGetCPUStats(int cpuNum ATTRIBUTE_UNUSED,
                virNodeCPUStatsPtr params ATTRIBUTE_UNUSED,
                int *nparams ATTRIBUTE_UNUSED,
                unsigned int flags)
1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
{
    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;
    }
1261 1262
#elif defined(__FreeBSD__)
    return freebsdNodeGetCPUStats(cpuNum, params, nparams);
1263
#else
1264 1265
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
                   _("node CPU stats not implemented on this platform"));
1266 1267 1268 1269
    return -1;
#endif
}

A
Andrea Bolognani 已提交
1270
int
1271
nodeGetMemoryStats(int cellNum ATTRIBUTE_UNUSED,
A
Andrea Bolognani 已提交
1272 1273 1274
                   virNodeMemoryStatsPtr params ATTRIBUTE_UNUSED,
                   int *nparams ATTRIBUTE_UNUSED,
                   unsigned int flags)
1275 1276 1277 1278 1279 1280 1281 1282
{
    virCheckFlags(0, -1);

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

1285
        if (cellNum == VIR_NODE_MEMORY_STATS_ALL_CELLS) {
1286
            if (VIR_STRDUP(meminfo_path, MEMINFO_PATH) < 0)
1287 1288
                return -1;
        } else {
1289
            if ((max_node = virNumaGetMaxNode()) < 0)
1290 1291
                return -1;

1292
            if (cellNum > max_node) {
1293 1294
                virReportInvalidArg(cellNum,
                                    _("cellNum in %s must be less than or equal to %d"),
1295
                                    __FUNCTION__, max_node);
1296 1297 1298
                return -1;
            }

1299 1300 1301
            if (virAsprintf(&meminfo_path,
                            SYSFS_SYSTEM_PATH "/node/node%d/meminfo",
                            cellNum) < 0)
1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
                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;
    }
1318 1319
#elif defined(__FreeBSD__)
    return freebsdNodeGetMemoryStats(params, nparams);
1320
#else
1321 1322
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
                   _("node memory stats not implemented on this platform"));
1323 1324 1325 1326
    return -1;
#endif
}

1327
int
1328
nodeGetCPUCount(void)
1329
{
R
Roman Bogorodskiy 已提交
1330
#if defined(__linux__)
1331 1332 1333 1334 1335
    /* To support older kernels that lack cpu/present, such as 2.6.18
     * in RHEL5, we fall back to count cpu/cpuNN entries; this assumes
     * that such kernels also lack hotplug, and therefore cpu/cpuNN
     * will be consecutive.
     */
1336
    char *present_path = NULL;
1337
    char *cpupath = NULL;
1338 1339
    int ncpu = -1;

1340
    if (!(present_path = linuxGetCPUPresentPath()))
1341
        return -1;
1342

1343
    if (virFileExists(present_path)) {
1344
        ncpu = linuxParseCPUCount(present_path);
1345 1346 1347
        goto cleanup;
    }

1348
    if (virAsprintf(&cpupath, "%s/cpu/cpu0", sysfs_system_path) < 0)
1349 1350
        goto cleanup;
    if (virFileExists(cpupath)) {
1351
        ncpu = 0;
1352
        do {
1353
            ncpu++;
1354 1355
            VIR_FREE(cpupath);
            if (virAsprintf(&cpupath, "%s/cpu/cpu%d",
1356
                            sysfs_system_path, ncpu) < 0) {
1357 1358 1359
                ncpu = -1;
                goto cleanup;
            }
1360 1361 1362 1363 1364 1365 1366
        } while (virFileExists(cpupath));
    } else {
        /* no cpu/cpu0: we give up */
        virReportError(VIR_ERR_NO_SUPPORT, "%s",
                       _("host cpu counting not supported on this node"));
    }

1367 1368
 cleanup:
    VIR_FREE(present_path);
1369
    VIR_FREE(cpupath);
1370
    return ncpu;
1371 1372
#elif defined(__FreeBSD__) || defined(__APPLE__)
    return appleFreebsdNodeGetCPUCount();
1373 1374 1375 1376 1377 1378 1379
#else
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
                   _("host cpu counting not implemented on this platform"));
    return -1;
#endif
}

1380
virBitmapPtr
1381
nodeGetPresentCPUBitmap(void)
1382
{
1383
#ifdef __linux__
1384
    virBitmapPtr present_cpus = NULL;
1385
    char *present_path = NULL;
1386
    int npresent_cpus;
1387

1388
    if ((npresent_cpus = nodeGetCPUCount()) < 0)
1389
        goto cleanup;
1390

1391
    if (!(present_path = linuxGetCPUPresentPath()))
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408
        goto cleanup;

    /* If the cpu/present file is available, parse it and exit */
    if (virFileExists(present_path)) {
        present_cpus = linuxParseCPUmap(npresent_cpus, present_path);
        goto cleanup;
    }

    /* If the file is not available, we can assume that the kernel is
     * too old to support non-consecutive CPU ids and just mark all
     * possible CPUs as present */
    if (!(present_cpus = virBitmapNew(npresent_cpus)))
        goto cleanup;

    virBitmapSetAll(present_cpus);

 cleanup:
1409
    VIR_FREE(present_path);
1410 1411

    return present_cpus;
1412 1413
#endif
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
1414
                   _("node present CPU map not implemented on this platform"));
1415 1416 1417
    return NULL;
}

H
Hu Tao 已提交
1418
virBitmapPtr
1419
nodeGetOnlineCPUBitmap(void)
1420 1421
{
#ifdef __linux__
1422
    char *online_path = NULL;
1423
    char *cpudir = NULL;
H
Hu Tao 已提交
1424
    virBitmapPtr cpumap;
1425 1426
    int present;

1427
    present = nodeGetCPUCount();
1428
    if (present < 0)
1429
        return NULL;
E
Eric Blake 已提交
1430

1431
    if (!(online_path = linuxGetCPUOnlinePath()))
1432 1433 1434
        return NULL;
    if (virFileExists(online_path)) {
        cpumap = linuxParseCPUmap(present, online_path);
E
Eric Blake 已提交
1435
    } else {
1436
        size_t i;
E
Eric Blake 已提交
1437 1438

        cpumap = virBitmapNew(present);
1439
        if (!cpumap)
1440
            goto cleanup;
1441

1442
        if (virAsprintf(&cpudir, "%s/cpu", sysfs_system_path) < 0)
1443 1444
            goto cleanup;

E
Eric Blake 已提交
1445
        for (i = 0; i < present; i++) {
1446
            int online = virNodeGetCpuValue(cpudir, i, "online", 1);
E
Eric Blake 已提交
1447 1448
            if (online < 0) {
                virBitmapFree(cpumap);
1449 1450
                cpumap = NULL;
                goto cleanup;
E
Eric Blake 已提交
1451 1452 1453 1454 1455
            }
            if (online)
                ignore_value(virBitmapSetBit(cpumap, i));
        }
    }
1456

1457 1458
 cleanup:
    VIR_FREE(online_path);
1459
    VIR_FREE(cpudir);
1460 1461
    return cpumap;
#else
1462
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
1463
                   _("node online CPU map not implemented on this platform"));
1464 1465 1466 1467
    return NULL;
#endif
}

1468
#ifdef __linux__
1469
static int
1470
nodeSetMemoryParameterValue(virTypedParameterPtr param)
1471 1472 1473 1474 1475 1476
{
    char *path = NULL;
    char *strval = NULL;
    int ret = -1;
    int rc = -1;

1477
    char *field = strchr(param->field, '_');
1478
    sa_assert(field);
1479
    field++;
1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491
    if (virAsprintf(&path, "%s/%s",
                    SYSFS_MEMORY_SHARED_PATH, field) < 0) {
        ret = -2;
        goto cleanup;
    }

    if (virAsprintf(&strval, "%u", param->value.ui) == -1) {
        ret = -2;
        goto cleanup;
    }

    if ((rc = virFileWriteStr(path, strval, 0)) < 0) {
1492
        virReportSystemError(-rc, _("failed to set %s"), param->field);
1493 1494 1495 1496
        goto cleanup;
    }

    ret = 0;
1497
 cleanup:
1498 1499 1500 1501
    VIR_FREE(path);
    VIR_FREE(strval);
    return ret;
}
1502 1503 1504 1505 1506 1507

static bool
nodeMemoryParametersIsAllSupported(virTypedParameterPtr params,
                                   int nparams)
{
    char *path = NULL;
1508
    size_t i;
1509 1510 1511 1512 1513

    for (i = 0; i < nparams; i++) {
        virTypedParameterPtr param = &params[i];

        char *field = strchr(param->field, '_');
1514
        sa_assert(field);
1515 1516
        field++;
        if (virAsprintf(&path, "%s/%s",
1517
                        SYSFS_MEMORY_SHARED_PATH, field) < 0)
1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532
            return false;

        if (!virFileExists(path)) {
            virReportError(VIR_ERR_OPERATION_INVALID,
                           _("Parameter '%s' is not supported by "
                             "this kernel"), param->field);
            VIR_FREE(path);
            return false;
        }

        VIR_FREE(path);
    }

    return true;
}
1533
#endif
1534 1535

int
1536
nodeSetMemoryParameters(virTypedParameterPtr params ATTRIBUTE_UNUSED,
1537
                        int nparams ATTRIBUTE_UNUSED,
1538 1539 1540 1541 1542
                        unsigned int flags)
{
    virCheckFlags(0, -1);

#ifdef __linux__
1543
    size_t i;
1544
    int rc;
1545

1546 1547 1548 1549 1550 1551 1552 1553
    if (virTypedParamsValidate(params, nparams,
                               VIR_NODE_MEMORY_SHARED_PAGES_TO_SCAN,
                               VIR_TYPED_PARAM_UINT,
                               VIR_NODE_MEMORY_SHARED_SLEEP_MILLISECS,
                               VIR_TYPED_PARAM_UINT,
                               VIR_NODE_MEMORY_SHARED_MERGE_ACROSS_NODES,
                               VIR_TYPED_PARAM_UINT,
                               NULL) < 0)
1554 1555
        return -1;

1556 1557
    if (!nodeMemoryParametersIsAllSupported(params, nparams))
        return -1;
1558

1559 1560
    for (i = 0; i < nparams; i++) {
        rc = nodeSetMemoryParameterValue(&params[i]);
1561

1562
        if (rc < 0)
1563
            return -1;
1564 1565
    }

1566
    return 0;
1567 1568 1569 1570 1571 1572 1573 1574
#else
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
                   _("node set memory parameters not implemented"
                     " on this platform"));
    return -1;
#endif
}

1575
#ifdef __linux__
1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586
static int
nodeGetMemoryParameterValue(const char *field,
                            void *value)
{
    char *path = NULL;
    char *buf = NULL;
    char *tmp = NULL;
    int ret = -1;
    int rc = -1;

    if (virAsprintf(&path, "%s/%s",
1587
                    SYSFS_MEMORY_SHARED_PATH, field) < 0)
1588 1589
        goto cleanup;

1590 1591 1592 1593 1594
    if (!virFileExists(path)) {
        ret = -2;
        goto cleanup;
    }

1595 1596 1597 1598 1599 1600
    if (virFileReadAll(path, 1024, &buf) < 0)
        goto cleanup;

    if ((tmp = strchr(buf, '\n')))
        *tmp = '\0';

1601 1602 1603
    if (STREQ(field, "pages_to_scan")   ||
        STREQ(field, "sleep_millisecs") ||
        STREQ(field, "merge_across_nodes"))
1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618
        rc = virStrToLong_ui(buf, NULL, 10, (unsigned int *)value);
    else if (STREQ(field, "pages_shared")    ||
             STREQ(field, "pages_sharing")   ||
             STREQ(field, "pages_unshared")  ||
             STREQ(field, "pages_volatile")  ||
             STREQ(field, "full_scans"))
        rc = virStrToLong_ull(buf, NULL, 10, (unsigned long long *)value);

    if (rc < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse %s"), field);
        goto cleanup;
    }

    ret = 0;
1619
 cleanup:
1620 1621 1622 1623
    VIR_FREE(path);
    VIR_FREE(buf);
    return ret;
}
1624
#endif
1625

1626
#define NODE_MEMORY_PARAMETERS_NUM 8
1627
int
1628
nodeGetMemoryParameters(virTypedParameterPtr params ATTRIBUTE_UNUSED,
1629
                        int *nparams ATTRIBUTE_UNUSED,
1630 1631 1632 1633 1634 1635 1636
                        unsigned int flags)
{
    virCheckFlags(VIR_TYPED_PARAM_STRING_OKAY, -1);

#ifdef __linux__
    unsigned int pages_to_scan;
    unsigned int sleep_millisecs;
1637
    unsigned int merge_across_nodes;
1638 1639 1640 1641 1642
    unsigned long long pages_shared;
    unsigned long long pages_sharing;
    unsigned long long pages_unshared;
    unsigned long long pages_volatile;
    unsigned long long full_scans = 0;
1643
    size_t i;
1644
    int ret;
1645 1646 1647 1648 1649 1650 1651 1652 1653

    if ((*nparams) == 0) {
        *nparams = NODE_MEMORY_PARAMETERS_NUM;
        return 0;
    }

    for (i = 0; i < *nparams && i < NODE_MEMORY_PARAMETERS_NUM; i++) {
        virTypedParameterPtr param = &params[i];

1654
        switch (i) {
1655
        case 0:
1656 1657 1658 1659
            ret = nodeGetMemoryParameterValue("pages_to_scan", &pages_to_scan);
            if (ret == -2)
                continue;
            else if (ret == -1)
1660 1661 1662 1663 1664 1665 1666 1667 1668
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_PAGES_TO_SCAN,
                                        VIR_TYPED_PARAM_UINT, pages_to_scan) < 0)
                return -1;

            break;

        case 1:
1669 1670 1671 1672
            ret = nodeGetMemoryParameterValue("sleep_millisecs", &sleep_millisecs);
            if (ret == -2)
                continue;
            else if (ret == -1)
1673 1674 1675 1676 1677 1678 1679 1680 1681
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_SLEEP_MILLISECS,
                                        VIR_TYPED_PARAM_UINT, sleep_millisecs) < 0)
                return -1;

            break;

        case 2:
1682 1683 1684 1685
            ret = nodeGetMemoryParameterValue("pages_shared", &pages_shared);
            if (ret == -2)
                continue;
            else if (ret == -1)
1686 1687 1688 1689 1690 1691 1692 1693 1694
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_PAGES_SHARED,
                                        VIR_TYPED_PARAM_ULLONG, pages_shared) < 0)
                return -1;

            break;

        case 3:
1695 1696 1697 1698
            ret = nodeGetMemoryParameterValue("pages_sharing", &pages_sharing);
            if (ret == -2)
                continue;
            else if (ret == -1)
1699 1700 1701 1702 1703 1704 1705 1706 1707
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_PAGES_SHARING,
                                        VIR_TYPED_PARAM_ULLONG, pages_sharing) < 0)
                return -1;

            break;

        case 4:
1708 1709 1710 1711
            ret = nodeGetMemoryParameterValue("pages_unshared", &pages_unshared);
            if (ret == -2)
                continue;
            else if (ret == -1)
1712 1713 1714 1715 1716 1717 1718 1719 1720
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_PAGES_UNSHARED,
                                        VIR_TYPED_PARAM_ULLONG, pages_unshared) < 0)
                return -1;

            break;

        case 5:
1721 1722 1723 1724
            ret = nodeGetMemoryParameterValue("pages_volatile", &pages_volatile);
            if (ret == -2)
                continue;
            else if (ret == -1)
1725 1726 1727 1728 1729 1730 1731 1732 1733
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_PAGES_VOLATILE,
                                        VIR_TYPED_PARAM_ULLONG, pages_volatile) < 0)
                return -1;

            break;

        case 6:
1734 1735 1736 1737
            ret = nodeGetMemoryParameterValue("full_scans", &full_scans);
            if (ret == -2)
                continue;
            else if (ret == -1)
1738 1739 1740 1741 1742 1743 1744 1745
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_FULL_SCANS,
                                        VIR_TYPED_PARAM_ULLONG, full_scans) < 0)
                return -1;

            break;

1746
        case 7:
1747 1748 1749 1750
            ret = nodeGetMemoryParameterValue("merge_across_nodes", &merge_across_nodes);
            if (ret == -2)
                continue;
            else if (ret == -1)
1751 1752 1753 1754 1755 1756 1757
                return -1;

            if (virTypedParameterAssign(param, VIR_NODE_MEMORY_SHARED_MERGE_ACROSS_NODES,
                                        VIR_TYPED_PARAM_UINT, merge_across_nodes) < 0)
                return -1;

            break;
1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769
        }
    }

    return 0;
#else
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
                   _("node get memory parameters not implemented"
                     " on this platform"));
    return -1;
#endif
}

1770
int
1771
nodeGetCPUMap(unsigned char **cpumap,
1772 1773
              unsigned int *online,
              unsigned int flags)
1774 1775 1776 1777 1778 1779 1780
{
    virBitmapPtr cpus = NULL;
    int ret = -1;
    int dummy;

    virCheckFlags(0, -1);

1781
    if (!cpumap && !online)
1782
        return nodeGetCPUCount();
1783

1784
    if (!(cpus = nodeGetOnlineCPUBitmap()))
1785 1786 1787 1788 1789 1790 1791
        goto cleanup;

    if (cpumap && virBitmapToData(cpus, cpumap, &dummy) < 0)
        goto cleanup;
    if (online)
        *online = virBitmapCountBits(cpus);

1792 1793
    ret = virBitmapSize(cpus);

1794
 cleanup:
1795 1796 1797 1798 1799 1800
    if (ret < 0 && cpumap)
        VIR_FREE(*cpumap);
    virBitmapFree(cpus);
    return ret;
}

1801
static int
1802
nodeCapsInitNUMAFake(const char *cpupath ATTRIBUTE_UNUSED,
1803
                     virCapsPtr caps ATTRIBUTE_UNUSED)
1804 1805 1806 1807 1808
{
    virNodeInfo nodeinfo;
    virCapsHostNUMACellCPUPtr cpus;
    int ncpus;
    int s, c, t;
1809 1810
    int id, cid;
    int onlinecpus ATTRIBUTE_UNUSED;
1811

1812
    if (nodeGetInfo(&nodeinfo) < 0)
1813 1814 1815
        return -1;

    ncpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
1816
    onlinecpus = nodeinfo.cpus;
1817

1818
    if (VIR_ALLOC_N(cpus, ncpus) < 0)
1819 1820
        return -1;

1821
    id = cid = 0;
1822 1823 1824
    for (s = 0; s < nodeinfo.sockets; s++) {
        for (c = 0; c < nodeinfo.cores; c++) {
            for (t = 0; t < nodeinfo.threads; t++) {
1825
#ifdef __linux__
1826
                if (virNodeGetCpuValue(cpupath, id, "online", 1)) {
1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838
#endif
                    cpus[cid].id = id;
                    cpus[cid].socket_id = s;
                    cpus[cid].core_id = c;
                    if (!(cpus[cid].siblings = virBitmapNew(ncpus)))
                        goto error;
                    ignore_value(virBitmapSetBit(cpus[cid].siblings, id));
                    cid++;
#ifdef __linux__
                }
#endif

1839 1840 1841 1842 1843 1844 1845
                id++;
            }
        }
    }

    if (virCapabilitiesAddHostNUMACell(caps, 0,
                                       nodeinfo.memory,
1846 1847 1848
#ifdef __linux__
                                       onlinecpus, cpus,
#else
1849
                                       ncpus, cpus,
1850
#endif
M
Michal Privoznik 已提交
1851
                                       0, NULL,
1852
                                       0, NULL) < 0)
1853 1854 1855 1856 1857
        goto error;

    return 0;

 error:
1858
    for (; id >= 0; id--)
1859 1860 1861 1862 1863 1864
        virBitmapFree(cpus[id].siblings);
    VIR_FREE(cpus);
    return -1;
}

static int
1865
nodeGetCellsFreeMemoryFake(unsigned long long *freeMems,
1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888
                           int startCell,
                           int maxCells ATTRIBUTE_UNUSED)
{
    double avail = physmem_available();

    if (startCell != 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("start cell %d out of range (0-%d)"),
                       startCell, 0);
        return -1;
    }

    freeMems[0] = (unsigned long long)avail;

    if (!freeMems[0]) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Cannot determine free memory"));
        return -1;
    }

    return 1;
}

1889 1890 1891
static int
nodeGetMemoryFake(unsigned long long *mem,
                  unsigned long long *freeMem)
1892
{
1893 1894
    int ret = -1;

W
Wojciech Macek 已提交
1895 1896 1897 1898 1899
#if defined(__FreeBSD__)
    unsigned long pagesize = getpagesize();
    u_int value;
    size_t value_size = sizeof(value);

1900 1901 1902 1903 1904 1905 1906 1907
    if (mem) {
        if (sysctlbyname("vm.stats.vm.v_page_count", &value,
                         &value_size, NULL, 0) < 0) {
            virReportSystemError(errno, "%s",
                                 _("sysctl failed for vm.stats.vm.v_page_count"));
            goto cleanup;
        }
        *mem = value * (unsigned long long)pagesize;
W
Wojciech Macek 已提交
1908 1909
    }

1910 1911 1912 1913 1914 1915 1916 1917 1918 1919
    if (freeMem) {
        if (sysctlbyname("vm.stats.vm.v_free_count", &value,
                         &value_size, NULL, 0) < 0) {
            virReportSystemError(errno, "%s",
                                 _("sysctl failed for vm.stats.vm.v_free_count"));
            goto cleanup;
        }

        *freeMem = value * (unsigned long long)pagesize;
    }
W
Wojciech Macek 已提交
1920 1921

#else
1922 1923 1924 1925 1926 1927 1928
    if (mem) {
        double total = physmem_total();
        if (!total) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("Cannot determine free memory"));
            goto cleanup;
        }
1929

1930
        *mem = (unsigned long long) total;
1931 1932
    }

1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943
    if (freeMem) {
        double avail = physmem_available();

        if (!avail) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("Cannot determine free memory"));
            goto cleanup;
        }

        *freeMem = (unsigned long long) avail;
    }
W
Wojciech Macek 已提交
1944
#endif
1945 1946 1947 1948

    ret = 0;
 cleanup:
    return ret;
1949 1950
}

1951 1952
/* returns 1 on success, 0 if the detection failed and -1 on hard error */
static int
1953 1954
virNodeCapsFillCPUInfo(const char *cpupath ATTRIBUTE_UNUSED,
                       int cpu_id ATTRIBUTE_UNUSED,
E
Eric Blake 已提交
1955
                       virCapsHostNUMACellCPUPtr cpu ATTRIBUTE_UNUSED)
1956
{
E
Eric Blake 已提交
1957
#ifdef __linux__
1958 1959 1960
    int tmp;
    cpu->id = cpu_id;

1961
    if ((tmp = virNodeGetCpuValue(cpupath, cpu_id,
1962 1963 1964 1965 1966
                                  "topology/physical_package_id", -1)) < 0)
        return 0;

    cpu->socket_id = tmp;

1967
    if ((tmp = virNodeGetCpuValue(cpupath, cpu_id,
1968 1969 1970 1971 1972
                                  "topology/core_id", -1)) < 0)
        return 0;

    cpu->core_id = tmp;

1973
    if (!(cpu->siblings = virNodeGetSiblingsList(cpupath, cpu_id)))
1974 1975 1976
        return -1;

    return 0;
E
Eric Blake 已提交
1977 1978 1979 1980 1981
#else
    virReportError(VIR_ERR_NO_SUPPORT, "%s",
                   _("node cpu info not implemented on this platform"));
    return -1;
#endif
1982 1983
}

1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030
static int
virNodeCapsGetSiblingInfo(int node,
                          virCapsHostNUMACellSiblingInfoPtr *siblings,
                          int *nsiblings)
{
    virCapsHostNUMACellSiblingInfoPtr tmp = NULL;
    int tmp_size = 0;
    int ret = -1;
    int *distances = NULL;
    int ndistances = 0;
    size_t i;

    if (virNumaGetDistances(node, &distances, &ndistances) < 0)
        goto cleanup;

    if (!distances) {
        *siblings = NULL;
        *nsiblings = 0;
        return 0;
    }

    if (VIR_ALLOC_N(tmp, ndistances) < 0)
        goto cleanup;

    for (i = 0; i < ndistances; i++) {
        if (!distances[i])
            continue;

        tmp[tmp_size].node = i;
        tmp[tmp_size].distance = distances[i];
        tmp_size++;
    }

    if (VIR_REALLOC_N(tmp, tmp_size) < 0)
        goto cleanup;

    *siblings = tmp;
    *nsiblings = tmp_size;
    tmp = NULL;
    tmp_size = 0;
    ret = 0;
 cleanup:
    VIR_FREE(distances);
    VIR_FREE(tmp);
    return ret;
}

M
Michal Privoznik 已提交
2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059
static int
virNodeCapsGetPagesInfo(int node,
                        virCapsHostNUMACellPageInfoPtr *pageinfo,
                        int *npageinfo)
{
    int ret = -1;
    unsigned int *pages_size = NULL, *pages_avail = NULL;
    size_t npages, i;

    if (virNumaGetPages(node, &pages_size, &pages_avail, NULL, &npages) < 0)
        goto cleanup;

    if (VIR_ALLOC_N(*pageinfo, npages) < 0)
        goto cleanup;
    *npageinfo = npages;

    for (i = 0; i < npages; i++) {
        (*pageinfo)[i].size = pages_size[i];
        (*pageinfo)[i].avail = pages_avail[i];
    }

    ret = 0;

 cleanup:
    VIR_FREE(pages_avail);
    VIR_FREE(pages_size);
    return ret;
}

2060
int
2061
nodeCapsInitNUMA(virCapsPtr caps)
2062 2063
{
    int n;
2064
    unsigned long long memory;
2065
    virCapsHostNUMACellCPUPtr cpus = NULL;
2066
    virBitmapPtr cpumap = NULL;
2067
    virCapsHostNUMACellSiblingInfoPtr siblings = NULL;
2068
    int nsiblings = 0;
M
Michal Privoznik 已提交
2069 2070
    virCapsHostNUMACellPageInfoPtr pageinfo = NULL;
    int npageinfo;
2071
    int ret = -1;
2072
    int ncpus = 0;
2073
    int cpu;
2074
    bool topology_failed = false;
2075
    int max_node;
2076

2077
    if (!virNumaIsAvailable()) {
2078
        ret = nodeCapsInitNUMAFake(SYSFS_SYSTEM_PATH "/cpu", caps);
2079 2080
        goto cleanup;
    }
2081

2082 2083 2084 2085
    if ((max_node = virNumaGetMaxNode()) < 0)
        goto cleanup;

    for (n = 0; n <= max_node; n++) {
2086
        size_t i;
2087

2088 2089 2090
        if ((ncpus = virNumaGetNodeCPUs(n, &cpumap)) < 0) {
            if (ncpus == -2)
                continue;
2091

2092 2093
            goto cleanup;
        }
2094 2095 2096

        if (VIR_ALLOC_N(cpus, ncpus) < 0)
            goto cleanup;
2097
        cpu = 0;
2098

2099
        for (i = 0; i < virBitmapSize(cpumap); i++) {
J
Ján Tomko 已提交
2100
            if (virBitmapIsBitSet(cpumap, i)) {
2101 2102
                if (virNodeCapsFillCPUInfo(SYSFS_SYSTEM_PATH "/cpu",
                                           i, cpus + cpu++) < 0) {
2103 2104 2105 2106 2107
                    topology_failed = true;
                    virResetLastError();
                }
            }
        }
2108

2109 2110 2111
        if (virNodeCapsGetSiblingInfo(n, &siblings, &nsiblings) < 0)
            goto cleanup;

M
Michal Privoznik 已提交
2112 2113 2114
        if (virNodeCapsGetPagesInfo(n, &pageinfo, &npageinfo) < 0)
            goto cleanup;

2115 2116 2117 2118
        /* Detect the amount of memory in the numa cell in KiB */
        virNumaGetNodeMemory(n, &memory, NULL);
        memory >>= 10;

2119 2120
        if (virCapabilitiesAddHostNUMACell(caps, n, memory,
                                           ncpus, cpus,
M
Michal Privoznik 已提交
2121 2122
                                           nsiblings, siblings,
                                           npageinfo, pageinfo) < 0)
2123
            goto cleanup;
2124 2125

        cpus = NULL;
2126
        siblings = NULL;
M
Michal Privoznik 已提交
2127
        pageinfo = NULL;
2128 2129
        virBitmapFree(cpumap);
        cpumap = NULL;
2130 2131 2132 2133
    }

    ret = 0;

2134
 cleanup:
2135
    if ((topology_failed || ret < 0) && cpus)
2136 2137
        virCapabilitiesClearHostNUMACellCPUTopology(cpus, ncpus);

2138 2139
    virBitmapFree(cpumap);
    VIR_FREE(cpus);
2140
    VIR_FREE(siblings);
M
Michal Privoznik 已提交
2141
    VIR_FREE(pageinfo);
2142 2143
    return ret;
}
2144 2145 2146


int
2147
nodeGetCellsFreeMemory(unsigned long long *freeMems,
2148 2149 2150
                       int startCell,
                       int maxCells)
{
2151
    unsigned long long mem;
2152 2153 2154 2155
    int n, lastCell, numCells;
    int ret = -1;
    int maxCell;

2156
    if (!virNumaIsAvailable())
2157
        return nodeGetCellsFreeMemoryFake(freeMems,
2158 2159
                                          startCell, maxCells);

2160 2161 2162
    if ((maxCell = virNumaGetMaxNode()) < 0)
        return 0;

2163
    if (startCell > maxCell) {
2164 2165 2166
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("start cell %d out of range (0-%d)"),
                       startCell, maxCell);
2167 2168 2169 2170 2171 2172
        goto cleanup;
    }
    lastCell = startCell + maxCells - 1;
    if (lastCell > maxCell)
        lastCell = maxCell;

2173
    for (numCells = 0, n = startCell; n <= lastCell; n++) {
2174
        virNumaGetNodeMemory(n, NULL, &mem);
2175

2176 2177 2178 2179
        freeMems[numCells++] = mem;
    }
    ret = numCells;

2180
 cleanup:
2181 2182 2183
    return ret;
}

2184 2185 2186
int
nodeGetMemory(unsigned long long *mem,
              unsigned long long *freeMem)
2187
{
2188
    int max_node;
2189 2190
    int n;

2191 2192 2193 2194 2195 2196
    if (mem)
        *mem = 0;

    if (freeMem)
        *freeMem = 0;

2197
    if (!virNumaIsAvailable())
2198
        return nodeGetMemoryFake(mem, freeMem);
2199

2200
    if ((max_node = virNumaGetMaxNode()) < 0)
2201
        return -1;
2202

2203
    for (n = 0; n <= max_node; n++) {
2204 2205 2206 2207 2208 2209 2210
        unsigned long long tmp_mem = 0, tmp_freeMem = 0;

        if (!virNumaNodeIsAvailable(n))
            continue;

        if (virNumaGetNodeMemory(n, &tmp_mem, &tmp_freeMem) < 0)
            return -1;
2211

2212 2213 2214 2215 2216
        if (mem)
            *mem += tmp_mem;

        if (freeMem)
            *freeMem += tmp_freeMem;
2217 2218
    }

2219
    return 0;
2220
}
2221 2222 2223 2224 2225 2226 2227 2228 2229

int
nodeGetFreePages(unsigned int npages,
                 unsigned int *pages,
                 int startCell,
                 unsigned int cellCount,
                 unsigned long long *counts)
{
    int ret = -1;
2230
    int cell, lastCell;
2231 2232
    size_t i, ncounts = 0;

2233 2234 2235 2236 2237 2238 2239 2240 2241 2242
    if ((lastCell = virNumaGetMaxNode()) < 0)
        return 0;

    if (startCell > lastCell) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("start cell %d out of range (0-%d)"),
                       startCell, lastCell);
        goto cleanup;
    }

2243
    lastCell = MIN(lastCell, startCell + (int) cellCount - 1);
2244

2245
    for (cell = startCell; cell <= lastCell; cell++) {
2246 2247 2248 2249
        for (i = 0; i < npages; i++) {
            unsigned int page_size = pages[i];
            unsigned int page_free;

2250
            if (virNumaGetPageInfo(cell, page_size, 0, NULL, &page_free) < 0)
2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266
                goto cleanup;

            counts[ncounts++] = page_free;
        }
    }

    if (!ncounts) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("no suitable info found"));
        goto cleanup;
    }

    ret = ncounts;
 cleanup:
    return ret;
}
2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307

int
nodeAllocPages(unsigned int npages,
               unsigned int *pageSizes,
               unsigned long long *pageCounts,
               int startCell,
               unsigned int cellCount,
               bool add)
{
    int ret = -1;
    int cell, lastCell;
    size_t i, ncounts = 0;

    if ((lastCell = virNumaGetMaxNode()) < 0)
        return 0;

    if (startCell > lastCell) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("start cell %d out of range (0-%d)"),
                       startCell, lastCell);
        goto cleanup;
    }

    lastCell = MIN(lastCell, startCell + (int) cellCount - 1);

    for (cell = startCell; cell <= lastCell; cell++) {
        for (i = 0; i < npages; i++) {
            unsigned int page_size = pageSizes[i];
            unsigned long long page_count = pageCounts[i];

            if (virNumaSetPagePoolSize(cell, page_size, page_count, add) < 0)
                goto cleanup;

            ncounts++;
        }
    }

    ret = ncounts;
 cleanup:
    return ret;
}
2308

2309 2310
#if HAVE_LINUX_KVM_H && defined(KVM_CAP_PPC_SMT)

2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359
/* Get the number of threads per subcore.
 *
 * This will be 2, 4 or 8 on POWER hosts, depending on the current
 * micro-threading configuration, and 0 everywhere else.
 *
 * Returns the number of threads per subcore if subcores are in use, zero
 * if subcores are not in use, and a negative value on error */
int
nodeGetThreadsPerSubcore(virArch arch)
{
    int threads_per_subcore = 0;
    const char *kvmpath = "/dev/kvm";
    int kvmfd;

    if (ARCH_IS_PPC64(arch)) {

        /* It's okay if /dev/kvm doesn't exist, because
         *   a. we might be running in a guest
         *   b. the kvm module might not be installed or enabled
         * In either case, falling back to the subcore-unaware thread
         * counting logic is the right thing to do */
        if (!virFileExists(kvmpath))
            goto out;

        if ((kvmfd = open(kvmpath, O_RDONLY)) < 0) {
            /* This can happen when running as a regular user if
             * permissions are tight enough, in which case erroring out
             * is better than silently falling back and reporting
             * different nodeinfo depending on the user */
            virReportSystemError(errno,
                                 _("Failed to open '%s'"),
                                 kvmpath);
            threads_per_subcore = -1;
            goto out;
        }

        /* For Phyp and KVM based guests the ioctl for KVM_CAP_PPC_SMT
         * returns zero and both primary and secondary threads will be
         * online */
        threads_per_subcore = ioctl(kvmfd,
                                    KVM_CHECK_EXTENSION,
                                    KVM_CAP_PPC_SMT);

        VIR_FORCE_CLOSE(kvmfd);
    }

 out:
    return threads_per_subcore;
}
2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371

#else

/* Fallback for nodeGetThreadsPerSubcore() used when KVM headers
 * are not available on the system */
int
nodeGetThreadsPerSubcore(virArch arch ATTRIBUTE_UNUSED)
{
    return 0;
}

#endif /* HAVE_LINUX_KVM_H && defined(KVM_CAP_PPC_SMT) */