1 #! /usr/bin/env python

     2 """Test program for the fcntl C module.

     3    OS/2+EMX doesn't support the file locking operations.

     4    Roger E. Masse

     5 """

     6 import struct

     7 import fcntl

     8 import os, sys

     9 from test.test_support import verbose, TESTFN

    10 

    11 filename = TESTFN

    12 

    13 try:

    14     os.O_LARGEFILE

    15 except AttributeError:

    16     start_len = "ll"

    17 else:

    18     start_len = "qq"

    19 

    20 if sys.platform.startswith('atheos'):

    21     start_len = "qq"

    22 

    23 if sys.platform in ('netbsd1', 'netbsd2', 'netbsd3',

    24                     'Darwin1.2', 'darwin',

    25                     'freebsd2', 'freebsd3', 'freebsd4', 'freebsd5',

    26                     'freebsd6', 'freebsd7',

    27                     'bsdos2', 'bsdos3', 'bsdos4',

    28                     'openbsd', 'openbsd2', 'openbsd3', 'openbsd4'):

    29     if struct.calcsize('l') == 8:

    30         off_t = 'l'

    31         pid_t = 'i'

    32     else:

    33         off_t = 'lxxxx'

    34         pid_t = 'l'

    35     lockdata = struct.pack(off_t+off_t+pid_t+'hh', 0, 0, 0, fcntl.F_WRLCK, 0)

    36 elif sys.platform in ['aix3', 'aix4', 'hp-uxB', 'unixware7']:

    37     lockdata = struct.pack('hhlllii', fcntl.F_WRLCK, 0, 0, 0, 0, 0, 0)

    38 elif sys.platform in ['os2emx']:

    39     lockdata = None

    40 else:

    41     lockdata = struct.pack('hh'+start_len+'hh', fcntl.F_WRLCK, 0, 0, 0, 0, 0)

    42 if lockdata:

    43     if verbose:

    44         print 'struct.pack: ', repr(lockdata)

    45 

    46 # the example from the library docs

    47 f = open(filename, 'w')

    48 rv = fcntl.fcntl(f.fileno(), fcntl.F_SETFL, os.O_NONBLOCK)

    49 if verbose:

    50     print 'Status from fcntl with O_NONBLOCK: ', rv

    51 

    52 if sys.platform not in ['os2emx']:

    53     rv = fcntl.fcntl(f.fileno(), fcntl.F_SETLKW, lockdata)

    54     if verbose:

    55         print 'String from fcntl with F_SETLKW: ', repr(rv)

    56 

    57 f.close()

    58 os.unlink(filename)

    59 

    60 

    61 # Again, but pass the file rather than numeric descriptor:

    62 f = open(filename, 'w')

    63 rv = fcntl.fcntl(f, fcntl.F_SETFL, os.O_NONBLOCK)

    64 

    65 if sys.platform not in ['os2emx']:

    66     rv = fcntl.fcntl(f, fcntl.F_SETLKW, lockdata)

    67 

    68 f.close()

    69 os.unlink(filename)