summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--libpit/source/libpit.h14
1 files changed, 7 insertions, 7 deletions
diff --git a/libpit/source/libpit.h b/libpit/source/libpit.h
index ccdea20..dbf731a 100644
--- a/libpit/source/libpit.h
+++ b/libpit/source/libpit.h
@@ -208,12 +208,12 @@ namespace libpit
void SetPartitionName(const char *partitionName)
{
// This isn't strictly necessary but ensures no junk is left in our PIT file.
- memset(this->partitionName, 0, 64);
+ memset(this->partitionName, 0, kPartitionNameMaxLength);
- if (strlen(partitionName) < 64)
+ if (strlen(partitionName) < kPartitionNameMaxLength)
strcpy(this->partitionName, partitionName);
else
- memcpy(this->partitionName, partitionName, 63);
+ memcpy(this->partitionName, partitionName, kPartitionNameMaxLength - 1);
}
const char *GetFlashFilename(void) const
@@ -226,10 +226,10 @@ namespace libpit
// This isn't strictly necessary but ensures no junk is left in our PIT file.
memset(this->flashFilename, 0, kFlashFilenameMaxLength);
- if (strlen(partitionName) < 32)
+ if (strlen(partitionName) < kFlashFilenameMaxLength)
strcpy(this->flashFilename, flashFilename);
else
- memcpy(this->flashFilename, flashFilename, 31);
+ memcpy(this->flashFilename, flashFilename, kFlashFilenameMaxLength - 1);
}
const char *GetFotaFilename(void) const
@@ -242,10 +242,10 @@ namespace libpit
// This isn't strictly necessary but ensures no junk is left in our PIT file.
memset(this->fotaFilename, 0, kFotaFilenameMaxLength);
- if (strlen(partitionName) < 32)
+ if (strlen(partitionName) < kFotaFilenameMaxLength)
strcpy(this->fotaFilename, fotaFilename);
else
- memcpy(this->fotaFilename, fotaFilename, 31);
+ memcpy(this->fotaFilename, fotaFilename, kFotaFilenameMaxLength - 1);
}
};