trunk/rdesktop/printer.c

#define MAX_PRINTERS    5

#include "rdesktop.h"

extern RDPDR_DEVICE g_rdpdr_device[];

PRINTER *
get_printer_data(HANDLE handle)
{
        int index;

        for (index = 0; index < RDPDR_MAX_DEVICES; index++)
        {
                if (handle == g_rdpdr_device[index].handle)
                        return (PRINTER *) g_rdpdr_device[index].pdevice_data;
        }
        return NULL;
}

int
printer_enum_devices(int *id, char *optarg)
{
        PRINTER *pprinter_data;

        char *pos = optarg;
        char *pos2;
        int count = 0;
        int already = 0;

        // we need to know how many printers we've already set up
        // supplied from other -r flags than this one.
        while (count < *id)
        {
                if (g_rdpdr_device[count].device_type == DEVICE_TYPE_PRINTER)
                        already++;
                count++;
        }

        count = 0;

        if (*optarg == ':')
                optarg++;

        while ((pos = next_arg(optarg, ',')) && *id < RDPDR_MAX_DEVICES)
        {
                pprinter_data = (PRINTER *) xmalloc(sizeof(PRINTER));

                strcpy(g_rdpdr_device[*id].name, "PRN");
                strcat(g_rdpdr_device[*id].name, ltoa(already + count + 1, 10));

                /* first printer is set as default printer */
                if ((already + count) == 0)
                        pprinter_data->default_printer = True;
                else
                        pprinter_data->default_printer = False;

                pos2 = next_arg(optarg, '=');
                if (*optarg == (char) 0x00)
                        pprinter_data->printer = "mydeskjet";   /* set default */
                else
                {
                        pprinter_data->printer = xmalloc(strlen(optarg) + 1);
                        strcpy(pprinter_data->printer, optarg);
                }

                if (!pos2 || (*pos2 == (char) 0x00))
                        pprinter_data->driver = "HP LaserJet IIIP";     /* no printer driver supplied set default */
                else
                {
                        pprinter_data->driver = xmalloc(strlen(pos2) + 1);
                        strcpy(pprinter_data->driver, pos2);
                }

                printf("PRINTER %s to %s driver %s\n", g_rdpdr_device[*id].name,
                       pprinter_data->printer, pprinter_data->driver);
                g_rdpdr_device[*id].device_type = DEVICE_TYPE_PRINTER;
                g_rdpdr_device[*id].pdevice_data = (void *) pprinter_data;
                count++;
                (*id)++;

                optarg = pos;
        }
        return count;
}

static NTSTATUS
printer_create(uint32 device_id, uint32 access, uint32 share_mode, uint32 disposition, uint32 flags,
               char *filename, HANDLE * handle)
{
        char cmd[256];
        PRINTER *pprinter_data;

        pprinter_data = (PRINTER *) g_rdpdr_device[device_id].pdevice_data;

        /* default printer name use default printer queue as well in unix */
        if (pprinter_data->printer == "mydeskjet")
        {
                pprinter_data->printer_fp = popen("lpr", "w");
        }
        else
        {
                sprintf(cmd, "lpr -P %s", pprinter_data->printer);
                pprinter_data->printer_fp = popen(cmd, "w");
        }

        g_rdpdr_device[device_id].handle = fileno(pprinter_data->printer_fp);
        *handle = g_rdpdr_device[device_id].handle;
        return STATUS_SUCCESS;
}

static NTSTATUS
printer_close(HANDLE handle)
{
        PRINTER *pprinter_data;

        pprinter_data = get_printer_data(handle);

        g_rdpdr_device[get_device_index(handle)].handle = 0;

        pclose(pprinter_data->printer_fp);

        return STATUS_SUCCESS;
}

static NTSTATUS
printer_write(HANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result)
{
        PRINTER *pprinter_data;

        pprinter_data = get_printer_data(handle);
        *result = length * fwrite(data, length, 1, pprinter_data->printer_fp);

        if (ferror(pprinter_data->printer_fp))
        {
                *result = 0;
                return STATUS_INVALID_HANDLE;
        }
        return STATUS_SUCCESS;
}

