Quelle  s390-cpumsf.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright IBM Corp. 2018
 * Auxtrace support for s390 CPU-Measurement Sampling Facility
 *
 * Author(s):  Thomas Richter <tmricht@linux.ibm.com>
 *
 * Auxiliary traces are collected during 'perf record' using rbd000 event.
 * Several PERF_RECORD_XXX are generated during recording:
 *
 * PERF_RECORD_AUX:
 * Records that new data landed in the AUX buffer part.
 * PERF_RECORD_AUXTRACE:
 * Defines auxtrace data. Followed by the actual data. The contents of
 * the auxtrace data is dependent on the event and the CPU.
 * This record is generated by perf record command. For details
 * see Documentation/perf.data-file-format.txt.
 * PERF_RECORD_AUXTRACE_INFO:
 * Defines a table of contains for PERF_RECORD_AUXTRACE records. This
 * record is generated during 'perf record' command. Each record contains
 * up to 256 entries describing offset and size of the AUXTRACE data in the
 * perf.data file.
 * PERF_RECORD_AUXTRACE_ERROR:
 * Indicates an error during AUXTRACE collection such as buffer overflow.
 * PERF_RECORD_FINISHED_ROUND:
 * Perf events are not necessarily in time stamp order, as they can be
 * collected in parallel on different CPUs. If the events should be
 * processed in time order they need to be sorted first.
 * Perf report guarantees that there is no reordering over a
 * PERF_RECORD_FINISHED_ROUND boundary event. All perf records with a
 * time stamp lower than this record are processed (and displayed) before
 * the succeeding perf record are processed.
 *
 * These records are evaluated during perf report command.
 *
 * 1. PERF_RECORD_AUXTRACE_INFO is used to set up the infrastructure for
 * auxiliary trace data processing. See s390_cpumsf_process_auxtrace_info()
 * below.
 * Auxiliary trace data is collected per CPU. To merge the data into the report
 * an auxtrace_queue is created for each CPU. It is assumed that the auxtrace
 * data is in ascending order.
 *
 * Each queue has a double linked list of auxtrace_buffers. This list contains
 * the offset and size of a CPU's auxtrace data. During auxtrace processing
 * the data portion is mmap()'ed.
 *
 * To sort the queues in chronological order, all queue access is controlled
 * by the auxtrace_heap. This is basically a stack, each stack element has two
 * entries, the queue number and a time stamp. However the stack is sorted by
 * the time stamps. The highest time stamp is at the bottom the lowest
 * (nearest) time stamp is at the top. That sort order is maintained at all
 * times!
 *
 * After the auxtrace infrastructure has been setup, the auxtrace queues are
 * filled with data (offset/size pairs) and the auxtrace_heap is populated.
 *
 * 2. PERF_RECORD_XXX processing triggers access to the auxtrace_queues.
 * Each record is handled by s390_cpumsf_process_event(). The time stamp of
 * the perf record is compared with the time stamp located on the auxtrace_heap
 * top element. If that time stamp is lower than the time stamp from the
 * record sample, the auxtrace queues will be processed. As auxtrace queues
 * control many auxtrace_buffers and each buffer can be quite large, the
 * auxtrace buffer might be processed only partially. In this case the
 * position in the auxtrace_buffer of that queue is remembered and the time
 * stamp of the last processed entry of the auxtrace_buffer replaces the
 * current auxtrace_heap top.
 *
 * 3. Auxtrace_queues might run of out data and are fed by the
 * PERF_RECORD_AUXTRACE handling, see s390_cpumsf_process_auxtrace_event().
 *
 * Event Generation
 * Each sampling-data entry in the auxiliary trace data generates a perf sample.
 * This sample is filled
 * with data from the auxtrace such as PID/TID, instruction address, CPU state,
 * etc. This sample is processed with perf_session__deliver_synth_event() to
 * be included into the GUI.
 *
 * 4. PERF_RECORD_FINISHED_ROUND event is used to process all the remaining
 * auxiliary traces entries until the time stamp of this record is reached
 * auxtrace_heap top. This is triggered by ordered_event->deliver().
 *
 *
 * Perf event processing.
 * Event processing of PERF_RECORD_XXX entries relies on time stamp entries.
 * This is the function call sequence:
 *
 * __cmd_report()
 * |
 * perf_session__process_events()
 * |
 * __perf_session__process_events()
 * |
 * perf_session__process_event()
 * |  This functions splits the PERF_RECORD_XXX records.
 * |  - Those generated by perf record command (type number equal or higher
 * |    than PERF_RECORD_USER_TYPE_START) are handled by
 * |    perf_session__process_user_event(see below)
 * |  - Those generated by the kernel are handled by
 * |    evlist__parse_sample_timestamp()
 * |
 * evlist__parse_sample_timestamp()
 * |  Extract time stamp from sample data.
 * |
 * perf_session__queue_event()
 * |  If timestamp is positive the sample is entered into an ordered_event
 * |  list, sort order is the timestamp. The event processing is deferred until
 * |  later (see perf_session__process_user_event()).
 * |  Other timestamps (0 or -1) are handled immediately by
 * |  perf_session__deliver_event(). These are events generated at start up
 * |  of command perf record. They create PERF_RECORD_COMM and PERF_RECORD_MMAP*
 * |  records. They are needed to create a list of running processes and its
 * |  memory mappings and layout. They are needed at the beginning to enable
 * |  command perf report to create process trees and memory mappings.
 * |
 * perf_session__deliver_event()
 * |  Delivers a PERF_RECORD_XXX entry for handling.
 * |
 * auxtrace__process_event()
 * |  The timestamp of the PERF_RECORD_XXX entry is taken to correlate with
 * |  time stamps from the auxiliary trace buffers. This enables
 * |  synchronization between auxiliary trace data and the events on the
 * |  perf.data file.
 * |
 * machine__deliver_event()
 * |  Handles the PERF_RECORD_XXX event. This depends on the record type.
 *    It might update the process tree, update a process memory map or enter
 *    a sample with IP and call back chain data into GUI data pool.
 *
 *
 * Deferred processing determined by perf_session__process_user_event() is
 * finally processed when a PERF_RECORD_FINISHED_ROUND is encountered. These
 * are generated during command perf record.
 * The timestamp of PERF_RECORD_FINISHED_ROUND event is taken to process all
 * PERF_RECORD_XXX entries stored in the ordered_event list. This list was
 * built up while reading the perf.data file.
 * Each event is now processed by calling perf_session__deliver_event().
 * This enables time synchronization between the data in the perf.data file and
 * the data in the auxiliary trace buffers.
 */

