分享

[转]伪终端驱动程序代码--取自Suse 8.0 Enterprise Server

 techres 2012-07-07
准备自己写一个字符设备的驱动程序(伪终端的),网上找了很多资料,但是没有找到比较全的源代码。网上流传的比较多的一份BSD的代码是很不完全的。
下面的代码都是直接从Suse 8.0 Enterprise Server上找的源代码:/usr/src/linux/drivers/pty.c的代码我都贴在下面。另外还有 /usr/src/linux/include/linux/tty.h和/usr/src/linux/include/linux/tty_driver.h中有几个比较重要的结构,对pty.c这段代码的阅读会很帮助。我把这两个文件当作附件传上来。(如果不行就贴几个重要结构的源代码)
另外还有一篇文章,对阅读这些代码也有帮助,可以当作工具书来使用:
http://freeseagull./612391.html

我自己也准备写一个这样的驱动程序,如果做好了,代码可能不能全部放上来(涉及公司的版权),但是具体的执行步骤我会写上来。呵呵,前提是“如果做好了”,因为我不敢保证会做完,第一,是因为这个工作在公司安排给我的任务里优先级最低的,我现在有点空才做的;第二,呵呵,先不说吧,以后告诉大家。

部分比较重要的数据结构,如果缺了,大家给我留言,我以后尽量贴上来
/*
 * Where all of the state associated with a tty is kept while the tty
 * is open.  Since the termios state should be kept even if the tty
 * has been closed --- for things like the baud rate, etc --- it is
 * not stored here, but rather a pointer to the real state is stored
 * here.  Possible the winsize structure should have the same
 * treatment, but (1) the default 80x24 is usually right and (2) it's
 * most often used by a windowing system, which will set the correct
 * size each time the window is created or resized anyway.
 * IMPORTANT: since this structure is dynamically allocated, it must
 * be no larger than 4096 bytes.  Changing TTY_FLIPBUF_SIZE will change
 * the size of this structure, and it needs to be done with care.
 *                         - TYT, 9/14/92
 */
struct tty_struct {
    int    magic;
    struct tty_driver driver;
    struct tty_ldisc ldisc;
    struct termios *termios, *termios_locked;
    int pgrp;
    int session;
    kdev_t    device;
    unsigned long flags;
    int count;
    struct winsize winsize;
    unsigned char stopped:1, hw_stopped:1, flow_stopped:1, packet:1;
    unsigned char low_latency:1, warned:1;
    unsigned char ctrl_status;

    struct tty_struct *link;
    struct fasync_struct *fasync;
    struct tty_flip_buffer flip;
    int max_flip_cnt;
    int alt_speed;        /* For magic substitution of 38400 bps */
    wait_queue_head_t write_wait;
    wait_queue_head_t read_wait;
    struct tq_struct tq_hangup;
    void *disc_data;
    void *driver_data;
    struct list_head tty_files;

#define N_TTY_BUF_SIZE 4096
   
    /*
     * The following is data for the N_TTY line discipline.  For
     * historical reasons, this is included in the tty structure.
     */
    unsigned int column;
    unsigned char lnext:1, erasing:1, raw:1, real_raw:1, icanon:1;
    unsigned char closing:1;
    unsigned short minimum_to_wake;
    unsigned overrun_time;
    int num_overrun;
    unsigned long process_char_map[256/(8*sizeof(unsigned long))];
    char *read_buf;
    int read_head;
    int read_tail;
    int read_cnt;
    unsigned long read_flags[N_TTY_BUF_SIZE/(8*sizeof(unsigned long))];
    int canon_data;
    unsigned long canon_head;
    unsigned int canon_column;
    struct semaphore atomic_read;
    struct semaphore atomic_write;
    spinlock_t read_lock;
    /* If the tty has a pending do_SAK, queue it here - akpm */
    struct tq_struct SAK_tq;
#ifdef CONFIG_TTY_LOG
        int log_fd;
#endif
};