DEVICE_FNS printer_fns = {
        printer_create,
        printer_close,
        NULL,                   /* read */
        printer_write,
        NULL                    /* device_control */
};
1	n-ki	569	#define MAX_PRINTERS 5
2
3	matthewc	432	#include "rdesktop.h"
4
5	n-ki	569	extern RDPDR_DEVICE g_rdpdr_device[];
6	matthewc	432
7	astrand	580	PRINTER *
8			get_printer_data(HANDLE handle)
9	n-ki	569	{
10			int index;
11
12			for (index = 0; index < RDPDR_MAX_DEVICES; index++)
13			{
14			if (handle == g_rdpdr_device[index].handle)
15			return (PRINTER *) g_rdpdr_device[index].pdevice_data;
16			}
17			return NULL;
18			}
19
20			int
21			printer_enum_devices(int id, char optarg)
22			{
23			PRINTER *pprinter_data;
24
25			char *pos = optarg;
26			char *pos2;
27			int count = 0;
28			int already = 0;
29
30			// we need to know how many printers we've already set up
31			// supplied from other -r flags than this one.
32			while (count < *id)
33			{
34			if (g_rdpdr_device[count].device_type == DEVICE_TYPE_PRINTER)
35			already++;
36			count++;
37			}
38
39			count = 0;
40
41			if (*optarg == ':')
42			optarg++;
43
44			while ((pos = next_arg(optarg, ',')) && *id < RDPDR_MAX_DEVICES)
45			{
46			pprinter_data = (PRINTER *) xmalloc(sizeof(PRINTER));
47
48			strcpy(g_rdpdr_device[*id].name, "PRN");
49			strcat(g_rdpdr_device[*id].name, ltoa(already + count + 1, 10));
50
51			/* first printer is set as default printer */
52			if ((already + count) == 0)
53			pprinter_data->default_printer = True;
54			else
55			pprinter_data->default_printer = False;
56
57	n-ki	582	pos2 = next_arg(optarg, '=');
58	n-ki	569	if (*optarg == (char) 0x00)
59			pprinter_data->printer = "mydeskjet"; /* set default */
60			else
61			{
62			pprinter_data->printer = xmalloc(strlen(optarg) + 1);
63			strcpy(pprinter_data->printer, optarg);
64			}
65
66			if (!pos2 \|\| (*pos2 == (char) 0x00))
67			pprinter_data->driver = "HP LaserJet IIIP"; /* no printer driver supplied set default */
68			else
69			{
70			pprinter_data->driver = xmalloc(strlen(pos2) + 1);
71			strcpy(pprinter_data->driver, pos2);
72			}
73
74			printf("PRINTER %s to %s driver %s\n", g_rdpdr_device[*id].name,
75			pprinter_data->printer, pprinter_data->driver);
76			g_rdpdr_device[*id].device_type = DEVICE_TYPE_PRINTER;
77			g_rdpdr_device[id].pdevice_data = (void ) pprinter_data;
78			count++;
79			(*id)++;
80
81			optarg = pos;
82			}
83			return count;
84			}
85
86	matthewc	432	static NTSTATUS
87	n-ki	569	printer_create(uint32 device_id, uint32 access, uint32 share_mode, uint32 disposition, uint32 flags,
88			char filename, HANDLE handle)
89	matthewc	432	{
90	n-ki	569	char cmd[256];
91			PRINTER *pprinter_data;
92
93			pprinter_data = (PRINTER *) g_rdpdr_device[device_id].pdevice_data;
94
95			/* default printer name use default printer queue as well in unix */
96			if (pprinter_data->printer == "mydeskjet")
97			{
98			pprinter_data->printer_fp = popen("lpr", "w");
99			}
100			else
101			{
102			sprintf(cmd, "lpr -P %s", pprinter_data->printer);
103			pprinter_data->printer_fp = popen(cmd, "w");
104			}
105
106	stargo	570	g_rdpdr_device[device_id].handle = fileno(pprinter_data->printer_fp);
107	n-ki	569	*handle = g_rdpdr_device[device_id].handle;
108	matthewc	432	return STATUS_SUCCESS;
109			}
110
111			static NTSTATUS
112			printer_close(HANDLE handle)
113			{
114	n-ki	569	PRINTER *pprinter_data;
115
116			pprinter_data = get_printer_data(handle);
117
118			g_rdpdr_device[get_device_index(handle)].handle = 0;
119
120			pclose(pprinter_data->printer_fp);
121
122	matthewc	432	return STATUS_SUCCESS;
123			}
124
125			static NTSTATUS
126	n-ki	569	printer_write(HANDLE handle, uint8 * data, uint32 length, uint32 offset, uint32 * result)
127	matthewc	432	{
128	n-ki	569	PRINTER *pprinter_data;
129
130			pprinter_data = get_printer_data(handle);
131			result = length fwrite(data, length, 1, pprinter_data->printer_fp);
132
133			if (ferror(pprinter_data->printer_fp))
134			{
135			*result = 0;
136			return STATUS_INVALID_HANDLE;
137			}
138	matthewc	432	return STATUS_SUCCESS;
139			}
140
141	astrand	435	DEVICE_FNS printer_fns = {
142	matthewc	432	printer_create,
143			printer_close,
144	astrand	435	NULL, /* read */
145	matthewc	432	printer_write,
146	astrand	435	NULL /* device_control */
147	matthewc	432	};