#include <endian.h>
#include <errno.h>
#include <byteswap.h>
#include <inttypes.h>
#include <linux/kernel.h>
#include <linux/types.h>
#include <linux/bitops.h>
#include <linux/log2.h>
#include <linux/zalloc.h>

#include <sys/stat.h>
#include <sys/types.h>

#include "color.h"
#include "evsel.h"
#include "evlist.h"
#include "machine.h"
#include "session.h"
#include "tool.h"
#include "debug.h"
#include "auxtrace.h"
#include "s390-cpumsf.h"
#include "s390-cpumsf-kernel.h"
#include "s390-cpumcf-kernel.h"
#include "config.h"
#include "util/sample.h"

struct s390_cpumsf {
 struct auxtrace  auxtrace;
 struct auxtrace_queues queues;
 struct auxtrace_heap heap;
 struct perf_session *session;
 struct machine  *machine;
 u32   auxtrace_type;
 u32   pmu_type;
 u16   machine_type;
 bool   data_queued;
 bool   use_logfile;
 char   *logdir;
};

struct s390_cpumsf_queue {
 struct s390_cpumsf *sf;
 unsigned int  queue_nr;
 struct auxtrace_buffer *buffer;
 int   cpu;
 FILE   *logfile;
 FILE   *logfile_ctr;
};

/* Check if the raw data should be dumped to file. If this is the case and
 * the file to dump to has not been opened for writing, do so.
 *
 * Return 0 on success and greater zero on error so processing continues.
 */
static int s390_cpumcf_dumpctr(struct s390_cpumsf *sf,
          struct perf_sample *sample)
{
 struct s390_cpumsf_queue *sfq;
 struct auxtrace_queue *q;
 int rc = 0;

 if (!sf->use_logfile || sf->queues.nr_queues <= sample->cpu)
  return rc;

 q = &sf->queues.queue_array[sample->cpu];
 sfq = q->priv;
 if (!sfq)  /* Queue not yet allocated */
  return rc;

 if (!sfq->logfile_ctr) {
  char *name;

  rc = (sf->logdir)
   ? asprintf(&name, "%s/aux.ctr.%02x",
     sf->logdir, sample->cpu)
   : asprintf(&name, "aux.ctr.%02x", sample->cpu);
  if (rc > 0)
   sfq->logfile_ctr = fopen(name, "w");
  if (sfq->logfile_ctr == NULL) {
   pr_err("Failed to open counter set log file %s, "
          "continue...\n", name);
   rc = 1;
  }
  free(name);
 }

 if (sfq->logfile_ctr) {
  /* See comment above for -4 */
  size_t n = fwrite(sample->raw_data, sample->raw_size - 4, 1,
      sfq->logfile_ctr);
  if (n != 1) {
   pr_err("Failed to write counter set data\n");
   rc = 1;
  }
 }
 return rc;
}

/* Display s390 CPU measurement facility basic-sampling data entry
 * Data written on s390 in big endian byte order and contains bit
 * fields across byte boundaries.
 */
static bool s390_cpumsf_basic_show(const char *color, size_t pos,
       struct hws_basic_entry *basicp)
{
 struct hws_basic_entry *basic = basicp;
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 struct hws_basic_entry local;
 unsigned long long word = be64toh(*(unsigned long long *)basicp);

