// SPDX-License-Identifier: GPL-2.0 
 | 
/* 
 | 
 * trace_seq.c 
 | 
 * 
 | 
 * Copyright (C) 2008-2014 Red Hat Inc, Steven Rostedt <srostedt@redhat.com> 
 | 
 * 
 | 
 * The trace_seq is a handy tool that allows you to pass a descriptor around 
 | 
 * to a buffer that other functions can write to. It is similar to the 
 | 
 * seq_file functionality but has some differences. 
 | 
 * 
 | 
 * To use it, the trace_seq must be initialized with trace_seq_init(). 
 | 
 * This will set up the counters within the descriptor. You can call 
 | 
 * trace_seq_init() more than once to reset the trace_seq to start 
 | 
 * from scratch. 
 | 
 *  
 | 
 * The buffer size is currently PAGE_SIZE, although it may become dynamic 
 | 
 * in the future. 
 | 
 * 
 | 
 * A write to the buffer will either succed or fail. That is, unlike 
 | 
 * sprintf() there will not be a partial write (well it may write into 
 | 
 * the buffer but it wont update the pointers). This allows users to 
 | 
 * try to write something into the trace_seq buffer and if it fails 
 | 
 * they can flush it and try again. 
 | 
 * 
 | 
 */ 
 | 
#include <linux/uaccess.h> 
 | 
#include <linux/seq_file.h> 
 | 
#include <linux/trace_seq.h> 
 | 
  
 | 
/* How much buffer is left on the trace_seq? */ 
 | 
#define TRACE_SEQ_BUF_LEFT(s) seq_buf_buffer_left(&(s)->seq) 
 | 
  
 | 
/* 
 | 
 * trace_seq should work with being initialized with 0s. 
 | 
 */ 
 | 
static inline void __trace_seq_init(struct trace_seq *s) 
 | 
{ 
 | 
    if (unlikely(!s->seq.size)) 
 | 
        trace_seq_init(s); 
 | 
} 
 | 
  
 | 
/** 
 | 
 * trace_print_seq - move the contents of trace_seq into a seq_file 
 | 
 * @m: the seq_file descriptor that is the destination 
 | 
 * @s: the trace_seq descriptor that is the source. 
 | 
 * 
 | 
 * Returns 0 on success and non zero on error. If it succeeds to 
 | 
 * write to the seq_file it will reset the trace_seq, otherwise 
 | 
 * it does not modify the trace_seq to let the caller try again. 
 | 
 */ 
 | 
int trace_print_seq(struct seq_file *m, struct trace_seq *s) 
 | 
{ 
 | 
    int ret; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    ret = seq_buf_print_seq(m, &s->seq); 
 | 
  
 | 
    /* 
 | 
     * Only reset this buffer if we successfully wrote to the 
 | 
     * seq_file buffer. This lets the caller try again or 
 | 
     * do something else with the contents. 
 | 
     */ 
 | 
    if (!ret) 
 | 
        trace_seq_init(s); 
 | 
  
 | 
    return ret; 
 | 
} 
 | 
  
 | 
/** 
 | 
 * trace_seq_printf - sequence printing of trace information 
 | 
 * @s: trace sequence descriptor 
 | 
 * @fmt: printf format string 
 | 
 * 
 | 
 * The tracer may use either sequence operations or its own 
 | 
 * copy to user routines. To simplify formating of a trace 
 | 
 * trace_seq_printf() is used to store strings into a special 
 | 
 * buffer (@s). Then the output may be either used by 
 | 
 * the sequencer or pulled into another buffer. 
 | 
 */ 
 | 
void trace_seq_printf(struct trace_seq *s, const char *fmt, ...) 
 | 
{ 
 | 
    unsigned int save_len = s->seq.len; 
 | 
    va_list ap; 
 | 
  
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    va_start(ap, fmt); 
 | 
    seq_buf_vprintf(&s->seq, fmt, ap); 
 | 
    va_end(ap); 
 | 
  
 | 
    /* If we can't write it all, don't bother writing anything */ 
 | 
    if (unlikely(seq_buf_has_overflowed(&s->seq))) { 
 | 
        s->seq.len = save_len; 
 | 
        s->full = 1; 
 | 
    } 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_printf); 
 | 
  
 | 
