forked from flipperdevices/flipperzero-firmware
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[DEVOPS-18]: Add map file parser, mariadb inserter (flipperdevices#2732)
- Loading branch information
1 parent
d9a9fa0
commit 72ad22b
Showing
3 changed files
with
414 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,139 @@ | ||
#!/usr/bin/env python3 | ||
|
||
# Requiremets: | ||
# mariadb==1.1.6 | ||
|
||
from datetime import datetime | ||
import argparse | ||
import mariadb | ||
import sys | ||
import os | ||
|
||
|
||
def parseArgs(): | ||
parser = argparse.ArgumentParser() | ||
parser.add_argument("db_user", help="MariaDB user") | ||
parser.add_argument("db_pass", help="MariaDB password") | ||
parser.add_argument("db_host", help="MariaDB hostname") | ||
parser.add_argument("db_port", type=int, help="MariaDB port") | ||
parser.add_argument("db_name", help="MariaDB database") | ||
parser.add_argument("report_file", help="Report file(.map.all)") | ||
args = parser.parse_args() | ||
return args | ||
|
||
|
||
def mariadbConnect(args): | ||
try: | ||
conn = mariadb.connect( | ||
user=args.db_user, | ||
password=args.db_pass, | ||
host=args.db_host, | ||
port=args.db_port, | ||
database=args.db_name, | ||
) | ||
except mariadb.Error as e: | ||
print(f"Error connecting to MariaDB: {e}") | ||
sys.exit(1) | ||
return conn | ||
|
||
|
||
def parseEnv(): | ||
outArr = [] | ||
outArr.append(datetime.now().strftime("%Y-%m-%d %H:%M:%S")) | ||
outArr.append(os.getenv("COMMIT_HASH", default=None)) | ||
outArr.append(os.getenv("COMMIT_MSG", default=None)) | ||
outArr.append(os.getenv("BRANCH_NAME", default=None)) | ||
outArr.append(os.getenv("BSS_SIZE", default=None)) | ||
outArr.append(os.getenv("TEXT_SIZE", default=None)) | ||
outArr.append(os.getenv("RODATA_SIZE", default=None)) | ||
outArr.append(os.getenv("DATA_SIZE", default=None)) | ||
outArr.append(os.getenv("FREE_FLASH_SIZE", default=None)) | ||
outArr.append(os.getenv("PULL_ID", default=None)) | ||
outArr.append(os.getenv("PULL_NAME", default=None)) | ||
return outArr | ||
|
||
|
||
def createTables(cur, conn): | ||
headerTable = "CREATE TABLE IF NOT EXISTS `header` ( \ | ||
`id` int(10) unsigned NOT NULL AUTO_INCREMENT, \ | ||
`datetime` datetime NOT NULL, \ | ||
`commit` varchar(40) NOT NULL, \ | ||
`commit_msg` text NOT NULL, \ | ||
`branch_name` text NOT NULL, \ | ||
`bss_size` int(10) unsigned NOT NULL, \ | ||
`text_size` int(10) unsigned NOT NULL, \ | ||
`rodata_size` int(10) unsigned NOT NULL, \ | ||
`data_size` int(10) unsigned NOT NULL, \ | ||
`free_flash_size` int(10) unsigned NOT NULL, \ | ||
`pullrequest_id` int(10) unsigned DEFAULT NULL, \ | ||
`pullrequest_name` text DEFAULT NULL, \ | ||
PRIMARY KEY (`id`), \ | ||
KEY `header_id_index` (`id`) )" | ||
dataTable = "CREATE TABLE IF NOT EXISTS `data` ( \ | ||
`header_id` int(10) unsigned NOT NULL, \ | ||
`id` int(10) unsigned NOT NULL AUTO_INCREMENT, \ | ||
`section` text NOT NULL, \ | ||
`address` text NOT NULL, \ | ||
`size` int(10) unsigned NOT NULL, \ | ||
`name` text NOT NULL, \ | ||
`lib` text NOT NULL, \ | ||
`obj_name` text NOT NULL, \ | ||
PRIMARY KEY (`id`), \ | ||
KEY `data_id_index` (`id`), \ | ||
KEY `data_header_id_index` (`header_id`), \ | ||
CONSTRAINT `data_header_id_foreign` FOREIGN KEY (`header_id`) REFERENCES `header` (`id`) )" | ||
cur.execute(headerTable) | ||
cur.execute(dataTable) | ||
conn.commit() | ||
|
||
|
||
def insertHeader(data, cur, conn): | ||
query = "INSERT INTO `header` ( \ | ||
datetime, commit, commit_msg, branch_name, bss_size, text_size, \ | ||
rodata_size, data_size, free_flash_size, pullrequest_id, pullrequest_name) \ | ||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)" | ||
cur.execute(query, data) | ||
conn.commit() | ||
return cur.lastrowid | ||
|
||
|
||
def parseFile(fileObj, headerID): | ||
arr = [] | ||
fileLines = fileObj.readlines() | ||
for line in fileLines: | ||
lineArr = [] | ||
tempLineArr = line.split("\t") | ||
lineArr.append(headerID) | ||
lineArr.append(tempLineArr[0]) # section | ||
lineArr.append(int(tempLineArr[2], 16)) # address hex | ||
lineArr.append(int(tempLineArr[3])) # size | ||
lineArr.append(tempLineArr[4]) # name | ||
lineArr.append(tempLineArr[5]) # lib | ||
lineArr.append(tempLineArr[6]) # obj_name | ||
arr.append(tuple(lineArr)) | ||
return arr | ||
|
||
|
||
def insertData(data, cur, conn): | ||
query = "INSERT INTO `data` ( \ | ||
header_id, section, address, size, \ | ||
name, lib, obj_name) \ | ||
VALUES (?, ?, ?, ?, ? ,?, ?)" | ||
cur.executemany(query, data) | ||
conn.commit() | ||
|
||
|
||
def main(): | ||
args = parseArgs() | ||
dbConn = mariadbConnect(args) | ||
reportFile = open(args.report_file) | ||
dbCurs = dbConn.cursor() | ||
createTables(dbCurs, dbConn) | ||
headerID = insertHeader(parseEnv(), dbCurs, dbConn) | ||
insertData(parseFile(reportFile, headerID), dbCurs, dbConn) | ||
reportFile.close() | ||
dbCurs.close() | ||
|
||
|
||
if __name__ == "__main__": | ||
main() |
Oops, something went wrong.