 memset(&local, 0, sizeof(local));
 local.def = be16toh(basicp->def);
 local.prim_asn = word & 0xffff;
 local.CL = word >> 30 & 0x3;
 local.I = word >> 32 & 0x1;
 local.AS = word >> 33 & 0x3;
 local.P = word >> 35 & 0x1;
 local.W = word >> 36 & 0x1;
 local.T = word >> 37 & 0x1;
 local.U = word >> 40 & 0xf;
 local.ia = be64toh(basicp->ia);
 local.gpp = be64toh(basicp->gpp);
 local.hpp = be64toh(basicp->hpp);
 basic = &local;
#endif
 if (basic->def != 1) {
  pr_err("Invalid AUX trace basic entry [%#08zx]\n", pos);
  return false;
 }
 color_fprintf(stdout, color, "    [%#08zx] Basic   Def:%04x Inst:%#04x"
        " %c%c%c%c AS:%d ASN:%#04x IA:%#018llx\n"
        "\t\tCL:%d HPP:%#018llx GPP:%#018llx\n",
        pos, basic->def, basic->U,
        basic->T ? 'T' : ' ',
        basic->W ? 'W' : ' ',
        basic->P ? 'P' : ' ',
        basic->I ? 'I' : ' ',
        basic->AS, basic->prim_asn, basic->ia, basic->CL,
        basic->hpp, basic->gpp);
 return true;
}

/* Display s390 CPU measurement facility diagnostic-sampling data entry.
 * Data written on s390 in big endian byte order and contains bit
 * fields across byte boundaries.
 */
static bool s390_cpumsf_diag_show(const char *color, size_t pos,
      struct hws_diag_entry *diagp)
{
 struct hws_diag_entry *diag = diagp;
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 struct hws_diag_entry local;
 unsigned long long word = be64toh(*(unsigned long long *)diagp);

 local.def = be16toh(diagp->def);
 local.I = word >> 32 & 0x1;
 diag = &local;
#endif
 if (diag->def < S390_CPUMSF_DIAG_DEF_FIRST) {
  pr_err("Invalid AUX trace diagnostic entry [%#08zx]\n", pos);
  return false;
 }
 color_fprintf(stdout, color, "    [%#08zx] Diag    Def:%04x %c\n",
        pos, diag->def, diag->I ? 'I' : ' ');
 return true;
}

/* Return TOD timestamp contained in an trailer entry */
static unsigned long long trailer_timestamp(struct hws_trailer_entry *te,
         int idx)
{
 /* te->t set: TOD in STCKE format, bytes 8-15
  * to->t not set: TOD in STCK format, bytes 0-7
 */
 unsigned long long ts;

 memcpy(&ts, &te->timestamp[idx], sizeof(ts));
 return be64toh(ts);
}

/* Display s390 CPU measurement facility trailer entry */
static bool s390_cpumsf_trailer_show(const char *color, size_t pos,
         struct hws_trailer_entry *te)
{
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 struct hws_trailer_entry local;
 const unsigned long long flags = be64toh(te->flags);

 memset(&local, 0, sizeof(local));
 local.f = flags >> 63 & 0x1;
 local.a = flags >> 62 & 0x1;
 local.t = flags >> 61 & 0x1;
 local.bsdes = be16toh((flags >> 16 & 0xffff));
 local.dsdes = be16toh((flags & 0xffff));
 memcpy(&local.timestamp, te->timestamp, sizeof(te->timestamp));
 local.overflow = be64toh(te->overflow);
 local.clock_base = be64toh(te->progusage[0]) >> 63 & 1;
 local.progusage2 = be64toh(te->progusage2);
 te = &local;
#endif
 if (te->bsdes != sizeof(struct hws_basic_entry)) {
  pr_err("Invalid AUX trace trailer entry [%#08zx]\n", pos);
  return false;
 }
 color_fprintf(stdout, color, "    [%#08zx] Trailer %c%c%c bsdes:%d"
        " dsdes:%d Overflow:%lld Time:%#llx\n"
        "\t\tC:%d TOD:%#llx\n",
        pos,
        te->f ? 'F' : ' ',
        te->a ? 'A' : ' ',
        te->t ? 'T' : ' ',
        te->bsdes, te->dsdes, te->overflow,
        trailer_timestamp(te, te->clock_base),
        te->clock_base, te->progusage2);
 return true;
}

/* Test a sample data block. It must be 4KB or a multiple thereof in size and
 * 4KB page aligned. Each sample data page has a trailer entry at the
 * end which contains the sample entry data sizes.
 *
 * Return true if the sample data block passes the checks and set the
 * basic set entry size and diagnostic set entry size.
 *
 * Return false on failure.
 *
 * Note: Old hardware does not set the basic or diagnostic entry sizes
 * in the trailer entry. Use the type number instead.
 */
static bool s390_cpumsf_validate(int machine_type,
     unsigned char *buf, size_t len,
     unsigned short *bsdes,
     unsigned short *dsdes)
{
 struct hws_basic_entry *basic = (struct hws_basic_entry *)buf;
 struct hws_trailer_entry *te;

 *dsdes = *bsdes = 0;
 if (len & (S390_CPUMSF_PAGESZ - 1)) /* Illegal size */
  return false;
 if (be16toh(basic->def) != 1) /* No basic set entry, must be first */
  return false;
 /* Check for trailer entry at end of SDB */
 te = (struct hws_trailer_entry *)(buf + S390_CPUMSF_PAGESZ
           - sizeof(*te));
 *bsdes = be16toh(te->bsdes);
 *dsdes = be16toh(te->dsdes);
 if (!te->bsdes && !te->dsdes) {
  /* Very old hardware, use CPUID */
  switch (machine_type) {
  case 2097:
  case 2098:
   *dsdes = 64;
   *bsdes = 32;
   break;
  case 2817:
  case 2818:
   *dsdes = 74;
   *bsdes = 32;
   break;
  case 2827:
  case 2828:
   *dsdes = 85;
   *bsdes = 32;
   break;
  case 2964:
  case 2965:
   *dsdes = 112;
   *bsdes = 32;
   break;
  default:
   /* Illegal trailer entry */
   return false;
  }
 }
 return true;
}