/** 
 | 
 * trace_seq_bitmask - write a bitmask array in its ASCII representation 
 | 
 * @s:        trace sequence descriptor 
 | 
 * @maskp:    points to an array of unsigned longs that represent a bitmask 
 | 
 * @nmaskbits:    The number of bits that are valid in @maskp 
 | 
 * 
 | 
 * Writes a ASCII representation of a bitmask string into @s. 
 | 
 */ 
 | 
void trace_seq_bitmask(struct trace_seq *s, const unsigned long *maskp, 
 | 
              int nmaskbits) 
 | 
{ 
 | 
    unsigned int save_len = s->seq.len; 
 | 
  
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    seq_buf_printf(&s->seq, "%*pb", nmaskbits, maskp); 
 | 
  
 | 
    if (unlikely(seq_buf_has_overflowed(&s->seq))) { 
 | 
        s->seq.len = save_len; 
 | 
        s->full = 1; 
 | 
    } 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_bitmask); 
 | 
  
 | 
/** 
 | 
 * trace_seq_vprintf - sequence printing of trace information 
 | 
 * @s: trace sequence descriptor 
 | 
 * @fmt: printf format string 
 | 
 * 
 | 
 * The tracer may use either sequence operations or its own 
 | 
 * copy to user routines. To simplify formating of a trace 
 | 
 * trace_seq_printf is used to store strings into a special 
 | 
 * buffer (@s). Then the output may be either used by 
 | 
 * the sequencer or pulled into another buffer. 
 | 
 */ 
 | 
void trace_seq_vprintf(struct trace_seq *s, const char *fmt, va_list args) 
 | 
{ 
 | 
    unsigned int save_len = s->seq.len; 
 | 
  
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    seq_buf_vprintf(&s->seq, fmt, args); 
 | 
  
 | 
    /* If we can't write it all, don't bother writing anything */ 
 | 
    if (unlikely(seq_buf_has_overflowed(&s->seq))) { 
 | 
        s->seq.len = save_len; 
 | 
        s->full = 1; 
 | 
    } 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_vprintf); 
 | 
  
 | 
/** 
 | 
 * trace_seq_bprintf - Write the printf string from binary arguments 
 | 
 * @s: trace sequence descriptor 
 | 
 * @fmt: The format string for the @binary arguments 
 | 
 * @binary: The binary arguments for @fmt. 
 | 
 * 
 | 
 * When recording in a fast path, a printf may be recorded with just 
 | 
 * saving the format and the arguments as they were passed to the 
 | 
 * function, instead of wasting cycles converting the arguments into 
 | 
 * ASCII characters. Instead, the arguments are saved in a 32 bit 
 | 
 * word array that is defined by the format string constraints. 
 | 
 * 
 | 
 * This function will take the format and the binary array and finish 
 | 
 * the conversion into the ASCII string within the buffer. 
 | 
 */ 
 | 
void trace_seq_bprintf(struct trace_seq *s, const char *fmt, const u32 *binary) 
 | 
{ 
 | 
    unsigned int save_len = s->seq.len; 
 | 
  
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    seq_buf_bprintf(&s->seq, fmt, binary); 
 | 
  
 | 
    /* If we can't write it all, don't bother writing anything */ 
 | 
    if (unlikely(seq_buf_has_overflowed(&s->seq))) { 
 | 
        s->seq.len = save_len; 
 | 
        s->full = 1; 
 | 
        return; 
 | 
    } 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_bprintf); 
 | 
  
 | 
