#include "includes.h"
#include "utils/net.h"
struct table_node {
const char *long_archi;
const char *short_archi;
int version;
};
static const struct table_node archi_table[]= {
{"Windows 4.0", "WIN40", 0 },
{"Windows NT x86", "W32X86", 2 },
{"Windows NT x86", "W32X86", 3 },
{"Windows NT R4000", "W32MIPS", 2 },
{"Windows NT Alpha_AXP", "W32ALPHA", 2 },
{"Windows NT PowerPC", "W32PPC", 2 },
{"Windows IA64", "IA64", 3 },
{"Windows x64", "x64", 3 },
{NULL, "", -1 }
};
char *get_sec_mask_str(uint32 type)
{
static fstring typestr="";
typestr[0] = 0;
if (type & GENERIC_ALL_ACCESS)
fstrcat(typestr, "Generic all access ");
if (type & GENERIC_EXECUTE_ACCESS)
fstrcat(typestr, "Generic execute access ");
if (type & GENERIC_WRITE_ACCESS)
fstrcat(typestr, "Generic write access ");
if (type & GENERIC_READ_ACCESS)
fstrcat(typestr, "Generic read access ");
if (type & MAXIMUM_ALLOWED_ACCESS)
fstrcat(typestr, "MAXIMUM_ALLOWED_ACCESS ");
if (type & SYSTEM_SECURITY_ACCESS)
fstrcat(typestr, "SYSTEM_SECURITY_ACCESS ");
if (type & SYNCHRONIZE_ACCESS)
fstrcat(typestr, "SYNCHRONIZE_ACCESS ");
if (type & WRITE_OWNER_ACCESS)
fstrcat(typestr, "WRITE_OWNER_ACCESS ");
if (type & WRITE_DAC_ACCESS)
fstrcat(typestr, "WRITE_DAC_ACCESS ");
if (type & READ_CONTROL_ACCESS)
fstrcat(typestr, "READ_CONTROL_ACCESS ");
if (type & DELETE_ACCESS)
fstrcat(typestr, "DELETE_ACCESS ");
printf("\t\tSpecific bits: 0x%lx\n", (unsigned long)type&SPECIFIC_RIGHTS_MASK);
return typestr;
}
void display_sec_ace(SEC_ACE *ace)
{
fstring sid_str;
printf("\tACE\n\t\ttype: ");
switch (ace->type) {
case SEC_ACE_TYPE_ACCESS_ALLOWED:
printf("ACCESS ALLOWED");
break;
case SEC_ACE_TYPE_ACCESS_DENIED:
printf("ACCESS DENIED");
break;
case SEC_ACE_TYPE_SYSTEM_AUDIT:
printf("SYSTEM AUDIT");
break;
case SEC_ACE_TYPE_SYSTEM_ALARM:
printf("SYSTEM ALARM");
break;
default:
printf("????");
break;
}
printf(" (%d) flags: %d\n", ace->type, ace->flags);
printf("\t\tPermissions: 0x%x: %s\n", ace->info.mask, get_sec_mask_str(ace->info.mask));
sid_to_string(sid_str, &ace->trustee);
printf("\t\tSID: %s\n\n", sid_str);
}
void display_sec_acl(SEC_ACL *sec_acl)
{
int i;
printf("\tACL\tNum ACEs:\t%d\trevision:\t%x\n",
sec_acl->num_aces, sec_acl->revision);
printf("\t---\n");
if (sec_acl->size != 0 && sec_acl->num_aces != 0)
for (i = 0; i < sec_acl->num_aces; i++)
display_sec_ace(&sec_acl->ace[i]);
}
void display_sec_desc(SEC_DESC *sec)
{
fstring sid_str;
if (sec == NULL)
return;
if (sec->sacl) {
printf("SACL\n");
display_sec_acl(sec->sacl);
}
if (sec->dacl) {
printf("DACL\n");
display_sec_acl(sec->dacl);
}
if (sec->owner_sid) {
sid_to_string(sid_str, sec->owner_sid);
printf("\tOwner SID:\t%s\n", sid_str);
}
if (sec->grp_sid) {
sid_to_string(sid_str, sec->grp_sid);
printf("\tParent SID:\t%s\n", sid_str);
}
}
static void display_print_driver_3(DRIVER_INFO_3 *i1)
{
fstring name = "";
fstring architecture = "";
fstring driverpath = "";
fstring datafile = "";
fstring configfile = "";
fstring helpfile = "";
fstring dependentfiles = "";
fstring monitorname = "";
fstring defaultdatatype = "";
int length=0;
BOOL valid = True;
if (i1 == NULL)
return;
rpcstr_pull(name, i1->name.buffer, sizeof(name), -1, STR_TERMINATE);
rpcstr_pull(architecture, i1->architecture.buffer, sizeof(architecture), -1, STR_TERMINATE);
rpcstr_pull(driverpath, i1->driverpath.buffer, sizeof(driverpath), -1, STR_TERMINATE);
rpcstr_pull(datafile, i1->datafile.buffer, sizeof(datafile), -1, STR_TERMINATE);
rpcstr_pull(configfile, i1->configfile.buffer, sizeof(configfile), -1, STR_TERMINATE);
rpcstr_pull(helpfile, i1->helpfile.buffer, sizeof(helpfile), -1, STR_TERMINATE);
rpcstr_pull(monitorname, i1->monitorname.buffer, sizeof(monitorname), -1, STR_TERMINATE);
rpcstr_pull(defaultdatatype, i1->defaultdatatype.buffer, sizeof(defaultdatatype), -1, STR_TERMINATE);
d_printf ("Printer Driver Info 3:\n");
d_printf ("\tVersion: [%x]\n", i1->version);
d_printf ("\tDriver Name: [%s]\n",name);
d_printf ("\tArchitecture: [%s]\n", architecture);
d_printf ("\tDriver Path: [%s]\n", driverpath);
d_printf ("\tDatafile: [%s]\n", datafile);
d_printf ("\tConfigfile: [%s]\n", configfile);
d_printf ("\tHelpfile: [%s]\n\n", helpfile);
while (valid) {
rpcstr_pull(dependentfiles, i1->dependentfiles+length, sizeof(dependentfiles), -1, STR_TERMINATE);
length+=strlen(dependentfiles)+1;
if (strlen(dependentfiles) > 0) {
d_printf ("\tDependentfiles: [%s]\n", dependentfiles);
} else {
valid = False;
}
}
printf ("\n");
d_printf ("\tMonitorname: [%s]\n", monitorname);
d_printf ("\tDefaultdatatype: [%s]\n\n", defaultdatatype);
return;
}
static void display_reg_value(pstring subkey, REGISTRY_VALUE value)
{
pstring text;
switch(value.type) {
case REG_DWORD:
d_printf("\t[%s:%s]: REG_DWORD: 0x%08x\n", subkey, value.valuename,
*((uint32 *) value.data_p));
break;
case REG_SZ:
rpcstr_pull(text, value.data_p, sizeof(text), value.size,
STR_TERMINATE);
d_printf("\t[%s:%s]: REG_SZ: %s\n", subkey, value.valuename, text);
break;
case REG_BINARY:
d_printf("\t[%s:%s]: REG_BINARY: unknown length value not displayed\n",
subkey, value.valuename);
break;
case REG_MULTI_SZ: {
uint16 *curstr = (uint16 *) value.data_p;
uint8 *start = value.data_p;
d_printf("\t[%s:%s]: REG_MULTI_SZ:\n", subkey, value.valuename);
while ((*curstr != 0) &&
((uint8 *) curstr < start + value.size)) {
rpcstr_pull(text, curstr, sizeof(text), -1,
STR_TERMINATE);
d_printf("%s\n", text);
curstr += strlen(text) + 1;
}
}
break;
default:
d_printf("\t%s: unknown type %d\n", value.valuename, value.type);
}
}
static NTSTATUS
net_copy_fileattr(TALLOC_CTX *mem_ctx,
struct cli_state *cli_share_src,
struct cli_state *cli_share_dst,
char *src_name, char *dst_name,
BOOL copy_acls, BOOL copy_attrs,
BOOL copy_timestamps, BOOL is_file)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
int fnum_src = 0;
int fnum_dst = 0;
SEC_DESC *sd = NULL;
uint16 attr;
time_t atime, ctime, mtime;
if (!copy_timestamps && !copy_acls && !copy_attrs)
return NT_STATUS_OK;
DEBUGADD(3,("opening %s %s on originating server\n",
is_file?"file":"dir", src_name));
fnum_src = cli_nt_create(cli_share_src, src_name, READ_CONTROL_ACCESS);
if (fnum_src == -1) {
DEBUGADD(0,("cannot open %s %s on originating server %s\n",
is_file?"file":"dir", src_name, cli_errstr(cli_share_src)));
nt_status = cli_nt_error(cli_share_src);
goto out;
}
if (copy_acls) {
sd = cli_query_secdesc(cli_share_src, fnum_src, mem_ctx);
if (!sd) {
DEBUG(0,("failed to get security descriptor: %s\n",
cli_errstr(cli_share_src)));
nt_status = cli_nt_error(cli_share_src);
goto out;
}
if (opt_verbose && SAMBA_DEBUGLEVEL >= 3)
display_sec_desc(sd);
}
if (copy_attrs || copy_timestamps) {
if (!cli_getattrE(cli_share_src, fnum_src, &attr, NULL,
&ctime, &atime, &mtime)) {
DEBUG(0,("failed to get file-attrs: %s\n",
cli_errstr(cli_share_src)));
nt_status = cli_nt_error(cli_share_src);
goto out;
}
}
fnum_dst = cli_nt_create(cli_share_dst, dst_name, WRITE_DAC_ACCESS | WRITE_OWNER_ACCESS);
if (fnum_dst == -1) {
DEBUG(0,("failed to open %s on the destination server: %s: %s\n",
is_file?"file":"dir", dst_name, cli_errstr(cli_share_dst)));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
if (copy_timestamps) {
if (!cli_setattrE(cli_share_dst, fnum_dst, ctime, atime, mtime)) {
DEBUG(0,("failed to set file-attrs (timestamps): %s\n",
cli_errstr(cli_share_dst)));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
}
if (copy_acls) {
if (!cli_set_secdesc(cli_share_dst, fnum_dst, sd)) {
DEBUG(0,("could not set secdesc on %s: %s\n",
dst_name, cli_errstr(cli_share_dst)));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
}
if (copy_attrs) {
if (!cli_setatr(cli_share_dst, dst_name, attr, 0)) {
DEBUG(0,("failed to set file-attrs: %s\n",
cli_errstr(cli_share_dst)));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
}
if (!cli_close(cli_share_src, fnum_src)) {
d_printf("could not close %s on originating server: %s\n",
is_file?"file":"dir", cli_errstr(cli_share_src));
nt_status = cli_nt_error(cli_share_src);
goto out;
}
if (!cli_close(cli_share_dst, fnum_dst)) {
d_printf("could not close %s on destination server: %s\n",
is_file?"file":"dir", cli_errstr(cli_share_dst));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
nt_status = NT_STATUS_OK;
out:
if (fnum_src)
cli_close(cli_share_src, fnum_src);
if (fnum_dst)
cli_close(cli_share_dst, fnum_dst);
return nt_status;
}
NTSTATUS net_copy_file(TALLOC_CTX *mem_ctx,
struct cli_state *cli_share_src,
struct cli_state *cli_share_dst,
char *src_name, char *dst_name,
BOOL copy_acls, BOOL copy_attrs,
BOOL copy_timestamps, BOOL is_file)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
int fnum_src = 0;
int fnum_dst = 0;
static int io_bufsize = 64512;
int read_size = io_bufsize;
char *data = NULL;
off_t start = 0;
off_t nread = 0;
if (!src_name || !dst_name)
goto out;
if (cli_share_dst == NULL || cli_share_dst == NULL)
goto out;
DEBUGADD(3,("opening %s %s on originating server\n",
is_file ? "file":"dir", src_name));
if (is_file)
fnum_src = cli_open(cli_share_src, src_name, O_RDONLY, DENY_NONE);
else
fnum_src = cli_nt_create(cli_share_src, src_name, READ_CONTROL_ACCESS);
if (fnum_src == -1) {
DEBUGADD(0,("cannot open file %s on originating server %s\n",
src_name, cli_errstr(cli_share_src)));
nt_status = cli_nt_error(cli_share_src);
goto out;
}
if (is_file) {
DEBUGADD(3,("opening file %s on destination server\n", dst_name));
fnum_dst = cli_open(cli_share_dst, dst_name,
O_RDWR|O_CREAT|O_TRUNC, DENY_NONE);
if (fnum_dst == -1) {
DEBUGADD(1,("cannot create file %s on destination server: %s\n",
dst_name, cli_errstr(cli_share_dst)));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
if (!(data = (char *)SMB_MALLOC(read_size))) {
d_printf("malloc fail for size %d\n", read_size);
nt_status = NT_STATUS_NO_MEMORY;
goto out;
}
}
if (opt_verbose) {
d_printf("copying [\\\\%s\\%s%s] => [\\\\%s\\%s%s] "
"%s ACLs and %s DOS Attributes %s\n",
cli_share_src->desthost, cli_share_src->share, src_name,
cli_share_dst->desthost, cli_share_dst->share, dst_name,
copy_acls ? "with" : "without",
copy_attrs ? "with" : "without",
copy_timestamps ? "(preserving timestamps)" : "" );
}
while (is_file) {
int n, ret;
n = cli_read(cli_share_src, fnum_src, data, nread + start,
read_size);
if (n <= 0)
break;
ret = cli_write(cli_share_dst, fnum_dst, 0, data,
nread + start, n);
if (n != ret) {
d_printf("Error writing file: %s\n",
cli_errstr(cli_share_dst));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
nread += n;
}
if (!is_file && !cli_chkpath(cli_share_dst, dst_name)) {
DEBUGADD(3,("creating dir %s on the destination server\n",
dst_name));
if (!cli_mkdir(cli_share_dst, dst_name)) {
DEBUG(0,("cannot create directory %s: %s\n",
dst_name, cli_errstr(cli_share_dst)));
nt_status = NT_STATUS_NO_SUCH_FILE;
}
if (!cli_chkpath(cli_share_dst, dst_name)) {
d_printf("cannot check for directory %s: %s\n",
dst_name, cli_errstr(cli_share_dst));
goto out;
}
}
if (!cli_close(cli_share_src, fnum_src)) {
d_printf("could not close file on originating server: %s\n",
cli_errstr(cli_share_src));
nt_status = cli_nt_error(cli_share_src);
goto out;
}
if (is_file && !cli_close(cli_share_dst, fnum_dst)) {
d_printf("could not close file on destination server: %s\n",
cli_errstr(cli_share_dst));
nt_status = cli_nt_error(cli_share_dst);
goto out;
}
nt_status = net_copy_fileattr(mem_ctx, cli_share_src, cli_share_dst,
src_name, dst_name, copy_acls,
copy_attrs, copy_timestamps, is_file);
if (!NT_STATUS_IS_OK(nt_status))
goto out;
nt_status = NT_STATUS_OK;
out:
if (fnum_src)
cli_close(cli_share_src, fnum_src);
if (fnum_dst)
cli_close(cli_share_dst, fnum_dst);
SAFE_FREE(data);
return nt_status;
}
static NTSTATUS net_copy_driverfile(TALLOC_CTX *mem_ctx,
struct cli_state *cli_share_src,
struct cli_state *cli_share_dst,
char *file, const char *short_archi) {
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
const char *p;
char *src_name;
char *dst_name;
fstring version;
fstring filename;
fstring tok;
p = file;
while (next_token(&p, tok, "\\", sizeof(tok))) {
if (strequal(tok, short_archi)) {
next_token(&p, version, "\\", sizeof(version));
next_token(&p, filename, "\\", sizeof(filename));
}
}
if (asprintf(&src_name, "\\%s\\%s\\%s", short_archi, version, filename) < 0 )
return NT_STATUS_NO_MEMORY;
if (asprintf(&dst_name, "\\%s\\%s", short_archi, filename) < 0 )
return NT_STATUS_NO_MEMORY;
nt_status = net_copy_file(mem_ctx, cli_share_src, cli_share_dst,
src_name, dst_name, False, False, False, True);
if (!NT_STATUS_IS_OK(nt_status))
goto out;
nt_status = NT_STATUS_OK;
out:
SAFE_FREE(src_name);
SAFE_FREE(dst_name);
return nt_status;
}
static NTSTATUS
check_arch_dir(struct cli_state *cli_share, const char *short_archi)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
char *dir;
if (asprintf(&dir, "\\%s", short_archi) < 0) {
return NT_STATUS_NO_MEMORY;
}
DEBUG(10,("creating print-driver dir for architecture: %s\n",
short_archi));
if (!cli_mkdir(cli_share, dir)) {
DEBUG(1,("cannot create directory %s: %s\n",
dir, cli_errstr(cli_share)));
nt_status = NT_STATUS_NO_SUCH_FILE;
}
if (!cli_chkpath(cli_share, dir)) {
d_printf("cannot check %s: %s\n",
dir, cli_errstr(cli_share));
goto out;
}
nt_status = NT_STATUS_OK;
out:
SAFE_FREE(dir);
return nt_status;
}
static NTSTATUS
copy_print_driver_3(TALLOC_CTX *mem_ctx,
struct cli_state *cli_share_src,
struct cli_state *cli_share_dst,
const char *short_archi, DRIVER_INFO_3 *i1)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
int length = 0;
BOOL valid = True;
fstring name = "";
fstring driverpath = "";
fstring datafile = "";
fstring configfile = "";
fstring helpfile = "";
fstring dependentfiles = "";
if (i1 == NULL)
return nt_status;
rpcstr_pull(name, i1->name.buffer, sizeof(name), -1, STR_TERMINATE);
rpcstr_pull(driverpath, i1->driverpath.buffer, sizeof(driverpath), -1, STR_TERMINATE);
rpcstr_pull(datafile, i1->datafile.buffer, sizeof(datafile), -1, STR_TERMINATE);
rpcstr_pull(configfile, i1->configfile.buffer, sizeof(configfile), -1, STR_TERMINATE);
rpcstr_pull(helpfile, i1->helpfile.buffer, sizeof(helpfile), -1, STR_TERMINATE);
if (opt_verbose)
d_printf("copying driver: [%s], for architecture: [%s], version: [%d]\n",
name, short_archi, i1->version);
nt_status = net_copy_driverfile(mem_ctx, cli_share_src, cli_share_dst,
driverpath, short_archi);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
nt_status = net_copy_driverfile(mem_ctx, cli_share_src, cli_share_dst,
datafile, short_archi);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
nt_status = net_copy_driverfile(mem_ctx, cli_share_src, cli_share_dst,
configfile, short_archi);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
nt_status = net_copy_driverfile(mem_ctx, cli_share_src, cli_share_dst,
helpfile, short_archi);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
while (valid) {
rpcstr_pull(dependentfiles, i1->dependentfiles+length, sizeof(dependentfiles), -1, STR_TERMINATE);
length+=strlen(dependentfiles)+1;
if (strlen(dependentfiles) > 0) {
nt_status = net_copy_driverfile(mem_ctx,
cli_share_src, cli_share_dst,
dependentfiles, short_archi);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
} else {
valid = False;
}
}
return NT_STATUS_OK;
}
static BOOL
net_spoolss_enum_printers(struct cli_state *cli, TALLOC_CTX *mem_ctx,
char *name, uint32 flags, uint32 level,
uint32 *num_printers, PRINTER_INFO_CTR *ctr)
{
WERROR result;
uint32 needed;
result = cli_spoolss_enum_printers(
cli, mem_ctx, 0, &needed, name, flags,
level, num_printers, ctr);
if (W_ERROR_V(result) == W_ERROR_V(WERR_INSUFFICIENT_BUFFER))
result = cli_spoolss_enum_printers(
cli, mem_ctx, needed, NULL, name, flags,
level, num_printers, ctr);
if (!W_ERROR_IS_OK(result)) {
printf("cannot enum printers: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_open_printer_ex(struct cli_state *cli, TALLOC_CTX *mem_ctx,
const char *printername, uint32 access_required,
const char *username, POLICY_HND *hnd)
{
WERROR result;
fstring servername, printername2;
slprintf(servername, sizeof(servername)-1, "\\\\%s", cli->desthost);
fstrcpy(printername2, servername);
fstrcat(printername2, "\\");
fstrcat(printername2, printername);
DEBUG(10,("connecting to: %s as %s for %s and access: %x\n",
servername, username, printername2, access_required));
result = cli_spoolss_open_printer_ex(cli, mem_ctx, printername2,
"", access_required,
servername, username, hnd);
if (W_ERROR_V(result) == W_ERROR_V(WERR_ACCESS_DENIED)) {
d_printf("no access to printer [%s] on [%s] for user [%s] granted\n",
printername2, servername, username);
return False;
}
if (!W_ERROR_IS_OK(result)) {
d_printf("cannot open printer %s on server %s: %s\n",
printername2, servername, dos_errstr(result));
return False;
}
DEBUG(2,("got printer handle for printer: %s, server: %s\n",
printername2, servername));
return True;
}
static BOOL
net_spoolss_getprinter(struct cli_state *cli, TALLOC_CTX *mem_ctx,
POLICY_HND *hnd, uint32 level,
PRINTER_INFO_CTR *ctr)
{
WERROR result;
uint32 needed;
result = cli_spoolss_getprinter(cli,
mem_ctx, 0, &needed, hnd, level, ctr);
if (W_ERROR_V(result) == W_ERROR_V(WERR_INSUFFICIENT_BUFFER))
result = cli_spoolss_getprinter(cli,
mem_ctx, needed, NULL, hnd, level, ctr);
if (!W_ERROR_IS_OK(result)) {
printf("cannot get printer-info: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_setprinter(struct cli_state *cli, TALLOC_CTX *mem_ctx,
POLICY_HND *hnd, uint32 level,
PRINTER_INFO_CTR *ctr)
{
WERROR result;
result = cli_spoolss_setprinter(cli, mem_ctx, hnd, level, ctr, 0);
if (!W_ERROR_IS_OK(result)) {
printf("cannot set printer-info: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_setprinterdata(struct cli_state *cli, TALLOC_CTX *mem_ctx,
POLICY_HND *hnd, REGISTRY_VALUE *value)
{
WERROR result;
result = cli_spoolss_setprinterdata(cli, mem_ctx, hnd, value);
if (!W_ERROR_IS_OK(result)) {
printf ("unable to set printerdata: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_enumprinterkey(struct cli_state *cli, TALLOC_CTX *mem_ctx,
POLICY_HND *hnd, const char *keyname,
uint16 **keylist)
{
WERROR result;
uint32 needed, len;
result = cli_spoolss_enumprinterkey(
cli, mem_ctx, 0, &needed, hnd, keyname, NULL, NULL);
if (W_ERROR_V(result) == W_ERROR_V(WERR_MORE_DATA))
result = cli_spoolss_enumprinterkey(
cli, mem_ctx, needed, NULL, hnd, keyname, keylist,
&len);
if (!W_ERROR_IS_OK(result)) {
printf("enumprinterkey failed: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_enumprinterdataex(struct cli_state *cli, TALLOC_CTX *mem_ctx,
uint32 offered,
POLICY_HND *hnd, const char *keyname,
REGVAL_CTR *ctr)
{
WERROR result;
uint32 needed;
result = cli_spoolss_enumprinterdataex(
cli, mem_ctx, 0, &needed, hnd, keyname, NULL);
if (W_ERROR_V(result) == W_ERROR_V(WERR_MORE_DATA))
result = cli_spoolss_enumprinterdataex(
cli, mem_ctx, needed, NULL, hnd, keyname, ctr);
if (!W_ERROR_IS_OK(result)) {
printf("enumprinterdataex failed: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_setprinterdataex(struct cli_state *cli, TALLOC_CTX *mem_ctx,
POLICY_HND *hnd, char *keyname,
REGISTRY_VALUE *value)
{
WERROR result;
result = cli_spoolss_setprinterdataex(cli, mem_ctx, hnd,
keyname, value);
if (!W_ERROR_IS_OK(result)) {
printf("could not set printerdataex: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_enumforms(struct cli_state *cli, TALLOC_CTX *mem_ctx,
POLICY_HND *hnd, int level, uint32 *num_forms,
FORM_1 **forms)
{
WERROR result;
uint32 needed;
result = cli_spoolss_enumforms(
cli, mem_ctx, 0, &needed, hnd, level, num_forms, forms);
if (W_ERROR_V(result) == W_ERROR_V(WERR_INSUFFICIENT_BUFFER))
result = cli_spoolss_enumforms(
cli, mem_ctx, needed, NULL, hnd, level,
num_forms, forms);
if (!W_ERROR_IS_OK(result)) {
printf("could not enum forms: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_enumprinterdrivers (struct cli_state *cli, TALLOC_CTX *mem_ctx,
uint32 level, const char *env,
uint32 *num_drivers,
PRINTER_DRIVER_CTR *ctr)
{
WERROR result;
uint32 needed;
result = cli_spoolss_enumprinterdrivers(
cli, mem_ctx, 0, &needed, level,
env, num_drivers, ctr);
if (W_ERROR_V(result) == W_ERROR_V(WERR_INSUFFICIENT_BUFFER))
result = cli_spoolss_enumprinterdrivers(
cli, mem_ctx, needed, NULL, level,
env, num_drivers, ctr);
if (!W_ERROR_IS_OK(result)) {
printf("cannot enum drivers: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
net_spoolss_getprinterdriver(struct cli_state *cli,
TALLOC_CTX *mem_ctx,
POLICY_HND *hnd, uint32 level,
const char *env, int version,
PRINTER_DRIVER_CTR *ctr)
{
WERROR result;
uint32 needed;
result = cli_spoolss_getprinterdriver(
cli, mem_ctx, 0, &needed, hnd, level,
env, version, ctr);
if (W_ERROR_V(result) == W_ERROR_V(WERR_INSUFFICIENT_BUFFER))
result = cli_spoolss_getprinterdriver(
cli, mem_ctx, needed, NULL, hnd, level,
env, version, ctr);
if (!W_ERROR_IS_OK(result)) {
DEBUG(1,("cannot get driver (for architecture: %s): %s\n",
env, dos_errstr(result)));
if (W_ERROR_V(result) != W_ERROR_V(WERR_UNKNOWN_PRINTER_DRIVER) &&
W_ERROR_V(result) != W_ERROR_V(WERR_INVALID_ENVIRONMENT)) {
printf("cannot get driver: %s\n", dos_errstr(result));
}
return False;
}
return True;
}
static BOOL
net_spoolss_addprinterdriver(struct cli_state *cli,
TALLOC_CTX *mem_ctx, uint32 level,
PRINTER_DRIVER_CTR *ctr)
{
WERROR result;
result = cli_spoolss_addprinterdriver(cli, mem_ctx, level, ctr);
if (W_ERROR_V(result) == W_ERROR_V(WERR_ACCESS_DENIED)) {
printf("You are not allowed to add drivers\n");
return False;
}
if (!W_ERROR_IS_OK(result)) {
printf("cannot add driver: %s\n", dos_errstr(result));
return False;
}
return True;
}
static BOOL
get_printer_info(struct cli_state *cli, TALLOC_CTX *mem_ctx,
int level, int argc, const char **argv,
uint32 *num_printers, PRINTER_INFO_CTR *ctr)
{
char *sharename;
POLICY_HND hnd;
if (argc == 0) {
if (!net_spoolss_enum_printers(cli, mem_ctx, NULL,
PRINTER_ENUM_LOCAL|PRINTER_ENUM_SHARED,
level, num_printers, ctr))
return False;
goto out;
}
sharename = SMB_STRDUP(argv[0]);
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd))
return False;
if (!net_spoolss_getprinter(cli, mem_ctx, &hnd, level, ctr)) {
cli_spoolss_close_printer(cli, mem_ctx, &hnd);
return False;
}
cli_spoolss_close_printer(cli, mem_ctx, &hnd);
*num_printers = 1;
out:
DEBUG(3,("got %d printers\n", *num_printers));
return True;
}
NTSTATUS rpc_printer_list_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i, num_printers;
uint32 level = 2;
pstring printername, sharename;
PRINTER_INFO_CTR ctr;
printf("listing printers\n");
if (!get_printer_info(cli, mem_ctx, level, argc, argv, &num_printers, &ctr))
return nt_status;
for (i = 0; i < num_printers; i++) {
rpcstr_pull(printername, ctr.printers_2[i].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, ctr.printers_2[i].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
d_printf("printer %d: %s, shared as: %s\n",
i+1, printername, sharename);
}
return NT_STATUS_OK;
}
NTSTATUS rpc_printer_driver_list_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i;
uint32 level = 3;
PRINTER_DRIVER_CTR drv_ctr_enum;
int d;
ZERO_STRUCT(drv_ctr_enum);
printf("listing printer-drivers\n");
for (i=0; archi_table[i].long_archi!=NULL; i++) {
int num_drivers;
if (!net_spoolss_enumprinterdrivers(cli, mem_ctx, level,
archi_table[i].long_archi,
&num_drivers, &drv_ctr_enum)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
if (num_drivers == 0) {
d_printf ("no drivers found on server for architecture: [%s].\n",
archi_table[i].long_archi);
continue;
}
d_printf("got %d printer-drivers for architecture: [%s]\n",
num_drivers, archi_table[i].long_archi);
for (d = 0; d < num_drivers; d++) {
display_print_driver_3(&(drv_ctr_enum.info3[d]));
}
}
nt_status = NT_STATUS_OK;
done:
return nt_status;
}
static NTSTATUS rpc_printer_publish_internals_args(struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv, uint32 action)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i, num_printers;
uint32 level = 7;
pstring printername, sharename;
PRINTER_INFO_CTR ctr, ctr_pub;
POLICY_HND hnd;
BOOL got_hnd = False;
WERROR result;
char *action_str;
if (!get_printer_info(cli, mem_ctx, 2, argc, argv, &num_printers, &ctr))
return nt_status;
for (i = 0; i < num_printers; i++) {
rpcstr_pull(printername, ctr.printers_2[i].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, ctr.printers_2[i].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
PRINTER_ALL_ACCESS, cli->user_name, &hnd))
goto done;
got_hnd = True;
if (!net_spoolss_getprinter(cli, mem_ctx, &hnd, level, &ctr_pub))
goto done;
switch (action) {
case SPOOL_DS_PUBLISH:
action_str = "published";
break;
case SPOOL_DS_UPDATE:
action_str = "updated";
break;
case SPOOL_DS_UNPUBLISH:
action_str = "unpublished";
break;
default:
printf("unkown action: %d\n", action);
break;
}
ctr_pub.printers_7->action = action;
result = cli_spoolss_setprinter(cli, mem_ctx, &hnd, level, &ctr_pub, 0);
if (!W_ERROR_IS_OK(result) && (W_ERROR_V(result) =! W_ERROR_V(WERR_IO_PENDING))) {
printf("cannot set printer-info: %s\n", dos_errstr(result));
goto done;
}
printf("successfully %s printer %s in Active Directory\n", action_str, sharename);
}
nt_status = NT_STATUS_OK;
done:
if (got_hnd)
cli_spoolss_close_printer(cli, mem_ctx, &hnd);
return nt_status;
}
NTSTATUS rpc_printer_publish_publish_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
return rpc_printer_publish_internals_args(cli, mem_ctx, argc, argv, SPOOL_DS_PUBLISH);
}
NTSTATUS rpc_printer_publish_unpublish_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
return rpc_printer_publish_internals_args(cli, mem_ctx, argc, argv, SPOOL_DS_UNPUBLISH);
}
NTSTATUS rpc_printer_publish_update_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
return rpc_printer_publish_internals_args(cli, mem_ctx, argc, argv, SPOOL_DS_UPDATE);
}
NTSTATUS rpc_printer_publish_list_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i, num_printers;
uint32 level = 7;
pstring printername, sharename;
pstring guid;
PRINTER_INFO_CTR ctr, ctr_pub;
POLICY_HND hnd;
BOOL got_hnd = False;
int state;
if (!get_printer_info(cli, mem_ctx, 2, argc, argv, &num_printers, &ctr))
return nt_status;
for (i = 0; i < num_printers; i++) {
ZERO_STRUCT(ctr_pub);
rpcstr_pull(printername, ctr.printers_2[i].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, ctr.printers_2[i].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
PRINTER_ALL_ACCESS, cli->user_name, &hnd))
goto done;
got_hnd = True;
if (!net_spoolss_getprinter(cli, mem_ctx, &hnd, level, &ctr_pub))
goto done;
rpcstr_pull(guid, ctr_pub.printers_7->guid.buffer, sizeof(guid), -1, STR_TERMINATE);
state = ctr_pub.printers_7->action;
switch (state) {
case SPOOL_DS_PUBLISH:
printf("printer [%s] is published", sharename);
if (opt_verbose)
printf(", guid: %s", guid);
printf("\n");
break;
case SPOOL_DS_UNPUBLISH:
printf("printer [%s] is unpublished\n", sharename);
break;
case SPOOL_DS_UPDATE:
printf("printer [%s] is currently updating\n", sharename);
break;
default:
printf("unkown state: %d\n", state);
break;
}
}
nt_status = NT_STATUS_OK;
done:
if (got_hnd)
cli_spoolss_close_printer(cli, mem_ctx, &hnd);
return nt_status;
}
NTSTATUS rpc_printer_migrate_security_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i = 0;
uint32 num_printers;
uint32 level = 2;
pstring printername = "", sharename = "";
BOOL got_hnd_src = False;
BOOL got_hnd_dst = False;
BOOL got_dst_spoolss_pipe = False;
POLICY_HND hnd_src, hnd_dst;
PRINTER_INFO_CTR ctr_src, ctr_dst, ctr_enum;
struct cli_state *cli_dst = NULL;
ZERO_STRUCT(ctr_src);
DEBUG(3,("copying printer ACLs\n"));
nt_status = connect_pipe(&cli_dst, PI_SPOOLSS, &got_dst_spoolss_pipe);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
if (!get_printer_info(cli, mem_ctx, level, argc, argv, &num_printers, &ctr_enum)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
if (!num_printers) {
printf ("no printers found on server.\n");
nt_status = NT_STATUS_OK;
goto done;
}
for (i = 0; i < num_printers; i++) {
rpcstr_pull(printername, ctr_enum.printers_2[i].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, ctr_enum.printers_2[i].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
nt_status = NT_STATUS_UNSUCCESSFUL;
d_printf("migrating printer ACLs for: [%s] / [%s]\n",
printername, sharename);
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
goto done;
got_hnd_src = True;
if (!net_spoolss_open_printer_ex(cli_dst, mem_ctx, sharename,
PRINTER_ALL_ACCESS, cli_dst->user_name, &hnd_dst))
goto done;
got_hnd_dst = True;
if (!net_spoolss_getprinter(cli_dst, mem_ctx, &hnd_dst, level, &ctr_dst))
goto done;
if (!net_spoolss_getprinter(cli, mem_ctx, &hnd_src, 3, &ctr_src))
goto done;
ctr_dst.printers_2->devmode = NULL;
ctr_dst.printers_2->secdesc = dup_sec_desc(mem_ctx, ctr_src.printers_3->secdesc);
if (opt_verbose)
display_sec_desc(ctr_dst.printers_2->secdesc);
if (!net_spoolss_setprinter(cli_dst, mem_ctx, &hnd_dst, 2, &ctr_dst))
goto done;
DEBUGADD(1,("\tSetPrinter of SECDESC succeeded\n"));
if (got_hnd_src) {
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
got_hnd_src = False;
}
if (got_hnd_dst) {
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
got_hnd_dst = False;
}
}
nt_status = NT_STATUS_OK;
done:
if (got_hnd_src)
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
if (got_hnd_dst)
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
if (got_dst_spoolss_pipe) {
cli_nt_session_close(cli_dst);
cli_shutdown(cli_dst);
}
return nt_status;
}
NTSTATUS rpc_printer_migrate_forms_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
WERROR result;
uint32 i, f;
uint32 num_printers;
uint32 level = 1;
pstring printername = "", sharename = "";
BOOL got_hnd_src = False;
BOOL got_hnd_dst = False;
BOOL got_dst_spoolss_pipe = False;
POLICY_HND hnd_src, hnd_dst;
PRINTER_INFO_CTR ctr_enum, ctr_dst;
uint32 num_forms;
FORM_1 *forms;
struct cli_state *cli_dst = NULL;
ZERO_STRUCT(ctr_enum);
DEBUG(3,("copying forms\n"));
nt_status = connect_pipe(&cli_dst, PI_SPOOLSS, &got_dst_spoolss_pipe);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
if (!get_printer_info(cli, mem_ctx, 2, argc, argv, &num_printers, &ctr_enum)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
if (!num_printers) {
printf ("no printers found on server.\n");
nt_status = NT_STATUS_OK;
goto done;
}
for (i = 0; i < num_printers; i++) {
rpcstr_pull(printername, ctr_enum.printers_2[i].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, ctr_enum.printers_2[i].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
nt_status = NT_STATUS_UNSUCCESSFUL;
d_printf("migrating printer forms for: [%s] / [%s]\n",
printername, sharename);
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
goto done;
got_hnd_src = True;
if (!net_spoolss_open_printer_ex(cli_dst, mem_ctx, sharename,
PRINTER_ALL_ACCESS, cli->user_name, &hnd_dst))
goto done;
got_hnd_dst = True;
if (!net_spoolss_getprinter(cli_dst, mem_ctx, &hnd_dst, level, &ctr_dst))
goto done;
if (!net_spoolss_enumforms(cli, mem_ctx, &hnd_src, level, &num_forms, &forms))
goto done;
DEBUG(1,("got %d forms for printer\n", num_forms));
for (f = 0; f < num_forms; f++) {
FORM form;
fstring form_name;
if (forms[f].flag != FORM_PRINTER)
continue;
if (forms[f].name.buffer)
rpcstr_pull(form_name, forms[f].name.buffer,
sizeof(form_name), -1, STR_TERMINATE);
if (opt_verbose)
d_printf("\tmigrating form # %d [%s] of type [%d]\n",
f, form_name, forms[f].flag);
form.flags = FORM_PRINTER;
form.size_x = forms[f].width;
form.size_y = forms[f].length;
form.left = forms[f].left;
form.top = forms[f].top;
form.right = forms[f].right;
form.bottom = forms[f].bottom;
init_unistr2(&form.name, form_name, UNI_STR_TERMINATE);
result = cli_spoolss_addform(cli_dst, mem_ctx,
&hnd_dst, 1, &form);
if (!W_ERROR_IS_OK(result)) {
d_printf("\tAddForm form %d: [%s] refused.\n",
f, form_name);
continue;
}
DEBUGADD(1,("\tAddForm of [%s] succeeded\n", form_name));
}
if (got_hnd_src) {
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
got_hnd_src = False;
}
if (got_hnd_dst) {
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
got_hnd_dst = False;
}
}
nt_status = NT_STATUS_OK;
done:
if (got_hnd_src)
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
if (got_hnd_dst)
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
if (got_dst_spoolss_pipe) {
cli_nt_session_close(cli_dst);
cli_shutdown(cli_dst);
}
return nt_status;
}
NTSTATUS rpc_printer_migrate_drivers_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i, p;
uint32 num_printers;
uint32 level = 3;
pstring printername = "", sharename = "";
BOOL got_hnd_src = False;
BOOL got_hnd_dst = False;
BOOL got_dst_spoolss_pipe = False;
BOOL got_src_driver_share = False;
BOOL got_dst_driver_share = False;
POLICY_HND hnd_src, hnd_dst;
PRINTER_DRIVER_CTR drv_ctr_src, drv_ctr_dst;
PRINTER_INFO_CTR info_ctr_enum, info_ctr_dst;
struct cli_state *cli_dst = NULL;
struct cli_state *cli_share_src = NULL;
struct cli_state *cli_share_dst = NULL;
fstring drivername = "";
ZERO_STRUCT(drv_ctr_src);
ZERO_STRUCT(drv_ctr_dst);
ZERO_STRUCT(info_ctr_enum);
ZERO_STRUCT(info_ctr_dst);
DEBUG(3,("copying printer-drivers\n"));
nt_status = connect_pipe(&cli_dst, PI_SPOOLSS, &got_dst_spoolss_pipe);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
nt_status = connect_to_service(&cli_share_src, &cli->dest_ip,
cli->desthost, "print$", "A:");
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
got_src_driver_share = True;
nt_status = connect_to_service(&cli_share_dst, &cli_dst->dest_ip,
cli_dst->desthost, "print$", "A:");
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
got_dst_driver_share = True;
if (!get_printer_info(cli, mem_ctx, 2, argc, argv, &num_printers, &info_ctr_enum)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
if (!num_printers) {
printf ("no printers found on server.\n");
nt_status = NT_STATUS_OK;
goto done;
}
for (p = 0; p < num_printers; p++) {
rpcstr_pull(printername, info_ctr_enum.printers_2[p].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, info_ctr_enum.printers_2[p].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
nt_status = NT_STATUS_UNSUCCESSFUL;
d_printf("migrating printer driver for: [%s] / [%s]\n",
printername, sharename);
if (!net_spoolss_open_printer_ex(cli_dst, mem_ctx, sharename,
PRINTER_ALL_ACCESS, cli->user_name, &hnd_dst))
goto done;
got_hnd_dst = True;
if (!net_spoolss_getprinter(cli_dst, mem_ctx, &hnd_dst, 2, &info_ctr_dst))
goto done;
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
goto done;
got_hnd_src = True;
for (i=0; archi_table[i].long_archi!=NULL; i++) {
if (!net_spoolss_getprinterdriver(cli, mem_ctx, &hnd_src,
level, archi_table[i].long_archi,
archi_table[i].version, &drv_ctr_src))
continue;
rpcstr_pull(drivername, drv_ctr_src.info3->name.buffer,
sizeof(drivername), -1, STR_TERMINATE);
if (opt_verbose)
display_print_driver_3(drv_ctr_src.info3);
nt_status = check_arch_dir(cli_share_dst, archi_table[i].short_archi);
if (!NT_STATUS_IS_OK(nt_status))
goto done;
nt_status = copy_print_driver_3(mem_ctx, cli_share_src, cli_share_dst,
archi_table[i].short_archi,
drv_ctr_src.info3);
if (!NT_STATUS_IS_OK(nt_status))
goto done;
if (!net_spoolss_addprinterdriver(cli_dst, mem_ctx, level, &drv_ctr_src)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
DEBUGADD(1,("Sucessfully added driver [%s] for printer [%s]\n",
drivername, printername));
}
init_unistr(&info_ctr_dst.printers_2->drivername, drivername);
if (!net_spoolss_setprinter(cli_dst, mem_ctx, &hnd_dst, 2, &info_ctr_dst)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
DEBUGADD(1,("Sucessfully set driver %s for printer %s\n",
drivername, printername));
if (got_hnd_dst) {
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
got_hnd_dst = False;
}
if (got_hnd_src) {
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
got_hnd_src = False;
}
}
nt_status = NT_STATUS_OK;
done:
if (got_hnd_src)
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
if (got_hnd_dst)
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
if (got_dst_spoolss_pipe) {
cli_nt_session_close(cli_dst);
cli_shutdown(cli_dst);
}
if (got_src_driver_share)
cli_shutdown(cli_share_src);
if (got_dst_driver_share)
cli_shutdown(cli_share_dst);
return nt_status;
}
NTSTATUS rpc_printer_migrate_printers_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
WERROR result;
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i = 0, num_printers;
uint32 level = 2;
PRINTER_INFO_CTR ctr_src, ctr_dst, ctr_enum;
struct cli_state *cli_dst = NULL;
POLICY_HND hnd_dst, hnd_src;
pstring printername, sharename;
BOOL got_hnd_src = False;
BOOL got_hnd_dst = False;
BOOL got_dst_spoolss_pipe = False;
DEBUG(3,("copying printers\n"));
nt_status = connect_pipe(&cli_dst, PI_SPOOLSS, &got_dst_spoolss_pipe);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
if (!get_printer_info(cli, mem_ctx, 2, argc, argv, &num_printers, &ctr_enum)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
if (!num_printers) {
printf ("no printers found on server.\n");
nt_status = NT_STATUS_OK;
goto done;
}
for (i = 0; i < num_printers; i++) {
rpcstr_pull(printername, ctr_enum.printers_2[i].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, ctr_enum.printers_2[i].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
nt_status = NT_STATUS_UNSUCCESSFUL;
d_printf("migrating printer queue for: [%s] / [%s]\n",
printername, sharename);
if (!net_spoolss_open_printer_ex(cli_dst, mem_ctx, sharename,
PRINTER_ALL_ACCESS, cli->user_name, &hnd_dst)) {
DEBUG(1,("could not open printer: %s\n", sharename));
} else {
got_hnd_dst = True;
}
if (!net_spoolss_getprinter(cli_dst, mem_ctx, &hnd_dst, level, &ctr_dst)) {
printf ("could not get printer, creating printer.\n");
} else {
DEBUG(1,("printer already exists: %s\n", sharename));
if (got_hnd_src) {
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
got_hnd_src = False;
}
if (got_hnd_dst) {
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
got_hnd_dst = False;
}
continue;
}
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
goto done;
got_hnd_src = True;
if (!net_spoolss_getprinter(cli, mem_ctx, &hnd_src, level, &ctr_src))
goto done;
d_printf("creating printer: %s\n", printername);
result = cli_spoolss_addprinterex (cli_dst, mem_ctx, level, &ctr_src);
if (W_ERROR_IS_OK(result))
d_printf ("printer [%s] successfully added.\n", printername);
else if (W_ERROR_V(result) == W_ERROR_V(WERR_PRINTER_ALREADY_EXISTS))
d_printf ("printer [%s] already exists.\n", printername);
else {
printf ("could not create printer\n");
goto done;
}
if (got_hnd_src) {
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
got_hnd_src = False;
}
if (got_hnd_dst) {
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
got_hnd_dst = False;
}
}
nt_status = NT_STATUS_OK;
done:
if (got_hnd_src)
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
if (got_hnd_dst)
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
if (got_dst_spoolss_pipe) {
cli_nt_session_close(cli_dst);
cli_shutdown(cli_dst);
}
return nt_status;
}
NTSTATUS rpc_printer_migrate_settings_internals(const DOM_SID *domain_sid, const char *domain_name,
struct cli_state *cli, TALLOC_CTX *mem_ctx,
int argc, const char **argv)
{
WERROR result;
NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
uint32 i = 0, p = 0, j = 0;
uint32 num_printers, val_needed, data_needed;
uint32 level = 2;
pstring printername = "", sharename = "";
BOOL got_hnd_src = False;
BOOL got_hnd_dst = False;
BOOL got_dst_spoolss_pipe = False;
POLICY_HND hnd_src, hnd_dst;
PRINTER_INFO_CTR ctr_enum, ctr_dst, ctr_dst_publish;
REGVAL_CTR reg_ctr;
struct cli_state *cli_dst = NULL;
char *devicename = NULL, *unc_name = NULL, *url = NULL;
fstring longname;
const char *keyname = NULL;
uint16 *keylist = NULL, *curkey;
ZERO_STRUCT(ctr_enum);
DEBUG(3,("copying printer settings\n"));
nt_status = connect_pipe(&cli_dst, PI_SPOOLSS, &got_dst_spoolss_pipe);
if (!NT_STATUS_IS_OK(nt_status))
return nt_status;
if (!get_printer_info(cli, mem_ctx, level, argc, argv, &num_printers, &ctr_enum)) {
nt_status = NT_STATUS_UNSUCCESSFUL;
goto done;
}
if (!num_printers) {
printf ("no printers found on server.\n");
nt_status = NT_STATUS_OK;
goto done;
}
get_mydnsfullname(longname);
for (i = 0; i < num_printers; i++) {
rpcstr_pull(printername, ctr_enum.printers_2[i].printername.buffer,
sizeof(printername), -1, STR_TERMINATE);
rpcstr_pull(sharename, ctr_enum.printers_2[i].sharename.buffer,
sizeof(sharename), -1, STR_TERMINATE);
keyname = "";
nt_status = NT_STATUS_UNSUCCESSFUL;
d_printf("migrating printer settings for: [%s] / [%s]\n",
printername, sharename);
if (!net_spoolss_open_printer_ex(cli, mem_ctx, sharename,
MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
goto done;
got_hnd_src = True;
if (!net_spoolss_open_printer_ex(cli_dst, mem_ctx, sharename,
PRINTER_ALL_ACCESS, cli_dst->user_name, &hnd_dst))
goto done;
got_hnd_dst = True;
if (!net_spoolss_getprinter(cli_dst, mem_ctx, &hnd_dst,
level, &ctr_dst))
goto done;
ctr_dst.printers_2 = &ctr_enum.printers_2[i];
init_unistr( &ctr_dst.printers_2->portname, SAMBA_PRINTER_PORT_NAME);
if (ctr_enum.printers_2[i].attributes & PRINTER_ATTRIBUTE_PUBLISHED) {
if (!net_spoolss_getprinter(cli_dst, mem_ctx, &hnd_dst, 7, &ctr_dst_publish))
goto done;
ctr_dst_publish.printers_7->action = SPOOL_DS_PUBLISH;
net_spoolss_setprinter(cli_dst, mem_ctx, &hnd_dst, 7, &ctr_dst_publish);
DEBUG(3,("republished printer\n"));
}
ctr_dst.printers_2->devmode = TALLOC_MEMDUP(mem_ctx,
ctr_enum.printers_2[i].devmode, sizeof(DEVICEMODE));
ctr_dst.printers_2->secdesc = NULL;
#if 0
if (asprintf(&devicename, "\\\\%s\\%s", longname, printername) < 0) {
nt_status = NT_STATUS_NO_MEMORY;
goto done;
}
init_unistr(&ctr_dst.printers_2->devmode->devicename, devicename);
#endif
if (!net_spoolss_setprinter(cli_dst, mem_ctx, &hnd_dst,
level, &ctr_dst))
goto done;
DEBUGADD(1,("\tSetPrinter of DEVICEMODE succeeded\n"));
result = cli_spoolss_enumprinterdata(cli, mem_ctx, &hnd_src, p, 0, 0,
&val_needed, &data_needed, NULL);
while (W_ERROR_IS_OK(result)) {
REGISTRY_VALUE value;
result = cli_spoolss_enumprinterdata(
cli, mem_ctx, &hnd_src, p++, val_needed,
data_needed, 0, 0, &value);
if (W_ERROR_IS_OK(result)) {
if (opt_verbose)
display_reg_value(NULL, value);
if (!net_spoolss_setprinterdata(cli_dst, mem_ctx,
&hnd_dst, &value))
goto done;
DEBUGADD(1,("\tSetPrinterData of [%s] succeeded\n",
value.valuename));
}
}
if (!net_spoolss_enumprinterkey(cli, mem_ctx, &hnd_src, keyname, &keylist)) {
printf("got no key-data\n");
continue;
}
if (keylist == NULL)
continue;
curkey = keylist;
while (*curkey != 0) {
pstring subkey;
rpcstr_pull(subkey, curkey, sizeof(subkey), -1, STR_TERMINATE);
curkey += strlen(subkey) + 1;
if (!net_spoolss_enumprinterdataex(cli, mem_ctx, 0,
&hnd_src, subkey,
®_ctr))
goto done;
for (j=0; j < reg_ctr.num_values; j++) {
REGISTRY_VALUE value;
UNISTR2 data;
if (strequal(reg_ctr.values[j]->valuename, SPOOL_REG_PORTNAME) ||
strequal(reg_ctr.values[j]->valuename, SPOOL_REG_UNCNAME) ||
strequal(reg_ctr.values[j]->valuename, SPOOL_REG_URL) ||
strequal(reg_ctr.values[j]->valuename, SPOOL_REG_SHORTSERVERNAME) ||
strequal(reg_ctr.values[j]->valuename, SPOOL_REG_SERVERNAME)) {
if (strequal(reg_ctr.values[j]->valuename, SPOOL_REG_PORTNAME)) {
init_unistr2(&data, SAMBA_PRINTER_PORT_NAME, UNI_STR_TERMINATE);
fstrcpy(value.valuename, SPOOL_REG_PORTNAME);
}
if (strequal(reg_ctr.values[j]->valuename, SPOOL_REG_UNCNAME)) {
if (asprintf(&unc_name, "\\\\%s\\%s", longname, sharename) < 0) {
nt_status = NT_STATUS_NO_MEMORY;
goto done;
}
init_unistr2(&data, unc_name, UNI_STR_TERMINATE);
fstrcpy(value.valuename, SPOOL_REG_UNCNAME);
}
if (strequal(reg_ctr.values[j]->valuename, SPOOL_REG_URL)) {
continue;
#if 0
if (asprintf(&url, "http://%s:631/printers/%s", longname, sharename) < 0) {
nt_status = NT_STATUS_NO_MEMORY;
goto done;
}
init_unistr2(&data, url, UNI_STR_TERMINATE);
fstrcpy(value.valuename, SPOOL_REG_URL);
#endif
}
if (strequal(reg_ctr.values[j]->valuename, SPOOL_REG_SERVERNAME)) {
init_unistr2(&data, longname, UNI_STR_TERMINATE);
fstrcpy(value.valuename, SPOOL_REG_SERVERNAME);
}
if (strequal(reg_ctr.values[j]->valuename, SPOOL_REG_SHORTSERVERNAME)) {
init_unistr2(&data, global_myname(), UNI_STR_TERMINATE);
fstrcpy(value.valuename, SPOOL_REG_SHORTSERVERNAME);
}
value.type = REG_SZ;
value.size = data.uni_str_len * 2;
value.data_p = TALLOC_MEMDUP(mem_ctx, data.buffer, value.size);
if (opt_verbose)
display_reg_value(subkey, value);
if (!net_spoolss_setprinterdataex(cli_dst, mem_ctx, &hnd_dst,
subkey, &value))
goto done;
} else {
if (opt_verbose)
display_reg_value(subkey, *(reg_ctr.values[j]));
if (!net_spoolss_setprinterdataex(cli_dst, mem_ctx, &hnd_dst,
subkey, reg_ctr.values[j]))
goto done;
}
DEBUGADD(1,("\tSetPrinterDataEx of key [%s\\%s] succeeded\n",
subkey, reg_ctr.values[j]->valuename));
}
regval_ctr_destroy(®_ctr);
}
safe_free(keylist);
if (got_hnd_src) {
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
got_hnd_src = False;
}
if (got_hnd_dst) {
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
got_hnd_dst = False;
}
}
nt_status = NT_STATUS_OK;
done:
SAFE_FREE(devicename);
SAFE_FREE(url);
SAFE_FREE(unc_name);
if (got_hnd_src)
cli_spoolss_close_printer(cli, mem_ctx, &hnd_src);
if (got_hnd_dst)
cli_spoolss_close_printer(cli_dst, mem_ctx, &hnd_dst);
if (got_dst_spoolss_pipe) {
cli_nt_session_close(cli_dst);
cli_shutdown(cli_dst);
}
return nt_status;
}