| 0,0 → 1,362 |
|---|
| /* This file is part of the SvarCOM project and is published under the terms |
| * of the MIT license. |
| * |
| * Copyright (C) 2021-2022 Mateusz Viste |
| * |
| * Permission is hereby granted, free of charge, to any person obtaining a |
| * copy of this software and associated documentation files (the "Software"), |
| * to deal in the Software without restriction, including without limitation |
| * the rights to use, copy, modify, merge, publish, distribute, sublicense, |
| * and/or sell copies of the Software, and to permit persons to whom the |
| * Software is furnished to do so, subject to the following conditions: |
| * |
| * The above copyright notice and this permission notice shall be included in |
| * all copies or substantial portions of the Software. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
| * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
| * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER |
| * DEALINGS IN THE SOFTWARE. |
| */ |
| |
| /* |
| * copy |
| */ |
| |
| /* /A - Used to copy ASCII files. Applies to the filename preceding it and to |
| * all following filenames. Files will be copied until an end-of-file mark is |
| * encountered in the file being copied. If an end-of-file mark is encountered |
| * in the file, the rest of the file is not copied. DOS will append an EOF |
| * mark at the end of the copied file. |
| * |
| * /B - Used to copy binary files. Applies to the filename preceding it and to |
| * all following filenames. Copied files will be read by size (according to |
| * the number of bytes indicated in the file`s directory listing). An EOF mark |
| * is not placed at the end of the copied file. |
| * |
| * /V - Checks after the copy to assure that a file was copied correctly. If |
| * the copy cannot be verified, the program will display an error message. |
| * Using this option will result in a slower copying process. |
| * |
| * special case: "COPY A+B+C+D" means "append B, C and D files to the A file" |
| * if A does not exist, then "append C and D to B", etc. |
| */ |
| |
| struct copy_setup { |
| const char *src[64]; |
| unsigned short src_count; /* how many sources are declared */ |
| char cursrc[256]; /* buffer for currently processed src */ |
| char dst[256]; |
| unsigned short dstlen; |
| char src_asciimode[64]; |
| char dst_asciimode; |
| char last_asciimode; /* /A or /B impacts the file preceding it and becomes the new default for all files that follow */ |
| char verifyflag; |
| char lastitemwasplus; |
| unsigned short databufsz; |
| char databuf[1]; |
| }; |
| |
| |
| /* copies src to dst, overwriting or appending to the destination. |
| * - copy is performed in ASCII mode if asciiflag set (stop at first EOF in src |
| * and append an EOF in dst). |
| * - returns zero on success, DOS error code on error */ |
| static unsigned short cmd_copy_internal(const char *dst, char dstascii, const char *src, char srcascii, unsigned char appendflag, void *buff, unsigned short buffsz) { |
| unsigned short errcode = 0; |
| unsigned short srch = 0xffff, dsth = 0xffff; |
| _asm { |
| |
| /* open src */ |
| OPENSRC: |
| mov ax, 0x3d00 /* DOS 2+ -- open an existing file, read access mode */ |
| mov dx, src /* ASCIIZ fname */ |
| int 0x21 /* CF clear on success, handle in AX */ |
| jc FAIL |
| mov [srch], ax /* store src handle in memory */ |
| |
| /* check appendflag so I know if I have to try opening dst for append */ |
| xor al, al |
| or al, [appendflag] |
| jz CREATEDST |
| |
| /* try opening dst first if appendflag set */ |
| mov ax, 0x3d01 /* DOS 2+ -- open an existing file, write access mode */ |
| mov dx, dst /* ASCIIZ fname */ |
| int 0x21 /* CF clear on success, handle in AX */ |
| jc CREATEDST /* failed to open file (file does not exist) */ |
| mov [dsth], ax /* store dst handle in memory */ |
| |
| /* got file open, LSEEK to end of it now so future data is appended */ |
| mov bx, ax /* file handle in BX (was still in AX) */ |
| mov ax, 0x4202 /* DOS 2+ -- set file pointer to end of file + CX:DX */ |
| xor cx, cx /* offset zero */ |
| xor dx, dx /* offset zero */ |
| int 0x21 /* CF set on error */ |
| jc FAIL |
| jmp COPY |
| |
| /* create dst */ |
| CREATEDST: |
| mov ah, 0x3c /* DOS 2+ -- create a file */ |
| mov dx, dst |
| xor cx, cx /* zero out attributes */ |
| int 0x21 /* handle in AX on success, CF set on error */ |
| jc FAIL |
| mov [dsth], ax /* store dst handle in memory */ |
| |
| /* perform actual copy */ |
| COPY: |
| /* read a block from src */ |
| mov ah, 0x3f /* DOS 2+ -- read from file */ |
| mov bx, [srch] |
| mov cx, [buffsz] |
| mov dx, [buff] /* DX points to buffer */ |
| int 0x21 /* CF set on error, bytes read in AX (0=EOF) */ |
| jc FAIL /* abort on error */ |
| /* EOF? (ax == 0) */ |
| test ax, ax |
| jz ENDOFFILE |
| /* write block of AX bytes to dst */ |
| mov cx, ax /* block length */ |
| mov ah, 0x40 /* DOS 2+ -- write to file (CX bytes from DS:DX) */ |
| mov bx, [dsth] /* file handle */ |
| /* mov dx, [buff] */ /* DX points to buffer already */ |
| int 0x21 /* CF clear and AX=CX on success */ |
| jc FAIL |
| cmp ax, cx /* sould be equal, otherwise failed */ |
| mov ax, 0x08 /* preset to DOS error "Insufficient memory" */ |
| jne FAIL |
| jmp COPY |
| |
| ENDOFFILE: |
| /* if dst ascii mode -> add an EOF (ASCII mode not supported for the time being) */ |
| |
| jmp CLOSESRC |
| |
| FAIL: |
| mov [errcode], ax |
| |
| CLOSESRC: |
| /* close src and dst */ |
| mov bx, [srch] |
| cmp bx, 0xffff |
| je CLOSEDST |
| mov ah, 0x3e /* DOS 2+ -- close a file handle */ |
| int 0x21 |
| |
| CLOSEDST: |
| mov bx, [dsth] |
| cmp bx, 0xffff |
| je DONE |
| mov ah, 0x3e /* DOS 2+ -- close a file handle */ |
| int 0x21 |
| |
| DONE: |
| } |
| return(errcode); |
| } |
| |
| |
| static enum cmd_result cmd_copy(struct cmd_funcparam *p) { |
| struct copy_setup *setup = (void *)(p->BUFFER); |
| unsigned short i; |
| unsigned short copiedcount_in = 0, copiedcount_out = 0; /* number of input/output copied files */ |
| struct DTA *dta = (void *)0x80; /* use DTA at default location in PSP */ |
| |
| if (cmd_ishlp(p)) { |
| nls_outputnl(38,0); /* "Copies one or more files to another location." */ |
| outputnl(""); |
| nls_outputnl(38,1); /* "COPY [/A|/B] source [/A|/B] [+source [/A|/B] [+...]] [destination [/A|/B]] [/V]" */ |
| outputnl(""); |
| nls_outputnl(38,2); /* "source Specifies the file or files to be copied" */ |
| nls_outputnl(38,3); /* "/A Indicates an ASCII text file" */ |
| nls_outputnl(38,4); /* "/B Indicates a binary file" */ |
| nls_outputnl(38,5); /* "destination Specifies the directory and/or filename for the new file(s)" */ |
| nls_outputnl(38,6); /* "/V Verifies that new files are written correctly" */ |
| outputnl(""); |
| nls_outputnl(38,7); /* "To append files, specify a single file for destination, but multiple (...)" */ |
| outputnl(""); |
| nls_outputnl(38,8); /* "NOTE: /A and /B are no-ops, provided only for compatibility reasons" */ |
| return(CMD_OK); |
| } |
| |
| /* parse cmdline and fill the setup struct accordingly */ |
| |
| memset(setup, 0, sizeof(*setup)); |
| setup->databufsz = p->BUFFERSZ - sizeof(*setup); |
| |
| for (i = 0; i < p->argc; i++) { |
| |
| /* switch? */ |
| if (p->argv[i][0] == '/') { |
| if ((imatch(p->argv[i], "/a")) || (imatch(p->argv[i], "/b"))) { |
| setup->last_asciimode = 'b'; |
| if (imatch(p->argv[i], "/a")) setup->last_asciimode = 'a'; |
| /* */ |
| if (setup->dst[0] != 0) { |
| setup->dst_asciimode = setup->last_asciimode; |
| } else if (setup->src_count != 0) { |
| setup->src_asciimode[setup->src_count - 1] = setup->last_asciimode; |
| } |
| } else if (imatch(p->argv[i], "/v")) { |
| setup->verifyflag = 1; |
| } else { |
| nls_outputnl(0,2); /* "Invalid switch" */ |
| return(CMD_FAIL); |
| } |
| continue; |
| } |
| |
| /* not a switch - must be either a source, a destination or a + */ |
| if (p->argv[i][0] == '+') { |
| /* a plus cannot appear after destination or before first source */ |
| if ((setup->dst[0] != 0) || (setup->src_count == 0)) { |
| nls_outputnl(0,1); /* "Invalid syntax" */ |
| return(CMD_FAIL); |
| } |
| setup->lastitemwasplus = 1; |
| /* a plus may be immediately followed by a filename - if so, emulate |
| * a new argument */ |
| if (p->argv[i][1] != 0) { |
| p->argv[i] += 1; |
| i--; |
| } |
| continue; |
| } |
| |
| /* src? (first non-switch or something that follows a +) */ |
| if ((setup->lastitemwasplus) || (setup->src_count == 0)) { |
| setup->src[setup->src_count] = p->argv[i]; |
| setup->src_asciimode[setup->src_count] = setup->last_asciimode; |
| setup->src_count++; |
| setup->lastitemwasplus = 0; |
| continue; |
| } |
| |
| /* must be a dst then */ |
| if (setup->dst[0] != 0) { |
| nls_outputnl(0,1); /* "Invalid syntax" */ |
| return(CMD_FAIL); |
| } |
| if (file_truename(p->argv[i], setup->dst) != 0) { |
| nls_outputnl(0,8); /* "Invalid destination" */ |
| return(CMD_FAIL); |
| } |
| setup->dst_asciimode = setup->last_asciimode; |
| /* if dst is a directory then append a backslash */ |
| setup->dstlen = path_appendbkslash_if_dir(setup->dst); |
| } |
| |
| /* DEBUG: output setup content ("if 1" to enable) */ |
| #if 0 |
| printf("src: "); |
| for (i = 0; i < setup->src_count; i++) { |
| if (i != 0) printf(", "); |
| printf("%s [%c]", setup->src[i], setup->src_asciimode[i]); |
| } |
| printf("\r\n"); |
| printf("dst: %s [%c]\r\n", setup->dst, setup->dst_asciimode); |
| printf("verify: %s\r\n", (setup->verifyflag)?"ON":"OFF"); |
| #endif |
| |
| /* must have at least one source */ |
| if (setup->src_count == 0) { |
| nls_outputnl(0,7); /* "Required parameter missing" */ |
| return(CMD_FAIL); |
| } |
| |
| /* perform the operation based on setup directives: |
| * iterate over every source and copy it to dest */ |
| |
| for (i = 0; i < setup->src_count; i++) { |
| unsigned short t; |
| unsigned short cursrclen; |
| unsigned short pathendoffset; |
| |
| /* resolve truename of src and write it to buffer */ |
| t = file_truename(setup->src[i], setup->cursrc); |
| if (t != 0) { |
| output(setup->src[i]); |
| output(" - "); |
| nls_outputnl_doserr(t); |
| continue; |
| } |
| cursrclen = strlen(setup->cursrc); /* remember cursrc length */ |
| |
| /* if length zero, skip (not sure why this would be possible, though) */ |
| if (cursrclen == 0) continue; |
| |
| /* if src does not end with a backslash AND it is a directory then append a backslash */ |
| cursrclen = path_appendbkslash_if_dir(setup->cursrc); |
| |
| /* if src ends with a '\' then append *.* */ |
| if (setup->cursrc[cursrclen - 1] == '\\') { |
| strcat(setup->cursrc, "*.*"); |
| } |
| |
| /* remember where the path in cursrc ends */ |
| for (t = 0; setup->cursrc[t] != 0; t++) { |
| if (setup->cursrc[t] == '\\') pathendoffset = t + 1; |
| } |
| |
| /* */ |
| if (findfirst(dta, setup->cursrc, 0) != 0) { |
| continue; |
| } |
| |
| do { |
| char appendflag; |
| if (dta->attr & DOS_ATTR_DIR) continue; /* skip directories */ |
| |
| /* compute full path/name of the file */ |
| strcpy(setup->cursrc + pathendoffset, dta->fname); |
| |
| /* if there was no destination, then YOU are the destination now! |
| * this handles situations like COPY a.txt+b.txt+c.txt */ |
| if (setup->dst[0] == NULL) { |
| strcpy(setup->dst, setup->cursrc); |
| setup->dstlen = strlen(setup->dst); |
| copiedcount_in++; |
| copiedcount_out++; |
| continue; |
| } |
| |
| /* is dst ending with a backslash? then append fname to it */ |
| if (setup->dst[setup->dstlen - 1] == '\\') strcpy(setup->dst + setup->dstlen, dta->fname); |
| |
| /* now cursrc contains the full source and dst contains the full dest... COPY TIME! */ |
| |
| /* if dst file exists already -> overwrite it or append? |
| - if dst is a dir (dstlen-1 points at a \\) -> overwrite |
| - otherwise: if copiedcount_in==0 overwrite, else append */ |
| output(setup->cursrc); |
| if ((setup->dst[setup->dstlen - 1] == '\\') || (copiedcount_in == 0)) { |
| appendflag = 0; |
| output(" > "); |
| copiedcount_out++; |
| } else { |
| appendflag = 1; |
| output(" >> "); |
| } |
| outputnl(setup->dst); |
| |
| t = cmd_copy_internal(setup->dst, 0, setup->cursrc, 0, appendflag, setup->databuf, setup->databufsz); |
| if (t != 0) { |
| nls_outputnl_doserr(t); |
| return(CMD_FAIL); |
| } |
| |
| copiedcount_in++; |
| } while (findnext(dta) == 0); |
| |
| } |
| |
| sprintf(setup->databuf, svarlang_str(38,9)/*"%u file(s) copied"*/, copiedcount_out); |
| outputnl(setup->databuf); |
| |
| return(CMD_OK); |
| } |