/**
***     QADRT/QADRTMAIN2 substitution program.
***     This is needed because the IBM-provided QADRTMAIN2 does not
***     properly translate arguments by default or if no locale is provided.
***
***     See Copyright for the status of this software.
***
***     Author: Patrick Monnerat <pm@datasphere.ch>, DATASPHERE S.A.
**/

#include <stdlib.h>
#include <string.h>
#include <iconv.h>
#include <errno.h>
#include <locale.h>

/* Do not use qadrt.h since it defines unneeded static procedures. */
extern void     QadrtInit(void);
extern int      QadrtFreeConversionTable(void);
extern int      QadrtFreeEnviron(void);
extern char *   setlocale_a(int, const char *);


/* The ASCII main program. */
extern int      main_a(int argc, char * * argv);

/* Global values of original EBCDIC arguments. */
int             ebcdic_argc;
char * *        ebcdic_argv;


int
main(int argc, char * * argv)

{
        int i;
        int j;
        iconv_t cd;
        size_t bytecount = 0;
        char * inbuf;
        char * outbuf;
        size_t inbytesleft;
        size_t outbytesleft;
        char dummybuf[128];
        char tocode[32];
        char fromcode[32];

        ebcdic_argc = argc;
        ebcdic_argv = argv;

        /* Build the encoding converter. */
        strncpy(tocode, "IBMCCSID01208", sizeof tocode);
        strncpy(fromcode, "IBMCCSID000000000010", sizeof fromcode);
        cd = iconv_open(tocode, fromcode);

        /* Measure the arguments. */
        for (i = 0; i < argc; i++) {
                inbuf = argv[i];
                do {
                        inbytesleft = 0;
                        outbuf = dummybuf;
                        outbytesleft = sizeof dummybuf;
                        j = iconv(cd,
                                  &inbuf, &inbytesleft, &outbuf, &outbytesleft);
                        bytecount += outbuf - dummybuf;
                } while (j == -1 && errno == E2BIG);
                /* Reset the shift state. */
                iconv(cd, NULL, &inbytesleft, &outbuf, &outbytesleft);
                }

        /* Allocate memory for the ASCII arguments and vector. */
        argv = (char * *) malloc((argc + 1) * sizeof *argv + bytecount);

        /* Build the vector and convert argument encoding. */
        outbuf = (char *) (argv + argc + 1);
        outbytesleft = bytecount;

        for (i = 0; i < argc; i++) {
                argv[i] = outbuf;
                inbuf = ebcdic_argv[i];
                inbytesleft = 0;
                iconv(cd, &inbuf, &inbytesleft, &outbuf, &outbytesleft);
                iconv(cd, NULL, &inbytesleft, &outbuf, &outbytesleft);
                }

        iconv_close(cd);
        argv[argc] = NULL;

        /* Try setting the locale regardless of QADRT_ENV_LOCALE. */
        setlocale_a(LC_ALL, "");

        /* Call the program. */
        i = main_a(argc, argv);

        /* Clean-up allocated items. */
        free((char *) argv);
        QadrtFreeConversionTable();
        QadrtFreeEnviron();

        /* Terminate. */
        return i;
}