tee.c

Go to the documentation of this file.

/*
 * TEE.C - external command.
 *
 * clone from 4nt tee command
 *
 * 01 Sep 1999 - Paolo Pantaleo <paolopan@freemail.it>
 *     started
 *
 *
 */


#include <windows.h>
#include <tchar.h>
#include <stdio.h>
#include <malloc.h>



#define TEE_BUFFER_SIZE 8192

/*these are function that emulate the ones used in cmd*/

/*many of them are just copied in this file from their
original location*/

VOID ConOutPuts (LPTSTR szText)
{
    DWORD dwWritten;

    WriteFile (GetStdHandle (STD_OUTPUT_HANDLE), szText, _tcslen(szText), &dwWritten, NULL);
    WriteFile (GetStdHandle (STD_OUTPUT_HANDLE), _T("\n"), 1, &dwWritten, NULL);
}


VOID ConErrPrintf (LPTSTR szFormat, ...)
{
    DWORD dwWritten;
    TCHAR szOut[4096];
    va_list arg_ptr;

    va_start (arg_ptr, szFormat);
    _vstprintf (szOut, szFormat, arg_ptr);
    va_end (arg_ptr);

    WriteFile (GetStdHandle (STD_ERROR_HANDLE), szOut, _tcslen(szOut), &dwWritten, NULL);
}



VOID error_sfile_not_found (LPTSTR f)
{
    ConErrPrintf (_T("Error opening file") _T(" - %s\n"), f);
}




VOID ConErrPuts (LPTSTR szText)
{
    ConErrPrintf(_T("%s\n"),szText );
}


INT main (int argc,char **p)
{
    /*reading/writing buffer*/
    TCHAR buff[TEE_BUFFER_SIZE];

    /*handle for file and console*/
    HANDLE hConsoleIn,hConsoleOut;

    /*bytes written by WriteFile and ReadFile*/
    DWORD dwRead,dwWritten;


    BOOL bRet,bAppend=FALSE;


    /*command line parsing stuff*/
    LPTSTR tmp;
    INT i;
    //BOOL bQuote;

    /*file list implementation*/
    LPTSTR *files;
    INT iFileCounter=0;
    HANDLE *hFile;

    /*used to remove '"' (if any)*/
    INT add;

    DWORD dw;


    if (argc < 2)
        return 1;

    if (_tcsncmp (p[1], _T("/?"), 2) == 0)
    {
        ConOutPuts (_T("Copy standard input to both standard output and a file.\n"
                       "\n"
                       "TEE [/A] file...\n"
                       "\n"
                       "  file  One or more files that will receive output.\n"
                       "  /A    Append output to files.\n"));
        return 0;
    }

    files = malloc(sizeof(LPTSTR)*argc);
    hFile = malloc(sizeof(HANDLE)*argc);

    hConsoleIn=GetStdHandle(STD_INPUT_HANDLE);
    hConsoleOut=GetStdHandle(STD_OUTPUT_HANDLE);

    /*parse command line for /a and file name(s)*/
    for(i=1;i <argc;i++)
    {
        //bQuote=FALSE;
        add=0;

        if(_tcsnicmp(p[i],_T("/a"),2) == 0)
        {
            bAppend = TRUE;
            continue;
        }

        /*remove quote if any*/
        if (p[i][0] == _T('"'))
        {
            tmp = _tcschr (p[i]+1, _T('"'));
            if (tmp != 0)
            {
                add = 1;
                *tmp= _T('\0');
            }
        }

        /*add filename to array of filename*/
/*
        if(  iFileCounter >= sizeof(files) / sizeof(*files)  )
        {
            ConErrPrintf("too many files, maximum is %d\n",sizeof(files) / sizeof(*files));
            return 1;
        }
        */

        files[iFileCounter++]= p[i]+add;
    }

    /*open file(s)*/
    for(i=0;i<iFileCounter;i++)
    {
        //l=0;
        hFile[i] = CreateFile(files[i],GENERIC_WRITE,
            0,NULL,
            CREATE_ALWAYS,
            FILE_ATTRIBUTE_NORMAL,NULL);

        if (hFile[i] == INVALID_HANDLE_VALUE)
        {
            error_sfile_not_found (files[i]);

            for(i=0;i<iFileCounter;i++)
                CloseHandle (hFile[i]);

            free (files);
            free (hFile);

            return 1;
        }

        /*set append mode*/
        if (bAppend)
        {
            if (GetFileType (hFile[i]) == FILE_TYPE_DISK)
            {
                dw = SetFilePointer (hFile[i],0,NULL,FILE_END);
                if (dw == 0xFFFFFFFF)
                {
                    ConErrPrintf(_T("error moving to end of file %s"),files[i]);

                    for(i=0;i<iFileCounter;i++)
                        CloseHandle (hFile[i]);

                    free (files);
                    free (hFile);

                    return 1;
                }

                ConErrPrintf(_T("SetFilePointer() = %d\n"),dw);
            }
        }
    }

    /*read and write*/
    do
    {
        bRet = ReadFile(hConsoleIn,buff,sizeof(buff),&dwRead,NULL);

        if (dwRead>0 && bRet)
        {
            for(i=0;i<iFileCounter;i++)
                WriteFile(hFile[i],buff,dwRead,&dwWritten,NULL);

            WriteFile(hConsoleOut,buff,dwRead,&dwWritten,NULL);
        }
    } while(dwRead>0 && bRet);

    for(i=0;i<iFileCounter;i++)
        CloseHandle (hFile[i]);

    free (files);
    free (hFile);

    return 0;
}

/* EOF */