injectCX.cpp

#include <iostream> 
#include <fstream> 
#include <string> 
#include <cstring> 
#include <cstdio>  

int main(int argc, char *argv[]) { 
    if (argc == 2 || argv[1] == "--help") { 
        std::cout << "INJECT \n\n" << "Usage: inject [File] [Line Number] [Content] \n";
        return -3; // Help Command 
    } 
    else if (argc < 4) { 
        std::cout << "Invalid, Not Enough Arguments; Run --help For More Info. \n"; 
        return -2; // Not Enough Args 
    } 
    std::ifstream file(argv[1]);
    std::string* pHemp = new std::string("temp_"); 
    *pHemp+=argv[1]; 
    char* THEMP = new char[(*pHemp).size()+1]; 
    strcpy(THEMP, (*pHemp).c_str()); 
    delete pHemp; 
    // std::ofstream tfile(THEMP); 
    std::string tfile; 
    long num = std::stoi(argv[2]); 
    std::string line; 
    long li = 0; 
    bool numEx = false; 
    while (std::getline(file, line)) { 
        if (li == num) { 
                for (short i=3;i<argc;i++) { tfile += argv[i]; } tfile += " \n"; numEx=1;  
        } 
        else { 
            tfile += line; tfile += " \n";  
        } 
        li++; 
    } 
    if (!numEx) { 
        if(li<1) { tfile += " \n"; }  
        for (short j=3;j<argc;j++) { 
            tfile += argv[j]; 
        } 
        tfile += " \n"; 
    }  
    if (file.good()) { 
        // std::remove(argv[1]); 
    }
    file.close(); 
    std::ofstream ofile(argv[1]); 
    if (ofile.is_open()) { 
        ofile << tfile; 
    } 
    else { 
        std::cerr << "Error Opening File For Output. \n"; 
    } 
    ofile.close(); 
    // tfile.close();  
    // std::rename(THEMP, argv[1]); 
    delete[] THEMP; 
    return 0; 
}