/** 
 | 
 * trace_seq_puts - trace sequence printing of simple string 
 | 
 * @s: trace sequence descriptor 
 | 
 * @str: simple string to record 
 | 
 * 
 | 
 * The tracer may use either the sequence operations or its own 
 | 
 * copy to user routines. This function records a simple string 
 | 
 * into a special buffer (@s) for later retrieval by a sequencer 
 | 
 * or other mechanism. 
 | 
 */ 
 | 
void trace_seq_puts(struct trace_seq *s, const char *str) 
 | 
{ 
 | 
    unsigned int len = strlen(str); 
 | 
  
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    if (len > TRACE_SEQ_BUF_LEFT(s)) { 
 | 
        s->full = 1; 
 | 
        return; 
 | 
    } 
 | 
  
 | 
    seq_buf_putmem(&s->seq, str, len); 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_puts); 
 | 
  
 | 
/** 
 | 
 * trace_seq_putc - trace sequence printing of simple character 
 | 
 * @s: trace sequence descriptor 
 | 
 * @c: simple character to record 
 | 
 * 
 | 
 * The tracer may use either the sequence operations or its own 
 | 
 * copy to user routines. This function records a simple charater 
 | 
 * into a special buffer (@s) for later retrieval by a sequencer 
 | 
 * or other mechanism. 
 | 
 */ 
 | 
void trace_seq_putc(struct trace_seq *s, unsigned char c) 
 | 
{ 
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    if (TRACE_SEQ_BUF_LEFT(s) < 1) { 
 | 
        s->full = 1; 
 | 
        return; 
 | 
    } 
 | 
  
 | 
    seq_buf_putc(&s->seq, c); 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_putc); 
 | 
  
 | 
/** 
 | 
 * trace_seq_putmem - write raw data into the trace_seq buffer 
 | 
 * @s: trace sequence descriptor 
 | 
 * @mem: The raw memory to copy into the buffer 
 | 
 * @len: The length of the raw memory to copy (in bytes) 
 | 
 * 
 | 
 * There may be cases where raw memory needs to be written into the 
 | 
 * buffer and a strcpy() would not work. Using this function allows 
 | 
 * for such cases. 
 | 
 */ 
 | 
void trace_seq_putmem(struct trace_seq *s, const void *mem, unsigned int len) 
 | 
{ 
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    if (len > TRACE_SEQ_BUF_LEFT(s)) { 
 | 
        s->full = 1; 
 | 
        return; 
 | 
    } 
 | 
  
 | 
    seq_buf_putmem(&s->seq, mem, len); 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_putmem); 
 | 
  
 | 
/** 
 | 
 * trace_seq_putmem_hex - write raw memory into the buffer in ASCII hex 
 | 
 * @s: trace sequence descriptor 
 | 
 * @mem: The raw memory to write its hex ASCII representation of 
 | 
 * @len: The length of the raw memory to copy (in bytes) 
 | 
 * 
 | 
 * This is similar to trace_seq_putmem() except instead of just copying the 
 | 
 * raw memory into the buffer it writes its ASCII representation of it 
 | 
 * in hex characters. 
 | 
 */ 
 | 
void trace_seq_putmem_hex(struct trace_seq *s, const void *mem, 
 | 
             unsigned int len) 
 | 
{ 
 | 
    unsigned int save_len = s->seq.len; 
 | 
  
 | 
    if (s->full) 
 | 
        return; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    /* Each byte is represented by two chars */ 
 | 
    if (len * 2 > TRACE_SEQ_BUF_LEFT(s)) { 
 | 
        s->full = 1; 
 | 
        return; 
 | 
    } 
 | 
  
 | 
    /* The added spaces can still cause an overflow */ 
 | 
    seq_buf_putmem_hex(&s->seq, mem, len); 
 | 
  
 | 
    if (unlikely(seq_buf_has_overflowed(&s->seq))) { 
 | 
        s->seq.len = save_len; 
 | 
        s->full = 1; 
 | 
        return; 
 | 
    } 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_putmem_hex); 
 | 
  
 | 
