spoolss.py

来自「samba-3.0.22.tar.gz 编译smb服务器的源码」· Python 代码 · 共 289 行 · 第 1/2 页

#!/usr/bin/python

import re
import comfychair, stf
from samba import spoolss

class PrintServerTest(comfychair.TestCase):
    """An abstract class requiring a print server."""
    def setUp(self):
        # TODO: create a test printer
        self.server = stf.get_server(platform = "nt")
        self.require(self.server != None, "print server required")
        # TODO: remove hardcoded printer name
        self.printername = "p"
        self.uncname = "\\\\%s\\%s" % \
                           (self.server["hostname"], self.printername)

class W2kPrintServerTest(comfychair.TestCase):
    """An abstract class requiring a print server."""
    def setUp(self):
        # TODO: create a test printer
        self.server = stf.get_server(platform = "nt5")
        self.require(self.server != None, "print server required")
        # TODO: remove hardcoded printer name
        self.printername = "p"
        self.uncname = "\\\\%s\\%s" % \
                           (self.server["hostname"], self.printername)

class CredentialTest(PrintServerTest):
    """An class that calls a function with various sets of credentials."""
    def runTest(self):

        bad_user_creds = {"username": "spotty",
                          "domain": "dog",
                          "password": "bone"}
                        
        cases = ((self.server["administrator"], "Admin credentials", 1),
                 (bad_user_creds,               "Bad credentials",   0))

        # TODO: add unpriv user case

        for creds, testname, result in cases:
            try:
                self.runTestArg(creds)
            except:
                if result:
                    import traceback
                    traceback.print_exc()
                    self.fail("rpc with creds %s failed when it "
                              "should have suceeded" % creds)
                return

            if not result:
                self.fail("rpc with creds %s suceeded when it should "
                          "have failed" % creds)
        
class ArgTestServer(PrintServerTest):
    """Test a RPC that takes a UNC print server name."""
    def runTest(self):

        # List of test cases, %s substituted for server name

        cases = (("",             "No server name",          0),
                 ("\\\\%s",       "Valid server name",       1),
                 ("\\%s",         "Invalid unc server name", 0),
                 ("\\\\%s__",     "Invalid unc server name", 0))

        for unc, testname, result in cases:
            unc = re.sub("%s", self.server["hostname"], unc)
            try:
                self.runTestArg(unc)
            except:
                if result:
                    self.fail("rpc(\"%s\") failed when it should have "
                              "suceeded" % unc)
                return

            if not result:
                # Suceeded when we should have failed
                self.fail("rpc(\"%s\") suceeded when it should have "
                          "failed" % unc)

class ArgTestServerAndPrinter(ArgTestServer):
    """Test a RPC that takes a UNC print server or UNC printer name."""
    def runTest(self):

        ArgTestServer.runTest(self)
        
        # List of test cases, %s substituted for server name, %p substituted
        # for printer name.

        cases = (("\\\\%s\\%p",   "Valid server and printer name",      1),
                 ("\\\\%s\\%p__", "Valid server, invalid printer name", 0),
                 ("\\\\%s__\\%p", "Invalid server, valid printer name", 0))

        for unc, testname, result in cases:
            unc = re.sub("%s", self.server["hostname"], unc)
            unc = re.sub("%p", self.printername, unc)
            try:
                self.runTestArg(unc)
            except:
                if result:
                    self.fail("openprinter(\"%s\") failed when it should have "
                              "suceeded" % unc)
                return

            if not result:
                # Suceeded when we should have failed
                self.fail("openprinter(\"%s\") suceeded when it should have "
                          "failed" % unc)

class OpenPrinterArg(ArgTestServerAndPrinter):
    """Test the OpenPrinter RPC with combinations of valid and invalid
    server and printer names."""
    def runTestArg(self, unc):
        spoolss.openprinter(unc)

class OpenPrinterCred(CredentialTest):
    """Test opening printer with good and bad credentials."""
    def runTestArg(self, creds):
        spoolss.openprinter(self.uncname, creds = creds)

class ClosePrinter(PrintServerTest):
    """Test the ClosePrinter RPC on a printer handle."""
    def runTest(self):
        hnd = spoolss.openprinter(self.uncname)
        spoolss.closeprinter(hnd)
        
class ClosePrinterServer(PrintServerTest):
    """Test the ClosePrinter RPC on a print server handle."""
    def runTest(self):
        hnd = spoolss.openprinter("\\\\%s" % self.server["hostname"])
        spoolss.closeprinter(hnd)
        