struct tty_driver {
    int    magic;        /* magic number for this structure */
    const char    *driver_name;
    const char    *name;
    int    name_base;    /* offset of printed name */
    short    major;        /* major device number */
    short    minor_start;    /* start of minor device number*/
    short    num;        /* number of devices */
    short    type;        /* type of tty driver */
    short    subtype;    /* subtype of tty driver */
    struct termios init_termios; /* Initial termios */
    int    flags;        /* tty driver flags */
    int    *refcount;    /* for loadable tty drivers */
    struct proc_dir_entry *proc_entry; /* /proc fs entry */
    struct tty_driver *other; /* only used for the PTY driver */

    /*
     * Pointer to the tty data structures
     */
    struct tty_struct **table;
    struct termios **termios;
    struct termios **termios_locked;
    void *driver_state;    /* only used for the PTY driver */
   
    /*
     * Interface routines from the upper tty layer to the tty
     * driver.
     */
    int  (*open)(struct tty_struct * tty, struct file * filp);
    void (*close)(struct tty_struct * tty, struct file * filp);
    int  (*write)(struct tty_struct * tty, int from_user,
              const unsigned char *buf, int count);
    void (*put_char)(struct tty_struct *tty, unsigned char ch);
    void (*flush_chars)(struct tty_struct *tty);
    int  (*write_room)(struct tty_struct *tty);
    int  (*chars_in_buffer)(struct tty_struct *tty);
    int  (*ioctl)(struct tty_struct *tty, struct file * file,
            unsigned int cmd, unsigned long arg);
    void (*set_termios)(struct tty_struct *tty, struct termios * old);
    void (*throttle)(struct tty_struct * tty);
    void (*unthrottle)(struct tty_struct * tty);
    void (*stop)(struct tty_struct *tty);
    void (*start)(struct tty_struct *tty);
    void (*hangup)(struct tty_struct *tty);
    void (*break_ctl)(struct tty_struct *tty, int state);
    void (*flush_buffer)(struct tty_struct *tty);
    void (*set_ldisc)(struct tty_struct *tty);
    void (*wait_until_sent)(struct tty_struct *tty, int timeout);
    void (*send_xchar)(struct tty_struct *tty, char ch);
    int (*read_proc)(char *page, char **start, off_t off,
              int count, int *eof, void *data);
    int (*write_proc)(struct file *file, const char *buffer,
              unsigned long count, void *data);

    /*
     * linked list pointers
     */
    struct tty_driver *next;
    struct tty_driver *prev;
};

pty.c的源代码:


/*
 *  linux/drivers/char/pty.c
 *
 *  Copyright (C) 1991, 1992  Linus Torvalds
 *
 *  Added support for a Unix98-style ptmx device.
 *    -- C. Scott Ananian <cananian@alumni.princeton.edu>, 14-Jan-1998
 *  Added TTY_DO_WRITE_WAKEUP to enable n_tty to send POLL_OUT to
 *      waiting writers -- Sapan Bhatia <sapan@corewars.org>
 *
 *
 */

#include <linux/config.h>
#include <linux/module.h>    /* For EXPORT_SYMBOL */

#include <linux/errno.h>
#include <linux/sched.h>
#include <linux/interrupt.h>
#include <linux/tty.h>
#include <linux/tty_flip.h>
#include <linux/fcntl.h>
#include <linux/string.h>
#include <linux/major.h>
#include <linux/mm.h>
#include <linux/init.h>
#include <linux/devfs_fs_kernel.h>

#include <asm/uaccess.h>
#include <asm/system.h>
#include <asm/bitops.h>

#define BUILDING_PTY_C 1
#include <linux/devpts_fs.h>

struct pty_struct {
    int    magic;
    wait_queue_head_t open_wait;
};

#define PTY_MAGIC 0x5001

static struct tty_driver pty_driver, pty_slave_driver;
static int pty_refcount;

/* Note: one set of tables for BSD and one for Unix98 */
static struct tty_struct *pty_table[NR_PTYS];
static struct termios *pty_termios[NR_PTYS];
static struct termios *pty_termios_locked[NR_PTYS];
static struct tty_struct *ttyp_table[NR_PTYS];
static struct termios *ttyp_termios[NR_PTYS];
static struct termios *ttyp_termios_locked[NR_PTYS];
static struct pty_struct pty_state[NR_PTYS];