/* Return true if there is room for another entry */
static bool s390_cpumsf_reached_trailer(size_t entry_sz, size_t pos)
{
 size_t payload = S390_CPUMSF_PAGESZ - sizeof(struct hws_trailer_entry);

 if (payload - (pos & (S390_CPUMSF_PAGESZ - 1)) < entry_sz)
  return false;
 return true;
}

/* Dump an auxiliary buffer. These buffers are multiple of
 * 4KB SDB pages.
 */
static void s390_cpumsf_dump(struct s390_cpumsf *sf,
        unsigned char *buf, size_t len)
{
 const char *color = PERF_COLOR_BLUE;
 struct hws_basic_entry *basic;
 struct hws_diag_entry *diag;
 unsigned short bsdes, dsdes;
 size_t pos = 0;

 color_fprintf(stdout, color,
        ". ... s390 AUX data: size %zu bytes\n",
        len);

 if (!s390_cpumsf_validate(sf->machine_type, buf, len, &bsdes,
      &dsdes)) {
  pr_err("Invalid AUX trace data block size:%zu"
         " (type:%d bsdes:%hd dsdes:%hd)\n",
         len, sf->machine_type, bsdes, dsdes);
  return;
 }

 /* s390 kernel always returns 4KB blocks fully occupied,
  * no partially filled SDBs.
 */
 while (pos < len) {
  /* Handle Basic entry */
  basic = (struct hws_basic_entry *)(buf + pos);
  if (s390_cpumsf_basic_show(color, pos, basic))
   pos += bsdes;
  else
   return;

  /* Handle Diagnostic entry */
  diag = (struct hws_diag_entry *)(buf + pos);
  if (s390_cpumsf_diag_show(color, pos, diag))
   pos += dsdes;
  else
   return;

  /* Check for trailer entry */
  if (!s390_cpumsf_reached_trailer(bsdes + dsdes, pos)) {
   /* Show trailer entry */
   struct hws_trailer_entry te;

   pos = (pos + S390_CPUMSF_PAGESZ)
          & ~(S390_CPUMSF_PAGESZ - 1);
   pos -= sizeof(te);
   memcpy(&te, buf + pos, sizeof(te));
   /* Set descriptor sizes in case of old hardware
    * where these values are not set.
 */
   te.bsdes = bsdes;
   te.dsdes = dsdes;
   if (s390_cpumsf_trailer_show(color, pos, &te))
    pos += sizeof(te);
   else
    return;
  }
 }
}

static void s390_cpumsf_dump_event(struct s390_cpumsf *sf, unsigned char *buf,
       size_t len)
{
 printf(".\n");
 s390_cpumsf_dump(sf, buf, len);
}

#define S390_LPP_PID_MASK 0xffffffff

static bool s390_cpumsf_make_event(size_t pos,
       struct hws_basic_entry *basic,
       struct s390_cpumsf_queue *sfq)
{
 struct perf_sample sample = {
    .ip = basic->ia,
    .pid = basic->hpp & S390_LPP_PID_MASK,
    .tid = basic->hpp & S390_LPP_PID_MASK,
    .cpumode = PERF_RECORD_MISC_CPUMODE_UNKNOWN,
    .cpu = sfq->cpu,
    .period = 1
       };
 union perf_event event;
 int ret;

 memset(&event, 0, sizeof(event));
 if (basic->CL == 1) /* Native LPAR mode */
  sample.cpumode = basic->P ? PERF_RECORD_MISC_USER
       : PERF_RECORD_MISC_KERNEL;
 else if (basic->CL == 2) /* Guest kernel/user space */
  sample.cpumode = basic->P ? PERF_RECORD_MISC_GUEST_USER
       : PERF_RECORD_MISC_GUEST_KERNEL;
 else if (basic->gpp || basic->prim_asn != 0xffff)
  /* Use heuristics on old hardware */
  sample.cpumode = basic->P ? PERF_RECORD_MISC_GUEST_USER
       : PERF_RECORD_MISC_GUEST_KERNEL;
 else
  sample.cpumode = basic->P ? PERF_RECORD_MISC_USER
       : PERF_RECORD_MISC_KERNEL;

 event.sample.header.type = PERF_RECORD_SAMPLE;
 event.sample.header.misc = sample.cpumode;
 event.sample.header.size = sizeof(struct perf_event_header);

 pr_debug4("%s pos:%#zx ip:%#" PRIx64 " P:%d CL:%d pid:%d.%d cpumode:%d cpu:%d\n",
   __func__, pos, sample.ip, basic->P, basic->CL, sample.pid,
   sample.tid, sample.cpumode, sample.cpu);
 ret = perf_session__deliver_synth_event(sfq->sf->session, &event, &sample);
 perf_sample__exit(&sample);
 if (ret) {
  pr_err("s390 Auxiliary Trace: failed to deliver event\n");
  return false;
 }
 return true;
}

