examples: factorize strcpy

example1/example1.cpp

@@ -20,6 +20,7 @@
 #include "dds_defs.h"
 
 #include "crn_core.h"
+#include "crn_strutils.h"
 #include "crn_file_utils.h"
 
 // stb_image, for loading/saving image files.
@@ -313,11 +314,7 @@ int main(int argc, char* argv[]) {
  if (++i >= argc)
  return error("Expected output filename!");
 
-#if defined(_WIN32)
- strcpy_s(out_filename, sizeof(out_filename), argv[i]);
-#else
- strncpy(out_filename, argv[i], sizeof(out_filename));
-#endif
+ strcpy_safe(out_filename, sizeof(out_filename), argv[i]);
  } else if (!crnlib_stricmp(argv[i], "-nonsrgb"))
  srgb_colorspace = false;
  else if (!crnlib_stricmp(argv[i], "-nomips"))
@@ -482,11 +479,7 @@ int main(int argc, char* argv[]) {
  char dst_filename[FILENAME_MAX];
  crnlib_snprintf(dst_filename, sizeof(dst_filename), "%s%s%s%s", drive.get_ptr(), dir.get_ptr(), fname.get_ptr(), output_crn ? ".crn" : ".dds");
  if (out_filename[0])
-#if defined(_WIN32)
- strcpy_s(dst_filename, sizeof(dst_filename), out_filename);
-#else
- strncpy(dst_filename, out_filename, sizeof(dst_filename));
-#endif
+ strcpy_safe(dst_filename, sizeof(dst_filename), out_filename);
 
  printf("Writing %s file: %s\n", output_crn ? "CRN" : "DDS", dst_filename);
  FILE* pFile = NULL;
@@ -516,11 +509,7 @@ int main(int argc, char* argv[]) {
  char dst_filename[FILENAME_MAX];
  crnlib_snprintf(dst_filename, sizeof(dst_filename), "%s%s%s.dds", drive.get_ptr(), dir.get_ptr(), fname.get_ptr());
  if (out_filename[0])
-#if defined(_WIN32)
- strcpy_s(dst_filename, sizeof(dst_filename), out_filename);
-#else
- strncpy(dst_filename, out_filename, sizeof(dst_filename));
-#endif
+ strcpy_safe(dst_filename, sizeof(dst_filename), out_filename);
 
  printf("Writing file: %s\n", dst_filename);
  FILE* pFile = NULL;

example2/example2.cpp

@@ -28,11 +28,29 @@
 
 using namespace crnlib;
 
-#if defined(_WIN32)
-#define example2_strncpy(out, in, size) strcpy_s(out, size, in)
-#else
-#define example2_strncpy(out, in, size) strncpy(out, in, size)
-#endif
+// Copied from crn_strutils.cpp without asserts.
+char* example2_strcpy_safe(char* pDst, uint dst_len, const char* pSrc) {
+ if (!dst_len)
+ return pDst;
+
+ char* q = pDst;
+ char c;
+
+ do {
+ if (dst_len == 1) {
+ *q++ = '\0';
+ break;
+ }
+
+ c = *pSrc++;
+ *q++ = c;
+
+ dst_len--;
+
+ } while (c);
+
+ return pDst;
+}
 
 static int print_usage() {
  printf("Description: Transcodes .CRN to .DDS files using crn_decomp.h.\n");
@@ -96,11 +114,7 @@ int main(int argc, char* argv[]) {
  if (++i >= argc)
  return error("Expected output filename!");
 
-#if defined(_WIN32)
- strcpy_s(out_filename, sizeof(out_filename), argv[i]);
-#else
- strncpy(out_filename, argv[i], sizeof(out_filename));
-#endif
+ example2_strcpy_safe(out_filename, sizeof(out_filename), argv[i]);
  } else
  return error("Invalid option: %s\n", argv[i]);
  }
@@ -143,7 +157,7 @@ int main(int argc, char* argv[]) {
  // Now create the DDS file.
  char dst_filename[FILENAME_MAX];
  if (out_filename[0]) {
- strcpy(dst_filename, out_filename);
+ example2_strcpy_safe(dst_filename, sizeof(dst_filename), out_filename);
  } else {
  unsigned int stripped_length = UINT32_MAX;
  const char* ext_begin = strrchr(pSrc_filename, '.');

example3/example3.cpp

@@ -19,6 +19,7 @@
 #include "dds_defs.h"
 
 #include "crn_core.h"
+#include "crn_strutils.h"
 #include "crn_file_utils.h"
 
 // stb_image, for loading/saving image files.
@@ -94,11 +95,7 @@ int main(int argc, char* argv[]) {
  if (++i >= argc)
  return error("Expected output filename!");
 
-#if defined(_WIN32)
- strcpy_s(out_filename, sizeof(out_filename), argv[i]);
-#else
- strncpy(out_filename, argv[i], sizeof(out_filename));
-#endif
+ strcpy_safe(out_filename, sizeof(out_filename), argv[i]);
  } else if (!crnlib_stricmp(argv[i], "-nonsrgb"))
  srgb_colorspace = false;
  else if (!crnlib_stricmp(argv[i], "-pixelformat")) {
@@ -233,11 +230,7 @@ int main(int argc, char* argv[]) {
  char dst_filename[FILENAME_MAX];
  crnlib_snprintf(dst_filename, sizeof(dst_filename), "%s%s%s.dds", drive.get_ptr(), dir.get_ptr(), fname.get_ptr());
  if (out_filename[0])
-#if defined(_WIN32)
- strcpy_s(dst_filename, sizeof(dst_filename), out_filename);
-#else
- strncpy(dst_filename, out_filename, sizeof(dst_filename));
-#endif
+ strcpy_safe(dst_filename, sizeof(dst_filename), out_filename);
 
  printf("Writing DDS file: %s\n", dst_filename);