1

import socket

     2

     3

import _utmp

     4

     5

class WtmpEntry (object) :

     6

    _ATTRS = (

     7

        "type", "pid", "line", "id", "user", "host", "exit", "session", "tv", "addr_v6"

     8

    )

     9

    10

    def __str__ (self) :

    11

        return " ".join("%s=%s" % (key, getattr(self, key)) for key in self._ATTRS)

    12

    13

def read_entry (file) :

    14

    bytes = file.read(_utmp.size())

    15

    16

    if not bytes :

    17

        return

    18

    19

    result = _utmp.parse(bytes)

    20

    21

    wtmp = WtmpEntry()

    22

    23

    for name, field in zip(wtmp._ATTRS, result) :

    24

        if isinstance(field, str) and name not in ("addr_v6", ) :

    25

            field = field.rstrip("\0")

    26

    27

        setattr(wtmp, name, field)

    28

    29

    wtmp.addr_v6 = socket.inet_ntop(socket.AF_INET6, wtmp.addr_v6)

    30

    31

    return wtmp

    32

    33

def read_entries (file) :

    34

    while True :

    35

        wtmp = read_entry(file)

    36

    37

        if wtmp :

    38

            yield wtmp

    39

        else :

    40

            return

    41

    42

def test_main () :

    43

    fh = open("/var/log/wtmp", "r")

    44

    45

    for item in read_entries(fh) :

    46

        print item

    47

    48

if __name__ == '__main__' :

    49

    test_main()

    50