static unsigned long long get_trailer_time(const unsigned char *buf)
{
 struct hws_trailer_entry *te;
 unsigned long long aux_time, progusage2;
 bool clock_base;

 te = (struct hws_trailer_entry *)(buf + S390_CPUMSF_PAGESZ
           - sizeof(*te));

#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 clock_base = be64toh(te->progusage[0]) >> 63 & 0x1;
 progusage2 = be64toh(te->progusage[1]);
#else
 clock_base = te->clock_base;
 progusage2 = te->progusage2;
#endif
 if (!clock_base) /* TOD_CLOCK_BASE value missing */
  return 0;

 /* Correct calculation to convert time stamp in trailer entry to
  * nano seconds (taken from arch/s390 function tod_to_ns()).
  * TOD_CLOCK_BASE is stored in trailer entry member progusage2.
 */
 aux_time = trailer_timestamp(te, clock_base) - progusage2;
 aux_time = (aux_time >> 9) * 125 + (((aux_time & 0x1ff) * 125) >> 9);
 return aux_time;
}

/* Process the data samples of a single queue. The first parameter is a
 * pointer to the queue, the second parameter is the time stamp. This
 * is the time stamp:
 * - of the event that triggered this processing.
 * - or the time stamp when the last processing of this queue stopped.
 *   In this case it stopped at a 4KB page boundary and record the
 *   position on where to continue processing on the next invocation
 *   (see buffer->use_data and buffer->use_size).
 *
 * When this function returns the second parameter is updated to
 * reflect the time stamp of the last processed auxiliary data entry
 * (taken from the trailer entry of that page). The caller uses this
 * returned time stamp to record the last processed entry in this
 * queue.
 *
 * The function returns:
 * 0:  Processing successful. The second parameter returns the
 *     time stamp from the trailer entry until which position
 *     processing took place. Subsequent calls resume from this
 *     position.
 * <0: An error occurred during processing. The second parameter
 *     returns the maximum time stamp.
 * >0: Done on this queue. The second parameter returns the
 *     maximum time stamp.
 */
static int s390_cpumsf_samples(struct s390_cpumsf_queue *sfq, u64 *ts)
{
 struct s390_cpumsf *sf = sfq->sf;
 unsigned char *buf = sfq->buffer->use_data;
 size_t len = sfq->buffer->use_size;
 struct hws_basic_entry *basic;
 unsigned short bsdes, dsdes;
 size_t pos = 0;
 int err = 1;
 u64 aux_ts;

 if (!s390_cpumsf_validate(sf->machine_type, buf, len, &bsdes,
      &dsdes)) {
  *ts = ~0ULL;
  return -1;
 }

 /* Get trailer entry time stamp and check if entries in
  * this auxiliary page are ready for processing. If the
  * time stamp of the first entry is too high, whole buffer
  * can be skipped. In this case return time stamp.
 */
 aux_ts = get_trailer_time(buf);
 if (!aux_ts) {
  pr_err("[%#08" PRIx64 "] Invalid AUX trailer entry TOD clock base\n",
         (s64)sfq->buffer->data_offset);
  aux_ts = ~0ULL;
  goto out;
 }
 if (aux_ts > *ts) {
  *ts = aux_ts;
  return 0;
 }

 while (pos < len) {
  /* Handle Basic entry */
  basic = (struct hws_basic_entry *)(buf + pos);
  if (s390_cpumsf_make_event(pos, basic, sfq))
   pos += bsdes;
  else {
   err = -EBADF;
   goto out;
  }

  pos += dsdes; /* Skip diagnostic entry */

  /* Check for trailer entry */
  if (!s390_cpumsf_reached_trailer(bsdes + dsdes, pos)) {
   pos = (pos + S390_CPUMSF_PAGESZ)
          & ~(S390_CPUMSF_PAGESZ - 1);
   /* Check existence of next page */
   if (pos >= len)
    break;
   aux_ts = get_trailer_time(buf + pos);
   if (!aux_ts) {
    aux_ts = ~0ULL;
    goto out;
   }
   if (aux_ts > *ts) {
    *ts = aux_ts;
    sfq->buffer->use_data += pos;
    sfq->buffer->use_size -= pos;
    return 0;
   }
  }
 }
out:
 *ts = aux_ts;
 sfq->buffer->use_size = 0;
 sfq->buffer->use_data = NULL;
 return err; /* Buffer completely scanned or error */
}

/* Run the s390 auxiliary trace decoder.
 * Select the queue buffer to operate on, the caller already selected
 * the proper queue, depending on second parameter 'ts'.
 * This is the time stamp until which the auxiliary entries should
 * be processed. This value is updated by called functions and
 * returned to the caller.
 *
 * Resume processing in the current buffer. If there is no buffer
 * get a new buffer from the queue and setup start position for
 * processing.
 * When a buffer is completely processed remove it from the queue
 * before returning.
 *
 * This function returns
 * 1: When the queue is empty. Second parameter will be set to
 *    maximum time stamp.
 * 0: Normal processing done.
 * <0: Error during queue buffer setup. This causes the caller
 *     to stop processing completely.
 */
