File: estimate-max-threads.c

package info (click to toggle)
libguestfs 1%3A1.44.0-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 118,932 kB
  • sloc: ansic: 458,017; ml: 51,424; sh: 13,191; java: 9,578; makefile: 7,931; cs: 6,328; haskell: 5,674; python: 3,871; perl: 3,528; erlang: 2,446; xml: 1,347; ruby: 350; pascal: 257; javascript: 157; lex: 135; yacc: 128; cpp: 10
file content (83 lines) | stat: -rw-r--r-- 2,283 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
/* libguestfs
 * Copyright (C) 2013 Red Hat Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <config.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <error.h>
#include <errno.h>
#include <libintl.h>

#include "guestfs.h"
#include "guestfs-utils.h"
#include "estimate-max-threads.h"

static char *read_line_from (const char *cmd);

/* The actual overhead is likely much smaller than this, but err on
 * the safe side.
 */
#define MBYTES_PER_THREAD 650

/**
 * This function uses the output of C<free -m> to estimate how many
 * libguestfs appliances could be safely started in parallel.  Note
 * that it always returns E<ge> 1.
 */
size_t
estimate_max_threads (void)
{
  CLEANUP_FREE char *mbytes_str = NULL;
  size_t mbytes;

  /* Choose the number of threads based on the amount of free memory. */
  mbytes_str = read_line_from ("LANG=C free -m | "
                               "grep '^Mem' | awk '{print $4+$6+$7}'");
  if (mbytes_str == NULL)
    return 1;

  if (sscanf (mbytes_str, "%zu", &mbytes) != 1)
    return 1;

  return MAX (1, mbytes / MBYTES_PER_THREAD);
}

/**
 * Run external command and read the first line of output.
 */
static char *
read_line_from (const char *cmd)
{
  FILE *pp;
  char *ret = NULL;
  size_t allocsize = 0;

  pp = popen (cmd, "r");
  if (pp == NULL)
    error (EXIT_FAILURE, errno, "%s: external command failed", cmd);

  if (getline (&ret, &allocsize, pp) == -1)
    error (EXIT_FAILURE, errno, "could not read line from external command");

  if (pclose (pp) == -1)
    error (EXIT_FAILURE, errno, "pclose");

  return ret;
}