1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626562756285629563056315632563356345635563656375638563956405641564256435644564556465647564856495650565156525653565456555656565756585659566056615662566356645665566656675668566956705671567256735674567556765677567856795680568156825683568456855686568756885689569056915692569356945695569656975698569957005701570257035704570557065707570857095710571157125713571457155716571757185719572057215722572357245725572657275728572957305731573257335734 |
- //! Program state processor
- use crate::{
- error::TokenError,
- instruction::{is_valid_signer_index, AuthorityType, TokenInstruction, MAX_SIGNERS},
- state::{Account, AccountState, Mint, Multisig},
- };
- use num_traits::FromPrimitive;
- use solana_program::{
- account_info::{next_account_info, AccountInfo},
- decode_error::DecodeError,
- entrypoint::ProgramResult,
- msg,
- program_error::{PrintProgramError, ProgramError},
- program_option::COption,
- program_pack::{IsInitialized, Pack},
- pubkey::Pubkey,
- sysvar::{rent::Rent, Sysvar},
- };
- /// Program state handler.
- pub struct Processor {}
- impl Processor {
- /// Processes an [InitializeMint](enum.TokenInstruction.html) instruction.
- pub fn process_initialize_mint(
- accounts: &[AccountInfo],
- decimals: u8,
- mint_authority: Pubkey,
- freeze_authority: COption<Pubkey>,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let mint_info = next_account_info(account_info_iter)?;
- let mint_data_len = mint_info.data_len();
- let rent = &Rent::from_account_info(next_account_info(account_info_iter)?)?;
- let mut mint = Mint::unpack_unchecked(&mint_info.data.borrow())?;
- if mint.is_initialized {
- return Err(TokenError::AlreadyInUse.into());
- }
- if !rent.is_exempt(mint_info.lamports(), mint_data_len) {
- return Err(TokenError::NotRentExempt.into());
- }
- mint.mint_authority = COption::Some(mint_authority);
- mint.decimals = decimals;
- mint.is_initialized = true;
- mint.freeze_authority = freeze_authority;
- Mint::pack(mint, &mut mint_info.data.borrow_mut())?;
- Ok(())
- }
- fn _process_initialize_account(
- accounts: &[AccountInfo],
- owner: Option<&Pubkey>,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let new_account_info = next_account_info(account_info_iter)?;
- let mint_info = next_account_info(account_info_iter)?;
- let owner = if let Some(owner) = owner {
- owner
- } else {
- next_account_info(account_info_iter)?.key
- };
- let new_account_info_data_len = new_account_info.data_len();
- let rent = &Rent::from_account_info(next_account_info(account_info_iter)?)?;
- let mut account = Account::unpack_unchecked(&new_account_info.data.borrow())?;
- if account.is_initialized() {
- return Err(TokenError::AlreadyInUse.into());
- }
- if !rent.is_exempt(new_account_info.lamports(), new_account_info_data_len) {
- return Err(TokenError::NotRentExempt.into());
- }
- if *mint_info.key != crate::native_mint::id() {
- let _ = Mint::unpack(&mint_info.data.borrow_mut())
- .map_err(|_| Into::<ProgramError>::into(TokenError::InvalidMint))?;
- }
- account.mint = *mint_info.key;
- account.owner = *owner;
- account.delegate = COption::None;
- account.delegated_amount = 0;
- account.state = AccountState::Initialized;
- if *mint_info.key == crate::native_mint::id() {
- let rent_exempt_reserve = rent.minimum_balance(new_account_info_data_len);
- account.is_native = COption::Some(rent_exempt_reserve);
- account.amount = new_account_info
- .lamports()
- .checked_sub(rent_exempt_reserve)
- .ok_or(TokenError::Overflow)?;
- } else {
- account.is_native = COption::None;
- account.amount = 0;
- };
- Account::pack(account, &mut new_account_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes an [InitializeAccount](enum.TokenInstruction.html) instruction.
- pub fn process_initialize_account(accounts: &[AccountInfo]) -> ProgramResult {
- Self::_process_initialize_account(accounts, None)
- }
- /// Processes an [InitializeAccount2](enum.TokenInstruction.html) instruction.
- pub fn process_initialize_account2(accounts: &[AccountInfo], owner: Pubkey) -> ProgramResult {
- Self::_process_initialize_account(accounts, Some(&owner))
- }
- /// Processes a [InitializeMultisig](enum.TokenInstruction.html) instruction.
- pub fn process_initialize_multisig(accounts: &[AccountInfo], m: u8) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let multisig_info = next_account_info(account_info_iter)?;
- let multisig_info_data_len = multisig_info.data_len();
- let rent = &Rent::from_account_info(next_account_info(account_info_iter)?)?;
- let mut multisig = Multisig::unpack_unchecked(&multisig_info.data.borrow())?;
- if multisig.is_initialized {
- return Err(TokenError::AlreadyInUse.into());
- }
- if !rent.is_exempt(multisig_info.lamports(), multisig_info_data_len) {
- return Err(TokenError::NotRentExempt.into());
- }
- let signer_infos = account_info_iter.as_slice();
- multisig.m = m;
- multisig.n = signer_infos.len() as u8;
- if !is_valid_signer_index(multisig.n as usize) {
- return Err(TokenError::InvalidNumberOfProvidedSigners.into());
- }
- if !is_valid_signer_index(multisig.m as usize) {
- return Err(TokenError::InvalidNumberOfRequiredSigners.into());
- }
- for (i, signer_info) in signer_infos.iter().enumerate() {
- multisig.signers[i] = *signer_info.key;
- }
- multisig.is_initialized = true;
- Multisig::pack(multisig, &mut multisig_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes a [Transfer](enum.TokenInstruction.html) instruction.
- pub fn process_transfer(
- program_id: &Pubkey,
- accounts: &[AccountInfo],
- amount: u64,
- expected_decimals: Option<u8>,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let source_account_info = next_account_info(account_info_iter)?;
- #[allow(clippy::manual_map)]
- let expected_mint_info = if let Some(expected_decimals) = expected_decimals {
- Some((next_account_info(account_info_iter)?, expected_decimals))
- } else {
- None
- };
- let dest_account_info = next_account_info(account_info_iter)?;
- let authority_info = next_account_info(account_info_iter)?;
- let mut source_account = Account::unpack(&source_account_info.data.borrow())?;
- let mut dest_account = Account::unpack(&dest_account_info.data.borrow())?;
- if source_account.is_frozen() || dest_account.is_frozen() {
- return Err(TokenError::AccountFrozen.into());
- }
- if source_account.amount < amount {
- return Err(TokenError::InsufficientFunds.into());
- }
- if source_account.mint != dest_account.mint {
- return Err(TokenError::MintMismatch.into());
- }
- if let Some((mint_info, expected_decimals)) = expected_mint_info {
- if source_account.mint != *mint_info.key {
- return Err(TokenError::MintMismatch.into());
- }
- let mint = Mint::unpack(&mint_info.data.borrow_mut())?;
- if expected_decimals != mint.decimals {
- return Err(TokenError::MintDecimalsMismatch.into());
- }
- }
- let self_transfer = source_account_info.key == dest_account_info.key;
- match source_account.delegate {
- COption::Some(ref delegate) if authority_info.key == delegate => {
- Self::validate_owner(
- program_id,
- delegate,
- authority_info,
- account_info_iter.as_slice(),
- )?;
- if source_account.delegated_amount < amount {
- return Err(TokenError::InsufficientFunds.into());
- }
- if !self_transfer {
- source_account.delegated_amount = source_account
- .delegated_amount
- .checked_sub(amount)
- .ok_or(TokenError::Overflow)?;
- if source_account.delegated_amount == 0 {
- source_account.delegate = COption::None;
- }
- }
- }
- _ => Self::validate_owner(
- program_id,
- &source_account.owner,
- authority_info,
- account_info_iter.as_slice(),
- )?,
- };
- // This check MUST occur just before the amounts are manipulated
- // to ensure self-transfers are fully validated
- if self_transfer {
- return Ok(());
- }
- source_account.amount = source_account
- .amount
- .checked_sub(amount)
- .ok_or(TokenError::Overflow)?;
- dest_account.amount = dest_account
- .amount
- .checked_add(amount)
- .ok_or(TokenError::Overflow)?;
- if source_account.is_native() {
- let source_starting_lamports = source_account_info.lamports();
- **source_account_info.lamports.borrow_mut() = source_starting_lamports
- .checked_sub(amount)
- .ok_or(TokenError::Overflow)?;
- let dest_starting_lamports = dest_account_info.lamports();
- **dest_account_info.lamports.borrow_mut() = dest_starting_lamports
- .checked_add(amount)
- .ok_or(TokenError::Overflow)?;
- }
- Account::pack(source_account, &mut source_account_info.data.borrow_mut())?;
- Account::pack(dest_account, &mut dest_account_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes an [Approve](enum.TokenInstruction.html) instruction.
- pub fn process_approve(
- program_id: &Pubkey,
- accounts: &[AccountInfo],
- amount: u64,
- expected_decimals: Option<u8>,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let source_account_info = next_account_info(account_info_iter)?;
- #[allow(clippy::manual_map)]
- let expected_mint_info = if let Some(expected_decimals) = expected_decimals {
- Some((next_account_info(account_info_iter)?, expected_decimals))
- } else {
- None
- };
- let delegate_info = next_account_info(account_info_iter)?;
- let owner_info = next_account_info(account_info_iter)?;
- let mut source_account = Account::unpack(&source_account_info.data.borrow())?;
- if source_account.is_frozen() {
- return Err(TokenError::AccountFrozen.into());
- }
- if let Some((mint_info, expected_decimals)) = expected_mint_info {
- if source_account.mint != *mint_info.key {
- return Err(TokenError::MintMismatch.into());
- }
- let mint = Mint::unpack(&mint_info.data.borrow_mut())?;
- if expected_decimals != mint.decimals {
- return Err(TokenError::MintDecimalsMismatch.into());
- }
- }
- Self::validate_owner(
- program_id,
- &source_account.owner,
- owner_info,
- account_info_iter.as_slice(),
- )?;
- source_account.delegate = COption::Some(*delegate_info.key);
- source_account.delegated_amount = amount;
- Account::pack(source_account, &mut source_account_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes an [Revoke](enum.TokenInstruction.html) instruction.
- pub fn process_revoke(program_id: &Pubkey, accounts: &[AccountInfo]) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let source_account_info = next_account_info(account_info_iter)?;
- let mut source_account = Account::unpack(&source_account_info.data.borrow())?;
- let owner_info = next_account_info(account_info_iter)?;
- if source_account.is_frozen() {
- return Err(TokenError::AccountFrozen.into());
- }
- Self::validate_owner(
- program_id,
- &source_account.owner,
- owner_info,
- account_info_iter.as_slice(),
- )?;
- source_account.delegate = COption::None;
- source_account.delegated_amount = 0;
- Account::pack(source_account, &mut source_account_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes a [SetAuthority](enum.TokenInstruction.html) instruction.
- pub fn process_set_authority(
- program_id: &Pubkey,
- accounts: &[AccountInfo],
- authority_type: AuthorityType,
- new_authority: COption<Pubkey>,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let account_info = next_account_info(account_info_iter)?;
- let authority_info = next_account_info(account_info_iter)?;
- if account_info.data_len() == Account::get_packed_len() {
- let mut account = Account::unpack(&account_info.data.borrow())?;
- if account.is_frozen() {
- return Err(TokenError::AccountFrozen.into());
- }
- match authority_type {
- AuthorityType::AccountOwner => {
- Self::validate_owner(
- program_id,
- &account.owner,
- authority_info,
- account_info_iter.as_slice(),
- )?;
- if let COption::Some(authority) = new_authority {
- account.owner = authority;
- } else {
- return Err(TokenError::InvalidInstruction.into());
- }
- }
- AuthorityType::CloseAccount => {
- let authority = account.close_authority.unwrap_or(account.owner);
- Self::validate_owner(
- program_id,
- &authority,
- authority_info,
- account_info_iter.as_slice(),
- )?;
- account.close_authority = new_authority;
- }
- _ => {
- return Err(TokenError::AuthorityTypeNotSupported.into());
- }
- }
- Account::pack(account, &mut account_info.data.borrow_mut())?;
- } else if account_info.data_len() == Mint::get_packed_len() {
- let mut mint = Mint::unpack(&account_info.data.borrow())?;
- match authority_type {
- AuthorityType::MintTokens => {
- // Once a mint's supply is fixed, it cannot be undone by setting a new
- // mint_authority
- let mint_authority = mint
- .mint_authority
- .ok_or(Into::<ProgramError>::into(TokenError::FixedSupply))?;
- Self::validate_owner(
- program_id,
- &mint_authority,
- authority_info,
- account_info_iter.as_slice(),
- )?;
- mint.mint_authority = new_authority;
- }
- AuthorityType::FreezeAccount => {
- // Once a mint's freeze authority is disabled, it cannot be re-enabled by
- // setting a new freeze_authority
- let freeze_authority = mint
- .freeze_authority
- .ok_or(Into::<ProgramError>::into(TokenError::MintCannotFreeze))?;
- Self::validate_owner(
- program_id,
- &freeze_authority,
- authority_info,
- account_info_iter.as_slice(),
- )?;
- mint.freeze_authority = new_authority;
- }
- _ => {
- return Err(TokenError::AuthorityTypeNotSupported.into());
- }
- }
- Mint::pack(mint, &mut account_info.data.borrow_mut())?;
- } else {
- return Err(ProgramError::InvalidArgument);
- }
- Ok(())
- }
- /// Processes a [MintTo](enum.TokenInstruction.html) instruction.
- pub fn process_mint_to(
- program_id: &Pubkey,
- accounts: &[AccountInfo],
- amount: u64,
- expected_decimals: Option<u8>,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let mint_info = next_account_info(account_info_iter)?;
- let dest_account_info = next_account_info(account_info_iter)?;
- let owner_info = next_account_info(account_info_iter)?;
- let mut dest_account = Account::unpack(&dest_account_info.data.borrow())?;
- if dest_account.is_frozen() {
- return Err(TokenError::AccountFrozen.into());
- }
- if dest_account.is_native() {
- return Err(TokenError::NativeNotSupported.into());
- }
- if mint_info.key != &dest_account.mint {
- return Err(TokenError::MintMismatch.into());
- }
- let mut mint = Mint::unpack(&mint_info.data.borrow())?;
- if let Some(expected_decimals) = expected_decimals {
- if expected_decimals != mint.decimals {
- return Err(TokenError::MintDecimalsMismatch.into());
- }
- }
- match mint.mint_authority {
- COption::Some(mint_authority) => Self::validate_owner(
- program_id,
- &mint_authority,
- owner_info,
- account_info_iter.as_slice(),
- )?,
- COption::None => return Err(TokenError::FixedSupply.into()),
- }
- dest_account.amount = dest_account
- .amount
- .checked_add(amount)
- .ok_or(TokenError::Overflow)?;
- mint.supply = mint
- .supply
- .checked_add(amount)
- .ok_or(TokenError::Overflow)?;
- Account::pack(dest_account, &mut dest_account_info.data.borrow_mut())?;
- Mint::pack(mint, &mut mint_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes a [Burn](enum.TokenInstruction.html) instruction.
- pub fn process_burn(
- program_id: &Pubkey,
- accounts: &[AccountInfo],
- amount: u64,
- expected_decimals: Option<u8>,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let source_account_info = next_account_info(account_info_iter)?;
- let mint_info = next_account_info(account_info_iter)?;
- let authority_info = next_account_info(account_info_iter)?;
- let mut source_account = Account::unpack(&source_account_info.data.borrow())?;
- let mut mint = Mint::unpack(&mint_info.data.borrow())?;
- if source_account.is_frozen() {
- return Err(TokenError::AccountFrozen.into());
- }
- if source_account.is_native() {
- return Err(TokenError::NativeNotSupported.into());
- }
- if source_account.amount < amount {
- return Err(TokenError::InsufficientFunds.into());
- }
- if mint_info.key != &source_account.mint {
- return Err(TokenError::MintMismatch.into());
- }
- if let Some(expected_decimals) = expected_decimals {
- if expected_decimals != mint.decimals {
- return Err(TokenError::MintDecimalsMismatch.into());
- }
- }
- match source_account.delegate {
- COption::Some(ref delegate) if authority_info.key == delegate => {
- Self::validate_owner(
- program_id,
- delegate,
- authority_info,
- account_info_iter.as_slice(),
- )?;
- if source_account.delegated_amount < amount {
- return Err(TokenError::InsufficientFunds.into());
- }
- source_account.delegated_amount = source_account
- .delegated_amount
- .checked_sub(amount)
- .ok_or(TokenError::Overflow)?;
- if source_account.delegated_amount == 0 {
- source_account.delegate = COption::None;
- }
- }
- _ => Self::validate_owner(
- program_id,
- &source_account.owner,
- authority_info,
- account_info_iter.as_slice(),
- )?,
- }
- source_account.amount = source_account
- .amount
- .checked_sub(amount)
- .ok_or(TokenError::Overflow)?;
- mint.supply = mint
- .supply
- .checked_sub(amount)
- .ok_or(TokenError::Overflow)?;
- Account::pack(source_account, &mut source_account_info.data.borrow_mut())?;
- Mint::pack(mint, &mut mint_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes a [CloseAccount](enum.TokenInstruction.html) instruction.
- pub fn process_close_account(program_id: &Pubkey, accounts: &[AccountInfo]) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let source_account_info = next_account_info(account_info_iter)?;
- let dest_account_info = next_account_info(account_info_iter)?;
- let authority_info = next_account_info(account_info_iter)?;
- let mut source_account = Account::unpack(&source_account_info.data.borrow())?;
- if !source_account.is_native() && source_account.amount != 0 {
- return Err(TokenError::NonNativeHasBalance.into());
- }
- let authority = source_account
- .close_authority
- .unwrap_or(source_account.owner);
- Self::validate_owner(
- program_id,
- &authority,
- authority_info,
- account_info_iter.as_slice(),
- )?;
- let dest_starting_lamports = dest_account_info.lamports();
- **dest_account_info.lamports.borrow_mut() = dest_starting_lamports
- .checked_add(source_account_info.lamports())
- .ok_or(TokenError::Overflow)?;
- **source_account_info.lamports.borrow_mut() = 0;
- source_account.amount = 0;
- Account::pack(source_account, &mut source_account_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes a [FreezeAccount](enum.TokenInstruction.html) or a
- /// [ThawAccount](enum.TokenInstruction.html) instruction.
- pub fn process_toggle_freeze_account(
- program_id: &Pubkey,
- accounts: &[AccountInfo],
- freeze: bool,
- ) -> ProgramResult {
- let account_info_iter = &mut accounts.iter();
- let source_account_info = next_account_info(account_info_iter)?;
- let mint_info = next_account_info(account_info_iter)?;
- let authority_info = next_account_info(account_info_iter)?;
- let mut source_account = Account::unpack(&source_account_info.data.borrow())?;
- if freeze && source_account.is_frozen() || !freeze && !source_account.is_frozen() {
- return Err(TokenError::InvalidState.into());
- }
- if source_account.is_native() {
- return Err(TokenError::NativeNotSupported.into());
- }
- if mint_info.key != &source_account.mint {
- return Err(TokenError::MintMismatch.into());
- }
- let mint = Mint::unpack(&mint_info.data.borrow_mut())?;
- match mint.freeze_authority {
- COption::Some(authority) => Self::validate_owner(
- program_id,
- &authority,
- authority_info,
- account_info_iter.as_slice(),
- ),
- COption::None => Err(TokenError::MintCannotFreeze.into()),
- }?;
- source_account.state = if freeze {
- AccountState::Frozen
- } else {
- AccountState::Initialized
- };
- Account::pack(source_account, &mut source_account_info.data.borrow_mut())?;
- Ok(())
- }
- /// Processes an [Instruction](enum.Instruction.html).
- pub fn process(program_id: &Pubkey, accounts: &[AccountInfo], input: &[u8]) -> ProgramResult {
- let instruction = TokenInstruction::unpack(input)?;
- match instruction {
- TokenInstruction::InitializeMint {
- decimals,
- mint_authority,
- freeze_authority,
- } => {
- msg!("Instruction: InitializeMint");
- Self::process_initialize_mint(accounts, decimals, mint_authority, freeze_authority)
- }
- TokenInstruction::InitializeAccount => {
- msg!("Instruction: InitializeAccount");
- Self::process_initialize_account(accounts)
- }
- TokenInstruction::InitializeAccount2 { owner } => {
- msg!("Instruction: InitializeAccount2");
- Self::process_initialize_account2(accounts, owner)
- }
- TokenInstruction::InitializeMultisig { m } => {
- msg!("Instruction: InitializeMultisig");
- Self::process_initialize_multisig(accounts, m)
- }
- TokenInstruction::Transfer { amount } => {
- msg!("Instruction: Transfer");
- Self::process_transfer(program_id, accounts, amount, None)
- }
- TokenInstruction::Approve { amount } => {
- msg!("Instruction: Approve");
- Self::process_approve(program_id, accounts, amount, None)
- }
- TokenInstruction::Revoke => {
- msg!("Instruction: Revoke");
- Self::process_revoke(program_id, accounts)
- }
- TokenInstruction::SetAuthority {
- authority_type,
- new_authority,
- } => {
- msg!("Instruction: SetAuthority");
- Self::process_set_authority(program_id, accounts, authority_type, new_authority)
- }
- TokenInstruction::MintTo { amount } => {
- msg!("Instruction: MintTo");
- Self::process_mint_to(program_id, accounts, amount, None)
- }
- TokenInstruction::Burn { amount } => {
- msg!("Instruction: Burn");
- Self::process_burn(program_id, accounts, amount, None)
- }
- TokenInstruction::CloseAccount => {
- msg!("Instruction: CloseAccount");
- Self::process_close_account(program_id, accounts)
- }
- TokenInstruction::FreezeAccount => {
- msg!("Instruction: FreezeAccount");
- Self::process_toggle_freeze_account(program_id, accounts, true)
- }
- TokenInstruction::ThawAccount => {
- msg!("Instruction: FreezeAccount");
- Self::process_toggle_freeze_account(program_id, accounts, false)
- }
- TokenInstruction::TransferChecked { amount, decimals } => {
- msg!("Instruction: TransferChecked");
- Self::process_transfer(program_id, accounts, amount, Some(decimals))
- }
- TokenInstruction::ApproveChecked { amount, decimals } => {
- msg!("Instruction: ApproveChecked");
- Self::process_approve(program_id, accounts, amount, Some(decimals))
- }
- TokenInstruction::MintToChecked { amount, decimals } => {
- msg!("Instruction: MintToChecked");
- Self::process_mint_to(program_id, accounts, amount, Some(decimals))
- }
- TokenInstruction::BurnChecked { amount, decimals } => {
- msg!("Instruction: BurnChecked");
- Self::process_burn(program_id, accounts, amount, Some(decimals))
- }
- }
- }
- /// Validates owner(s) are present
- pub fn validate_owner(
- program_id: &Pubkey,
- expected_owner: &Pubkey,
- owner_account_info: &AccountInfo,
- signers: &[AccountInfo],
- ) -> ProgramResult {
- if expected_owner != owner_account_info.key {
- return Err(TokenError::OwnerMismatch.into());
- }
- if program_id == owner_account_info.owner
- && owner_account_info.data_len() == Multisig::get_packed_len()
- {
- let multisig = Multisig::unpack(&owner_account_info.data.borrow())?;
- let mut num_signers = 0;
- let mut matched = [false; MAX_SIGNERS];
- for signer in signers.iter() {
- for (position, key) in multisig.signers[0..multisig.n as usize].iter().enumerate() {
- if key == signer.key && !matched[position] {
- if !signer.is_signer {
- return Err(ProgramError::MissingRequiredSignature);
- }
- matched[position] = true;
- num_signers += 1;
- }
- }
- }
- if num_signers < multisig.m {
- return Err(ProgramError::MissingRequiredSignature);
- }
- return Ok(());
- } else if !owner_account_info.is_signer {
- return Err(ProgramError::MissingRequiredSignature);
- }
- Ok(())
- }
- }
- impl PrintProgramError for TokenError {
- fn print<E>(&self)
- where
- E: 'static + std::error::Error + DecodeError<E> + PrintProgramError + FromPrimitive,
- {
- match self {
- TokenError::NotRentExempt => msg!("Error: Lamport balance below rent-exempt threshold"),
- TokenError::InsufficientFunds => msg!("Error: insufficient funds"),
- TokenError::InvalidMint => msg!("Error: Invalid Mint"),
- TokenError::MintMismatch => msg!("Error: Account not associated with this Mint"),
- TokenError::OwnerMismatch => msg!("Error: owner does not match"),
- TokenError::FixedSupply => msg!("Error: the total supply of this token is fixed"),
- TokenError::AlreadyInUse => msg!("Error: account or token already in use"),
- TokenError::InvalidNumberOfProvidedSigners => {
- msg!("Error: Invalid number of provided signers")
- }
- TokenError::InvalidNumberOfRequiredSigners => {
- msg!("Error: Invalid number of required signers")
- }
- TokenError::UninitializedState => msg!("Error: State is uninitialized"),
- TokenError::NativeNotSupported => {
- msg!("Error: Instruction does not support native tokens")
- }
- TokenError::NonNativeHasBalance => {
- msg!("Error: Non-native account can only be closed if its balance is zero")
- }
- TokenError::InvalidInstruction => msg!("Error: Invalid instruction"),
- TokenError::InvalidState => msg!("Error: Invalid account state for operation"),
- TokenError::Overflow => msg!("Error: Operation overflowed"),
- TokenError::AuthorityTypeNotSupported => {
- msg!("Error: Account does not support specified authority type")
- }
- TokenError::MintCannotFreeze => msg!("Error: This token mint cannot freeze accounts"),
- TokenError::AccountFrozen => msg!("Error: Account is frozen"),
- TokenError::MintDecimalsMismatch => {
- msg!("Error: decimals different from the Mint decimals")
- }
- }
- }
- }
- #[cfg(test)]
- mod tests {
- use super::*;
- use crate::instruction::*;
- use solana_program::{
- account_info::IntoAccountInfo, clock::Epoch, instruction::Instruction, sysvar::rent,
- };
- use solana_sdk::account::{
- create_account, create_is_signer_account_infos, Account as SolanaAccount,
- };
- fn do_process_instruction(
- instruction: Instruction,
- accounts: Vec<&mut SolanaAccount>,
- ) -> ProgramResult {
- let mut meta = instruction
- .accounts
- .iter()
- .zip(accounts)
- .map(|(account_meta, account)| (&account_meta.pubkey, account_meta.is_signer, account))
- .collect::<Vec<_>>();
- let account_infos = create_is_signer_account_infos(&mut meta);
- Processor::process(&instruction.program_id, &account_infos, &instruction.data)
- }
- fn do_process_instruction_dups(
- instruction: Instruction,
- account_infos: Vec<AccountInfo>,
- ) -> ProgramResult {
- Processor::process(&instruction.program_id, &account_infos, &instruction.data)
- }
- fn return_token_error_as_program_error() -> ProgramError {
- TokenError::MintMismatch.into()
- }
- fn rent_sysvar() -> SolanaAccount {
- create_account(&Rent::default(), 42)
- }
- fn mint_minimum_balance() -> u64 {
- Rent::default().minimum_balance(Mint::get_packed_len())
- }
- fn account_minimum_balance() -> u64 {
- Rent::default().minimum_balance(Account::get_packed_len())
- }
- fn multisig_minimum_balance() -> u64 {
- Rent::default().minimum_balance(Multisig::get_packed_len())
- }
- #[test]
- fn test_print_error() {
- let error = return_token_error_as_program_error();
- error.print::<TokenError>();
- }
- #[test]
- #[should_panic(expected = "Custom(3)")]
- fn test_error_unwrap() {
- Err::<(), ProgramError>(return_token_error_as_program_error()).unwrap();
- }
- #[test]
- fn test_unique_account_sizes() {
- assert_ne!(Mint::get_packed_len(), 0);
- assert_ne!(Mint::get_packed_len(), Account::get_packed_len());
- assert_ne!(Mint::get_packed_len(), Multisig::get_packed_len());
- assert_ne!(Account::get_packed_len(), 0);
- assert_ne!(Account::get_packed_len(), Multisig::get_packed_len());
- assert_ne!(Multisig::get_packed_len(), 0);
- }
- #[test]
- fn test_pack_unpack() {
- // Mint
- let check = Mint {
- mint_authority: COption::Some(Pubkey::new(&[1; 32])),
- supply: 42,
- decimals: 7,
- is_initialized: true,
- freeze_authority: COption::Some(Pubkey::new(&[2; 32])),
- };
- let mut packed = vec![0; Mint::get_packed_len() + 1];
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- Mint::pack(check, &mut packed)
- );
- let mut packed = vec![0; Mint::get_packed_len() - 1];
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- Mint::pack(check, &mut packed)
- );
- let mut packed = vec![0; Mint::get_packed_len()];
- Mint::pack(check, &mut packed).unwrap();
- let expect = vec![
- 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
- 1, 1, 1, 1, 1, 1, 1, 42, 0, 0, 0, 0, 0, 0, 0, 7, 1, 1, 0, 0, 0, 2, 2, 2, 2, 2, 2, 2, 2,
- 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
- ];
- assert_eq!(packed, expect);
- let unpacked = Mint::unpack(&packed).unwrap();
- assert_eq!(unpacked, check);
- // Account
- let check = Account {
- mint: Pubkey::new(&[1; 32]),
- owner: Pubkey::new(&[2; 32]),
- amount: 3,
- delegate: COption::Some(Pubkey::new(&[4; 32])),
- state: AccountState::Frozen,
- is_native: COption::Some(5),
- delegated_amount: 6,
- close_authority: COption::Some(Pubkey::new(&[7; 32])),
- };
- let mut packed = vec![0; Account::get_packed_len() + 1];
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- Account::pack(check, &mut packed)
- );
- let mut packed = vec![0; Account::get_packed_len() - 1];
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- Account::pack(check, &mut packed)
- );
- let mut packed = vec![0; Account::get_packed_len()];
- Account::pack(check, &mut packed).unwrap();
- let expect = vec![
- 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
- 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
- 2, 2, 2, 2, 2, 2, 3, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
- 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 2, 1, 0, 0, 0, 5, 0, 0,
- 0, 0, 0, 0, 0, 6, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
- 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
- ];
- assert_eq!(packed, expect);
- let unpacked = Account::unpack(&packed).unwrap();
- assert_eq!(unpacked, check);
- // Multisig
- let check = Multisig {
- m: 1,
- n: 2,
- is_initialized: true,
- signers: [Pubkey::new(&[3; 32]); MAX_SIGNERS],
- };
- let mut packed = vec![0; Multisig::get_packed_len() + 1];
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- Multisig::pack(check, &mut packed)
- );
- let mut packed = vec![0; Multisig::get_packed_len() - 1];
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- Multisig::pack(check, &mut packed)
- );
- let mut packed = vec![0; Multisig::get_packed_len()];
- Multisig::pack(check, &mut packed).unwrap();
- let expect = vec![
- 1, 2, 1, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
- 3, 3, 3, 3, 3, 3, 3,
- ];
- assert_eq!(packed, expect);
- let unpacked = Multisig::unpack(&packed).unwrap();
- assert_eq!(unpacked, check);
- }
- #[test]
- fn test_initialize_mint() {
- let program_id = Pubkey::new_unique();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account = SolanaAccount::new(42, Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut mint2_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // mint is not rent exempt
- assert_eq!(
- Err(TokenError::NotRentExempt.into()),
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar]
- )
- );
- mint_account.lamports = mint_minimum_balance();
- // create new mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create twice
- assert_eq!(
- Err(TokenError::AlreadyInUse.into()),
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2,).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar]
- )
- );
- // create another mint that can freeze
- do_process_instruction(
- initialize_mint(&program_id, &mint2_key, &owner_key, Some(&owner_key), 2).unwrap(),
- vec![&mut mint2_account, &mut rent_sysvar],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint2_account.data).unwrap();
- assert_eq!(mint.freeze_authority, COption::Some(owner_key));
- }
- #[test]
- fn test_initialize_mint_account() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(42, Account::get_packed_len(), &program_id);
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // account is not rent exempt
- assert_eq!(
- Err(TokenError::NotRentExempt.into()),
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar
- ],
- )
- );
- account_account.lamports = account_minimum_balance();
- // mint is not valid (not initialized)
- assert_eq!(
- Err(TokenError::InvalidMint.into()),
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar
- ],
- )
- );
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create twice
- assert_eq!(
- Err(TokenError::AlreadyInUse.into()),
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar
- ],
- )
- );
- }
- #[test]
- fn test_transfer_dups() {
- let program_id = Pubkey::new_unique();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut account2_info: AccountInfo = (&account2_key, false, &mut account2_account).into();
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_info: AccountInfo = (&account3_key, false, &mut account3_account).into();
- let account4_key = Pubkey::new_unique();
- let mut account4_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account4_info: AccountInfo = (&account4_key, true, &mut account4_account).into();
- let multisig_key = Pubkey::new_unique();
- let mut multisig_account = SolanaAccount::new(
- multisig_minimum_balance(),
- Multisig::get_packed_len(),
- &program_id,
- );
- let multisig_info: AccountInfo = (&multisig_key, true, &mut multisig_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, false, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction_dups(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- account2_info.clone(),
- mint_info.clone(),
- owner_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // mint to account
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- )
- .unwrap();
- // source-owner transfer
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-owner TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-delegate transfer
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.amount = 1000;
- account.delegated_amount = 1000;
- account.delegate = COption::Some(account1_key);
- account.owner = owner_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-delegate TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // test destination-owner transfer
- do_process_instruction_dups(
- initialize_account(&program_id, &account3_key, &mint_key, &account2_key).unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account3_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account3_info.clone(), owner_info.clone()],
- )
- .unwrap();
- account1_info.is_signer = false;
- account2_info.is_signer = true;
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account3_key,
- &account2_key,
- &account2_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- account2_info.clone(),
- account2_info.clone(),
- ],
- )
- .unwrap();
- // destination-owner TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account3_key,
- &mint_key,
- &account2_key,
- &account2_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account2_info.clone(),
- ],
- )
- .unwrap();
- // test source-multisig signer
- do_process_instruction_dups(
- initialize_multisig(&program_id, &multisig_key, &[&account4_key], 1).unwrap(),
- vec![
- multisig_info.clone(),
- rent_info.clone(),
- account4_info.clone(),
- ],
- )
- .unwrap();
- do_process_instruction_dups(
- initialize_account(&program_id, &account4_key, &mint_key, &multisig_key).unwrap(),
- vec![
- account4_info.clone(),
- mint_info.clone(),
- multisig_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account4_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account4_info.clone(), owner_info.clone()],
- )
- .unwrap();
- // source-multisig-signer transfer
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account4_key,
- &account2_key,
- &multisig_key,
- &[&account4_key],
- 500,
- )
- .unwrap(),
- vec![
- account4_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account4_info.clone(),
- ],
- )
- .unwrap();
- // source-multisig-signer TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account4_key,
- &mint_key,
- &account2_key,
- &multisig_key,
- &[&account4_key],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account4_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account4_info.clone(),
- ],
- )
- .unwrap();
- }
- #[test]
- fn test_transfer() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let mismatch_key = Pubkey::new_unique();
- let mut mismatch_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create mismatch account
- do_process_instruction(
- initialize_account(&program_id, &mismatch_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut mismatch_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&mismatch_account.data).unwrap();
- account.mint = mint2_key;
- Account::pack(account, &mut mismatch_account.data).unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- // missing signer
- let mut instruction = transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 1000,
- )
- .unwrap();
- instruction.accounts[2].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- );
- // mismatch mint
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &mismatch_key,
- &owner_key,
- &[],
- 1000
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mismatch_account,
- &mut owner_account,
- ],
- )
- );
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner2_key,
- &[],
- 1000
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner2_account,
- ],
- )
- );
- // transfer
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 1000,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // insufficient funds
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(&program_id, &account_key, &account2_key, &owner_key, &[], 1).unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- );
- // transfer half back
- do_process_instruction(
- transfer(
- &program_id,
- &account2_key,
- &account_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut account_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- transfer_checked(
- &program_id,
- &account2_key,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 1,
- 10 // <-- incorrect decimals
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut account_account,
- &mut owner_account,
- ],
- )
- );
- // incorrect mint
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- transfer_checked(
- &program_id,
- &account2_key,
- &account3_key, // <-- incorrect mint
- &account_key,
- &owner_key,
- &[],
- 1,
- 2
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut account3_account, // <-- incorrect mint
- &mut account_account,
- &mut owner_account,
- ],
- )
- );
- // transfer rest with explicit decimals
- do_process_instruction(
- transfer_checked(
- &program_id,
- &account2_key,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut account_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // insufficient funds
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(&program_id, &account2_key, &account_key, &owner_key, &[], 1).unwrap(),
- vec![
- &mut account2_account,
- &mut account_account,
- &mut owner_account,
- ],
- )
- );
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // transfer via delegate
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &delegate_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut delegate_account,
- ],
- )
- .unwrap();
- // insufficient funds approved via delegate
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &delegate_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut delegate_account,
- ],
- )
- );
- // transfer rest
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 900,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // insufficient funds in source account via delegate
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &delegate_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut delegate_account,
- ],
- )
- );
- }
- #[test]
- fn test_self_transfer() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- let account_info = (&account_key, false, &mut account_account).into_account_info();
- let account3_info = (&account3_key, false, &mut account3_account).into_account_info();
- let delegate_info = (&delegate_key, true, &mut delegate_account).into_account_info();
- let owner_info = (&owner_key, true, &mut owner_account).into_account_info();
- let owner2_info = (&owner2_key, true, &mut owner2_account).into_account_info();
- let mint_info = (&mint_key, false, &mut mint_account).into_account_info();
- // transfer
- let instruction = transfer(
- &program_id,
- &account_info.key,
- &account_info.key,
- &owner_info.key,
- &[],
- 1000,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- // transfer checked
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &owner_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- // missing signer
- let mut owner_no_sign_info = owner_info.clone();
- let mut instruction = transfer(
- &program_id,
- &account_info.key,
- &account_info.key,
- &owner_no_sign_info.key,
- &[],
- 1000,
- )
- .unwrap();
- instruction.accounts[2].is_signer = false;
- owner_no_sign_info.is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account_info.clone(),
- owner_no_sign_info.clone(),
- ],
- &instruction.data,
- )
- );
- // missing signer checked
- let mut instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &owner_no_sign_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- instruction.accounts[3].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_no_sign_info,
- ],
- &instruction.data,
- )
- );
- // missing owner
- let instruction = transfer(
- &program_id,
- &account_info.key,
- &account_info.key,
- &owner2_info.key,
- &[],
- 1000,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account_info.clone(),
- owner2_info.clone(),
- ],
- &instruction.data,
- )
- );
- // missing owner checked
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &owner2_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner2_info.clone(),
- ],
- &instruction.data,
- )
- );
- // insufficient funds
- let instruction = transfer(
- &program_id,
- &account_info.key,
- &account_info.key,
- &owner_info.key,
- &[],
- 1001,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // insufficient funds checked
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &owner_info.key,
- &[],
- 1001,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // incorrect decimals
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &owner_info.key,
- &[],
- 1,
- 10, // <-- incorrect decimals
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // incorrect mint
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &account3_info.key, // <-- incorrect mint
- &account_info.key,
- &owner_info.key,
- &[],
- 1,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account3_info.clone(), // <-- incorrect mint
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // approve delegate
- let instruction = approve(
- &program_id,
- &account_info.key,
- &delegate_info.key,
- &owner_info.key,
- &[],
- 100,
- )
- .unwrap();
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- delegate_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- .unwrap();
- // delegate transfer
- let instruction = transfer(
- &program_id,
- &account_info.key,
- &account_info.key,
- &delegate_info.key,
- &[],
- 100,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &instruction.data,
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- assert_eq!(account.delegated_amount, 100);
- // delegate transfer checked
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &delegate_info.key,
- &[],
- 100,
- 2,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &instruction.data,
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- assert_eq!(account.delegated_amount, 100);
- // delegate insufficient funds
- let instruction = transfer(
- &program_id,
- &account_info.key,
- &account_info.key,
- &delegate_info.key,
- &[],
- 101,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &instruction.data,
- )
- );
- // delegate insufficient funds checked
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &delegate_info.key,
- &[],
- 101,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &instruction.data,
- )
- );
- // owner transfer with delegate assigned
- let instruction = transfer(
- &program_id,
- &account_info.key,
- &account_info.key,
- &owner_info.key,
- &[],
- 1000,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- // owner transfer with delegate assigned checked
- let instruction = transfer_checked(
- &program_id,
- &account_info.key,
- &mint_info.key,
- &account_info.key,
- &owner_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- Processor::process(
- &instruction.program_id,
- &[
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &instruction.data,
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- }
- #[test]
- fn test_mintable_token_with_zero_supply() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint-able token with zero supply
- let decimals = 2;
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, decimals).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(
- mint,
- Mint {
- mint_authority: COption::Some(owner_key),
- supply: 0,
- decimals,
- is_initialized: true,
- freeze_authority: COption::None,
- }
- );
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // mint to
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- let _ = Mint::unpack(&mint_account.data).unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // mint to 2, with incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- mint_to_checked(
- &program_id,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 42,
- decimals + 1
- )
- .unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- );
- let _ = Mint::unpack(&mint_account.data).unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // mint to 2
- do_process_instruction(
- mint_to_checked(
- &program_id,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 42,
- decimals,
- )
- .unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- let _ = Mint::unpack(&mint_account.data).unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 84);
- }
- #[test]
- fn test_approve_dups() {
- let program_id = Pubkey::new_unique();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_info: AccountInfo = (&account2_key, false, &mut account2_account).into();
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_info: AccountInfo = (&account3_key, true, &mut account3_account).into();
- let multisig_key = Pubkey::new_unique();
- let mut multisig_account = SolanaAccount::new(
- multisig_minimum_balance(),
- Multisig::get_packed_len(),
- &program_id,
- );
- let multisig_info: AccountInfo = (&multisig_key, true, &mut multisig_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, false, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction_dups(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- account2_info.clone(),
- mint_info.clone(),
- owner_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // mint to account
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- )
- .unwrap();
- // source-owner approve
- do_process_instruction_dups(
- approve(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-owner approve_checked
- do_process_instruction_dups(
- approve_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-owner revoke
- do_process_instruction_dups(
- revoke(&program_id, &account1_key, &account1_key, &[]).unwrap(),
- vec![account1_info.clone(), account1_info.clone()],
- )
- .unwrap();
- // test source-multisig signer
- do_process_instruction_dups(
- initialize_multisig(&program_id, &multisig_key, &[&account3_key], 1).unwrap(),
- vec![
- multisig_info.clone(),
- rent_info.clone(),
- account3_info.clone(),
- ],
- )
- .unwrap();
- do_process_instruction_dups(
- initialize_account(&program_id, &account3_key, &mint_key, &multisig_key).unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- multisig_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account3_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account3_info.clone(), owner_info.clone()],
- )
- .unwrap();
- // source-multisig-signer approve
- do_process_instruction_dups(
- approve(
- &program_id,
- &account3_key,
- &account2_key,
- &multisig_key,
- &[&account3_key],
- 500,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account3_info.clone(),
- ],
- )
- .unwrap();
- // source-multisig-signer approve_checked
- do_process_instruction_dups(
- approve_checked(
- &program_id,
- &account3_key,
- &mint_key,
- &account2_key,
- &multisig_key,
- &[&account3_key],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account3_info.clone(),
- ],
- )
- .unwrap();
- // source-owner multisig-signer
- do_process_instruction_dups(
- revoke(&program_id, &account3_key, &multisig_key, &[&account3_key]).unwrap(),
- vec![
- account3_info.clone(),
- multisig_info.clone(),
- account3_info.clone(),
- ],
- )
- .unwrap();
- }
- #[test]
- fn test_approve() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- // missing signer
- let mut instruction = approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap();
- instruction.accounts[2].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- );
- // no owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner2_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner2_account,
- ],
- )
- );
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // approve delegate 2, with incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- approve_checked(
- &program_id,
- &account_key,
- &mint_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- 0 // <-- incorrect decimals
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- );
- // approve delegate 2, with incorrect mint
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- approve_checked(
- &program_id,
- &account_key,
- &account2_key, // <-- bad mint
- &delegate_key,
- &owner_key,
- &[],
- 100,
- 0
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account, // <-- bad mint
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- );
- // approve delegate 2
- do_process_instruction(
- approve_checked(
- &program_id,
- &account_key,
- &mint_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- 2,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // revoke delegate
- do_process_instruction(
- revoke(&program_id, &account_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut owner_account],
- )
- .unwrap();
- }
- #[test]
- fn test_set_authority_dups() {
- let program_id = Pubkey::new_unique();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &mint_key, Some(&mint_key), 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // set mint_authority when currently self
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner_key),
- AuthorityType::MintTokens,
- &mint_key,
- &[],
- )
- .unwrap(),
- vec![mint_info.clone(), mint_info.clone()],
- )
- .unwrap();
- // set freeze_authority when currently self
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner_key),
- AuthorityType::FreezeAccount,
- &mint_key,
- &[],
- )
- .unwrap(),
- vec![mint_info.clone(), mint_info.clone()],
- )
- .unwrap();
- // set account owner when currently self
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &account1_key,
- Some(&owner_key),
- AuthorityType::AccountOwner,
- &account1_key,
- &[],
- )
- .unwrap(),
- vec![account1_info.clone(), account1_info.clone()],
- )
- .unwrap();
- // set close_authority when currently self
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.close_authority = COption::Some(account1_key);
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &account1_key,
- Some(&owner_key),
- AuthorityType::CloseAccount,
- &account1_key,
- &[],
- )
- .unwrap(),
- vec![account1_info.clone(), account1_info.clone()],
- )
- .unwrap();
- }
- #[test]
- fn test_set_authority() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let owner3_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut mint2_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create mint with owner and freeze_authority
- do_process_instruction(
- initialize_mint(&program_id, &mint2_key, &owner_key, Some(&owner_key), 2).unwrap(),
- vec![&mut mint2_account, &mut rent_sysvar],
- )
- .unwrap();
- // invalid account
- assert_eq!(
- Err(ProgramError::UninitializedAccount),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- )
- );
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint2_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint2_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner_key),
- AuthorityType::AccountOwner,
- &owner2_key,
- &[]
- )
- .unwrap(),
- vec![&mut account_account, &mut owner2_account],
- )
- );
- // owner did not sign
- let mut instruction = set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[],
- )
- .unwrap();
- instruction.accounts[1].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(instruction, vec![&mut account_account, &mut owner_account,],)
- );
- // wrong authority type
- assert_eq!(
- Err(TokenError::AuthorityTypeNotSupported.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- )
- );
- // account owner may not be set to None
- assert_eq!(
- Err(TokenError::InvalidInstruction.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- None,
- AuthorityType::AccountOwner,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- )
- );
- // set owner
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- )
- .unwrap();
- // set close_authority
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::CloseAccount,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner2_account],
- )
- .unwrap();
- // close_authority may be set to None
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- None,
- AuthorityType::CloseAccount,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner2_account],
- )
- .unwrap();
- // wrong owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner3_key),
- AuthorityType::MintTokens,
- &owner2_key,
- &[]
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner2_account],
- )
- );
- // owner did not sign
- let mut instruction = set_authority(
- &program_id,
- &mint_key,
- Some(&owner2_key),
- AuthorityType::MintTokens,
- &owner_key,
- &[],
- )
- .unwrap();
- instruction.accounts[1].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(instruction, vec![&mut mint_account, &mut owner_account],)
- );
- // cannot freeze
- assert_eq!(
- Err(TokenError::MintCannotFreeze.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- )
- );
- // set owner
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner2_key),
- AuthorityType::MintTokens,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- )
- .unwrap();
- // set owner to None
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- None,
- AuthorityType::MintTokens,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner2_account],
- )
- .unwrap();
- // test unsetting mint_authority is one-way operation
- assert_eq!(
- Err(TokenError::FixedSupply.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- Some(&owner2_key),
- AuthorityType::MintTokens,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- )
- );
- // set freeze_authority
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint2_account, &mut owner_account],
- )
- .unwrap();
- // test unsetting freeze_authority is one-way operation
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- None,
- AuthorityType::FreezeAccount,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint2_account, &mut owner2_account],
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::MintCannotFreeze.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint2_account, &mut owner2_account],
- )
- );
- }
- #[test]
- fn test_mint_to_dups() {
- let program_id = Pubkey::new_unique();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &mint_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &owner_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- owner_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // mint_to when mint_authority is self
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &mint_key, &[], 42).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), mint_info.clone()],
- )
- .unwrap();
- // mint_to_checked when mint_authority is self
- do_process_instruction_dups(
- mint_to_checked(&program_id, &mint_key, &account1_key, &mint_key, &[], 42, 2).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), mint_info.clone()],
- )
- .unwrap();
- // mint_to when mint_authority is account owner
- let mut mint = Mint::unpack_unchecked(&mint_info.data.borrow()).unwrap();
- mint.mint_authority = COption::Some(account1_key);
- Mint::pack(mint, &mut mint_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- mint_to(
- &program_id,
- &mint_key,
- &account1_key,
- &account1_key,
- &[],
- 42,
- )
- .unwrap(),
- vec![
- mint_info.clone(),
- account1_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // mint_to_checked when mint_authority is account owner
- do_process_instruction_dups(
- mint_to(
- &program_id,
- &mint_key,
- &account1_key,
- &account1_key,
- &[],
- 42,
- )
- .unwrap(),
- vec![
- mint_info.clone(),
- account1_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- }
- #[test]
- fn test_mint_to() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mismatch_key = Pubkey::new_unique();
- let mut mismatch_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let uninitialized_key = Pubkey::new_unique();
- let mut uninitialized_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create mismatch account
- do_process_instruction(
- initialize_account(&program_id, &mismatch_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut mismatch_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&mismatch_account.data).unwrap();
- account.mint = mint2_key;
- Account::pack(account, &mut mismatch_account.data).unwrap();
- // mint to
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 42);
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // mint to another account to test supply accumulation
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account2_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account2_account, &mut owner_account],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 84);
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // missing signer
- let mut instruction =
- mint_to(&program_id, &mint_key, &account2_key, &owner_key, &[], 42).unwrap();
- instruction.accounts[2].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![&mut mint_account, &mut account2_account, &mut owner_account],
- )
- );
- // mismatch account
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &mismatch_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut mismatch_account, &mut owner_account],
- )
- );
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account2_key, &owner2_key, &[], 42).unwrap(),
- vec![
- &mut mint_account,
- &mut account2_account,
- &mut owner2_account,
- ],
- )
- );
- // uninitialized destination account
- assert_eq!(
- Err(ProgramError::UninitializedAccount),
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &uninitialized_key,
- &owner_key,
- &[],
- 42
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut uninitialized_account,
- &mut owner_account,
- ],
- )
- );
- // unset mint_authority and test minting fails
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- None,
- AuthorityType::MintTokens,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::FixedSupply.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account2_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account2_account, &mut owner_account],
- )
- );
- }
- #[test]
- fn test_burn_dups() {
- let program_id = Pubkey::new_unique();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // mint to account
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- )
- .unwrap();
- // source-owner burn
- do_process_instruction_dups(
- burn(
- &program_id,
- &mint_key,
- &account1_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-owner burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // mint-owner burn
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.owner = mint_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- burn(&program_id, &account1_key, &mint_key, &mint_key, &[], 500).unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- )
- .unwrap();
- // mint-owner burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &mint_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- )
- .unwrap();
- // source-delegate burn
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.delegated_amount = 1000;
- account.delegate = COption::Some(account1_key);
- account.owner = owner_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- burn(
- &program_id,
- &account1_key,
- &mint_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-delegate burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // mint-delegate burn
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.delegated_amount = 1000;
- account.delegate = COption::Some(mint_key);
- account.owner = owner_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- burn(&program_id, &account1_key, &mint_key, &mint_key, &[], 500).unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- )
- .unwrap();
- // mint-delegate burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &mint_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- )
- .unwrap();
- }
- #[test]
- fn test_burn() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let mismatch_key = Pubkey::new_unique();
- let mut mismatch_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut rent_sysvar = rent_sysvar();
- // create new mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create mismatch account
- do_process_instruction(
- initialize_account(&program_id, &mismatch_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut mismatch_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- // mint to mismatch account and change mint key
- do_process_instruction(
- mint_to(&program_id, &mint_key, &mismatch_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut mismatch_account, &mut owner_account],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&mismatch_account.data).unwrap();
- account.mint = mint2_key;
- Account::pack(account, &mut mismatch_account.data).unwrap();
- // missing signer
- let mut instruction =
- burn(&program_id, &account_key, &mint_key, &delegate_key, &[], 42).unwrap();
- instruction.accounts[1].is_signer = false;
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- instruction,
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account
- ],
- )
- );
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner2_key, &[], 42).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- )
- );
- // mint mismatch
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- burn(&program_id, &mismatch_key, &mint_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mismatch_account, &mut mint_account, &mut owner_account],
- )
- );
- // burn
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 21).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- .unwrap();
- // burn_checked, with incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- burn_checked(&program_id, &account_key, &mint_key, &owner_key, &[], 21, 3).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- );
- // burn_checked
- do_process_instruction(
- burn_checked(&program_id, &account_key, &mint_key, &owner_key, &[], 21, 2).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 2000 - 42);
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 1000 - 42);
- // insufficient funds
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &owner_key,
- &[],
- 100_000_000
- )
- .unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- );
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 84,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- // not a delegate of source account
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &owner_key,
- &[],
- 100_000_000
- )
- .unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- );
- // burn via delegate
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &delegate_key, &[], 84).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account,
- ],
- )
- .unwrap();
- // match
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 2000 - 42 - 84);
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 1000 - 42 - 84);
- // insufficient funds approved via delegate
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &delegate_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account
- ],
- )
- );
- }
- #[test]
- fn test_multisig() {
- let program_id = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let account_key = Pubkey::new_unique();
- let mut account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let multisig_key = Pubkey::new_unique();
- let mut multisig_account = SolanaAccount::new(42, Multisig::get_packed_len(), &program_id);
- let multisig_delegate_key = Pubkey::new_unique();
- let mut multisig_delegate_account = SolanaAccount::new(
- multisig_minimum_balance(),
- Multisig::get_packed_len(),
- &program_id,
- );
- let signer_keys = vec![Pubkey::new_unique(); MAX_SIGNERS];
- let signer_key_refs: Vec<&Pubkey> = signer_keys.iter().collect();
- let mut signer_accounts = vec![SolanaAccount::new(0, 0, &program_id); MAX_SIGNERS];
- let mut rent_sysvar = rent_sysvar();
- // multisig is not rent exempt
- let account_info_iter = &mut signer_accounts.iter_mut();
- assert_eq!(
- Err(TokenError::NotRentExempt.into()),
- do_process_instruction(
- initialize_multisig(&program_id, &multisig_key, &[&signer_keys[0]], 1).unwrap(),
- vec![
- &mut multisig_account,
- &mut rent_sysvar,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- );
- multisig_account.lamports = multisig_minimum_balance();
- // single signer
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- initialize_multisig(&program_id, &multisig_key, &[&signer_keys[0]], 1).unwrap(),
- vec![
- &mut multisig_account,
- &mut rent_sysvar,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // multiple signer
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- initialize_multisig(
- &program_id,
- &multisig_delegate_key,
- &signer_key_refs,
- MAX_SIGNERS as u8,
- )
- .unwrap(),
- vec![
- &mut multisig_delegate_account,
- &mut rent_sysvar,
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // create new mint with multisig owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &multisig_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account with multisig owner
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &multisig_key).unwrap(),
- vec![
- &mut account,
- &mut mint_account,
- &mut multisig_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account with multisig owner
- do_process_instruction(
- initialize_account(
- &program_id,
- &account2_key,
- &mint_key,
- &multisig_delegate_key,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut multisig_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // mint to account
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &multisig_key,
- &[&signer_keys[0]],
- 1000,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // approve
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &multisig_delegate_key,
- &multisig_key,
- &[&signer_keys[0]],
- 100,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut multisig_delegate_account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // transfer
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &multisig_key,
- &[&signer_keys[0]],
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut account2_account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // transfer via delegate
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &multisig_delegate_key,
- &signer_key_refs,
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut account2_account,
- &mut multisig_delegate_account,
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // mint to
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account2_key,
- &multisig_key,
- &[&signer_keys[0]],
- 42,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account2_account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // burn
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &multisig_key,
- &[&signer_keys[0]],
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut mint_account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // burn via delegate
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &multisig_delegate_key,
- &signer_key_refs,
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut mint_account,
- &mut multisig_delegate_account,
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // freeze account
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mint2_key = Pubkey::new_unique();
- let mut mint2_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- do_process_instruction(
- initialize_mint(
- &program_id,
- &mint2_key,
- &multisig_key,
- Some(&multisig_key),
- 2,
- )
- .unwrap(),
- vec![&mut mint2_account, &mut rent_sysvar],
- )
- .unwrap();
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint2_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint2_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- mint_to(
- &program_id,
- &mint2_key,
- &account3_key,
- &multisig_key,
- &[&signer_keys[0]],
- 1000,
- )
- .unwrap(),
- vec![
- &mut mint2_account,
- &mut account3_account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- freeze_account(
- &program_id,
- &account3_key,
- &mint2_key,
- &multisig_key,
- &[&signer_keys[0]],
- )
- .unwrap(),
- vec![
- &mut account3_account,
- &mut mint2_account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // do SetAuthority on mint
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner_key),
- AuthorityType::MintTokens,
- &multisig_key,
- &[&signer_keys[0]],
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- // do SetAuthority on account
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner_key),
- AuthorityType::AccountOwner,
- &multisig_key,
- &[&signer_keys[0]],
- )
- .unwrap(),
- vec![
- &mut account,
- &mut multisig_account,
- &mut account_info_iter.next().unwrap(),
- ],
- )
- .unwrap();
- }
- #[test]
- fn test_validate_owner() {
- let program_id = Pubkey::new_unique();
- let owner_key = Pubkey::new_unique();
- let mut signer_keys = [Pubkey::default(); MAX_SIGNERS];
- for signer_key in signer_keys.iter_mut().take(MAX_SIGNERS) {
- *signer_key = Pubkey::new_unique();
- }
- let mut signer_lamports = 0;
- let mut signer_data = vec![];
- let mut signers = vec![
- AccountInfo::new(
- &owner_key,
- true,
- false,
- &mut signer_lamports,
- &mut signer_data,
- &program_id,
- false,
- Epoch::default(),
- );
- MAX_SIGNERS + 1
- ];
- for (signer, key) in signers.iter_mut().zip(&signer_keys) {
- signer.key = key;
- }
- let mut lamports = 0;
- let mut data = vec![0; Multisig::get_packed_len()];
- let mut multisig = Multisig::unpack_unchecked(&data).unwrap();
- multisig.m = MAX_SIGNERS as u8;
- multisig.n = MAX_SIGNERS as u8;
- multisig.signers = signer_keys;
- multisig.is_initialized = true;
- Multisig::pack(multisig, &mut data).unwrap();
- let owner_account_info = AccountInfo::new(
- &owner_key,
- false,
- false,
- &mut lamports,
- &mut data,
- &program_id,
- false,
- Epoch::default(),
- );
- // full 11 of 11
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers).unwrap();
- // 1 of 11
- {
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 1;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- }
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers).unwrap();
- // 2:1
- {
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 2;
- multisig.n = 1;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- }
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers)
- );
- // 0:11
- {
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 0;
- multisig.n = 11;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- }
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers).unwrap();
- // 2:11 but 0 provided
- {
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 2;
- multisig.n = 11;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- }
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &[])
- );
- // 2:11 but 1 provided
- {
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 2;
- multisig.n = 11;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- }
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers[0..1])
- );
- // 2:11, 2 from middle provided
- {
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 2;
- multisig.n = 11;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- }
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers[5..7])
- .unwrap();
- // 11:11, one is not a signer
- {
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 11;
- multisig.n = 11;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- }
- signers[5].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers)
- );
- signers[5].is_signer = true;
- // 11:11, single signer signs multiple times
- {
- let mut signer_lamports = 0;
- let mut signer_data = vec![];
- let signers = vec![
- AccountInfo::new(
- &signer_keys[5],
- true,
- false,
- &mut signer_lamports,
- &mut signer_data,
- &program_id,
- false,
- Epoch::default(),
- );
- MAX_SIGNERS + 1
- ];
- let mut multisig =
- Multisig::unpack_unchecked(&owner_account_info.data.borrow()).unwrap();
- multisig.m = 11;
- multisig.n = 11;
- Multisig::pack(multisig, &mut owner_account_info.data.borrow_mut()).unwrap();
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- Processor::validate_owner(&program_id, &owner_key, &owner_account_info, &signers)
- );
- }
- }
- #[test]
- fn test_close_account_dups() {
- let program_id = Pubkey::new_unique();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_info: AccountInfo = (&account2_key, true, &mut account2_account).into();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, false, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // source-owner close
- do_process_instruction_dups(
- close_account(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // source-close-authority close
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.close_authority = COption::Some(account1_key);
- account.owner = owner_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- close_account(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- }
- #[test]
- fn test_close_account() {
- let program_id = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance() + 42,
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mut rent_sysvar = rent_sysvar();
- // uninitialized
- assert_eq!(
- Err(ProgramError::UninitializedAccount),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner2_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner2_account,
- ],
- )
- );
- // initialize and mint to non-native account
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 42).unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // initialize native account
- do_process_instruction(
- initialize_account(
- &program_id,
- &account2_key,
- &crate::native_mint::id(),
- &owner_key,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 42);
- // close non-native account with balance
- assert_eq!(
- Err(TokenError::NonNativeHasBalance.into()),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner_account,
- ],
- )
- );
- assert_eq!(account_account.lamports, account_minimum_balance());
- // empty account
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 42).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- .unwrap();
- // wrong owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner2_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner2_account,
- ],
- )
- );
- // close account
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- assert_eq!(account_account.lamports, 0);
- assert_eq!(account3_account.lamports, 2 * account_minimum_balance());
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 0);
- // fund and initialize new non-native account to test close authority
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- account_account.lamports = 2;
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::CloseAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- )
- .unwrap();
- // account owner cannot authorize close if close_authority is set
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner_account,
- ],
- )
- );
- // close non-native account with close_authority
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner2_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner2_account,
- ],
- )
- .unwrap();
- assert_eq!(account_account.lamports, 0);
- assert_eq!(account3_account.lamports, 2 * account_minimum_balance() + 2);
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 0);
- // close native account
- do_process_instruction(
- close_account(&program_id, &account2_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account2_account,
- &mut account3_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account_account.lamports, 0);
- assert_eq!(account.amount, 0);
- assert_eq!(
- account3_account.lamports,
- 3 * account_minimum_balance() + 2 + 42
- );
- }
- #[test]
- fn test_native_token() {
- let program_id = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance() + 40,
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(account_minimum_balance(), 0, &program_id);
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mut rent_sysvar = rent_sysvar();
- // initialize native account
- do_process_instruction(
- initialize_account(
- &program_id,
- &account_key,
- &crate::native_mint::id(),
- &owner_key,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 40);
- // initialize native account
- do_process_instruction(
- initialize_account(
- &program_id,
- &account2_key,
- &crate::native_mint::id(),
- &owner_key,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 0);
- // mint_to unsupported
- assert_eq!(
- Err(TokenError::NativeNotSupported.into()),
- do_process_instruction(
- mint_to(
- &program_id,
- &crate::native_mint::id(),
- &account_key,
- &owner_key,
- &[],
- 42
- )
- .unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- );
- // burn unsupported
- let bogus_mint_key = Pubkey::new_unique();
- let mut bogus_mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- do_process_instruction(
- initialize_mint(&program_id, &bogus_mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut bogus_mint_account, &mut rent_sysvar],
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::NativeNotSupported.into()),
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &bogus_mint_key,
- &owner_key,
- &[],
- 42
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut bogus_mint_account,
- &mut owner_account
- ],
- )
- );
- // ensure can't transfer below rent-exempt reserve
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 50,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- );
- // transfer between native accounts
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 40,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- assert_eq!(account_account.lamports, account_minimum_balance());
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 0);
- assert_eq!(account2_account.lamports, account_minimum_balance() + 40);
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 40);
- // close native account
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner_account,
- ],
- )
- .unwrap();
- assert_eq!(account_account.lamports, 0);
- assert_eq!(account3_account.lamports, 2 * account_minimum_balance());
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 0);
- }
- #[test]
- fn test_overflow() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_owner_key = Pubkey::new_unique();
- let mut mint_owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &mint_owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create an account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner2_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner2_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // mint the max to an account
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &mint_owner_key,
- &[],
- u64::MAX,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut mint_owner_account,
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX);
- // attempt to mint one more to account
- assert_eq!(
- Err(TokenError::Overflow.into()),
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &mint_owner_key,
- &[],
- 1,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut mint_owner_account,
- ],
- )
- );
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX);
- // atttempt to mint one more to the other account
- assert_eq!(
- Err(TokenError::Overflow.into()),
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account2_key,
- &mint_owner_key,
- &[],
- 1,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account2_account,
- &mut mint_owner_account,
- ],
- )
- );
- // burn some of the supply
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 100).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX - 100);
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &mint_owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut mint_owner_account,
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX);
- // manipulate account balance to attempt overflow transfer
- let mut account = Account::unpack_unchecked(&account2_account.data).unwrap();
- account.amount = 1;
- Account::pack(account, &mut account2_account.data).unwrap();
- assert_eq!(
- Err(TokenError::Overflow.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account2_key,
- &account_key,
- &owner2_key,
- &[],
- 1,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut account_account,
- &mut owner2_account,
- ],
- )
- );
- }
- #[test]
- fn test_frozen() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint and fund first account
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // fund first account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- // no transfer if either account is frozen
- let mut account = Account::unpack_unchecked(&account2_account.data).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account2_account.data).unwrap();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- );
- let mut account = Account::unpack_unchecked(&account_account.data).unwrap();
- account.state = AccountState::Initialized;
- Account::pack(account, &mut account_account.data).unwrap();
- let mut account = Account::unpack_unchecked(&account2_account.data).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account2_account.data).unwrap();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- )
- );
- // no approve if account is frozen
- let mut account = Account::unpack_unchecked(&account_account.data).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account_account.data).unwrap();
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- )
- );
- // no revoke if account is frozen
- let mut account = Account::unpack_unchecked(&account_account.data).unwrap();
- account.delegate = COption::Some(delegate_key);
- account.delegated_amount = 100;
- Account::pack(account, &mut account_account.data).unwrap();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- revoke(&program_id, &account_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut owner_account],
- )
- );
- // no set authority if account is frozen
- let new_owner_key = Pubkey::new_unique();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&new_owner_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account,],
- )
- );
- // no mint_to if destination account is frozen
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 100).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account,],
- )
- );
- // no burn if account is frozen
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 100).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- );
- }
- #[test]
- fn test_freeze_thaw_dups() {
- let program_id = Pubkey::new_unique();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, Some(&account1_key), 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- )
- .unwrap();
- // freeze where mint freeze_authority is account
- do_process_instruction_dups(
- freeze_account(&program_id, &account1_key, &mint_key, &account1_key, &[]).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- // thaw where mint freeze_authority is account
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- thaw_account(&program_id, &account1_key, &mint_key, &account1_key, &[]).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- )
- .unwrap();
- }
- #[test]
- fn test_freeze_account() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account_owner_key = Pubkey::new_unique();
- let mut account_owner_account = SolanaAccount::default();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner different from account owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &account_owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut account_owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- )
- .unwrap();
- // mint cannot freeze
- assert_eq!(
- Err(TokenError::MintCannotFreeze.into()),
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- );
- // missing freeze_authority
- let mut mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- mint.freeze_authority = COption::Some(owner_key);
- Mint::pack(mint, &mut mint_account.data).unwrap();
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner2_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- )
- );
- // check explicit thaw
- assert_eq!(
- Err(TokenError::InvalidState.into()),
- do_process_instruction(
- thaw_account(&program_id, &account_key, &mint_key, &owner2_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- )
- );
- // freeze
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.state, AccountState::Frozen);
- // check explicit freeze
- assert_eq!(
- Err(TokenError::InvalidState.into()),
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- );
- // check thaw authority
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- thaw_account(&program_id, &account_key, &mint_key, &owner2_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- )
- );
- // thaw
- do_process_instruction(
- thaw_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.state, AccountState::Initialized);
- }
- #[test]
- fn test_initialize_account2() {
- let program_id = Pubkey::new_unique();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- )
- .unwrap();
- do_process_instruction(
- initialize_account2(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![&mut account2_account, &mut mint_account, &mut rent_sysvar],
- )
- .unwrap();
- assert_eq!(account_account, account2_account);
- }
- }
|