/** 
 | 
 * trace_seq_path - copy a path into the sequence buffer 
 | 
 * @s: trace sequence descriptor 
 | 
 * @path: path to write into the sequence buffer. 
 | 
 * 
 | 
 * Write a path name into the sequence buffer. 
 | 
 * 
 | 
 * Returns 1 if we successfully written all the contents to 
 | 
 *   the buffer. 
 | 
 * Returns 0 if we the length to write is bigger than the 
 | 
 *   reserved buffer space. In this case, nothing gets written. 
 | 
 */ 
 | 
int trace_seq_path(struct trace_seq *s, const struct path *path) 
 | 
{ 
 | 
    unsigned int save_len = s->seq.len; 
 | 
  
 | 
    if (s->full) 
 | 
        return 0; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    if (TRACE_SEQ_BUF_LEFT(s) < 1) { 
 | 
        s->full = 1; 
 | 
        return 0; 
 | 
    } 
 | 
  
 | 
    seq_buf_path(&s->seq, path, "\n"); 
 | 
  
 | 
    if (unlikely(seq_buf_has_overflowed(&s->seq))) { 
 | 
        s->seq.len = save_len; 
 | 
        s->full = 1; 
 | 
        return 0; 
 | 
    } 
 | 
  
 | 
    return 1; 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_path); 
 | 
  
 | 
/** 
 | 
 * trace_seq_to_user - copy the squence buffer to user space 
 | 
 * @s: trace sequence descriptor 
 | 
 * @ubuf: The userspace memory location to copy to 
 | 
 * @cnt: The amount to copy 
 | 
 * 
 | 
 * Copies the sequence buffer into the userspace memory pointed to 
 | 
 * by @ubuf. It starts from the last read position (@s->readpos) 
 | 
 * and writes up to @cnt characters or till it reaches the end of 
 | 
 * the content in the buffer (@s->len), which ever comes first. 
 | 
 * 
 | 
 * On success, it returns a positive number of the number of bytes 
 | 
 * it copied. 
 | 
 * 
 | 
 * On failure it returns -EBUSY if all of the content in the 
 | 
 * sequence has been already read, which includes nothing in the 
 | 
 * sequenc (@s->len == @s->readpos). 
 | 
 * 
 | 
 * Returns -EFAULT if the copy to userspace fails. 
 | 
 */ 
 | 
int trace_seq_to_user(struct trace_seq *s, char __user *ubuf, int cnt) 
 | 
{ 
 | 
    __trace_seq_init(s); 
 | 
    return seq_buf_to_user(&s->seq, ubuf, cnt); 
 | 
} 
 | 
EXPORT_SYMBOL_GPL(trace_seq_to_user); 
 | 
  
 | 
int trace_seq_hex_dump(struct trace_seq *s, const char *prefix_str, 
 | 
               int prefix_type, int rowsize, int groupsize, 
 | 
               const void *buf, size_t len, bool ascii) 
 | 
{ 
 | 
    unsigned int save_len = s->seq.len; 
 | 
  
 | 
    if (s->full) 
 | 
        return 0; 
 | 
  
 | 
    __trace_seq_init(s); 
 | 
  
 | 
    if (TRACE_SEQ_BUF_LEFT(s) < 1) { 
 | 
        s->full = 1; 
 | 
        return 0; 
 | 
    } 
 | 
  
 | 
    seq_buf_hex_dump(&(s->seq), prefix_str, 
 | 
           prefix_type, rowsize, groupsize, 
 | 
           buf, len, ascii); 
 | 
  
 | 
    if (unlikely(seq_buf_has_overflowed(&s->seq))) { 
 | 
        s->seq.len = save_len; 
 | 
        s->full = 1; 
 | 
        return 0; 
 | 
    } 
 | 
  
 | 
    return 1; 
 | 
} 
 | 
EXPORT_SYMBOL(trace_seq_hex_dump); 
 |