#ifdef CONFIG_UNIX98_PTYS
/* These are global because they are accessed in tty_io.c */
struct tty_driver ptm_driver[UNIX98_NR_MAJORS];
struct tty_driver pts_driver[UNIX98_NR_MAJORS];

static struct tty_struct *ptm_table[UNIX98_NR_MAJORS][NR_PTYS];
static struct termios *ptm_termios[UNIX98_NR_MAJORS][NR_PTYS];
static struct termios *ptm_termios_locked[UNIX98_NR_MAJORS][NR_PTYS];
static struct tty_struct *pts_table[UNIX98_NR_MAJORS][NR_PTYS];
static struct termios *pts_termios[UNIX98_NR_MAJORS][NR_PTYS];
static struct termios *pts_termios_locked[UNIX98_NR_MAJORS][NR_PTYS];
static struct pty_struct ptm_state[UNIX98_NR_MAJORS][NR_PTYS];
#endif

#define MIN(a,b)    ((a) < (b) ? (a) : (b))

static void pty_close(struct tty_struct * tty, struct file * filp)
{
    if (!tty)
        return;
    if (tty->driver.subtype == PTY_TYPE_MASTER) {
        if (tty->count > 1)
            printk("master pty_close: count = %d!!\n", tty->count);
    } else {
        if (tty->count > 2)
            return;
    }
    wake_up_interruptible(&tty->read_wait);
    wake_up_interruptible(&tty->write_wait);
    tty->packet = 0;
    if (!tty->link)
        return;
    tty->link->packet = 0;
    wake_up_interruptible(&tty->link->read_wait);
    wake_up_interruptible(&tty->link->write_wait);
    set_bit(TTY_OTHER_CLOSED, &tty->link->flags);
    if (tty->driver.subtype == PTY_TYPE_MASTER) {
        set_bit(TTY_OTHER_CLOSED, &tty->flags);
#ifdef CONFIG_UNIX98_PTYS
        {
            unsigned int major = MAJOR(tty->device) - UNIX98_PTY_MASTER_MAJOR;
            if ( major < UNIX98_NR_MAJORS ) {
                devpts_pty_kill( MINOR(tty->device)
              - tty->driver.minor_start + tty->driver.name_base );
            }
        }
#endif
        tty_unregister_devfs (&tty->link->driver, MINOR (tty->device));
        tty_vhangup(tty->link);
    }
}

/*
 * The unthrottle routine is called by the line discipline to signal
 * that it can receive more characters.  For PTY's, the TTY_THROTTLED
 * flag is always set, to force the line discipline to always call the
 * unthrottle routine when there are fewer than TTY_THRESHOLD_UNTHROTTLE
 * characters in the queue.  This is necessary since each time this
 * happens, we need to wake up any sleeping processes that could be
 * (1) trying to send data to the pty, or (2) waiting in wait_until_sent()
 * for the pty buffer to be drained.
 */
static void pty_unthrottle(struct tty_struct * tty)
{
    struct tty_struct *o_tty = tty->link;

    if (!o_tty)
        return;

    if ((o_tty->flags & (1 << TTY_DO_WRITE_WAKEUP)) &&
        o_tty->ldisc.write_wakeup)
        (o_tty->ldisc.write_wakeup)(o_tty);
    wake_up_interruptible(&o_tty->write_wait);
    set_bit(TTY_THROTTLED, &tty->flags);
}

/*
 * WSH 05/24/97: modified to
 *   (1) use space in tty->flip instead of a shared temp buffer
 *     The flip buffers aren't being used for a pty, so there's lots
 *     of space available.  The buffer is protected by a per-pty
 *     semaphore that should almost never come under contention.
 *   (2) avoid redundant copying for cases where count >> receive_room
 * N.B. Calls from user space may now return an error code instead of
 * a count.
 */