static int s390_cpumsf_run_decoder(struct s390_cpumsf_queue *sfq,
       u64 *ts)
{

 struct auxtrace_buffer *buffer;
 struct auxtrace_queue *queue;
 int err;

 queue = &sfq->sf->queues.queue_array[sfq->queue_nr];

 /* Get buffer and last position in buffer to resume
  * decoding the auxiliary entries. One buffer might be large
  * and decoding might stop in between. This depends on the time
  * stamp of the trailer entry in each page of the auxiliary
  * data and the time stamp of the event triggering the decoding.
 */
 if (sfq->buffer == NULL) {
  sfq->buffer = buffer = auxtrace_buffer__next(queue,
            sfq->buffer);
  if (!buffer) {
   *ts = ~0ULL;
   return 1; /* Processing done on this queue */
  }
  /* Start with a new buffer on this queue */
  if (buffer->data) {
   buffer->use_size = buffer->size;
   buffer->use_data = buffer->data;
  }
  if (sfq->logfile) { /* Write into log file */
   size_t rc = fwrite(buffer->data, buffer->size, 1,
        sfq->logfile);
   if (rc != 1)
    pr_err("Failed to write auxiliary data\n");
  }
 } else
  buffer = sfq->buffer;

 if (!buffer->data) {
  int fd = perf_data__fd(sfq->sf->session->data);

  buffer->data = auxtrace_buffer__get_data(buffer, fd);
  if (!buffer->data)
   return -ENOMEM;
  buffer->use_size = buffer->size;
  buffer->use_data = buffer->data;

  if (sfq->logfile) { /* Write into log file */
   size_t rc = fwrite(buffer->data, buffer->size, 1,
        sfq->logfile);
   if (rc != 1)
    pr_err("Failed to write auxiliary data\n");
  }
 }
 pr_debug4("%s queue_nr:%d buffer:%" PRId64 " offset:%#" PRIx64 " size:%#zx rest:%#zx\n",
    __func__, sfq->queue_nr, buffer->buffer_nr, buffer->offset,
    buffer->size, buffer->use_size);
 err = s390_cpumsf_samples(sfq, ts);

 /* If non-zero, there is either an error (err < 0) or the buffer is
  * completely done (err > 0). The error is unrecoverable, usually
  * some descriptors could not be read successfully, so continue with
  * the next buffer.
  * In both cases the parameter 'ts' has been updated.
 */
 if (err) {
  sfq->buffer = NULL;
  list_del_init(&buffer->list);
  auxtrace_buffer__free(buffer);
  if (err > 0)  /* Buffer done, no error */
   err = 0;
 }
 return err;
}

static struct s390_cpumsf_queue *
s390_cpumsf_alloc_queue(struct s390_cpumsf *sf, unsigned int queue_nr)
{
 struct s390_cpumsf_queue *sfq;

 sfq = zalloc(sizeof(struct s390_cpumsf_queue));
 if (sfq == NULL)
  return NULL;

 sfq->sf = sf;
 sfq->queue_nr = queue_nr;
 sfq->cpu = -1;
 if (sf->use_logfile) {
  char *name;
  int rc;

  rc = (sf->logdir)
   ? asprintf(&name, "%s/aux.smp.%02x",
     sf->logdir, queue_nr)
   : asprintf(&name, "aux.smp.%02x", queue_nr);
  if (rc > 0)
   sfq->logfile = fopen(name, "w");
  if (sfq->logfile == NULL) {
   pr_err("Failed to open auxiliary log file %s,"
          "continue...\n", name);
   sf->use_logfile = false;
  }
  free(name);
 }
 return sfq;
}

static int s390_cpumsf_setup_queue(struct s390_cpumsf *sf,
       struct auxtrace_queue *queue,
       unsigned int queue_nr, u64 ts)
{
 struct s390_cpumsf_queue *sfq = queue->priv;

 if (list_empty(&queue->head))
  return 0;

 if (sfq == NULL) {
  sfq = s390_cpumsf_alloc_queue(sf, queue_nr);
  if (!sfq)
   return -ENOMEM;
  queue->priv = sfq;

  if (queue->cpu != -1)
   sfq->cpu = queue->cpu;
 }
 return auxtrace_heap__add(&sf->heap, queue_nr, ts);
}

static int s390_cpumsf_setup_queues(struct s390_cpumsf *sf, u64 ts)
{
 unsigned int i;
 int ret = 0;

 for (i = 0; i < sf->queues.nr_queues; i++) {
  ret = s390_cpumsf_setup_queue(sf, &sf->queues.queue_array[i],
           i, ts);
  if (ret)
   break;
 }
 return ret;
}

static int s390_cpumsf_update_queues(struct s390_cpumsf *sf, u64 ts)
{
 if (!sf->queues.new_data)
  return 0;

 sf->queues.new_data = false;
 return s390_cpumsf_setup_queues(sf, ts);
}

