/* misc.c - miscellaneous functions Copyright (c) 1999 - 2004 NoisyB Copyright (c) 2001 - 2004 dbjh Copyright (c) 2002 - 2004 Jan-Erik Karlsson (Amiga code) This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ #ifdef HAVE_CONFIG_H #include "config.h" // USE_ZLIB #endif #include #include #include #ifdef HAVE_UNISTD_H #include #endif #include #include #include #include #include // va_arg() #include // for S_IFLNK #ifdef __MSDOS__ #include // delay(), milliseconds #elif defined __unix__ #include // usleep(), microseconds #elif defined __BEOS__ #include // snooze(), microseconds // Include OS.h before misc.h, because OS.h includes StorageDefs.h which // includes param.h which unconditionally defines MIN and MAX. #elif defined AMIGA #include #include #include #include #include // GetKey() #include #include #elif defined _WIN32 #include // Sleep(), milliseconds #endif #ifdef USE_ZLIB #include "archive.h" #endif #include "file.h" #include "misc.h" #include "getopt.h" // struct option #ifdef __CYGWIN__ // under Cygwin (gcc for Windows) we #define USE_POLL // need poll() for kbhit(). poll() #include // is available on Linux, not on #endif // BeOS. DOS already has kbhit() #if (defined __unix__ && !defined __MSDOS__) || defined __BEOS__ || \ defined __APPLE__ // Mac OS X actually #include typedef struct termios tty_t; #endif #ifdef DJGPP #include // needed for __dpmi_int() by ansi_init() #ifdef DLL #include "dxedll_priv.h" #endif #endif #ifdef MAXBUFSIZE #undef MAXBUFSIZE #endif // MAXBUFSIZE #define MAXBUFSIZE 32768 extern int errno; typedef struct st_func_node { void (*func) (void); struct st_func_node *next; } st_func_node_t; static st_func_node_t func_list = { NULL, NULL }; static int func_list_locked = 0; static int misc_ansi_color = 0; #if (defined __unix__ && !defined __MSDOS__) || defined __BEOS__ || \ defined __APPLE__ // Mac OS X actually static void set_tty (tty_t *param); #endif #if defined _WIN32 && defined USE_ANSI_COLOR int vprintf2 (const char *format, va_list argptr) // Cheap hack to get the Visual C++ and MinGW ports support "ANSI colors". // Cheap, because it only supports the ANSI escape sequences uCON64 uses. { #undef printf #undef fprintf int n_chars = 0, n_ctrl = 0, n_print, done = 0; char output[MAXBUFSIZE], *ptr, *ptr2; HANDLE stdout_handle; CONSOLE_SCREEN_BUFFER_INFO info; WORD org_attr, new_attr = 0; n_chars = _vsnprintf (output, MAXBUFSIZE, format, argptr); if (n_chars == -1) { fprintf (stderr, "INTERNAL ERROR: Output buffer in vprintf2() is too small (%d bytes).\n" " Please send a bug report\n", MAXBUFSIZE); exit (1); } if ((ptr = strchr (output, 0x1b)) == NULL) fputs (output, stdout); else { stdout_handle = GetStdHandle (STD_OUTPUT_HANDLE); GetConsoleScreenBufferInfo (stdout_handle, &info); org_attr = info.wAttributes; if (ptr > output) { *ptr = 0; fputs (output, stdout); *ptr = 0x1b; } while (!done) { if (memcmp (ptr, "\x1b[0m", 4) == 0) { new_attr = org_attr; n_ctrl = 4; } else if (memcmp (ptr, "\x1b[01;31m", 8) == 0) { new_attr = FOREGROUND_INTENSITY | FOREGROUND_RED; n_ctrl = 8; } else if (memcmp (ptr, "\x1b[01;32m", 8) == 0) { new_attr = FOREGROUND_INTENSITY | FOREGROUND_GREEN; n_ctrl = 8; } else if (memcmp (ptr, "\x1b[01;33m", 8) == 0) // bright yellow { new_attr = FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN; n_ctrl = 8; } else if (memcmp (ptr, "\x1b[31;41m", 8) == 0) { new_attr = FOREGROUND_RED | BACKGROUND_RED; n_ctrl = 8; } else if (memcmp (ptr, "\x1b[32;42m", 8) == 0) { new_attr = FOREGROUND_GREEN | BACKGROUND_GREEN; n_ctrl = 8; } else if (memcmp (ptr, "\x1b[30;41m", 8) == 0) // 30 = foreground black { new_attr = BACKGROUND_RED; n_ctrl = 8; } else if (memcmp (ptr, "\x1b[30;42m", 8) == 0) { new_attr = BACKGROUND_GREEN; n_ctrl = 8; } else if (*ptr == 0x1b) { new_attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; SetConsoleTextAttribute (stdout_handle, new_attr); printf ("\n" "INTERNAL WARNING: vprintf2() encountered an unsupported ANSI escape sequence\n" " Please send a bug report\n"); n_ctrl = 0; } SetConsoleTextAttribute (stdout_handle, new_attr); ptr2 = strchr (ptr + 1, 0x1b); if (ptr2) n_print = ptr2 - ptr; else { n_print = strlen (ptr); done = 1; } ptr[n_print] = 0; ptr += n_ctrl; fputs (ptr, stdout); (ptr - n_ctrl)[n_print] = 0x1b; ptr = ptr2; } } return n_chars; #define printf printf2 #define fprintf fprintf2 } int printf2 (const char *format, ...) { va_list argptr; int n_chars; va_start (argptr, format); n_chars = vprintf2 (format, argptr); va_end (argptr); return n_chars; } int fprintf2 (FILE *file, const char *format, ...) { va_list argptr; int n_chars; va_start (argptr, format); if (file != stdout) n_chars = vfprintf (file, format, argptr); else n_chars = vprintf2 (format, argptr); va_end (argptr); return n_chars; } #endif // defined _WIN32 && defined USE_ANSI_COLOR void clear_line (void) /* This function is used to fix a problem when using the MinGW or Visual C++ port under Windows 98 (probably Windows 95 too) while ANSI.SYS is not loaded. If a line contains colors, printed with printf() or fprintf() (actually printf2() or fprintf2()), it cannot be cleared by printing spaces on the same line. A solution is using SetConsoleTextAttribute(). The problem doesn't occur if ANSI.SYS is loaded. It also doesn't occur under Windows XP, even if ANSI.SYS isn't loaded. We print 79 spaces (not 80), because under command.com the cursor advances to the next line if we print something on the 80th column (in 80 column mode). This doesn't happen under xterm. */ { #if !defined _WIN32 || !defined USE_ANSI_COLOR fputs ("\r \r", stdout); #else WORD org_attr; CONSOLE_SCREEN_BUFFER_INFO info; HANDLE stdout_handle = GetStdHandle (STD_OUTPUT_HANDLE); GetConsoleScreenBufferInfo (stdout_handle, &info); org_attr = info.wAttributes; SetConsoleTextAttribute (stdout_handle, FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE); fputs ("\r \r", stdout); SetConsoleTextAttribute (stdout_handle, org_attr); #endif } int ansi_init (void) { int result = isatty (STDOUT_FILENO); #ifdef DJGPP if (result) { // Don't use __MSDOS__, because __dpmi_regs and __dpmi_int are DJGPP specific __dpmi_regs reg; reg.x.ax = 0x1a00; // DOS 4.0+ ANSI.SYS installation check __dpmi_int (0x2f, ®); if (reg.h.al != 0xff) // AL == 0xff if installed result = 0; } #endif misc_ansi_color = result; return result; } void dumper (FILE *output, const void *buffer, size_t bufferlen, int virtual_start, unsigned int flags) // Do NOT use DUMPER_PRINT in uCON64 code - dbjh { #define DUMPER_REPLACER ('.') size_t pos; char buf[17]; const unsigned char *p = (const unsigned char *) buffer; memset (buf, 0, sizeof (buf)); for (pos = 0; pos < bufferlen; pos++, p++) if (flags & DUMPER_PRINT) { fprintf (output, "%c", isprint (*p) || #ifdef USE_ANSI_COLOR *p == 0x1b || // ESC #endif isspace (*p) ? *p : DUMPER_REPLACER); } else if (flags & DUMPER_DUAL) { if (!(pos & 3)) fprintf (output, (flags & DUMPER_DEC_COUNT ? "%010d " : "%08x "), (int) (pos + virtual_start)); fprintf (output, "%02x %08d ", *p, ((*p >> 7) & 1) * 10000000 + ((*p >> 6) & 1) * 1000000 + ((*p >> 5) & 1) * 100000 + ((*p >> 4) & 1) * 10000 + ((*p >> 3) & 1) * 1000 + ((*p >> 2) & 1) * 100 + ((*p >> 1) & 1) * 10 + (*p & 1)); *(buf + (pos & 3)) = isprint (*p) ? *p : DUMPER_REPLACER; if (!((pos + 1) & 3)) fprintf (output, "%s\n", buf); } else if (flags & DUMPER_CODE) { fprintf (output, "0x%02x, ", *p); if (!((pos + 1) & 7)) fprintf (output, (flags & DUMPER_DEC_COUNT ? "// (%d) 0x%x\n" : "// 0x%x (%d)\n"), (int) (pos + virtual_start + 1), (int) (pos + virtual_start + 1)); } else // if (flags & DUMPER_HEX) // default { if (!(pos & 15)) fprintf (output, (flags & DUMPER_DEC_COUNT ? "%08d " : "%08x "), (int) (pos + virtual_start)); fprintf (output, (pos + 1) & 3 ? "%02x " : "%02x ", *p); *(buf + (pos & 15)) = isprint (*p) ? *p : DUMPER_REPLACER; if (!((pos + 1) & 15)) fprintf (output, "%s\n", buf); } if (flags & DUMPER_PRINT) return; else if (flags & DUMPER_DUAL) { if (pos & 3) { *(buf + (pos & 3)) = 0; fprintf (output, "%s\n", buf); } } else if (flags & DUMPER_CODE) return; else // if (flags & DUMPER_HEX) // default { if (pos & 15) { *(buf + (pos & 15)) = 0; fprintf (output, "%s\n", buf); } } } int change_mem (char *buf, int bufsize, char *searchstr, int strsize, char wc, char esc, char *newstr, int newsize, int offset, ...) // convenience wrapper for change_mem2() { va_list argptr; int i, n_esc = 0, retval; st_cm_set_t *sets; va_start (argptr, offset); for (i = 0; i < strsize; i++) if (searchstr[i] == esc) n_esc++; sets = (st_cm_set_t *) malloc (n_esc * sizeof (st_cm_set_t)); va_start (argptr, offset); for (i = 0; i < n_esc; i++) { sets[i].data = va_arg (argptr, char *); // get next set of characters sets[i].size = va_arg (argptr, int); // get set size } va_end (argptr); retval = change_mem2 (buf, bufsize, searchstr, strsize, wc, esc, newstr, newsize, offset, sets); free (sets); return retval; } int change_mem2 (char *buf, int bufsize, char *searchstr, int strsize, char wc, char esc, char *newstr, int newsize, int offset, st_cm_set_t *sets) /* Search for all occurrences of string searchstr in buf and replace newsize bytes in buf by copying string newstr to the end of the found search string in buf plus offset. If searchstr contains wildcard characters wc, then n wildcard characters in searchstr match any n characters in buf. If searchstr contains escape characters esc, sets must point to an array of sets. sets must contain as many elements as there are escape characters in searchstr. searchstr matches for an escape character if one of the characters in sets[i]->data matches. Note that searchstr is not necessarily a C string; it may contain one or more zero bytes as strsize indicates the length. offset is the relative offset from the last character in searchstring and may have a negative value. The return value is the number of times a match was found. This function was written to patch SNES ROM dumps. It does basically the same as the old uCON does, with one exception, the line with: bufpos -= n_wc; As stated in the comment, this causes the search to restart at the first wildcard character of the sequence of wildcards that was most recently skipped if the current character in buf didn't match the current character in searchstr. This makes change_mem() behave a bit more intuitive. For example char str[] = "f foobar means..."; change_mem (str, strlen (str), "f**bar", 6, '*', '!', "XXXXXXXX", 8, 2, NULL); finds and changes "foobar means..." into "foobar XXXXXXXX", while with uCON's algorithm it would not (but does the job good enough for patching SNES ROMs). One example of using sets: char str[] = "fu-bar is the same as foobar "; st_cm_set_t sets[] = {{"o-", 2}, {"uo", 2}}; change_mem (str, strlen (str), "f!!", 3, '*', '!', "fighter", 7, 1, sets); This changes str into "fu-fighter is the same as foofighter". */ { char *set; int bufpos, strpos = 0, pos_1st_esc = -1, setsize, i, n_wc, n_matches = 0, setindex = 0; for (bufpos = 0; bufpos < bufsize; bufpos++) { if (strpos == 0 && searchstr[0] != esc && searchstr[0] != wc) while (bufpos < bufsize && searchstr[0] != buf[bufpos]) bufpos++; // handle escape character in searchstr while (searchstr[strpos] == esc && bufpos < bufsize) { if (strpos == pos_1st_esc) setindex = 0; // reset argument pointer if (pos_1st_esc == -1) pos_1st_esc = strpos; set = sets[setindex].data; // get next set of characters setsize = sets[setindex].size; // get set size setindex++; i = 0; // see if buf[bufpos] matches with any character in current set while (i < setsize && buf[bufpos] != set[i]) i++; if (i == setsize) break; // buf[bufpos] didn't match with any char if (strpos == strsize - 1) // check if we are at the end of searchstr { memcpy (buf + bufpos + offset, newstr, newsize); n_matches++; break; } strpos++; bufpos++; } if (searchstr[strpos] == esc) { strpos = 0; continue; } // skip wildcards in searchstr n_wc = 0; while (searchstr[strpos] == wc && bufpos < bufsize) { if (strpos == strsize - 1) // check if at end of searchstr { memcpy (buf + bufpos + offset, newstr, newsize); n_matches++; break; } strpos++; bufpos++; n_wc++; } if (bufpos == bufsize) break; if (searchstr[strpos] == wc) { strpos = 0; continue; } if (searchstr[strpos] == esc) { bufpos--; // current char has to be checked, but `for' continue; // increments bufpos } // no escape char, no wildcard -> normal character if (searchstr[strpos] == buf[bufpos]) { if (strpos == strsize - 1) // check if at end of searchstr { memcpy (buf + bufpos + offset, newstr, newsize); n_matches++; strpos = 0; } else strpos++; } else { bufpos -= n_wc; // scan the most recent wildcards too if if (strpos > 0) // the character didn't match { bufpos--; // current char has to be checked, but `for' strpos = 0; // increments bufpos } } } return n_matches; } int build_cm_patterns (st_cm_pattern_t **patterns, const char *filename, int verbose) /* This function goes a bit over the top what memory allocation technique concerns, but at least it's stable. Note the important difference between (*patterns)[0].n_sets and patterns[0]->n_sets (not especially that member). I (dbjh) am too ashamed to tell how long it took me to finally realise that... */ { char src_name[FILENAME_MAX], line[MAXBUFSIZE], buffer[MAXBUFSIZE], *token, *last, *ptr; int line_num = 0, n_sets, n_codes = 0, n, currentsize1, requiredsize1, currentsize2, requiredsize2, currentsize3, requiredsize3; FILE *srcfile; strcpy (src_name, filename); if (access (src_name, F_OK | R_OK)) return -1; // NOT an error, it's optional if ((srcfile = fopen (src_name, "r")) == NULL) // open in text mode { fprintf (stderr, "ERROR: Can't open \"%s\" for reading\n", src_name); return -1; } *patterns = NULL; currentsize1 = requiredsize1 = 0; while (fgets (line, sizeof line, srcfile) != NULL) { line_num++; n_sets = 0; ptr = line + strspn (line, "\t "); if (*ptr == '#' || *ptr == '\n' || *ptr == '\r') continue; if ((ptr = strpbrk (line, "\n\r#"))) // text after # is comment *ptr = 0; requiredsize1 += sizeof (st_cm_pattern_t); if (requiredsize1 > currentsize1) { currentsize1 = requiredsize1 + 10 * sizeof (st_cm_pattern_t); if (!(*patterns = (st_cm_pattern_t *) realloc (*patterns, currentsize1))) { fprintf (stderr, "ERROR: Not enough memory for buffer (%d bytes)\n", currentsize1); return -1; } } (*patterns)[n_codes].search = NULL; currentsize2 = 0; requiredsize2 = 1; // for string terminator n = 0; strcpy (buffer, line); token = strtok (buffer, ":"); token = strtok (token, " "); // printf ("token: \"%s\"\n", token); last = token; // token is never NULL here (yes, tested with empty files and such) do { requiredsize2++; if (requiredsize2 > currentsize2) { currentsize2 = requiredsize2 + 10; if (!((*patterns)[n_codes].search = (char *) realloc ((*patterns)[n_codes].search, currentsize2))) { fprintf (stderr, "ERROR: Not enough memory for buffer (%d bytes)\n", currentsize2); free (*patterns); *patterns = NULL; return -1; } } (*patterns)[n_codes].search[n] = (unsigned char) strtol (token, NULL, 16); n++; } while ((token = strtok (NULL, " "))); (*patterns)[n_codes].search_size = n; // size in bytes strcpy (buffer, line); token = strtok (last, ":"); token = strtok (NULL, ":"); token = strtok (token, " "); last = token; if (!token) { printf ("WARNING: Line %d is invalid, no wildcard value is specified\n", line_num); continue; } (*patterns)[n_codes].wildcard = (char) strtol (token, NULL, 16); strcpy (buffer, line); token = strtok (last, ":"); token = strtok (NULL, ":"); token = strtok (token, " "); last = token; if (!token) { printf ("WARNING: Line %d is invalid, no escape value is specified\n", line_num); continue; } (*patterns)[n_codes].escape = (char) strtol (token, NULL, 16); strcpy (buffer, line); token = strtok (last, ":"); token = strtok (NULL, ":"); token = strtok (token, " "); last = token; if (!token) { printf ("WARNING: Line %d is invalid, no replacement is specified\n", line_num); continue; } (*patterns)[n_codes].replace = NULL; currentsize2 = 0; requiredsize2 = 1; // for string terminator n = 0; do { requiredsize2++; if (requiredsize2 > currentsize2) { currentsize2 = requiredsize2 + 10; if (!((*patterns)[n_codes].replace = (char *) realloc ((*patterns)[n_codes].replace, currentsize2))) { fprintf (stderr, "ERROR: Not enough memory for buffer (%d bytes)\n", currentsize2); free ((*patterns)[n_codes].search); free (*patterns); *patterns = NULL; return -1; } } (*patterns)[n_codes].replace[n] = (unsigned char) strtol (token, NULL, 16); n++; } while ((token = strtok (NULL, " "))); (*patterns)[n_codes].replace_size = n; // size in bytes strcpy (buffer, line); token = strtok (last, ":"); token = strtok (NULL, ":"); token = strtok (token, " "); last = token; if (!token) { printf ("WARNING: Line %d is invalid, no offset is specified\n", line_num); continue; } (*patterns)[n_codes].offset = strtol (token, NULL, 10); // yes, offset is decimal if (verbose) { printf ("\n" "line: %d\n" "searchstring: ", line_num); for (n = 0; n < (*patterns)[n_codes].search_size; n++) printf ("%02x ", (unsigned char) (*patterns)[n_codes].search[n]); printf ("(%d)\n" "wildcard: %02x\n" "escape: %02x\n" "replacement: ", (*patterns)[n_codes].search_size, (unsigned char) (*patterns)[n_codes].wildcard, (unsigned char) (*patterns)[n_codes].escape); for (n = 0; n < (*patterns)[n_codes].replace_size; n++) printf ("%02x ", (unsigned char) (*patterns)[n_codes].replace[n]); printf ("(%d)\n" "offset: %d\n", (*patterns)[n_codes].replace_size, (*patterns)[n_codes].offset); } (*patterns)[n_codes].sets = NULL; currentsize2 = 0; requiredsize2 = 1; // for string terminator strcpy (buffer, line); token = strtok (last, ":"); token = strtok (NULL, ":"); last = token; while (token) { requiredsize2 += sizeof (st_cm_set_t); if (requiredsize2 > currentsize2) { currentsize2 = requiredsize2 + 10 * sizeof (st_cm_set_t); if (!((*patterns)[n_codes].sets = (st_cm_set_t *) realloc ((*patterns)[n_codes].sets, currentsize2))) { fprintf (stderr, "ERROR: Not enough memory for buffer (%d bytes)\n", currentsize2); free ((*patterns)[n_codes].replace); free ((*patterns)[n_codes].search); free (*patterns); *patterns = NULL; return -1; } } (*patterns)[n_codes].sets[n_sets].data = NULL; currentsize3 = 0; requiredsize3 = 1; // for string terminator n = 0; token = strtok (token, " "); do { requiredsize3++; if (requiredsize3 > currentsize3) { currentsize3 = requiredsize3 + 10; if (!((*patterns)[n_codes].sets[n_sets].data = (char *) realloc ((*patterns)[n_codes].sets[n_sets].data, currentsize3))) { fprintf (stderr, "ERROR: Not enough memory for buffer (%d bytes)\n", currentsize3); free ((*patterns)[n_codes].sets); free ((*patterns)[n_codes].replace); free ((*patterns)[n_codes].search); free (*patterns); *patterns = NULL; return -1; } } (*patterns)[n_codes].sets[n_sets].data[n] = (unsigned char) strtol (token, NULL, 16); n++; } while ((token = strtok (NULL, " "))); (*patterns)[n_codes].sets[n_sets].size = n; if (verbose) { printf ("set: "); for (n = 0; n < (*patterns)[n_codes].sets[n_sets].size; n++) printf ("%02x ", (unsigned char) (*patterns)[n_codes].sets[n_sets].data[n]); printf ("(%d)\n", (*patterns)[n_codes].sets[n_sets].size); } strcpy (buffer, line); token = strtok (last, ":"); token = strtok (NULL, ":"); last = token; n_sets++; } (*patterns)[n_codes].n_sets = n_sets; n_codes++; } fclose (srcfile); return n_codes; } void cleanup_cm_patterns (st_cm_pattern_t **patterns, int n_patterns) { int n, m; for (n = 0; n < n_patterns; n++) { free ((*patterns)[n].search); (*patterns)[n].search = NULL; free ((*patterns)[n].replace); (*patterns)[n].replace = NULL; for (m = 0; m < (*patterns)[n].n_sets; m++) { free ((*patterns)[n].sets[m].data); (*patterns)[n].sets[m].data = NULL; } free ((*patterns)[n].sets); (*patterns)[n].sets = NULL; } free (*patterns); *patterns = NULL; } #if 1 int gauge (FILE *output, time_t start_time, int pos, int size, unsigned int flags) { #define GAUGE_LENGTH ((int64_t) 24) int curr, bps, left, p, percentage; char progress[1024]; if (pos > size || !size) return -1; percentage = (int) ((((int64_t) 100) * pos) / size); if (/* output != stdout && */ flags & GAUGE_PERCENT) { fprintf (output, "%u\n", percentage); fflush (output); return 0; } if ((curr = time (0) - start_time) == 0) curr = 1; // `round up' to at least 1 sec (no division // by zero below) bps = pos / curr; // # bytes/second (average transfer speed) left = size - pos; left /= bps ? bps : 1; p = (int) ((GAUGE_LENGTH * pos) / size); *progress = 0; strncat (progress, "========================", p); // if (flags & GAUGE_ANSI) if (misc_ansi_color) { progress[p] = 0; if (p < GAUGE_LENGTH) strcat (progress, "\x1b[31;41m"); } strncat (&progress[p], "------------------------", (int) (GAUGE_LENGTH - p)); fprintf (output, // (flags & GAUGE_ANSI) ? misc_ansi_color ? "\r%10d Bytes [\x1b[32;42m%s\x1b[0m] %d%%, BPS=%d, " : "\r%10d Bytes [%s] %d%%, BPS=%d, ", pos, progress, percentage, bps); if (pos == size) fprintf (output, "TOTAL=%02d:%02d", curr / 60, curr % 60); // DON'T print a newline else if (pos) // -> gauge can be cleared fprintf (output, "ETA=%02d:%02d ", left / 60, left % 60); else // don't display a nonsense ETA fputs ("ETA=? ", stdout); fflush (output); return 0; } #else int gauge (time_t init_time, int pos, int size, unsigned int flags) { #define GAUGE_LENGTH ((int64_t) 24) int curr, bps, left, p, percentage; char progress[MAXBUFSIZE]; if (pos > size || !size) return -1; if ((curr = time (0) - init_time) == 0) curr = 1; // `round up' to at least 1 sec (no division // by zero below) bps = pos / curr; // # bytes/second (average transfer speed) left = size - pos; left /= bps ? bps : 1; p = (int) ((GAUGE_LENGTH * pos) / size); *progress = 0; strncat (progress, "========================", p); if (misc_ansi_color) { progress[p] = 0; if (p < GAUGE_LENGTH) strcat (progress, "\x1b[31;41m"); } strncat (&progress[p], "------------------------", (int) (GAUGE_LENGTH - p)); percentage = (int) ((((int64_t) 100) * pos) / size); printf ( misc_ansi_color ? "\r%10d Bytes [\x1b[32;42m%s\x1b[0m] %d%%, BPS=%d, " : "\r%10d Bytes [%s] %d%%, BPS=%d, ", pos, progress, percentage, bps); if (pos == size) printf ("TOTAL=%02d:%02d", curr / 60, curr % 60); // DON'T print a newline else if (pos) // -> gauge can be cleared printf ("ETA=%02d:%02d ", left / 60, left % 60); else // don't display a nonsense ETA fputs ("ETA=? ", stdout); fflush (stdout); return 0; } #endif #ifdef __CYGWIN__ /* Weird problem with combination Cygwin uCON64 exe and cmd.exe (Bash is ok): When a string with "e (e with diaeresis, one character) is read from an environment variable, the character isn't the right character for accessing the file system. We fix this. TODO: fix the same problem for other non-ASCII characters (> 127). */ char * fix_character_set (char *str) { int n, l = strlen (str); unsigned char *ptr = (unsigned char *) str; for (n = 0; n < l; n++) { if (ptr[n] == 0x89) // e diaeresis ptr[n] = 0xeb; else if (ptr[n] == 0x84) // a diaeresis ptr[n] = 0xe4; else if (ptr[n] == 0x8b) // i diaeresis ptr[n] = 0xef; else if (ptr[n] == 0x94) // o diaeresis ptr[n] = 0xf6; else if (ptr[n] == 0x81) // u diaeresis ptr[n] = 0xfc; } return str; } #endif char * getenv2 (const char *variable) /* getenv() suitable for enviroments w/o HOME, TMP or TEMP variables. The caller should copy the returned string to it's own memory, because this function will overwrite that memory on the next call. Note that this function never returns NULL. */ { char *tmp; static char value[MAXBUFSIZE]; #if defined __CYGWIN__ || defined __MSDOS__ /* Under DOS and Windows the environment variables are not stored in a case sensitive manner. The run-time systems of DJGPP and Cygwin act as if they are stored in upper case. Their getenv() however *is* case sensitive. We fix this by changing all characters of the search string (variable) to upper case. Note that under Cygwin's Bash environment variables *are* stored in a case sensitive manner. */ char tmp2[MAXBUFSIZE]; strcpy (tmp2, variable); variable = strupr (tmp2); // DON'T copy the string into variable #endif // (variable itself is local) *value = 0; if ((tmp = getenv (variable)) != NULL) strcpy (value, tmp); else { if (!strcmp (variable, "HOME")) { if ((tmp = getenv ("USERPROFILE")) != NULL) strcpy (value, tmp); else if ((tmp = getenv ("HOMEDRIVE")) != NULL) { strcpy (value, tmp); tmp = getenv ("HOMEPATH"); strcat (value, tmp ? tmp : FILE_SEPARATOR_S); } else /* Don't just use C:\\ under DOS, the user might not have write access there (Windows NT DOS-box). Besides, it would make uCON64 behave differently on DOS than on the other platforms. Returning the current directory when none of the above environment variables are set can be seen as a feature. A frontend could execute uCON64 with an environment without any of the environment variables set, so that the directory from where uCON64 starts will be used. */ { char c; getcwd (value, FILENAME_MAX); c = toupper (*value); // if current dir is root dir strip problematic ending slash (DJGPP) if (c >= 'A' && c <= 'Z' && value[1] == ':' && value[2] == '/' && value[3] == 0) value[2] = 0; } } if (!strcmp (variable, "TEMP") || !strcmp (variable, "TMP")) { #if defined __MSDOS__ || defined __CYGWIN__ /* DJGPP and (yet another) Cygwin quirck A trailing backslash is used to check for a directory. Normally DJGPP's run-time system is able to handle forward slashes in paths, but access() won't differentiate between files and dirs if a forward slash is used. Cygwin's run-time system seems to handle paths with forward slashes quite different from paths with backslashes. This trick seems to work only if a backslash is used. */ if (access ("\\tmp\\", R_OK | W_OK) == 0) #else // trailing file separator to force it to be a directory if (access (FILE_SEPARATOR_S"tmp"FILE_SEPARATOR_S, R_OK | W_OK) == 0) #endif strcpy (value, FILE_SEPARATOR_S"tmp"); else getcwd (value, FILENAME_MAX); } } #ifdef __CYGWIN__ /* Under certain circumstances Cygwin's run-time system returns "/" as value of HOME while that var has not been set. To specify a root dir a path like /cygdrive/ or simply a drive letter should be used. */ if (!strcmp (variable, "HOME") && !strcmp (value, "/")) getcwd (value, FILENAME_MAX); return fix_character_set (value); #else return value; #endif } long int strtol2 (const char *str, char **tail) { long int i; return ((i = strtol (str, tail, 10))) ? i : strtol (str, tail, 16); } #if (defined __unix__ && !defined __MSDOS__) || defined __BEOS__ || \ defined __APPLE__ // Mac OS X actually static int oldtty_set = 0, stdin_tty = 1; // 1 => stdin is a tty, 0 => it's not static tty_t oldtty, newtty; void set_tty (tty_t *param) { if (stdin_tty && tcsetattr (STDIN_FILENO, TCSANOW, param) == -1) { fprintf (stderr, "ERROR: Could not set tty parameters\n"); exit (100); } } /* This code compiles with DJGPP, but is not neccesary. Our kbhit() conflicts with DJGPP's one, so it won't be used for that function. Perhaps it works for making getchar() behave like getch(), but that's a bit pointless. */ void init_conio (void) { if (!isatty (STDIN_FILENO)) { stdin_tty = 0; return; // rest is nonsense if not a tty } if (tcgetattr (STDIN_FILENO, &oldtty) == -1) { fprintf (stderr, "ERROR: Could not get tty parameters\n"); exit (101); } oldtty_set = 1; if (register_func (deinit_conio) == -1) { fprintf (stderr, "ERROR: Could not register function with register_func()\n"); exit (102); } newtty = oldtty; newtty.c_lflag &= ~(ICANON | ECHO); newtty.c_lflag |= ISIG; newtty.c_cc[VMIN] = 1; // if VMIN != 0, read calls newtty.c_cc[VTIME] = 0; // block (wait for input) set_tty (&newtty); } void deinit_conio (void) { if (oldtty_set) { tcsetattr (STDIN_FILENO, TCSAFLUSH, &oldtty); oldtty_set = 0; } } #if defined __CYGWIN__ && !defined USE_POLL #warning kbhit() does not work properly in Cygwin executable if USE_POLL is not defined #endif // this kbhit() conflicts with DJGPP's one int kbhit (void) { #ifdef USE_POLL struct pollfd fd; fd.fd = STDIN_FILENO; fd.events = POLLIN; fd.revents = 0; return poll (&fd, 1, 0) > 0; #else tty_t tmptty = newtty; int ch, key_pressed; tmptty.c_cc[VMIN] = 0; // doesn't work as expected under set_tty (&tmptty); // Cygwin (define USE_POLL) if ((ch = fgetc (stdin)) != EOF) { key_pressed = 1; ungetc (ch, stdin); } else key_pressed = 0; set_tty (&newtty); return key_pressed; #endif } #elif defined AMIGA // (__unix__ && !__MSDOS__) || int // __BEOS__ ||__APPLE__ kbhit (void) { return GetKey () != 0xff ? 1 : 0; } int getch (void) { BPTR con_fileh; int temp; con_fileh = Input (); // put the console into RAW mode which makes getchar() behave like getch()? if (con_fileh) SetMode (con_fileh, 1); temp = getchar (); // put the console out of RAW mode (might make sense) if (con_fileh) SetMode (con_fileh, 0); return temp; } #endif // AMIGA #if defined __unix__ && !defined __MSDOS__ int drop_privileges (void) { uid_t uid; gid_t gid; uid = getuid (); if (setuid (uid) == -1) { fprintf (stderr, "ERROR: Could not set uid\n"); return 1; } gid = getgid (); // This shouldn't be necessary if (setgid (gid) == -1) // if `make install' was { // used, but just in case fprintf (stderr, "ERROR: Could not set gid\n"); // (root did `chmod +s') return 1; } return 0; } #endif int register_func (void (*func) (void)) { st_func_node_t *func_node = &func_list, *new_node; while (func_node->next != NULL) func_node = func_node->next; if ((new_node = (st_func_node_t *) malloc (sizeof (st_func_node_t))) == NULL) return -1; new_node->func = func; new_node->next = NULL; func_node->next = new_node; return 0; } int unregister_func (void (*func) (void)) { st_func_node_t *func_node = &func_list, *prev_node = &func_list; while (func_node->next != NULL && func_node->func != func) { prev_node = func_node; func_node = func_node->next; } if (func_node->func != func) return -1; if (!func_list_locked) { prev_node->next = func_node->next; free (func_node); return 0; } else return -1; } void handle_registered_funcs (void) { st_func_node_t *func_node = &func_list; func_list_locked = 1; while (func_node->next != NULL) { func_node = func_node->next; // first node contains no valid address if (func_node->func != NULL) func_node->func (); } func_list_locked = 0; } void wait2 (int nmillis) { #ifdef __MSDOS__ delay (nmillis); #elif defined __unix__ || defined __APPLE__ // Mac OS X actually usleep (nmillis * 1000); #elif defined __BEOS__ snooze (nmillis * 1000); #elif defined AMIGA Delay (nmillis * 1000); #elif defined _WIN32 Sleep (nmillis); #else #ifdef __GNUC__ #warning Please provide a wait2() implementation #else #pragma message ("Please provide a wait2() implementation") #endif volatile int n; for (n = 0; n < nmillis * 65536; n++) ; #endif } #ifdef _WIN32 int truncate (const char *path, off_t size) { int retval; HANDLE file = CreateFile (path, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); if (file == INVALID_HANDLE_VALUE) return -1; SetFilePointer (file, size, 0, FILE_BEGIN); retval = SetEndOfFile (file); // returns nonzero on success CloseHandle (file); return retval ? 0 : -1; // truncate() returns zero on success } int sync (void) { _commit (fileno (stdout)); _commit (fileno (stderr)); fflush (NULL); // flushes all streams opened for output return 0; } #if defined __MINGW32__ && defined DLL // Ugly hack in order to fix something in zlib FILE * fdopen (int fd, const char *mode) { return _fdopen (fd, mode); } #endif #elif defined AMIGA // _WIN32 int truncate (const char *path, off_t size) { BPTR fh; ULONG newsize; if (!(fh = Open (path, MODE_OLDFILE))) return -1; newsize = SetFileSize (fh, size, OFFSET_BEGINNING); Close (fh); return newsize == (ULONG) size ? 0 : -1; // truncate() returns zero on success } int chmod (const char *path, mode_t mode) { if (!SetProtection ((STRPTR) path, ((mode & S_IRUSR ? 0 : FIBF_READ) | (mode & S_IWUSR ? 0 : FIBF_WRITE | FIBF_DELETE) | (mode & S_IXUSR ? 0 : FIBF_EXECUTE) | (mode & S_IRGRP ? FIBF_GRP_READ : 0) | (mode & S_IWGRP ? FIBF_GRP_WRITE | FIBF_GRP_DELETE : 0) | (mode & S_IXGRP ? FIBF_GRP_EXECUTE : 0) | (mode & S_IROTH ? FIBF_OTR_READ : 0) | (mode & S_IWOTH ? FIBF_OTR_WRITE | FIBF_OTR_DELETE : 0) | (mode & S_IXOTH ? FIBF_OTR_EXECUTE : 0)))) return -1; else return 0; } void sync (void) { } int readlink (const char *path, char *buf, int bufsize) { (void) path; // warning remover (void) buf; // idem (void) bufsize; // idem // always return -1 as if anything passed to it isn't a soft link return -1; } // custom _popen() and _pclose(), because the standard ones (named popen() and // pclose()) are buggy FILE * _popen (const char *path, const char *mode) { int fd; BPTR fh; long fhflags; char *apipe = malloc (strlen (path) + 7); if (!apipe) return NULL; strcpy (apipe, "APIPE:"); strcat (apipe, path); if (*mode == 'w') fhflags = MODE_NEWFILE; else fhflags = MODE_OLDFILE; if (!(fh = Open (apipe, fhflags))) return NULL; return fdopen (fd, mode); } int _pclose (FILE *stream) { return fclose (stream); } #endif // AMIGA