static int pty_write(struct tty_struct * tty, int from_user,
               const unsigned char *buf, int count)
{
    struct tty_struct *to = tty->link;
    int    c=0, n, room;
    char    *temp_buffer;

    if (!to || tty->stopped)
        return 0;

    if (from_user) {
        down(&tty->flip.pty_sem);
        temp_buffer = &tty->flip.char_buf[0];
        while (count > 0) {
            /* check space so we don't copy needlessly */
            n = to->ldisc.receive_room(to);
            if (n > count)
                n = count;
            if (!n) break;

            n  = MIN(n, PTY_BUF_SIZE);
            n -= copy_from_user(temp_buffer, buf, n);
            if (!n) {
                if (!c)
                    c = -EFAULT;
                break;
            }

            /* check again in case the buffer filled up */
            room = to->ldisc.receive_room(to);
            if (n > room)
                n = room;
            if (!n) break;
            buf   += n;
            c     += n;
            count -= n;
            to->ldisc.receive_buf(to, temp_buffer, 0, n);
        }
        up(&tty->flip.pty_sem);
    } else {
        c = to->ldisc.receive_room(to);
        if (c > count)
            c = count;
        to->ldisc.receive_buf(to, buf, 0, c);
    }
    
    return c;
}

static int pty_write_room(struct tty_struct *tty)
{
    struct tty_struct *to = tty->link;

    if (!to || tty->stopped)
        return 0;

    return to->ldisc.receive_room(to);
}

/*
 *    WSH 05/24/97:  Modified for asymmetric MASTER/SLAVE behavior
 *    The chars_in_buffer() value is used by the ldisc select() function
 *    to hold off writing when chars_in_buffer > WAKEUP_CHARS (== 256).
 *    The pty driver chars_in_buffer() Master/Slave must behave differently:
 *
 *      The Master side needs to allow typed-ahead commands to accumulate
 *      while being canonicalized, so we report "our buffer" as empty until
 *    some threshold is reached, and then report the count. (Any count >
 *    WAKEUP_CHARS is regarded by select() as "full".)  To avoid deadlock
 *    the count returned must be 0 if no canonical data is available to be
 *    read. (The N_TTY ldisc.chars_in_buffer now knows this.)
 *  
 *    The Slave side passes all characters in raw mode to the Master side's
 *    buffer where they can be read immediately, so in this case we can
 *    return the true count in the buffer.
 */
static int pty_chars_in_buffer(struct tty_struct *tty)
{
    struct tty_struct *to = tty->link;
    int count;

    if (!to || !to->ldisc.chars_in_buffer)
        return 0;

    /* The ldisc must report 0 if no characters available to be read */
    count = to->ldisc.chars_in_buffer(to);

    if (tty->driver.subtype == PTY_TYPE_SLAVE) return count;

    /* Master side driver ... if the other side's read buffer is less than
     * half full, return 0 to allow writers to proceed; otherwise return
     * the count.  This leaves a comfortable margin to avoid overflow,
     * and still allows half a buffer's worth of typed-ahead commands.
     */
    return ((count < N_TTY_BUF_SIZE/2) ? 0 : count);
}

/*
 * Return the device number of a Unix98 PTY (only!).  This lets us open a
 * master pty with the multi-headed ptmx device, then find out which
 * one we got after it is open, with an ioctl.
 */
#ifdef CONFIG_UNIX98_PTYS
static int pty_get_device_number(struct tty_struct *tty, unsigned int *value)
{
    unsigned int result = MINOR(tty->device)
        - tty->driver.minor_start + tty->driver.name_base;
    return put_user(result, value);
}
#endif

/* Set the lock flag on a pty */
static int pty_set_lock(struct tty_struct *tty, int * arg)
{
    int val;
    if (get_user(val,arg))
        return -EFAULT;
    if (val)
        set_bit(TTY_PTY_LOCK, &tty->flags);
    else
        clear_bit(TTY_PTY_LOCK, &tty->flags);
    return 0;
}

static int pty_bsd_ioctl(struct tty_struct *tty, struct file *file,
            unsigned int cmd, unsigned long arg)
{
    if (!tty) {
        printk("pty_ioctl called with NULL tty!\n");
        return -EIO;
    }
    switch(cmd) {
    case TIOCSPTLCK: /* Set PT Lock (disallow slave open) */
        return pty_set_lock(tty, (int *) arg);
    }
    return -ENOIOCTLCMD;
}

#ifdef CONFIG_UNIX98_PTYS
static int pty_unix98_ioctl(struct tty_struct *tty, struct file *file,
                unsigned int cmd, unsigned long arg)
{
    if (!tty) {
        printk("pty_unix98_ioctl called with NULL tty!\n");
        return -EIO;
    }
    switch(cmd) {
    case TIOCGPTN: /* Get PT Number */
        return pty_get_device_number(tty, (unsigned int *)arg);
    }

    return pty_bsd_ioctl(tty,file,cmd,arg);
}
#endif

