source: branches/smbpdr-1.0/smbport.c

/* SMBPORT
 *
 * Simple program to create a new SMB printer port using the specified
 * host/queue/password.
 *
 * Syntax: smbport <port-name> <workgroup> <host> <printer> [<user-id> [<password>]]
 */

#define INCL_DOSERRORS
#define INCL_DOSMODULEMGR
#define INCL_DOSPROCESS
#define INCL_GPI
#define INCL_SPL
#define INCL_SPLERRORS
#define INCL_WIN
#include <os2.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define STR_LEN_PORTNAME          64
#define STR_LEN_PORTDESC          256
#define STR_LEN_DESC              81

#define APPNAME_LEAD_STR          "PM_"
#define APPNAME_PM_PORT_DRIVER    "PM_PORT_DRIVER"
#define APPNAME_PM_SPOOLER_PORT   "PM_SPOOLER_PORT"

#define KEY_DESCRIPTION           "DESCRIPTION"
#define KEY_INITIALIZATION        "INITIALIZATION"
#define KEY_TERMINATION           "TERMINATION"
#define KEY_PORTDRIVER            "PORTDRIVER"
#define KEY_TIMEOUT               "TIMEOUT"

/* Driver-specific port setting structure for SplPdSet & SplPdQuery.
 */
typedef struct _PORTSETTINGS {
    CHAR  szPortData[ STR_LEN_PORTDESC ];
} PORTSETTINGS, *PPORTSETTINGS;


/* Various items that should be defined in the toolkit header pmsplb.h,
 * if it actually existed...
 */
#define TYPE_LONG_WAIT   2
#define BIDI_SET_PORTDRV 0x19

#pragma import ( PrtSet,, "PMSPL", 604 )
extern ULONG APIENTRY PrtSet( PSZ pszComputerName, PSZ pszDeviceName, ULONG ulType, ULONG ulCommand, PVOID pInData, ULONG cbInData );


/* Copied from smb.c
 */

static char tohex (unsigned char b)
{
        b &= 0xF;
        
        if (b <= 9)
        {
                return b + '0';
        }
        
        return 'A' + (b - 0xA);
}

static void encryptPassword (const char *pszPlain, char *pszCrypt)
{
        /* A simple "encryption" encode each character as hex value. */
        const char *s = pszPlain;
        char *d = pszCrypt;
        
        while (*s)
        {
                *d++ = tohex ((*s) >> 4);
                *d++ = tohex (*s);
                s++;
        }
        
        *d++ = 0;
}