static int s390_cpumsf_process_queues(struct s390_cpumsf *sf, u64 timestamp)
{
 unsigned int queue_nr;
 u64 ts;
 int ret;

 while (1) {
  struct auxtrace_queue *queue;
  struct s390_cpumsf_queue *sfq;

  if (!sf->heap.heap_cnt)
   return 0;

  if (sf->heap.heap_array[0].ordinal >= timestamp)
   return 0;

  queue_nr = sf->heap.heap_array[0].queue_nr;
  queue = &sf->queues.queue_array[queue_nr];
  sfq = queue->priv;

  auxtrace_heap__pop(&sf->heap);
  if (sf->heap.heap_cnt) {
   ts = sf->heap.heap_array[0].ordinal + 1;
   if (ts > timestamp)
    ts = timestamp;
  } else {
   ts = timestamp;
  }

  ret = s390_cpumsf_run_decoder(sfq, &ts);
  if (ret < 0) {
   auxtrace_heap__add(&sf->heap, queue_nr, ts);
   return ret;
  }
  if (!ret) {
   ret = auxtrace_heap__add(&sf->heap, queue_nr, ts);
   if (ret < 0)
    return ret;
  }
 }
 return 0;
}

static int s390_cpumsf_synth_error(struct s390_cpumsf *sf, int code, int cpu,
       pid_t pid, pid_t tid, u64 ip, u64 timestamp)
{
 char msg[MAX_AUXTRACE_ERROR_MSG];
 union perf_event event;
 int err;

 strncpy(msg, "Lost Auxiliary Trace Buffer", sizeof(msg) - 1);
 auxtrace_synth_error(&event.auxtrace_error, PERF_AUXTRACE_ERROR_ITRACE,
        code, cpu, pid, tid, ip, msg, timestamp);

 err = perf_session__deliver_synth_event(sf->session, &event, NULL);
 if (err)
  pr_err("s390 Auxiliary Trace: failed to deliver error event,"
   "error %d\n", err);
 return err;
}

static int s390_cpumsf_lost(struct s390_cpumsf *sf, struct perf_sample *sample)
{
 return s390_cpumsf_synth_error(sf, 1, sample->cpu,
           sample->pid, sample->tid, 0,
           sample->time);
}

static int
s390_cpumsf_process_event(struct perf_session *session,
     union perf_event *event,
     struct perf_sample *sample,
     const struct perf_tool *tool)
{
 struct s390_cpumsf *sf = container_of(session->auxtrace,
           struct s390_cpumsf,
           auxtrace);
 u64 timestamp = sample->time;
 struct evsel *ev_bc000;

 int err = 0;

 if (dump_trace)
  return 0;

 if (!tool->ordered_events) {
  pr_err("s390 Auxiliary Trace requires ordered events\n");
  return -EINVAL;
 }

 if (event->header.type == PERF_RECORD_SAMPLE &&
     sample->raw_size) {
  /* Handle event with raw data */
  ev_bc000 = evlist__event2evsel(session->evlist, event);
  if (ev_bc000 &&
      ev_bc000->core.attr.config == PERF_EVENT_CPUM_CF_DIAG)
   err = s390_cpumcf_dumpctr(sf, sample);
  return err;
 }

 if (event->header.type == PERF_RECORD_AUX &&
     event->aux.flags & PERF_AUX_FLAG_TRUNCATED)
  return s390_cpumsf_lost(sf, sample);

 if (timestamp) {
  err = s390_cpumsf_update_queues(sf, timestamp);
  if (!err)
   err = s390_cpumsf_process_queues(sf, timestamp);
 }
 return err;
}

static int
s390_cpumsf_process_auxtrace_event(struct perf_session *session,
       union perf_event *event __maybe_unused,
       const struct perf_tool *tool __maybe_unused)
{
 struct s390_cpumsf *sf = container_of(session->auxtrace,
           struct s390_cpumsf,
           auxtrace);

 int fd = perf_data__fd(session->data);
 struct auxtrace_buffer *buffer;
 off_t data_offset;
 int err;

 if (sf->data_queued)
  return 0;

 if (perf_data__is_pipe(session->data)) {
  data_offset = 0;
 } else {
  data_offset = lseek(fd, 0, SEEK_CUR);
  if (data_offset == -1)
   return -errno;
 }

 err = auxtrace_queues__add_event(&sf->queues, session, event,
      data_offset, &buffer);
 if (err)
  return err;

 /* Dump here after copying piped trace out of the pipe */
 if (dump_trace) {
  if (auxtrace_buffer__get_data(buffer, fd)) {
   s390_cpumsf_dump_event(sf, buffer->data,
            buffer->size);
   auxtrace_buffer__put_data(buffer);
  }
 }
 return 0;
}

static void s390_cpumsf_free_events(struct perf_session *session __maybe_unused)
{
}

static int s390_cpumsf_flush(struct perf_session *session __maybe_unused,
        const struct perf_tool *tool __maybe_unused)
{
 return 0;
}