static void pty_flush_buffer(struct tty_struct *tty)
{
    struct tty_struct *to = tty->link;
    
    if (!to)
        return;
    
    if (to->ldisc.flush_buffer)
        to->ldisc.flush_buffer(to);
    
    if (to->packet) {
        tty->ctrl_status |= TIOCPKT_FLUSHWRITE;
        wake_up_interruptible(&to->read_wait);
    }
}

static int pty_open(struct tty_struct *tty, struct file * filp)
{
    int    retval;
    int    line;
    struct    pty_struct *pty;

    retval = -ENODEV;
    if (!tty || !tty->link)
        goto out;
    line = MINOR(tty->device) - tty->driver.minor_start;
    if ((line < 0) || (line >= NR_PTYS))
        goto out;
    pty = (struct pty_struct *)(tty->driver.driver_state) + line;
    tty->driver_data = pty;

    retval = -EIO;
    if (test_bit(TTY_OTHER_CLOSED, &tty->flags))
        goto out;
    if (test_bit(TTY_PTY_LOCK, &tty->link->flags))
        goto out;
    if (tty->link->count != 1)
        goto out;

    clear_bit(TTY_OTHER_CLOSED, &tty->link->flags);
    wake_up_interruptible(&pty->open_wait);
    set_bit(TTY_THROTTLED, &tty->flags);
    set_bit(TTY_DO_WRITE_WAKEUP, &tty->flags);

    /*  Register a slave for the master  */
    if (tty->driver.major == PTY_MASTER_MAJOR)
        tty_register_devfs(&tty->link->driver,
                   DEVFS_FL_CURRENT_OWNER | DEVFS_FL_WAIT,
                   tty->link->driver.minor_start +
                   MINOR(tty->device)-tty->driver.minor_start);
    retval = 0;
out:
    return retval;
}

static void pty_set_termios(struct tty_struct *tty, struct termios *old_termios)
{
        tty->termios->c_cflag &= ~(CSIZE | PARENB);
        tty->termios->c_cflag |= (CS8 | CREAD);
}