class GetPrinterInfo(PrintServerTest):
    """Retrieve printer info at various levels."""

    # Sample printer data

    sample_info = {
        0: {'printer_errors': 0, 'unknown18': 0, 'unknown13': 0, 'unknown26': 0, 'cjobs': 0, 'unknown11': 0, 'server_name': '\\\\win2kdc1', 'total_pages': 0, 'unknown15': 586, 'unknown16': 0, 'month': 2, 'unknown20': 0, 'second': 23, 'unknown22': 983040, 'unknown25': 0, 'total_bytes': 0, 'unknown27': 0, 'year': 2003, 'build_version': 2195, 'unknown28': 0, 'global_counter': 4, 'day': 13, 'minute': 53, 'total_jobs': 0, 'unknown29': 1114112, 'name': '\\\\win2kdc1\\p', 'hour': 2, 'level': 0, 'c_setprinter': 0, 'change_id': 522454169, 'major_version': 5, 'unknown23': 15, 'day_of_week': 4, 'unknown14': 1, 'session_counter': 2, 'status': 1, 'unknown7': 1, 'unknown8': 0, 'unknown9': 0, 'milliseconds': 421, 'unknown24': 0},
        1: {'comment': "I'm a teapot!", 'level': 1, 'flags': 8388608, 'name': '\\\\win2kdc1\\p', 'description': '\\\\win2kdc1\\p,HP LaserJet 4,Canberra office'},
        2: {'comment': "I'm a teapot!", 'status': 1, 'print_processor': 'WinPrint', 'until_time': 0, 'share_name': 'p', 'start_time': 0, 'device_mode': {'icm_method': 1, 'bits_per_pel': 0, 'log_pixels': 0, 'orientation': 1, 'panning_width': 0, 'color': 2, 'pels_width': 0, 'print_quality': 600, 'driver_version': 24, 'display_flags': 0, 'y_resolution': 600, 'media_type': 0, 'display_frequency': 0, 'icm_intent': 0, 'pels_height': 0, 'reserved1': 0, 'size': 220, 'scale': 100, 'dither_type': 0, 'panning_height': 0, 'default_source': 7, 'duplex': 1, 'fields': 16131, 'spec_version': 1025, 'copies': 1, 'device_name': '\\\\win2kdc1\\p', 'paper_size': 1, 'paper_length': 0, 'private': 'private', 'collate': 0, 'paper_width': 0, 'form_name': 'Letter', 'reserved2': 0, 'tt_option': 0}, 'port_name': 'LPT1:', 'sepfile': '', 'parameters': '', 'security_descriptor': {'group_sid': 'S-1-5-21-1606980848-1677128483-854245398-513', 'sacl': None, 'dacl': {'ace_list': [{'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-32-544'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-5-32-544'}, {'flags': 0, 'type': 0, 'mask': 131080, 'trustee': 'S-1-5-21-1606980848-1677128483-854245398-1121'}, {'flags': 10, 'type': 0, 'mask': 131072, 'trustee': 'S-1-3-0'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-3-0'}, {'flags': 0, 'type': 0, 'mask': 131080, 'trustee': 'S-1-5-21-1606980848-1677128483-854245398-1124'}, {'flags': 0, 'type': 0, 'mask': 131080, 'trustee': 'S-1-1-0'}, {'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-32-550'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-5-32-550'}, {'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-32-549'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-5-32-549'}, {'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-21-1606980848-1677128483-854245398-1106'}], 'revision': 2}, 'owner_sid': 'S-1-5-32-544', 'revision': 1}, 'name': '\\\\win2kdc1\\p', 'server_name': '\\\\win2kdc1', 'level': 2, 'datatype': 'RAW', 'cjobs': 0, 'average_ppm': 0, 'priority': 1, 'driver_name': 'HP LaserJet 4', 'location': 'Canberra office', 'attributes': 8776, 'default_priority': 0},
        3: {'flags': 4, 'security_descriptor': {'group_sid': 'S-1-5-21-1606980848-1677128483-854245398-513', 'sacl': None, 'dacl': {'ace_list': [{'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-32-544'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-5-32-544'}, {'flags': 0, 'type': 0, 'mask': 131080, 'trustee': 'S-1-5-21-1606980848-1677128483-854245398-1121'}, {'flags': 10, 'type': 0, 'mask': 131072, 'trustee': 'S-1-3-0'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-3-0'}, {'flags': 0, 'type': 0, 'mask': 131080, 'trustee': 'S-1-5-21-1606980848-1677128483-854245398-1124'}, {'flags': 0, 'type': 0, 'mask': 131080, 'trustee': 'S-1-1-0'}, {'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-32-550'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-5-32-550'}, {'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-32-549'}, {'flags': 9, 'type': 0, 'mask': 983056, 'trustee': 'S-1-5-32-549'}, {'flags': 0, 'type': 0, 'mask': 983052, 'trustee': 'S-1-5-21-1606980848-1677128483-854245398-1106'}], 'revision': 2}, 'owner_sid': 'S-1-5-32-544', 'revision': 1}, 'level': 3}
        }