static void s390_cpumsf_free_queues(struct perf_session *session)
{
 struct s390_cpumsf *sf = container_of(session->auxtrace,
           struct s390_cpumsf,
           auxtrace);
 struct auxtrace_queues *queues = &sf->queues;
 unsigned int i;

 for (i = 0; i < queues->nr_queues; i++) {
  struct s390_cpumsf_queue *sfq = (struct s390_cpumsf_queue *)
      queues->queue_array[i].priv;

  if (sfq != NULL) {
   if (sfq->logfile) {
    fclose(sfq->logfile);
    sfq->logfile = NULL;
   }
   if (sfq->logfile_ctr) {
    fclose(sfq->logfile_ctr);
    sfq->logfile_ctr = NULL;
   }
  }
  zfree(&queues->queue_array[i].priv);
 }
 auxtrace_queues__free(queues);
}

static void s390_cpumsf_free(struct perf_session *session)
{
 struct s390_cpumsf *sf = container_of(session->auxtrace,
           struct s390_cpumsf,
           auxtrace);

 auxtrace_heap__free(&sf->heap);
 s390_cpumsf_free_queues(session);
 session->auxtrace = NULL;
 zfree(&sf->logdir);
 free(sf);
}

static bool
s390_cpumsf_evsel_is_auxtrace(struct perf_session *session __maybe_unused,
         struct evsel *evsel)
{
 return evsel->core.attr.type == PERF_TYPE_RAW &&
        evsel->core.attr.config == PERF_EVENT_CPUM_SF_DIAG;
}

static int s390_cpumsf_get_type(const char *cpuid)
{
 int ret, family = 0;

 ret = sscanf(cpuid, "%*[^,],%u", &family);
 return (ret == 1) ? family : 0;
}

/* Check itrace options set on perf report command.
 * Return true, if none are set or all options specified can be
 * handled on s390 (currently only option 'd' for logging.
 * Return false otherwise.
 */
static bool check_auxtrace_itrace(struct itrace_synth_opts *itops)
{
 bool ison = false;

 if (!itops || !itops->set)
  return true;
 ison = itops->inject || itops->instructions || itops->branches ||
  itops->transactions || itops->ptwrites ||
  itops->pwr_events || itops->errors ||
  itops->dont_decode || itops->calls || itops->returns ||
  itops->callchain || itops->thread_stack ||
  itops->last_branch || itops->add_callchain ||
  itops->add_last_branch;
 if (!ison)
  return true;
 pr_err("Unsupported --itrace options specified\n");
 return false;
}

/* Check for AUXTRACE dump directory if it is needed.
 * On failure print an error message but continue.
 * Return 0 on wrong keyword in config file and 1 otherwise.
 */
static int s390_cpumsf__config(const char *var, const char *value, void *cb)
{
 struct s390_cpumsf *sf = cb;
 struct stat stbuf;
 int rc;

 if (strcmp(var, "auxtrace.dumpdir"))
  return 0;
 sf->logdir = strdup(value);
 if (sf->logdir == NULL) {
  pr_err("Failed to find auxtrace log directory %s,"
         " continue with current directory...\n", value);
  return 1;
 }
 rc = stat(sf->logdir, &stbuf);
 if (rc == -1 || !S_ISDIR(stbuf.st_mode)) {
  pr_err("Missing auxtrace log directory %s,"
         " continue with current directory...\n", value);
  zfree(&sf->logdir);
 }
 return 1;
}

int s390_cpumsf_process_auxtrace_info(union perf_event *event,
          struct perf_session *session)
{
 struct perf_record_auxtrace_info *auxtrace_info = &event->auxtrace_info;
 struct s390_cpumsf *sf;
 int err;

 if (auxtrace_info->header.size < sizeof(struct perf_record_auxtrace_info))
  return -EINVAL;

 sf = zalloc(sizeof(struct s390_cpumsf));
 if (sf == NULL)
  return -ENOMEM;

 if (!check_auxtrace_itrace(session->itrace_synth_opts)) {
  err = -EINVAL;
  goto err_free;
 }
 sf->use_logfile = session->itrace_synth_opts->log;
 if (sf->use_logfile)
  perf_config(s390_cpumsf__config, sf);

 err = auxtrace_queues__init(&sf->queues);
 if (err)
  goto err_free;

 sf->session = session;
 sf->machine = &session->machines.host; /* No kvm support */
 sf->auxtrace_type = auxtrace_info->type;
 sf->pmu_type = PERF_TYPE_RAW;
 sf->machine_type = s390_cpumsf_get_type(perf_session__env(session)->cpuid);

 sf->auxtrace.process_event = s390_cpumsf_process_event;
 sf->auxtrace.process_auxtrace_event = s390_cpumsf_process_auxtrace_event;
 sf->auxtrace.flush_events = s390_cpumsf_flush;
 sf->auxtrace.free_events = s390_cpumsf_free_events;
 sf->auxtrace.free = s390_cpumsf_free;
 sf->auxtrace.evsel_is_auxtrace = s390_cpumsf_evsel_is_auxtrace;
 session->auxtrace = &sf->auxtrace;

 if (dump_trace)
  return 0;

 err = auxtrace_queues__process_index(&sf->queues, session);
 if (err)
  goto err_free_queues;

 if (sf->queues.populated)
  sf->data_queued = true;

 return 0;

err_free_queues:
 auxtrace_queues__free(&sf->queues);
 session->auxtrace = NULL;
err_free:
 zfree(&sf->logdir);
 free(sf);
 return err;
}