int __init pty_init(void)
{
    int i;

    /* Traditional BSD devices */

    memset(&pty_state, 0, sizeof(pty_state));
    for (i = 0; i < NR_PTYS; i++)
        init_waitqueue_head(&pty_state[i].open_wait);
    memset(&pty_driver, 0, sizeof(struct tty_driver));
    pty_driver.magic = TTY_DRIVER_MAGIC;
    pty_driver.driver_name = "pty_master";
#ifdef CONFIG_DEVFS_FS
    pty_driver.name = "pty/m%d";
#else
    pty_driver.name = "pty";
#endif
    pty_driver.major = PTY_MASTER_MAJOR;
    pty_driver.minor_start = 0;
    pty_driver.num = NR_PTYS;
    pty_driver.type = TTY_DRIVER_TYPE_PTY;
    pty_driver.subtype = PTY_TYPE_MASTER;
    pty_driver.init_termios = tty_std_termios;
    pty_driver.init_termios.c_iflag = 0;
    pty_driver.init_termios.c_oflag = 0;
    pty_driver.init_termios.c_cflag = B38400 | CS8 | CREAD;
    pty_driver.init_termios.c_lflag = 0;
    pty_driver.flags = TTY_DRIVER_RESET_TERMIOS | TTY_DRIVER_REAL_RAW;
    pty_driver.refcount = &pty_refcount;
    pty_driver.table = pty_table;
    pty_driver.termios = pty_termios;
    pty_driver.termios_locked = pty_termios_locked;
    pty_driver.driver_state = pty_state;
    pty_driver.other = &pty_slave_driver;

    pty_driver.open = pty_open;
    pty_driver.close = pty_close;
    pty_driver.write = pty_write;
    pty_driver.write_room = pty_write_room;
    pty_driver.flush_buffer = pty_flush_buffer;
    pty_driver.chars_in_buffer = pty_chars_in_buffer;
    pty_driver.unthrottle = pty_unthrottle;
    pty_driver.set_termios = pty_set_termios;

    pty_slave_driver = pty_driver;
    pty_slave_driver.driver_name = "pty_slave";
    pty_slave_driver.proc_entry = 0;
#ifdef CONFIG_DEVFS_FS
    pty_slave_driver.name = "pty/s%d";
#else
    pty_slave_driver.name = "ttyp";
#endif
    pty_slave_driver.subtype = PTY_TYPE_SLAVE;
    pty_slave_driver.major = PTY_SLAVE_MAJOR;
    pty_slave_driver.minor_start = 0;
    pty_slave_driver.init_termios = tty_std_termios;
    pty_slave_driver.init_termios.c_cflag = B38400 | CS8 | CREAD;
    /* Slave ptys are registered when their corresponding master pty
     * is opened, and unregistered when the pair is closed.
     */
    pty_slave_driver.flags |= TTY_DRIVER_NO_DEVFS;
    pty_slave_driver.table = ttyp_table;
    pty_slave_driver.termios = ttyp_termios;
    pty_slave_driver.termios_locked = ttyp_termios_locked;
    pty_slave_driver.driver_state = pty_state;
    pty_slave_driver.other = &pty_driver;

    if (tty_register_driver(&pty_driver))
        panic("Couldn't register pty driver");
    if (tty_register_driver(&pty_slave_driver))
        panic("Couldn't register pty slave driver");

    /*
     * only the master pty gets this ioctl (which is why we
     * assign it here, instead of up with the rest of the
     * pty_driver initialization. <cananian@alumni.princeton.edu>
     */
    pty_driver.ioctl = pty_bsd_ioctl;

    /* Unix98 devices */
#ifdef CONFIG_UNIX98_PTYS
    devfs_mk_dir (NULL, "pts", NULL);
    printk("pty: %d Unix98 ptys configured\n", UNIX98_NR_MAJORS*NR_PTYS);
    for ( i = 0 ; i < UNIX98_NR_MAJORS ; i++ ) {
        int j;

        ptm_driver[i] = pty_driver;
        ptm_driver[i].name = "ptm";
        ptm_driver[i].proc_entry = 0;
        ptm_driver[i].major = UNIX98_PTY_MASTER_MAJOR+i;
        ptm_driver[i].minor_start = 0;
        ptm_driver[i].name_base = i*NR_PTYS;
        ptm_driver[i].num = NR_PTYS;
        ptm_driver[i].other = &pts_driver[i];
        ptm_driver[i].flags |= TTY_DRIVER_NO_DEVFS;
        ptm_driver[i].table = ptm_table[i];
        ptm_driver[i].termios = ptm_termios[i];
        ptm_driver[i].termios_locked = ptm_termios_locked[i];
        ptm_driver[i].driver_state = ptm_state[i];

        for (j = 0; j < NR_PTYS; j++)
            init_waitqueue_head(&ptm_state[i][j].open_wait);
        
        pts_driver[i] = pty_slave_driver;
#ifdef CONFIG_DEVFS_FS
        pts_driver[i].name = "pts/%d";
#else
        pts_driver[i].name = "pts";
#endif
        pts_driver[i].proc_entry = 0;
        pts_driver[i].major = UNIX98_PTY_SLAVE_MAJOR+i;
        pts_driver[i].minor_start = 0;
        pts_driver[i].name_base = i*NR_PTYS;
        pts_driver[i].num = ptm_driver[i].num;
        pts_driver[i].other = &ptm_driver[i];
        pts_driver[i].table = pts_table[i];
        pts_driver[i].termios = pts_termios[i];
        pts_driver[i].termios_locked = pts_termios_locked[i];
        pts_driver[i].driver_state = ptm_state[i];
        
        ptm_driver[i].ioctl = pty_unix98_ioctl;
        
        if (tty_register_driver(&ptm_driver[i]))
            panic("Couldn't register Unix98 ptm driver major %d",
                  ptm_driver[i].major);
        if (tty_register_driver(&pts_driver[i]))
            panic("Couldn't register Unix98 pts driver major %d",
                  pts_driver[i].major);
    }
#endif
    return 0;
}

    本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多