int main( int argc, char *argv[] )
{
    PORTSETTINGS settings;

    HAB     hab;
    ULONG   ulRC;
    PSZ     pszPortName    = NULL,
            pszHostName    = NULL,
            pszPrinterName = NULL,
            pszWorkgroup   = NULL,
            pszUserID      = NULL,
            pszPassword    = NULL,
            pszCopies      = NULL;
    CHAR    szPathName[ CCHMAXPATH + 1 ];
    CHAR    szIniApp[ STR_LEN_PORTNAME ];
    CHAR    szIniVal[ STR_LEN_PORTDESC ];
    CHAR    pwBuffer[ 256 ];
    HMODULE hPdr = NULLHANDLE;
    PFN     pfn_SmbInstallPort,
            pfn_SmbSet;


    if ( argc < 5 ) {
        printf("SMBPORT - Create a new Samba printer port for the specified SMB network printer\n");
        printf("Syntax:\n smbport <port> <workgroup> <host> <printer> [<userid>] [<password>] [<copies>]\n");
        printf("  <port>       Name of the new port (must be \"SMB<x>\", where <x> = 1-64)\n");
        printf("  <workgroup>  Name of the SMB workgroup to which the host and printer belong\n");
        printf("  <host>       SMB host name of the server where the printer is defined\n");
        printf("  <printer>    SMB name of the destination printer on <host>\n");
        printf("  <userid>     SMB user name authorized to access the printer\n");
        printf("  <password>   The password for <user-id>\n");
        printf("  <copies>     Number of copies of each job to print\n");
        printf(" To leave a parameter empty, specify \"\"\n");
        printf("\nExamples:\n    smbport SMB1 \"\" printsrv hplj4a guest guest\n");
        printf("    smbport SMB2 yggdrasill valhalla printer6 odin all-seeing 1\n");
        return 1;
    }
    pszPortName    = strupr( argv[ 1 ] );
    pszWorkgroup   = argv[ 2 ];
    pszHostName    = argv[ 3 ];
    pszPrinterName = argv[ 4 ];
    if ( argc > 4 ) pszUserID   = argv[ 5 ];
    if ( argc > 5 ) pszPassword = argv[ 6 ];
    if ( argc > 6 ) pszCopies   = argv[ 7 ];

    /* Get the path to the installed SMB.PDR */
    hab = WinInitialize( 0 );
    ulRC = PrfQueryProfileString( HINI_SYSTEMPROFILE, APPNAME_PM_PORT_DRIVER,
                                  "SMB", NULL, (PVOID) szPathName, CCHMAXPATH );
    if ( !ulRC ) {
        ulRC = ulRC = WinGetLastError( hab );
        printf("SMB port driver is not installed.\n");
        WinTerminate( hab );
        return ulRC;
    }

    /* Now load the port driver DLL and register the install and setup functions */
    ulRC = DosLoadModule( NULL, 0, szPathName, &hPdr );
    if ( ulRC != NO_ERROR ) {
        printf("Failed to load SMB.PDR, RC=%u.\n", ulRC );
        WinTerminate( hab );
        return ulRC;
    }
    ulRC = DosQueryProcAddr( hPdr, 0, "SPLPDINSTALLPORT", &pfn_SmbInstallPort );
    if ( ulRC != NO_ERROR ) {
        printf("Failed to resolve proc address, RC=%u.\n", ulRC );
        goto finish;
    }
    ulRC = DosQueryProcAddr( hPdr, 0, "SPLPDSET", &pfn_SmbSet );
    if ( ulRC != NO_ERROR ) {
        printf("Failed to resolve proc address, RC=%u.\n", ulRC );
        goto finish;
    }

    /* Create the new port (it will be given default settings by the PDR) */
    ulRC = pfn_SmbInstallPort( hab, pszPortName );
    if ( ulRC != NO_ERROR ) {
        printf("Failed to create port, RC=0x%X.\n", ulRC );
        goto finish;
    }
    printf("Port %s was created successfully.\n", pszPortName );

    /* Now update the port driver settings */
    strncpy( settings.szPortData, pszHostName, STR_LEN_PORTDESC-1 );
    strncat( settings.szPortData, "#", STR_LEN_PORTDESC-1 );
    strncat( settings.szPortData, pszPrinterName, STR_LEN_PORTDESC-1 );
    strncat( settings.szPortData, "#", STR_LEN_PORTDESC-1 );
    if ( pszWorkgroup && (strlen( pszWorkgroup )))
        strncat( settings.szPortData, pszWorkgroup, STR_LEN_PORTDESC-1 );
    strncat( settings.szPortData, "#", STR_LEN_PORTDESC-1 );
    if ( pszUserID && ( strlen( pszUserID )))
        strncat( settings.szPortData, pszUserID, STR_LEN_PORTDESC-1 );
    strncat( settings.szPortData, "#", STR_LEN_PORTDESC-1 );
    if ( pszCopies && ( strlen( pszCopies )))
        strncat( settings.szPortData, pszCopies, STR_LEN_PORTDESC-1 );
    strncat( settings.szPortData, "#", STR_LEN_PORTDESC-1 );
    if ( pszPassword && ( strlen( pszPassword ))) {
        encryptPassword( pszPassword, pwBuffer );
        strncat( settings.szPortData, pwBuffer, STR_LEN_PORTDESC-1 );
    }
    printf("Setting port data: %s\n", settings.szPortData );
    ulRC = pfn_SmbSet( pszPortName, TYPE_LONG_WAIT, BIDI_SET_PORTDRV,
                       &settings, sizeof( settings ));
    if ( ulRC ) {
        printf("Failed to update port settings, RC=0x%X.\n", ulRC );
        goto finish;
    }

    /* Make sure the settings actually were updated (older versions of SMB.PDR
     * don't do anything when SplPdSet is called, but return success anyway).
     * What we do here is read the port settings directly from the INI file
     * and compare it with the data we sent.
     */
    strcpy( szIniApp, APPNAME_LEAD_STR );
    strncat( szIniApp, pszPortName, STR_LEN_PORTNAME-1 );
    ulRC = PrfQueryProfileString( HINI_SYSTEMPROFILE, szIniApp,
                                  KEY_INITIALIZATION, NULL,
                                  (PVOID) szIniVal, STR_LEN_PORTDESC );
    if ( !ulRC ) {
        ulRC = 1;
        printf("Failed to get SMB port settings for %s: 0x%X\n", szIniApp, WinGetLastError( hab ));
        goto finish;
    }
    if ( strncmp( szIniVal, settings.szPortData, strlen( settings.szPortData ))) {
        printf("Unable to set SMB port settings.\n");
        ulRC = 1;
    }
    else {
        printf("Port settings were updated successfully.\n");
        ulRC = 0;
    }

finish:
    if ( hPdr ) DosFreeModule( hPdr );
    WinTerminate( hab );

    return ulRC;
}