From 8dd5dc662e7d69cab06b36074d2bf4992a20b7b7 Mon Sep 17 00:00:00 2001 From: Ray Jones Date: Tue, 9 Jul 2019 22:19:21 +1000 Subject: [PATCH] Added extra CRC checking to OTA --- AfterburnerCRC.exe | Bin 0 -> 13312 bytes add_CRC.py | 12 +++ platformio.ini | 2 + .../src/Libraries/esp32FOTA/src/esp32fota.cpp | 8 +- src/Afterburner/src/Protocol/Protocol.cpp | 25 ++---- src/Afterburner/src/Protocol/Protocol.h | 39 +--------- src/Afterburner/src/Utility/MODBUS-CRC16.cpp | 55 ++++++++++++++ src/Afterburner/src/Utility/MODBUS-CRC16.h | 71 ++++++++++++++++++ src/Afterburner/src/WiFi/BTCWebServer.cpp | 30 ++++++-- src/Afterburner/src/WiFi/BTCota.cpp | 40 ++++++++++ src/Afterburner/src/WiFi/BTCota.h | 1 + 11 files changed, 218 insertions(+), 65 deletions(-) create mode 100644 AfterburnerCRC.exe create mode 100644 add_CRC.py create mode 100644 src/Afterburner/src/Utility/MODBUS-CRC16.cpp create mode 100644 src/Afterburner/src/Utility/MODBUS-CRC16.h diff --git a/AfterburnerCRC.exe b/AfterburnerCRC.exe new file mode 100644 index 0000000000000000000000000000000000000000..efcdfcbea57a2d1e75f5cba73a2ccfe3e26362ab GIT binary patch literal 13312 zcmeHt4|r5XmTxubgcbrFF~O(^jf6o3Vt3yp(D}!0LI+TS4G9U+gr-Su=nhGDru#Mo z7shToIB75KuroM|JLtm|c9|JxhnYbPih7$M2@Wdk%yV(!xlz%%n7lD!7UGz(->>R+ zu*2x=x8Hl;_rBeo`cmgqopb8csZ&*_s&2aCfn6kp5TZqn#|h~Irio|p&x2vGC(n3s zGC6$18#B8!E8dt{<#PKCZ9eZtpR?7_;PiOBqG5wz@C7^ux5u#L{z^luw^7KQG-*Ok zLiK0T);TYHe6u*3U-^EA_%iArK9U(Y!tj*9K89}!yu|R-z%PN%Jw7$?Jn#!2PZcA; zXCIl)>SrHOMB47`BU!BdYIlQ+>b`y?b{iopG^ymr9Xa)*ygo8cGf^{HPo{!0O{HGW z2F^e>(mJn}#}Sf7t4O1m1W}<*L`orihzZhmv@yfi5G9h`g@hcYh4qB=X=vfP)J8}# z`r-2kLGc^YA@KQRt`U{5)0dFR6@_-{jQmUhY8M@A!fupUGQg@_U!zlW5_0d2AW@xD zkZ(m!N<6yGRmD^tb}(TxnU^GlB2S2%>+|~>P-bIYj{=p;MNUdQ)#nphybz>gX9K6U z(f*SXPsrk=>R*cg7ar&utmH}e>1%0?G*B%Cbj_N0){v1W;S<_uMf$2{Qf*hx#*x~z zabrAdkg>AY)ptX_Iqu5H;Ys5&)C_xgp!gZBohRID5!+y8b&Wl08)|%(vclVS<=SWBiSj)tL=WNK_@&L5Zz+EmdZV^p*9mareyu(JcKAekczZ7i+={BJmo|EU zHPUr`34Cc0nl+DK6|FekW4k~*>9O_cflk^^6QvK#9@_{dM{TFW+xs$H7okC4W;+e5 zPC&a<(TfgtoC<89TnI6q+RtU$L7+FDbrD8MYx<}qf5`CVN)@LUOR>-&=OzIv6Ck%|$Y&~^Qhnz;u%QePwY*aoOEoZO^;+jfz=60sfQ z;t|`$NI;2fA6RwLHljIcI}X?PK?nMN>|AGkcRcF?YEZR94m3aZMLcjqf27AA&pM}4 zgWHc2abob-akC8JPTTRKFGXGF^m6FzzbR_#wpT|hE-G0maPRg3DWI?+CF_UEuX+&o zN%a>a^~d_hapxsl*WAxA;ngmM`u%;31U@)QPl+8@#GEGmj3tCTmJ)qXV~#b&F6s0~ z7uSC!~C8w*E*roR4a(Usp*By;45)GNaL1 zITk))*l|jnH}`XRI{uZXd;Y}JHLH}r$JpQ=Iz|71`Ds&Iak5k~)EJpW=cU%M?m?+y zkb6T}Ox^WiJ@!Sl^r@7#Ll#S%gGb0tlMW1iETY8uFF;1FZJ{G4Y4>-+d3ANdFrCwLbI(C2Su55=H&frL$eMaQ5T5$BTMLsk*3kLeV+>;&a?7 zsXHud^xuzNeP`mRe`ZImv%;^(i(>ln<8~#JZf)!{EoO8p<@EJMmW}k^i0Z^u$`x!4 z@UY-y)l_@LHUeuS>oS%5F=DBH1U5^yVQO}GIKy{WbeTp<`v%Y5!aTQ|>C`zQ)1{N6 z!&E0gsRQd;up9MXhGc^aYoyQ7wRH2`;}P4iv{XCycnR}NM;h8l_v@l0)t~6 zqLtq5*F~pID0)}F_}$PKY<H^YzT*gd9BrEw{-`ffKGJ{F zj`QNAN@WN{ct9G253WeIbg5!kS~HUJW~5^1G194lC6d1AOh7+3W?xmUSSPbd?4LRJ z>}pias8WfF5wIoO&?+7ak%?qW!h$;g7pQf*W_J+f3M zjZd)+Ut1*!_tAZl?7q{k4D9ZBRvug2;J9+WsuxY72f^jQxXay_FTH?OfW;aQ=#&H4 zD7Eb9=$}OEaysg;RZlBN6P1tZyK!K|^&M_(q)STys+%>+KY*(|H6G`T@=vt*M*wM1 z=>Vl!6G>CDSkz5l*CC7k^Y1HrZy?0=69eMU^Vpcmde+cOdq97IRZ1-g%0U41KGC&u zIt1^dT&0aNHc?=sAP2y%bTF0)WvEY8=A)`ipzJ-^-D zN9fzGte~yCn=@U#7M>`LungmwhjG<*U}e~`NX+Lu>wgy(Mu_qOJk)8s@XT{KoVin0 zY^08RAb>uWMK=G+Q^)y0sxlM&X6lB+R8eih ze(7X?zEm+FX(SS{^&zl6H-U1yB|%X*wYfRC{dWWkJ_2X7m8}o4J1LT%qO)-kW2;tv z1hIIaPk9pfw2}_mB@v#YcNeT(Ej@Vxx_H(+6kIoGi5$<;p@=g!9f&Q|&_E8Rr1@s$ z2vlIPq&APMedDfYcdg<{WYSIwvpOlfy@SI1AcZB{0GoDZ9b{>l8#P5116THkSDBp&Zc?X&Z^-)qP-l=z^8SW(S+sl9^7-P`_g`_ zQh2Ev?vXxK)}nb&S{?WZtM$jVB<$1@DHe9>NRL(92zrl(a_OZo(PTDj(*6xif~*PU zvL@1}(Q<81S~2Qj0|~F!5=1mzSk~H=hp~@pYZKi@t=b+9m9i_Xlzx3_%Ks?aG~KqaLS>SxK1-3lq$Vlqp%Y*nz2C zOA-ddxN2!xaZ0cL1#4z>Rd&i56w)>g-kEZ_=xy;9sUp4T-gMstX|XP4ae7g&FFkzx z&Z4&iA9FG;&eHMz&#RRuu%5A;42{pwcjhaHusbEYwqILxM$Ev~ruhq(gW}-ZbVr~DWLmIY*G3sE+ z^Jqmx?IF#nZhFthMJ_e`r6#alnW$RwtsPeaceuJBrTh#O8YcCw6SVrYS{2L9a{qno ziXROO)r9+Iyh^2;4YhN7agqN`FPBQAe#AC}QxxavfddC#rR`M$ny56#$y92J27|&z zO;w6u>gi^ks~sL!Xtbwsu&d$$LYgcXB2NPK&lCfNx@d*I-UBO+PDD&>vUFxsb}44# zDCN;(Qdx)fOJlq^C)}Pxw#`fgi9?$H8Io;4LQoAYgj!j@J0`_wWJw5^aUC2RMBs7d z6No~WK1f}E11BG)^YmTStHBaF5kyQ2ZcitHDf9@_!z_x&1J~9jyw5x)!8_Dj;rC`h z;5WVD;do%YVn;~rpQt*$6y@5h=W1`0ub!D@)3J%8x?aR`RLSR-C1x`-EDX`58y1FX ztTGs}c5K5Ob^J&?YZ|I1TF6!lQQH87GMUg|v|@y_4RE#*E_SM0jRn;;ude)vf9VyN z9xr-J)b4mQ@GWIGdX;*M-U^IAdOw)`lR=#@>Bgmm#QHKO_C{C~xq8-t`?HPEj7m zJoIRDPLIRoi$&39M2TJk(j z7PNpdJ+^~cSi(KF=U_2*%;9*}o$y{WiLQ-bJs-6l#FkP=aO-(Y@|Sp3(8DWg8^+Yq z(+mDg$H}V0pr3?>+6NKCUDyirm?MlY<{ecxl`e2Mi(MfL+f282shi1~9hef{Do);< z9zLo-1rbv)cV@ZUm2aT4N&BPT#%_~HXkmlDwJ$I`+@6sVm?80*C5F@=+#=pqGLxdr zk~=7xR6_VCHDVek(<4LiVzj5jv->b>q`R{El_28pOB57+=D(#zD%}wmCrEE4t{eXx zKB+DG%y$XbP%7I=WhwagWWTzix{7Av(?mR<4n{_mvE9*m5SY|%#U*KwyJHFzvtkhb?B;D)X)6+8jSrsTUl z!gBhmv0cpfh(Ze%5B90rG+KWll7fLxv<*+}iTJgV`*e}Y^gtQ3%pP16*O$n6|A@0n z)~Agl;dU3?k5Y5K~CtMyW)BP2smR5mx9=OAxjXjYr4mbDOz&{95kR z*nY2dDb@E~?IswsI63^*xP*=rseaJ0?#J|7rP!p7U)z&P#|}XPI?;(MDasI+a;Nl~ zGZEWBWX&KA@)d*OzO?XgS|Ils>b7vNPHl(vXE8PZhO1Rgck`XjiGACo(&_I(3 z8T}J-cYO8=Bi8=xWeip^$TL{Xpq;@+2G=uqn!#QMw=pO)xR=2b3?63i5QEP#xSK(R z!JQ0tGI)%^GYobxC^G0`aDc%U2KO-7#o#^$_cM5q!3+kMFt`?w?h<@CyX5)=k*eb2 zr3_%HDn}PxoG#sanCn%FE`xCI2udcDx^`jbUHSy!1fKd!`mf=EOe-OG@`RKlKY;8( zehhg(@{7o4kUvHykV~f+X<4y~e+&EyashZdfj1++i@YD12aTrPINCdr-z_ENOJtgQ z6IiClKGNT!-j3XiTmV`XaM~h5evfh~@CxKQia{?$HY4AG`lGgNxccDyE z5ao8{HsmJcI?#WQJPqXuw2n-ZG@WO7-OVbdm7gMzS0~k(KmzQt^MpRZ?D;%-?N*-5 zyNxGnJexe;Egpl=-XOGzZm$Qjdy@DYoQ(!&3;wO)RGAmA=lOijtp<;<#n2*nHj20- z5t3Jc`xK6@rS8^N!QeDBxmyH-=rwHd`ZgK1xJ8$ttwnJ9g_)4`ZNP2VueMn4YL~!8@7sq-_YdqwyIKul&vl^lo)0k^VUuvn}r6^ z>npA*MQ2_^i`TP}5Wnbidp0H!sDUIVhhn@&(=De2$61eSPReP)@zx%a8hb5tg4b$J z%jv;M);EYg_EzXkelnR5dp-1Pexl}-d_z!Y?bV!x*l$Dslh4xpvwU-Ks`YKnKgm;qH(7r}Zj7A`o#8VygR(JbvVKaq z*k_^3yqOflu7s}g3pD>G=LhqwL!>x168a-wqKV6kf~D3kHGhx`gGJUah$Z%UXqaC} z?u;p+OZ;7$etB;2ZtKUI-^;ndd#s<3`LX{U8sO(?F3LH<+13xqoY?O|AMxKJcf{Tg zeZbGsyeH2L8mxVqee(LC)A|f)h&>znJAMOciv4})ANY-$1F{fowC*P@vFAcR<6AX9 zm%Txc^&d4qk(+~?tUo30*pEX8c^7Gqb%l=dTQx7qe-(Vl`l99)`8&aft*?^pv6n-~ z_-&fQvKS0lpC?;lFNBWpn~5*>%h0cQzvdV6w}WlgL&P5275Zy_mF5R>WpK6i`_vkI0LIW!5mU#zLVW z&$A_v0Hgmyr6=-QZj+sMv|dL*NZv$HS-0F=^kJ+JCG< z?`A>lD{3eGnl&bA#?=0wRPxYYBlh}eqFsonZ%fc9e>CBh6Q^Y!Ik^p`Wkc?{Bt6vr zW2pBg>7p{DLrB(@+>Yw9LvA=p7rhMW;FE0RWPMbQ`3d}FeaUU9zIMnhL)+0rZCH!? z`UH*gM^mz%(Kbah5x;W?SzWoLa^e?HwZHM}A?puU-uC>0OP~LS>MAR&TjLje{yK!n zI=?sIYY^&uLYvoLS8@N6#cL`Z2wtXwy43=UhGU4iZH*gP*Rug>rsJ7~ytJ%p1^rG> z7bN?Ccup$~*@VCj`cP15?dt!XVMMN*{*yb0vaoK{e|6^?jSWqRjoQ(A+3KpCwPjW+ zqPvbpiRoi$_^yF*kD)DRZ8NWRjOPkXlh?*TcyhFqH@$G)^QFdP5o`fKH$W>nJF!=NR2-U z8b@i%K%<)`LCa$_++D`uDUhIH6i9Z=0nq4g9kNlU9kDvxunTqi@5I~)9X1 zj^8}wP%okmT9EmU(U?t?etn`(X1Y!fmUJ7&HU$XmTwgj3Rw6746Ttq1Ky676-+wt;pmiS{vQB4})lBs#g%iUMd~m;SroqaPo)$gx(vzB&Y- z&+Fs;Vk6JjTG!eaFX0~`6_wRxc2k})x3Q%Kk&ieWW$o?d_?ein%;{-t5q#K@GeJcw z$F+O3qruy@6>ITN=#2vIq~5LAyi`q<%jex<`|9$&jI;`^enF(sJfY0tQhUjR%GOrg z(5uTXH_^zkxv32|VX=u65r^Be+1cW5 zbhJ6?AVk6E@Oa&xCNGdr5CcA(DBmGX^fK=76OlAEw0QjjjT=oH1+jqyNR!}gBIIGx z)D}QpM;r}!Bj6F>BDauM_FSMpb8`3vmjjo7+JVq6kR_BNI-G574smOnfS*voM)Zd> zvehZN;Ch1gFxb=N-Wc!+4v!PFvBlx^ZQRWK;r6&iH!OeX8cXnOcKf`ZR>1?0-Ax>@ z52IsJqpg!g7Sgf1MHtjdEMJu!0`7i7yBj)YQ?H{?rP|s=fAdF3DlBbtz{kFg4fJ<& z*J`lqW>~DYor;p$dJ%E>gpF=L9l7Z83C>0beL}TRH#nU58Jk;VF5BR2*hJ5T#F$l4 zCMn>@ZOmKns)HGU^+Ii6Yk=C6O`3cHaDh6GgkQWKrd*^iD!e$8>_~9v>R?Z~8R!)& zMzx0A&TPQKY-wu4q~p6Sb&}fyHgUGO=ePRjZ*hC(qs4qIbGLVXi)p@TzPcr@6T*KL#Ms5?PnB{Myp5T!D493{A*G zuxN>}A+T|y;2X6>q&h0Rje(X#aH=8XKHKV*wiTRNjbG(7!Zgs>w3L=wy_>O6{!^x3 z-16!Y`hOpKM$@~+Xr^DQY06z>t;Y8Y2Mb7QOee{_Yq9!kc&r`$w&9=~P(2(HVqLZjX1-i+w7QSe`5CbMnr zXiATX6~bnr1>g1nOXfKJ%RQUDn*`q+L%?lqprNv4PLs36FU&FAb4^e2J>T3*@jYW~ zD!%91;LyDIo>9M{a@|~fZ8ZaDC+1CKyv6s#-wpdfvLf5FoW zezl-$!OIKYUGTwz!3B3OOtGX}ZnjLflvZA-i5VaubIsO1UE zQ@<~Z* z>7vqQrFEqnN~5K}Dm_+uyma3pdC}{O=v|)}@a-7rZle(%2pJzWJ_#TE()gP3J>!?g z`%HFIv*}UOZqv`)%FK6ef zxjJqG=i)q^$UVdbxktI3-1oUX+|%5C?kC(K?g)2`JI?iTZ*sro{)Ow~KH>(r%iJ(G zkG@3l&e#n^OgdAhDcfW;<(U?m@S_*gT2q~Ay-75+o3@$unD&|m@&@yU^Xl@~=eOmH z`P=e$=kL!ynBSFuEMLy=&F{+}%pc0v7U&An3o;6_3ycMP!IFZig0%(f3)&0zSh_5- z<$|TpqF4qk!%hz#yq1PUZUpx KCH?32!2bc%8GBy< literal 0 HcmV?d00001 diff --git a/add_CRC.py b/add_CRC.py new file mode 100644 index 0000000..c2e76f5 --- /dev/null +++ b/add_CRC.py @@ -0,0 +1,12 @@ +Import("env") + +# access to global build environment +print env + +# Add CRC from BIN +env.AddPostAction( + "$BUILD_DIR/${PROGNAME}.bin", + env.VerboseAction(" ".join([ + "AfterburnerCRC.exe", "$BUILD_DIR/${PROGNAME}.bin" + ]), "Creating CRC from $BUILD_DIR/${PROGNAME}.bin, adding to $BUILD_DIR/${PROGNAME}.bin.crc") +) diff --git a/platformio.ini b/platformio.ini index 4cac612..ae83de3 100644 --- a/platformio.ini +++ b/platformio.ini @@ -20,3 +20,5 @@ upload_protocol = espota upload_flags = --port=3232 monitor_speed = 115200 +extra_scripts = post:add_CRC.py + diff --git a/src/Afterburner/src/Libraries/esp32FOTA/src/esp32fota.cpp b/src/Afterburner/src/Libraries/esp32FOTA/src/esp32fota.cpp index 9ed58c9..80492b8 100644 --- a/src/Afterburner/src/Libraries/esp32FOTA/src/esp32fota.cpp +++ b/src/Afterburner/src/Libraries/esp32FOTA/src/esp32fota.cpp @@ -12,6 +12,7 @@ #include #include #include "../../ArduinoJson/ArduinoJson.h" +#include "../../../WiFi/BTCota.h" extern void forceBootInit(); @@ -143,6 +144,10 @@ void esp32FOTA::execOTA() // execOTA(); } + if(!CheckFirmwareCRC(contentLength)) { + Update.abort(); + } + if (Update.end()) { Serial.println("OTA done!"); @@ -272,4 +277,5 @@ String esp32FOTA::getDeviceID() sprintf(deviceid, "%" PRIu64, chipid); String thisID(deviceid); return thisID; -} \ No newline at end of file +} + diff --git a/src/Afterburner/src/Protocol/Protocol.cpp b/src/Afterburner/src/Protocol/Protocol.cpp index f51ffd3..b97bb2f 100644 --- a/src/Afterburner/src/Protocol/Protocol.cpp +++ b/src/Afterburner/src/Protocol/Protocol.cpp @@ -27,28 +27,11 @@ #include "../Utility/macros.h" -uint16_t -CProtocol::CalcCRC(int len) const -{ - // calculate a CRC-16/MODBUS checksum using the first 22 bytes of the data array - uint16_t wCRCWord = 0xFFFF; - - int wLength = len; - const uint8_t* pData = Data; - while (wLength--) - { - uint8_t nTemp = *pData++ ^ wCRCWord; - wCRCWord >>= 8; - wCRCWord ^= wCRCTable[nTemp]; - } - - return wCRCWord; -} - void CProtocol::setCRC() { - setCRC(CalcCRC(22)); + CModBusCRC16 CRCengine; + setCRC(CRCengine.process(22, Data)); } void @@ -73,7 +56,9 @@ CProtocol::getCRC() const bool CProtocol::verifyCRC(bool bSilent) const { - uint16_t CRC = CalcCRC(22); // calculate CRC based on first 22 bytes + CModBusCRC16 CRCengine; + uint16_t CRC = CRCengine.process(22, Data); // calculate CRC based on first 22 bytes of our data buffer + uint16_t FrameCRC = getCRC(); bool bOK = (FrameCRC == CRC); if(!bOK && !bSilent) { diff --git a/src/Afterburner/src/Protocol/Protocol.h b/src/Afterburner/src/Protocol/Protocol.h index 6c06dec..70e3530 100644 --- a/src/Afterburner/src/Protocol/Protocol.h +++ b/src/Afterburner/src/Protocol/Protocol.h @@ -23,8 +23,10 @@ #define _CPROTOCOL_H_ #include "../Utility/UtilClasses.h" +#include "../Utility/MODBUS-CRC16.h" class CProtocol { + // CModBusCRC16 _CRCengine; public: union { uint8_t Data[24]; @@ -83,40 +85,6 @@ public: }; static const int CtrlMode = 1; static const int HeatMode = 2; - const uint16_t wCRCTable[256] = { - 0X0000, 0XC0C1, 0XC181, 0X0140, 0XC301, 0X03C0, 0X0280, 0XC241, - 0XC601, 0X06C0, 0X0780, 0XC741, 0X0500, 0XC5C1, 0XC481, 0X0440, - 0XCC01, 0X0CC0, 0X0D80, 0XCD41, 0X0F00, 0XCFC1, 0XCE81, 0X0E40, - 0X0A00, 0XCAC1, 0XCB81, 0X0B40, 0XC901, 0X09C0, 0X0880, 0XC841, - 0XD801, 0X18C0, 0X1980, 0XD941, 0X1B00, 0XDBC1, 0XDA81, 0X1A40, - 0X1E00, 0XDEC1, 0XDF81, 0X1F40, 0XDD01, 0X1DC0, 0X1C80, 0XDC41, - 0X1400, 0XD4C1, 0XD581, 0X1540, 0XD701, 0X17C0, 0X1680, 0XD641, - 0XD201, 0X12C0, 0X1380, 0XD341, 0X1100, 0XD1C1, 0XD081, 0X1040, - 0XF001, 0X30C0, 0X3180, 0XF141, 0X3300, 0XF3C1, 0XF281, 0X3240, - 0X3600, 0XF6C1, 0XF781, 0X3740, 0XF501, 0X35C0, 0X3480, 0XF441, - 0X3C00, 0XFCC1, 0XFD81, 0X3D40, 0XFF01, 0X3FC0, 0X3E80, 0XFE41, - 0XFA01, 0X3AC0, 0X3B80, 0XFB41, 0X3900, 0XF9C1, 0XF881, 0X3840, - 0X2800, 0XE8C1, 0XE981, 0X2940, 0XEB01, 0X2BC0, 0X2A80, 0XEA41, - 0XEE01, 0X2EC0, 0X2F80, 0XEF41, 0X2D00, 0XEDC1, 0XEC81, 0X2C40, - 0XE401, 0X24C0, 0X2580, 0XE541, 0X2700, 0XE7C1, 0XE681, 0X2640, - 0X2200, 0XE2C1, 0XE381, 0X2340, 0XE101, 0X21C0, 0X2080, 0XE041, - 0XA001, 0X60C0, 0X6180, 0XA141, 0X6300, 0XA3C1, 0XA281, 0X6240, - 0X6600, 0XA6C1, 0XA781, 0X6740, 0XA501, 0X65C0, 0X6480, 0XA441, - 0X6C00, 0XACC1, 0XAD81, 0X6D40, 0XAF01, 0X6FC0, 0X6E80, 0XAE41, - 0XAA01, 0X6AC0, 0X6B80, 0XAB41, 0X6900, 0XA9C1, 0XA881, 0X6840, - 0X7800, 0XB8C1, 0XB981, 0X7940, 0XBB01, 0X7BC0, 0X7A80, 0XBA41, - 0XBE01, 0X7EC0, 0X7F80, 0XBF41, 0X7D00, 0XBDC1, 0XBC81, 0X7C40, - 0XB401, 0X74C0, 0X7580, 0XB541, 0X7700, 0XB7C1, 0XB681, 0X7640, - 0X7200, 0XB2C1, 0XB381, 0X7340, 0XB101, 0X71C0, 0X7080, 0XB041, - 0X5000, 0X90C1, 0X9181, 0X5140, 0X9301, 0X53C0, 0X5280, 0X9241, - 0X9601, 0X56C0, 0X5780, 0X9741, 0X5500, 0X95C1, 0X9481, 0X5440, - 0X9C01, 0X5CC0, 0X5D80, 0X9D41, 0X5F00, 0X9FC1, 0X9E81, 0X5E40, - 0X5A00, 0X9AC1, 0X9B81, 0X5B40, 0X9901, 0X59C0, 0X5880, 0X9841, - 0X8801, 0X48C0, 0X4980, 0X8941, 0X4B00, 0X8BC1, 0X8A81, 0X4A40, - 0X4E00, 0X8EC1, 0X8F81, 0X4F40, 0X8D01, 0X4DC0, 0X4C80, 0X8C41, - 0X4400, 0X84C1, 0X8581, 0X4540, 0X8701, 0X47C0, 0X4680, 0X8641, - 0X8201, 0X42C0, 0X4380, 0X8341, 0X4100, 0X81C1, 0X8081, 0X4040 - }; public: CProtocol() { Init(0); }; @@ -124,11 +92,10 @@ public: void Init(int Txmode); // CRC handlers - uint16_t CalcCRC(int len) const; // calculate the CRC upon len bytes void setCRC(); // calculate and set the CRC in the buffer void setCRC(uint16_t CRC); // set the CRC in the buffer uint16_t getCRC() const; // extract CRC value from buffer - bool verifyCRC(bool silent=false) const; // return true for CRC match + bool verifyCRC(bool silent=false) const; // return true for CRC match void setActiveMode() { Controller.Byte0 = 0x76; }; // this allows heater to save tuning params to EEPROM void setPassiveMode() { Controller.Byte0 = 0x78; }; // this prevents heater saving tuning params to EEPROM diff --git a/src/Afterburner/src/Utility/MODBUS-CRC16.cpp b/src/Afterburner/src/Utility/MODBUS-CRC16.cpp new file mode 100644 index 0000000..a5217d8 --- /dev/null +++ b/src/Afterburner/src/Utility/MODBUS-CRC16.cpp @@ -0,0 +1,55 @@ +/* + * This file is part of the "bluetoothheater" distribution + * (https://gitlab.com/mrjones.id.au/bluetoothheater) + * + * Copyright (C) 2018 Ray Jones + * + * 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 3 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, see . + * + */ + +#include "MODBUS-CRC16.h" + +CModBusCRC16::CModBusCRC16() +{ + _CRC = 0xffff; +} + +void +CModBusCRC16::reset() +{ + _CRC = 0xffff; +} + +uint16_t +CModBusCRC16::process(int len, const uint8_t* pData) +{ + // calculate a CRC-16/MODBUS checksum using the bytes of the data array + // carries on from prior developed CRC value, or 0xffff if first pass + + while (len--) + { + uint8_t nTemp = *pData++ ^ _CRC; + _CRC >>= 8; + _CRC ^= _CRCTable[nTemp]; + } + + return _CRC; +} + +uint16_t +CModBusCRC16::get() const +{ + return _CRC; +} \ No newline at end of file diff --git a/src/Afterburner/src/Utility/MODBUS-CRC16.h b/src/Afterburner/src/Utility/MODBUS-CRC16.h new file mode 100644 index 0000000..593387a --- /dev/null +++ b/src/Afterburner/src/Utility/MODBUS-CRC16.h @@ -0,0 +1,71 @@ +/* + * This file is part of the "bluetoothheater" distribution + * (https://gitlab.com/mrjones.id.au/bluetoothheater) + * + * Copyright (C) 2018 Ray Jones + * + * 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 3 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, see . + * + */ + +#ifndef __BTC_MODBUSCRC_H__ +#define __BTC_MODBUSCRC_H__ + +#include + +class CModBusCRC16 { + uint16_t _CRC; + const uint16_t _CRCTable[256] = { + 0X0000, 0XC0C1, 0XC181, 0X0140, 0XC301, 0X03C0, 0X0280, 0XC241, + 0XC601, 0X06C0, 0X0780, 0XC741, 0X0500, 0XC5C1, 0XC481, 0X0440, + 0XCC01, 0X0CC0, 0X0D80, 0XCD41, 0X0F00, 0XCFC1, 0XCE81, 0X0E40, + 0X0A00, 0XCAC1, 0XCB81, 0X0B40, 0XC901, 0X09C0, 0X0880, 0XC841, + 0XD801, 0X18C0, 0X1980, 0XD941, 0X1B00, 0XDBC1, 0XDA81, 0X1A40, + 0X1E00, 0XDEC1, 0XDF81, 0X1F40, 0XDD01, 0X1DC0, 0X1C80, 0XDC41, + 0X1400, 0XD4C1, 0XD581, 0X1540, 0XD701, 0X17C0, 0X1680, 0XD641, + 0XD201, 0X12C0, 0X1380, 0XD341, 0X1100, 0XD1C1, 0XD081, 0X1040, + 0XF001, 0X30C0, 0X3180, 0XF141, 0X3300, 0XF3C1, 0XF281, 0X3240, + 0X3600, 0XF6C1, 0XF781, 0X3740, 0XF501, 0X35C0, 0X3480, 0XF441, + 0X3C00, 0XFCC1, 0XFD81, 0X3D40, 0XFF01, 0X3FC0, 0X3E80, 0XFE41, + 0XFA01, 0X3AC0, 0X3B80, 0XFB41, 0X3900, 0XF9C1, 0XF881, 0X3840, + 0X2800, 0XE8C1, 0XE981, 0X2940, 0XEB01, 0X2BC0, 0X2A80, 0XEA41, + 0XEE01, 0X2EC0, 0X2F80, 0XEF41, 0X2D00, 0XEDC1, 0XEC81, 0X2C40, + 0XE401, 0X24C0, 0X2580, 0XE541, 0X2700, 0XE7C1, 0XE681, 0X2640, + 0X2200, 0XE2C1, 0XE381, 0X2340, 0XE101, 0X21C0, 0X2080, 0XE041, + 0XA001, 0X60C0, 0X6180, 0XA141, 0X6300, 0XA3C1, 0XA281, 0X6240, + 0X6600, 0XA6C1, 0XA781, 0X6740, 0XA501, 0X65C0, 0X6480, 0XA441, + 0X6C00, 0XACC1, 0XAD81, 0X6D40, 0XAF01, 0X6FC0, 0X6E80, 0XAE41, + 0XAA01, 0X6AC0, 0X6B80, 0XAB41, 0X6900, 0XA9C1, 0XA881, 0X6840, + 0X7800, 0XB8C1, 0XB981, 0X7940, 0XBB01, 0X7BC0, 0X7A80, 0XBA41, + 0XBE01, 0X7EC0, 0X7F80, 0XBF41, 0X7D00, 0XBDC1, 0XBC81, 0X7C40, + 0XB401, 0X74C0, 0X7580, 0XB541, 0X7700, 0XB7C1, 0XB681, 0X7640, + 0X7200, 0XB2C1, 0XB381, 0X7340, 0XB101, 0X71C0, 0X7080, 0XB041, + 0X5000, 0X90C1, 0X9181, 0X5140, 0X9301, 0X53C0, 0X5280, 0X9241, + 0X9601, 0X56C0, 0X5780, 0X9741, 0X5500, 0X95C1, 0X9481, 0X5440, + 0X9C01, 0X5CC0, 0X5D80, 0X9D41, 0X5F00, 0X9FC1, 0X9E81, 0X5E40, + 0X5A00, 0X9AC1, 0X9B81, 0X5B40, 0X9901, 0X59C0, 0X5880, 0X9841, + 0X8801, 0X48C0, 0X4980, 0X8941, 0X4B00, 0X8BC1, 0X8A81, 0X4A40, + 0X4E00, 0X8EC1, 0X8F81, 0X4F40, 0X8D01, 0X4DC0, 0X4C80, 0X8C41, + 0X4400, 0X84C1, 0X8581, 0X4540, 0X8701, 0X47C0, 0X4680, 0X8641, + 0X8201, 0X42C0, 0X4380, 0X8341, 0X4100, 0X81C1, 0X8081, 0X4040 + }; + +public: + CModBusCRC16(); + void reset(); + uint16_t process(int len, const uint8_t* pData); + uint16_t get() const; +}; + +#endif diff --git a/src/Afterburner/src/WiFi/BTCWebServer.cpp b/src/Afterburner/src/WiFi/BTCWebServer.cpp index 2c23288..ff97a17 100644 --- a/src/Afterburner/src/WiFi/BTCWebServer.cpp +++ b/src/Afterburner/src/WiFi/BTCWebServer.cpp @@ -25,6 +25,7 @@ #include #include "BTCWifi.h" #include "BTCWebServer.h" +#include "BTCota.h" #include "../Utility/DebugPort.h" #include "../Protocol/TxManage.h" #include "../Utility/helpers.h" @@ -56,6 +57,8 @@ bool bRxWebData = false; // flags for OLED animation bool bTxWebData = false; bool bUpdateAccessed = false; // flag used to ensure web update always starts via /update. direct accesses to /updatenow will FAIL bool bFormatAccessed = false; +uint16_t fileCRC = 0; +int nUploadSize = 0; long _SuppliedFileSize = 0; void webSocketEvent(uint8_t num, WStype_t type, uint8_t * payload, size_t length); @@ -349,13 +352,13 @@ void rootRedirect() bool sendWebSocketString(const char* Str) { - CProfile profile; +// CProfile profile; if(webSocket.connectedClients()) { - unsigned long tCon = profile.elapsed(true); +// unsigned long tCon = profile.elapsed(true); bTxWebData = true; // OLED tx data animation flag webSocket.broadcastTXT(Str); - unsigned long tWeb = profile.elapsed(true); - DebugPort.printf("Websend times : %ld,%ld\r\n", tCon, tWeb); +// unsigned long tWeb = profile.elapsed(true); +// DebugPort.printf("Websend times : %ld,%ld\r\n", tCon, tWeb); return true; } return false; @@ -538,6 +541,7 @@ void onErase() // function called upon completion of file (form) upload void onUploadCompletion() { + _SuppliedFileSize = 0; DebugPort.println("WEB: POST /updatenow completion"); // completion functionality if(SPIFFSupload) { @@ -596,10 +600,17 @@ void onUploadProgression() HTTPUpload& upload = server.upload(); if (upload.status == UPLOAD_FILE_START) { String filename = upload.filename; + // CRC checking + nUploadSize = upload.totalSize; + DebugPort.setDebugOutput(true); if(filename.endsWith(".bin")) { - DebugPort.printf("Update: %s %d\r\n", filename.c_str(), upload.totalSize); - if (!Update.begin()) { //start with max available size + DebugPort.printf("Update: %s\r\n", filename.c_str()); + int sizetouse = -1; //start with max available size + if(_SuppliedFileSize) { + sizetouse = _SuppliedFileSize; // adapt to websocket supplied size + } + if (!Update.begin(sizetouse)) { Update.printError(DebugPort); } } @@ -608,7 +619,6 @@ void onUploadProgression() DebugPort.printf("handleFileUpload Name: %s\r\n", filename.c_str()); fsUploadFile = SPIFFS.open(filename, "w"); // Open the file for writing in SPIFFS (create if it doesn't exist) SPIFFSupload = fsUploadFile ? 1 : 2; - //filename = String(); } } @@ -640,6 +650,7 @@ void onUploadProgression() // handle end of upload else if (upload.status == UPLOAD_FILE_END) { + delay(2000); if(SPIFFSupload) { if(fsUploadFile) { fsUploadFile.close(); // Close the file again @@ -647,7 +658,10 @@ void onUploadProgression() } } else { - if (Update.end(true)) { //true to set the size to the current progress + if(!CheckFirmwareCRC(_SuppliedFileSize)) + Update.abort(); + + if (Update.end()) { DebugPort.printf("Update Success: %u\r\nRebooting...\r\n", upload.totalSize); } else { Update.printError(DebugPort); diff --git a/src/Afterburner/src/WiFi/BTCota.cpp b/src/Afterburner/src/WiFi/BTCota.cpp index 00ca3ed..a1db343 100644 --- a/src/Afterburner/src/WiFi/BTCota.cpp +++ b/src/Afterburner/src/WiFi/BTCota.cpp @@ -28,6 +28,8 @@ #include #include #include +#include "../Utility/MODBUS-CRC16.h" +#include "esp_ota_ops.h" esp32FOTA FOTA("afterburner-fota-http", int(getVersion()*1000)); @@ -142,4 +144,42 @@ bool isUpdateAvailable(bool test) void checkFOTA() { FOTAtime = millis(); +} + +const int CRCbufsize = 1024; +uint8_t CRCReadBuff[CRCbufsize]; + +bool CheckFirmwareCRC(int filesize) +{ + const esp_partition_t* pUsePartition = esp_ota_get_next_update_partition(NULL); + if(NULL == pUsePartition) { + DebugPort.println("CheckCRC: FAILED - bad partition?"); + return false; + } + int size = (filesize >> 2) << 2; // mod 4 + if((filesize - size) != 2) { + // we expect 2 extra bytes where the custom CRC is added + // all normal applications without CRC are multiples of 4 + DebugPort.println("CheckCRC: FAILED - bad source file size"); + return false; + } + + CModBusCRC16 CRCengine; + + int processed = 0; + while(processed < size) { + int toRead = size - processed; + if(toRead > CRCbufsize) + toRead = CRCbufsize; + + ESP.flashRead(pUsePartition->address + processed, (uint32_t*)CRCReadBuff, toRead); + CRCengine.process(toRead, CRCReadBuff); + processed += toRead; + } + ESP.flashRead(pUsePartition->address + processed, (uint32_t*)CRCReadBuff, 4); + uint32_t fileCRC = CRCReadBuff[0] + (CRCReadBuff[1]<<8); + + DebugPort.printf("Upload CRC TEST: calcCRC= %04X, fileCRC = %08X\r\n", CRCengine.get(), fileCRC); + + return fileCRC == CRCengine.get(); } \ No newline at end of file diff --git a/src/Afterburner/src/WiFi/BTCota.h b/src/Afterburner/src/WiFi/BTCota.h index 3b6069a..d992f63 100644 --- a/src/Afterburner/src/WiFi/BTCota.h +++ b/src/Afterburner/src/WiFi/BTCota.h @@ -24,5 +24,6 @@ void initOTA(); void DoOTA(); +bool CheckFirmwareCRC(int size); #endif \ No newline at end of file