Quelle  adi.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * Privileged ADI driver for sparc64
 *
 * Author: Tom Hromatka <tom.hromatka@oracle.com>
 */
#include <linux/kernel.h>
#include <linux/miscdevice.h>
#include <linux/module.h>
#include <linux/proc_fs.h>
#include <linux/slab.h>
#include <linux/uaccess.h>
#include <asm/asi.h>

#define MAX_BUF_SZ PAGE_SIZE

static int read_mcd_tag(unsigned long addr)
{
 long err;
 int ver;

 __asm__ __volatile__(
  "1: ldxa [%[addr]] %[asi], %[ver]\n"
  " mov 0, %[err]\n"
  "2:\n"
  " .section .fixup,#alloc,#execinstr\n"
  " .align 4\n"
  "3: sethi %%hi(2b), %%g1\n"
  " jmpl %%g1 + %%lo(2b), %%g0\n"
  " mov %[invalid], %[err]\n"
  " .previous\n"
  " .section __ex_table, \"a\"\n"
  " .align 4\n"
  " .word 1b, 3b\n"
  " .previous\n"
  : [ver] "=r" (ver), [err] "=r" (err)
  : [addr] "r"  (addr), [invalid] "i" (EFAULT),
    [asi] "i" (ASI_MCD_REAL)
  : "memory", "g1"
  );

 if (err)
  return -EFAULT;
 else
  return ver;
}

static ssize_t adi_read(struct file *file, char __user *buf,
   size_t count, loff_t *offp)
{
 size_t ver_buf_sz, bytes_read = 0;
 int ver_buf_idx = 0;
 loff_t offset;
 u8 *ver_buf;
 ssize_t ret;

 ver_buf_sz = min_t(size_t, count, MAX_BUF_SZ);
 ver_buf = kmalloc(ver_buf_sz, GFP_KERNEL);
 if (!ver_buf)
  return -ENOMEM;

 offset = (*offp) * adi_blksize();

 while (bytes_read < count) {
  ret = read_mcd_tag(offset);
  if (ret < 0)
   goto out;

  ver_buf[ver_buf_idx] = (u8)ret;
  ver_buf_idx++;
  offset += adi_blksize();

  if (ver_buf_idx >= ver_buf_sz) {
   if (copy_to_user(buf + bytes_read, ver_buf,
      ver_buf_sz)) {
    ret = -EFAULT;
    goto out;
   }

   bytes_read += ver_buf_sz;
   ver_buf_idx = 0;

   ver_buf_sz = min(count - bytes_read,
      (size_t)MAX_BUF_SZ);
  }
 }

 (*offp) += bytes_read;
 ret = bytes_read;
out:
 kfree(ver_buf);
 return ret;
}

static int set_mcd_tag(unsigned long addr, u8 ver)
{
 long err;

 __asm__ __volatile__(
  "1: stxa %[ver], [%[addr]] %[asi]\n"
  " mov 0, %[err]\n"
  "2:\n"
  " .section .fixup,#alloc,#execinstr\n"
  " .align 4\n"
  "3: sethi %%hi(2b), %%g1\n"
  " jmpl %%g1 + %%lo(2b), %%g0\n"
  " mov %[invalid], %[err]\n"
  " .previous\n"
  " .section __ex_table, \"a\"\n"
  " .align 4\n"
  " .word 1b, 3b\n"
  " .previous\n"
  : [err] "=r" (err)
  : [ver] "r" (ver), [addr] "r"  (addr),
    [invalid] "i" (EFAULT), [asi] "i" (ASI_MCD_REAL)
  : "memory", "g1"
  );

 if (err)
  return -EFAULT;
 else
  return ver;
}

static ssize_t adi_write(struct file *file, const char __user *buf,
    size_t count, loff_t *offp)
{
 size_t ver_buf_sz, bytes_written = 0;
 loff_t offset;
 u8 *ver_buf;
 ssize_t ret;
 int i;

 if (count <= 0)
  return -EINVAL;

 ver_buf_sz = min_t(size_t, count, MAX_BUF_SZ);
 ver_buf = kmalloc(ver_buf_sz, GFP_KERNEL);
 if (!ver_buf)
  return -ENOMEM;

 offset = (*offp) * adi_blksize();

 do {
  if (copy_from_user(ver_buf, &buf[bytes_written],
       ver_buf_sz)) {
   ret = -EFAULT;
   goto out;
  }

  for (i = 0; i < ver_buf_sz; i++) {
   ret = set_mcd_tag(offset, ver_buf[i]);
   if (ret < 0)
    goto out;

   offset += adi_blksize();
  }

  bytes_written += ver_buf_sz;
  ver_buf_sz = min(count - bytes_written, (size_t)MAX_BUF_SZ);
 } while (bytes_written < count);

 (*offp) += bytes_written;
 ret = bytes_written;
out:
 __asm__ __volatile__("membar #Sync");
 kfree(ver_buf);
 return ret;
}

static loff_t adi_llseek(struct file *file, loff_t offset, int whence)
{
 loff_t ret = -EINVAL;

 switch (whence) {
 case SEEK_END:
 case SEEK_DATA:
 case SEEK_HOLE:
  /* unsupported */
  return -EINVAL;
 case SEEK_CUR:
  if (offset == 0)
   return file->f_pos;

  offset += file->f_pos;
  break;
 case SEEK_SET:
  break;
 }

 if (offset != file->f_pos) {
  file->f_pos = offset;
  ret = offset;
 }

 return ret;
}

static const struct file_operations adi_fops = {
 .owner  = THIS_MODULE,
 .llseek  = adi_llseek,
 .read  = adi_read,
 .write  = adi_write,
 .fop_flags = FOP_UNSIGNED_OFFSET,
};

static struct miscdevice adi_miscdev = {
 .minor = MISC_DYNAMIC_MINOR,
 .name = KBUILD_MODNAME,
 .fops = &adi_fops,
};

static int __init adi_init(void)
{
 if (!adi_capable())
  return -EPERM;

 return misc_register(&adi_miscdev);
}

static void __exit adi_exit(void)
{
 misc_deregister(&adi_miscdev);
}

module_init(adi_init);
module_exit(adi_exit);

MODULE_AUTHOR("Tom Hromatka ");
MODULE_DESCRIPTION("Privileged interface to ADI");
MODULE_VERSION("1.0");
MODULE_LICENSE("GPL v2");